fix: fix crash

unit-mesh · Aug 9, 2023 · 24c3cd7 · 24c3cd7
1 parent 4d65801
commit 24c3cd7
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/src/main/kotlin/cc/unitmesh/devti/custom/CustomIntention.kt b/src/main/kotlin/cc/unitmesh/devti/custom/CustomIntention.kt
@@ -58,7 +58,7 @@ class CustomIntention(private val intentionConfig: CustomIntentionConfig) : Abst
             SelectionVariableResolver(psiElement.language.displayName ?: "", selectedText),
         ) + SpecResolverService.getInstance().createResolvers()
 
-        val resolverMap = LinkedHashMap<String, VariableResolver>(10)
+        val resolverMap = LinkedHashMap<String, VariableResolver>(20)
         for (resolver in variableResolvers) {
             resolverMap[resolver.variableName()] = resolver
         }

diff --git a/src/main/kotlin/cc/unitmesh/devti/custom/variable/SimilarChunkVariableResolver.kt b/src/main/kotlin/cc/unitmesh/devti/custom/variable/SimilarChunkVariableResolver.kt
@@ -8,8 +8,12 @@ class SimilarChunkVariableResolver(val element: PsiElement) : VariableResolver {
     override val type: CustomIntentionVariableType get() = CustomIntentionVariableType.SIMILAR_CHUNK
 
     override fun resolve(): String {
-        return ReadAction.compute<String, Throwable> {
-            SimilarChunksWithPaths.createQuery(element, 256)
+        return try {
+            ReadAction.compute<String, Throwable> {
+                SimilarChunksWithPaths.createQuery(element, 256)
+            }
+        } catch (e: Throwable) {
+            ""
         }
     }
 }
diff --git a/src/main/resources/intentionDescriptions/AutoDevIntentionHelper/description.html b/src/main/resources/intentionDescriptions/AutoDevIntentionHelper/description.html
@@ -0,0 +1 @@
+Invoke <b>AutoDev</b> Action in here.