Прогресс загрузки с RxJava, OkHttp и Okio в Android

Question

В нашем приложении я загружаю файл изображения с этим кодом. Мне нужно показать прогресс загрузки (загруженных байтов в процентах) на пользовательском интерфейсе. Как я могу получить прогресс загрузки в этом коде? Я искал решение, но все еще не мог сделать это самостоятельно.

Observable<String> downloadObservable = Observable.create(
                    sub -> {
                        Request request = new Request.Builder()
                                .url(media.getMediaUrl())
                                .build();
                        Response response = null;
                        try {
                            response = http_client.newCall(request).execute();
                            if (response.isSuccessful()) {
                                Log.d(TAG, "response.isSuccessful()");
                                String mimeType = MimeTypeMap.getFileExtensionFromUrl(media.getMediaUrl());
                                File file = new File(helper.getTmpFolder() + "/" + helper.generateUniqueName() + "test." + mimeType);
                                BufferedSink sink = Okio.buffer(Okio.sink(file));
                                sink.writeAll(response.body().source());
                                sink.close();
                                sub.onNext(response.toString());
                                sub.onCompleted();
                            } else {
                                sub.onError(new IOException());
                            }
                        } catch (IOException e) {
                            e.printStackTrace();
                        }

                    }
            );

            Subscriber<String> mySubscriber = new Subscriber<String>() {
                @Override
                public void onNext(String responseString) {
                    Log.d(TAG, "works: " + responseString);
                }
            };
            downloadObservable
                    .subscribeOn(Schedulers.newThread())
                    .observeOn(AndroidSchedulers.mainThread())
                    .subscribe(mySubscriber);

Answer 1

Это то, что я бы сделал, чтобы показать прогресс.

Observable<String> downloadObservable = Observable.create(
  sub -> {
          InputStream input = null;
          OutputStream output = null;
          try {
          Response response = http_client.newCall(request).execute();
           if (response.isSuccessful()) {                 
             input = response.body().byteStream();
             long tlength= response.body().contentLength();

             output = new FileOutputStream("/pathtofile");
             byte data[] = new byte[1024];

             sub.onNext("0%");
             long total = 0;
             int count;
             while ((count = input.read(data)) != -1) {
               total += count;

               sub.onNext(String.valueOf(total*100/tlength) + "%");

               output.write(data, 0, count);
             }
             output.flush();
             output.close();
             input.close();
           }
          } catch(IOException e){
            sub.onError(e);
          } finally {
                if (input != null){
                    try {
                        input.close();
                    }catch(IOException ioe){}
                }
                if (out != null){
                    try{
                        output.close();
                    }catch (IOException e){}                        
                }
          }
        sub.onCompleted();
   }
);

И используйте Подписчика, у которого есть полные абстрактные методы.

Subscriber<String> mySubscriber = new Subscriber<String>() {

@Override
public void onCompleted() {
  // hide progress bar
}

@Override
public void onError(Throwable e) {
  // hide progress bar
}

@Override
public void onNext(String percentProgress) {
  // show percentage progress
}
};

Answer 2

принятый ответ показывает только прогресс сохранения загруженного файла, но не фактический прогресс загрузки.

Создан файл загрузки Rx с решением прогресса на Kotlin, надеюсь, это поможет:

      private const val FULL_PROGRESS = 100L

class DownloadWithProgress {

    fun start(url: String, directory: File, fileName: String) =
        Observable.create<Download> { emitter ->
            try {
                val request = Request.Builder()
                    .url(url)
                    .build()

                OkHttpClient.Builder()
                    .addNetworkInterceptor { chain: Interceptor.Chain ->
                        val originalResponse = chain.proceed(chain.request())
                        originalResponse.newBuilder()
                            .body(originalResponse.body()?.let { responseBody ->
                                ProgressResponseBody(responseBody) { emitter.onNext(Download(it)) }
                            })
                            .build()
                    }
                    .build()
                    .newCall(request).execute().use { response ->
                        if (!response.isSuccessful) throw IOException("Unexpected code $response")
                        val bufferedSource = response.body()?.source() ?: throw NullPointerException("Response is null")

                        try {
                            val packageFile = File(directory, fileName).apply {
                                sink().buffer().run {
                                    writeAll(bufferedSource)
                                    close()
                                }
                            }

                            emitter.onNext(Download(FULL_PROGRESS, packageFile))
                        } catch (ioException: Exception) {
                            throw AppUpdateException("Io exception: " + ioException.localizedMessage)
                        }

                        emitter.onComplete()
                    }
            } catch (exception: Exception) {
                emitter.onError(exception)
            }
        }

    private class ProgressResponseBody(
        private val responseBody: ResponseBody,
        private val onProgress: (progress: Long) -> Unit
    ) : ResponseBody() {

        private var bufferedSource: BufferedSource? = null

        override fun contentType() = responseBody.contentType()

        override fun contentLength() = responseBody.contentLength()

        override fun source() = bufferedSource ?: source(responseBody.source()).buffer().also { bufferedSource = it }

        private fun source(source: Source): Source {
            return object : ForwardingSource(source) {
                var totalBytesRead = 0L

                override fun read(sink: Buffer, byteCount: Long): Long {
                    return super.read(sink, byteCount).apply {
                        val bytesRead = this
                        totalBytesRead += if (bytesRead != -1L) bytesRead else 0

                        if (bytesRead != -1L) {
                            val progress = FULL_PROGRESS * totalBytesRead / responseBody.contentLength()
                            onProgress(progress)
                        }
                    }
                }
            }
        }
    }
}

data class Download(
    val progress: Long,
    val file: File? = null
) {
    val isDownloaded = progress == FULL_PROGRESS && file != null
}