🐛 Fix findAgainArgument

maiqingqiang · Jul 19, 2023 · 55ed4bb · 55ed4bb
1 parent e29ee21
commit 55ed4bb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...iang/goormhelper/orm/goframe/codeInsights/completion/GoFrameColumnCompletionProvider.java b/...iang/goormhelper/orm/goframe/codeInsights/completion/GoFrameColumnCompletionProvider.java
@@ -149,9 +149,10 @@ protected GoCompositeElement customFindGoCompositeElementByGoCallExpr(GoCallExpr
     @Override
     protected GoCompositeElement findAgainArgument(GoCompositeElement argument) {
         if (argument instanceof GoCallExpr goCallExpr) {
-            return findGoTypeByReceiver(goCallExpr, Set.of(GoTypeSpecDescriptor.of("builtin.string")));
+            GoType newArgument = findGoTypeByReceiver(goCallExpr, Set.of(GoTypeSpecDescriptor.of("builtin.string")));
+            if (newArgument != null) return newArgument;
         }
-        return null;
+        return argument;
     }
 
 }