fix: fix null issue #19 #14

phodal · phodal · commit 1fe63fe94364 · 2023-08-07T17:59:20.000+08:00
diff --git a/java/src/main/kotlin/cc/unitmesh/idea/spring/JavaSpringCodeCreator.kt b/java/src/main/kotlin/cc/unitmesh/idea/spring/JavaSpringCodeCreator.kt
@@ -72,7 +72,7 @@ class JavaSpringCodeCreator(val project: Project) : SpringBaseCrud {
         val allControllerFiles = getAllControllerFiles()
         if (allControllerFiles.isEmpty()) return
 
-        val targetControllerFile = allControllerFiles.first { it.name == "$targetController.java" }
+        val targetControllerFile = allControllerFiles.firstOrNull { it.name == "$targetController.java" } ?: return
 
         updateCodeMethod(targetControllerFile, targetController, code, project, psiElementFactory)
     }

Original file line number	Diff line number	Diff line change
`@@ -72,7 +72,7 @@ class JavaSpringCodeCreator(val project: Project) : SpringBaseCrud {`
`72`	`72`	`val allControllerFiles = getAllControllerFiles()`
`73`	`73`	`if (allControllerFiles.isEmpty()) return`
`74`	`74`
`75`		`- val targetControllerFile = allControllerFiles.first { it.name == "$targetController.java" }`
	`75`	`+ val targetControllerFile = allControllerFiles.firstOrNull { it.name == "$targetController.java" } ?: return`
`76`	`76`
`77`	`77`	`updateCodeMethod(targetControllerFile, targetController, code, project, psiElementFactory)`
`78`	`78`	`}`