feat: 当解压临时文件异常后需要删除创建的文件#2406

TencentBlueKing · Jul 29, 2024 · ae038ca · ae038ca
1 parent cb89ee8
commit ae038ca
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/src/backend/rpm/biz-rpm/src/main/kotlin/com/tencent/bkrepo/rpm/util/GZipUtils.kt b/src/backend/rpm/biz-rpm/src/main/kotlin/com/tencent/bkrepo/rpm/util/GZipUtils.kt
@@ -79,15 +79,20 @@ object GZipUtils {
     fun InputStream.unGzipInputStream(): File {
         GZIPInputStream(this).use { gZIPInputStream ->
             val file = File.createTempFile("rpm_", ".xmlStream")
-            BufferedOutputStream(FileOutputStream(file)).use { bufferedOutputStream ->
-                var len: Int
-                val buffer = ByteArray(1 * 1024 * 1024)
-                while (gZIPInputStream.read(buffer).also { len = it } > 0) {
-                    bufferedOutputStream.write(buffer, 0, len)
+            try {
+                BufferedOutputStream(FileOutputStream(file)).use { bufferedOutputStream ->
+                    var len: Int
+                    val buffer = ByteArray(1 * 1024 * 1024)
+                    while (gZIPInputStream.read(buffer).also { len = it } > 0) {
+                        bufferedOutputStream.write(buffer, 0, len)
+                    }
+                    bufferedOutputStream.flush()
                 }
-                bufferedOutputStream.flush()
+                return file
+            } catch (e: Exception) {
+                file.delete()
+                throw e
             }
-            return file
         }
     }
 }