feat: 実際にダウンロードしたファイルと実際のファイルのサイズをチェックするようにした。

pantasystem · Jun 26, 2023 · a91a4e6 · a91a4e6
1 parent 5b0d082
commit a91a4e6
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/...c/main/java/net/pantasystem/milktea/data/infrastructure/image/ImageCacheRepositoryImpl.kt b/...c/main/java/net/pantasystem/milktea/data/infrastructure/image/ImageCacheRepositoryImpl.kt
@@ -146,9 +146,14 @@ class ImageCacheRepositoryImpl @Inject constructor(
     private suspend fun downloadAndSaveFile(url: String, file: File) {
         file.outputStream().use { out ->
             val req = Request.Builder().url(url).build()
-            okHttpClientProvider.get().newCall(req).execute().body?.byteStream()
+            val response = okHttpClientProvider.get().newCall(req).execute()
+            val contentLength = response.header("Content-Length")?.toLongOrNull()
+            response.body?.byteStream()
                 ?.use { inStream ->
-                    inStream.copyTo(out)
+                    val bytesCopied = inStream.copyTo(out)
+                    if (contentLength != null && bytesCopied != contentLength) {
+                        throw Exception("Download failed: url=$url")
+                    }
                 }
             val options = BitmapFactory.Options()
             options.inJustDecodeBounds = true