fix could not import a partially corrupted backup

sunny-chung · Feb 16, 2024 · e8cb50a · e8cb50a
1 parent cc2f476
commit e8cb50a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/...in/kotlin/com/sunnychung/application/multiplatform/hellohttp/importer/DataDumpImporter.kt b/...in/kotlin/com/sunnychung/application/multiplatform/hellohttp/importer/DataDumpImporter.kt
@@ -106,14 +106,14 @@ class DataDumpImporter {
             }
         }
 
-        fun copyWithNewId(current: TreeObject): TreeObject {
+        fun copyWithNewId(current: TreeObject): TreeObject? {
             return when (current) {
                 is TreeFolder -> current.copy(
                     id = uuidString(),
-                    childs = current.childs.map { copyWithNewId(it) }.toMutableList(),
+                    childs = current.childs.mapNotNull { copyWithNewId(it) }.toMutableList(),
                 )
                 is TreeRequest -> current.copy(
-                    id = newRequestTemplateIds[current.id]!!
+                    id = newRequestTemplateIds[current.id] ?: return null
                 )
             }
         }
@@ -129,7 +129,7 @@ class DataDumpImporter {
                         environments = it.environments.map {
                             it.copy(id = uuidString())
                         }.toMutableList(),
-                        treeObjects = it.treeObjects.map { copyWithNewId(it) }.toMutableList(),
+                        treeObjects = it.treeObjects.mapNotNull { copyWithNewId(it) }.toMutableList(),
                         grpcApiSpecIds = it.grpcApiSpecIds.mapNotNull { newApiSpecIds[it] }.toMutableSet(),
                     )
                 }.toMutableList()