Created
January 27, 2017 11:33
-
-
Save Anurag--Singh/a49f03c10241a64c1def32e7d03073da to your computer and use it in GitHub Desktop.
Okhttp3 file download with progress return
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import java.io.IOException; | |
import okhttp3.Interceptor; | |
import okhttp3.MediaType; | |
import okhttp3.OkHttpClient; | |
import okhttp3.Request; | |
import okhttp3.Response; | |
import okhttp3.ResponseBody; | |
import okio.Buffer; | |
import okio.BufferedSource; | |
import okio.ForwardingSource; | |
import okio.Okio; | |
import okio.Source; | |
public final class Progress { | |
public void run() throws Exception { | |
Request request = new Request.Builder() | |
.url("https://your_file_hosted_url/your_file.txt") | |
.build(); | |
final ProgressListener progressListener = new ProgressListener() { | |
@Override public void update(long bytesRead, long contentLength, boolean done) { | |
System.out.println(bytesRead); | |
System.out.println(contentLength); | |
System.out.println(done); | |
System.out.format("%d%% done\n", (100 * bytesRead) / contentLength); | |
} | |
}; | |
OkHttpClient client = new OkHttpClient.Builder() | |
.addNetworkInterceptor(new Interceptor() { | |
@Override public Response intercept(Chain chain) throws IOException { | |
Response originalResponse = chain.proceed(chain.request()); | |
return originalResponse.newBuilder() | |
.body(new ProgressResponseBody(originalResponse.body(), progressListener)) | |
.build(); | |
} | |
}) | |
.build(); | |
try (Response response = client.newCall(request).execute()) { | |
if (!response.isSuccessful()) throw new IOException("Unexpected code " + response); | |
System.out.println(response.body().string()); | |
} | |
} | |
public static void main(String... args) throws Exception { | |
new Progress().run(); | |
} | |
private static class ProgressResponseBody extends ResponseBody { | |
private final ResponseBody responseBody; | |
private final ProgressListener progressListener; | |
private BufferedSource bufferedSource; | |
public ProgressResponseBody(ResponseBody responseBody, ProgressListener progressListener) { | |
this.responseBody = responseBody; | |
this.progressListener = progressListener; | |
} | |
@Override public MediaType contentType() { | |
return responseBody.contentType(); | |
} | |
@Override public long contentLength() { | |
return responseBody.contentLength(); | |
} | |
@Override public BufferedSource source() { | |
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; | |
progressListener.update(totalBytesRead, responseBody.contentLength(), bytesRead == -1); | |
return bytesRead; | |
} | |
}; | |
} | |
} | |
interface ProgressListener { | |
void update(long bytesRead, long contentLength, boolean done); | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment