feature: 下载进度回调再主线程中进行

tangwanchao · Feb 2, 2023 · 24c0c29 · 24c0c29
1 parent 29a00b3
commit 24c0c29
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/download/src/main/java/me/twc/source/download/DownloadCallback.kt b/download/src/main/java/me/twc/source/download/DownloadCallback.kt
@@ -1,5 +1,6 @@
 package me.twc.source.download
 
+import androidx.annotation.MainThread
 import okhttp3.ResponseBody
 import java.io.File
 
@@ -31,11 +32,13 @@ interface DownloadCallback {
     /**
      * 开始下载
      */
+    @MainThread
     fun onStart() {}
 
     /**
      * 下载中
      */
+    @MainThread
     fun onDownload(downloadLength: Int, totalLength: Long) {}
 
     /**

diff --git a/download/src/main/java/me/twc/source/download/Downloader.kt b/download/src/main/java/me/twc/source/download/Downloader.kt
@@ -26,7 +26,9 @@ class Downloader(
             bs = responseBody.byteStream()
             outputFile = callback.genSaveFile() ?: genSaveFile()
             fos = FileOutputStream(outputFile)
-            callback.onStart()
+            withContext(Dispatchers.Main){
+                callback.onStart()
+            }
             val totalLength = responseBody.contentLength()
             var downloadLength = 0
             val buffer = ByteArray(8 * 1024)