fix: fix mvc issues

unit-mesh · Aug 1, 2023 · 8c9fec0 · 8c9fec0
1 parent d4764d2
commit 8c9fec0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/java/src/main/kotlin/cc/unitmesh/idea/MvcUtil.kt b/java/src/main/kotlin/cc/unitmesh/idea/MvcUtil.kt
@@ -2,9 +2,9 @@ package cc.unitmesh.idea
 
 object MvcUtil {
     fun isController(fileName: String, lang: String): Boolean {
-        return fileName.endsWith("Controller.$lang")
+        return fileName.endsWith("Controller.${lang.lowercase()}")
     }
 
     fun isService(fileName: String, lang: String) =
-        fileName.endsWith("Service.$lang") || fileName.endsWith("ServiceImpl.$lang")
+        fileName.endsWith("Service.$${lang.lowercase()}") || fileName.endsWith("ServiceImpl.$${lang.lowercase()}")
 }