J2K: Speed-up add imports

2020-07-17 23:03:27 +03:00
parent cd0c644654
commit c5005f8695
1 changed files with 15 additions and 8 deletions
@@ -92,14 +92,21 @@ class NewJavaToKotlinConverter(

    private fun KtFile.addImports(imports: Collection<FqName>) {
        val factory = KtPsiFactory(this)
-        var importList = importList
-        for (import in imports) {
-            val importDirective = factory.createImportDirective(ImportPath(import, isAllUnder = false))
-            if (importList == null) {
-                importList = addImportList(importDirective.parent as KtImportList)
-            } else {
-                importList.add(importDirective)
-            }
+
+
+        if (imports.isEmpty()) return
+        val importPsi = factory.createImportDirectives(
+            imports.map { ImportPath(it, isAllUnder = false) }
+        )
+        val createdImportList = importPsi.first().parent as KtImportList
+        val importList = importList
+        if (importList == null) {
+            addImportList(createdImportList)
+        } else {
+            val updatedList = if(importList.firstChild != null) {
+                createdImportList.addRangeBefore(importList.firstChild, importList.lastChild, createdImportList.firstChild)
+            } else createdImportList
+            importList.replace(updatedList)
        }
    }