优化

gedoor · Feb 4, 2020 · 0139150 · 0139150
1 parent 0f2533e
commit 0139150
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/app/src/main/java/com/kunfei/bookshelf/help/storage/Backup.kt b/app/src/main/java/com/kunfei/bookshelf/help/storage/Backup.kt
@@ -105,13 +105,17 @@ object Backup {
     }
 
     private fun copyBackup(context: Context, uri: Uri) {
-        DocumentFile.fromTreeUri(context, uri)?.let { treeDoc ->
-            for (fileName in backupFileNames) {
-                val doc = treeDoc.findFile(fileName) ?: treeDoc.createFile("", fileName)
-                doc?.let {
-                    DocumentUtil.writeBytes(context, FileHelp.getFile(backupPath + File.separator + fileName).readBytes(), doc)
+        try {
+            DocumentFile.fromTreeUri(context, uri)?.let { treeDoc ->
+                for (fileName in backupFileNames) {
+                    val doc = treeDoc.findFile(fileName) ?: treeDoc.createFile("", fileName)
+                    doc?.let {
+                        DocumentUtil.writeBytes(context, FileHelp.getFile(backupPath + File.separator + fileName).readBytes(), doc)
+                    }
                 }
             }
+        } catch (e: Exception) {
+            copyBackup()
         }
     }