okhttp瓜挽、retrofit 2未提供上傳晨抡、下載的進(jìn)度回調(diào)派昧，但是很多應(yīng)用在UI顯示方面需要加入進(jìn)度顯示特笋。實(shí)現(xiàn)方式如下：
========
下載進(jìn)度及攔截器：
public class DownloadProgressInterceptor implements Interceptor {
private DownloadProgressListener progressListener;
public DownloadProgressInterceptor(DownloadProgressListener progressListener) {
this.progressListener = progressListener;
}
@Override public Response intercept(Chain chain) throws IOException {
Response originalResponse = chain.proceed(chain.request());
return originalResponse.newBuilder()
.body(new DownloadProgressResponseBody(originalResponse.body(), progressListener))
.build();
}
private static class DownloadProgressResponseBody extends ResponseBody {
private final ResponseBody responseBody;
private final DownloadProgressListener progressListener;
private BufferedSource bufferedSource;
public DownloadProgressResponseBody(ResponseBody responseBody,
DownloadProgressListener progressListener) {
this.responseBody = responseBody;
this.progressListener = progressListener;
}
@Override public MediaType contentType() {
return responseBody.contentType();
}
@Override public long contentLength() throws IOException {
return responseBody.contentLength();
}
@Override public BufferedSource source() throws IOException {
if (bufferedSource == null) {
bufferedSource = Okio.buffer(source(responseBody.source()));
}
return bufferedSource;
}
private Source source(Source source) {
return new ForwardingSource(source) {
long totalBytesRead = 0L;
@Override public long read(Buffer sink, long byteCount) throws IOException {
long bytesRead = super.read(sink, byteCount);
// read() returns the number of bytes read, or -1 if this source is exhausted.
totalBytesRead += bytesRead != -1 ? bytesRead : 0;
if (null != progressListener) {
progressListener.update(totalBytesRead, responseBody.contentLength(), bytesRead == -1);
}
return bytesRead;
}
};
}
}
public interface DownloadProgressListener {
void update(long bytesRead, long contentLength, boolean done);
}
}
上傳進(jìn)度
/**
* Decorates an OkHttp request body to count the number of bytes written when writing it. Can
* decorate any request body, but is most useful for tracking the upload progress of large
* multipart requests.
*
* @author Leo Nikkil?
*/
public class CountingRequestBody extends RequestBody {
protected RequestBody delegate;
protected Listener listener;
protected CountingSink countingSink;
public CountingRequestBody(RequestBody delegate, Listener listener) {
this.delegate = delegate;
this.listener = listener;
}
@Override public MediaType contentType() {
return delegate.contentType();
}
@Override public long contentLength() {
try {
return delegate.contentLength();
} catch (IOException e) {
e.printStackTrace();
}
return -1;
}
@Override public void writeTo(BufferedSink sink) throws IOException {
BufferedSink bufferedSink;
countingSink = new CountingSink(sink);
bufferedSink = Okio.buffer(countingSink);
delegate.writeTo(bufferedSink);
bufferedSink.flush();
}
protected final class CountingSink extends ForwardingSink {
private long bytesWritten = 0;
public CountingSink(Sink delegate) {
super(delegate);
}
@Override public void write(Buffer source, long byteCount) throws IOException {
super.write(source, byteCount);
bytesWritten += byteCount;
listener.onRequestProgress(bytesWritten, contentLength());
}
}
public static interface Listener {
public void onRequestProgress(long bytesWritten, long contentLength);
}
}
上傳進(jìn)度攔截器
public class UpLoadProgressInterceptor implements Interceptor {
private CountingRequestBody.Listener progressListener;
public UpLoadProgressInterceptor(CountingRequestBody.Listener progressListener) {
this.progressListener = progressListener;
}
@Override public Response intercept(Chain chain) throws IOException {
Request originalRequest = chain.request();
if (originalRequest.body() == null) {
return chain.proceed(originalRequest);
}
Request progressRequest = originalRequest.newBuilder()
.method(originalRequest.method(),
new CountingRequestBody(originalRequest.body(), progressListener))
.build();
return chain.proceed(progressRequest);
}
}
====
使用方式：
如果只是在okhttp中使用剃浇，上傳可以使用CountingRequestBody類巾兆，下載可以把DownloadProgressResponseBody單獨(dú)抽出來(lái)使用。使用時(shí)new出來(lái)虎囚，傳入相應(yīng)參數(shù)即可角塑。
如果是在retrofit 2中使用，可以使用攔截器的方式淘讥。類似DownloadProgressInterceptor圃伶，UpLoadProgressInterceptor，關(guān)鍵代碼如下：
OkHttpClientManager.getInstance()
.getOkHttpClient()
.networkInterceptors()
.add(upLoadProgressInterceptor);
/*上傳圖片請(qǐng)求*/
@Multipart @POST(ConstantsNetInterface.COURSE_UPLOAD_PIC) Call<UploadPicBean> uploadPic(
@PartMap Map<String, ?> params);
RequestBody fileBody = RequestBody.create(MediaType.parse("image/*"), imgFile);
mParams = new HashMap<>();//請(qǐng)求參數(shù)
mParams.put("file\"; filename=\"" + imgFile.getName() + "", fileBody);
===
關(guān)于Batow提的問(wèn)題（為方便例子使用了Rxjava,不使用Rxjava則是用下面的Call來(lái)處理）：
@GET("{path}") @Streaming Call<ResponseBody> download(@Path("path") String path);
@GET("{path}") @Streaming Observable<ResponseBody> downloadFile(@Path("path") String path);
public static void downloadFile(final String path, final Subscriber<Boolean> subscriber) {
RetrofitManager.getInstance()
.createDownloadApiService(Api.class)
.downloadFile(path)
.map(new Func1<ResponseBody, Boolean>() {
@Override public Boolean call(ResponseBody responseBody) {
return writeFileToSD(path, responseBody);
}
})
.subscribeOn(Schedulers.io())
.observeOn(AndroidSchedulers.mainThread())
.subscribe(subscriber);
}
private static boolean writeFileToSD(String pathName, ResponseBody responseBody) {
String sdStatus = Environment.getExternalStorageState();
if (!sdStatus.equals(Environment.MEDIA_MOUNTED)) {
Logger.d("SD card is not avaiable/writeable right now.");
return false;
}
try {
String fileName = pathName.substring(pathName.lastIndexOf("/"));
Logger.d("fileName-->" + fileName);
File path = Environment.getExternalStorageDirectory();
File file = new File(path.getPath() + fileName);
if (!file.exists()) {
Logger.d("Create the file:" + file.getPath());
file.createNewFile();
}
FileOutputStream stream = new FileOutputStream(file);
byte[] buf = responseBody.bytes();
stream.write(buf);
stream.close();
return true;
} catch (Exception e) {
Logger.e("Error on writeFilToSD.");
e.printStackTrace();
}
return false;
}
NetworkWrapper.downloadFile(path, new Subscriber<Boolean>() {
@Override public void onCompleted() {
Logger.d("onCompleted");
}
@Override public void onError(Throwable e) {
Logger.d("onError:" + e);
}
@Override public void onNext(Boolean aBoolean) {
Logger.d("onNext" + aBoolean);
if (aBoolean) {
Toast.makeText(DownloadActivity.this, "pic download finish", Toast.LENGTH_LONG)
.show();
}
}
});
===
