✨ Compatibility GoParamDefinition and Save

Author: johnmai-dev

example/goframe/main.go

@@ -6,7 +6,7 @@ import (
 
 type User struct {
 	g.Meta `orm:"table:goframe_users, do:true"`
-	ID     uint // 主键
+	ID     uint    // 主键
 	Name   string  `orm:"user_name"` // 姓名
 	Email  *string // 邮箱
 	Age    int32   // 年龄

src/main/java/com/github/maiqingqiang/goormhelper/orm/ORMCompletionProvider.java

@@ -136,6 +136,18 @@ protected void addCompletions(@NotNull CompletionParameters parameters, @NotNull
                             handleGoTypeReferenceExpression(parameters, result, descriptor, goCompositeLit.getTypeReferenceExpression());
                         }
                     }
+                } else if(goReferenceExpression.resolve() instanceof GoParamDefinition goParamDefinition){
+                    GoType goType = goParamDefinition.getGoTypeInner(ResolveState.initial());
+
+                    if (goType instanceof GoArrayOrSliceType goArrayOrSliceType) {
+                        goType = goArrayOrSliceType.getType();
+                    }
+
+                    if (goType instanceof GoPointerType goPointerType) {
+                        goType = goPointerType.getType();
+                    }
+
+                    handleGoType(parameters, result, descriptor, goType);
                 }
             } else if (argument instanceof GoCompositeLit goCompositeLit) {
                 handleGoTypeReferenceExpression(parameters, result, descriptor, goCompositeLit.getTypeReferenceExpression());

src/main/java/com/github/maiqingqiang/goormhelper/orm/gorm/GormTypes.java

@@ -66,14 +66,16 @@ public interface GormTypes {
             Map.entry(GoMethodDescriptor.of("(*gorm.io/gorm.DB).Take"), 0),
             Map.entry(GoMethodDescriptor.of("(*gorm.io/gorm.DB).Last"), 0),
             Map.entry(GoMethodDescriptor.of("(*gorm.io/gorm.DB).Create"), 0),
+            Map.entry(GoMethodDescriptor.of("(*gorm.io/gorm.DB).Save"), 0),
             Map.entry(GoMethodDescriptor.of("(*github.com/jinzhu/gorm.DB).First"), 0),
             Map.entry(GoMethodDescriptor.of("(*github.com/jinzhu/gorm.DB).Find"), 0),
             Map.entry(GoMethodDescriptor.of("(*github.com/jinzhu/gorm.DB).Delete"), 0),
             Map.entry(GoMethodDescriptor.of("(*github.com/jinzhu/gorm.DB).FirstOrInit"), 0),
             Map.entry(GoMethodDescriptor.of("(*github.com/jinzhu/gorm.DB).FirstOrCreate"), 0),
             Map.entry(GoMethodDescriptor.of("(*github.com/jinzhu/gorm.DB).Take"), 0),
             Map.entry(GoMethodDescriptor.of("(*github.com/jinzhu/gorm.DB).Last"), 0),
-            Map.entry(GoMethodDescriptor.of("(*github.com/jinzhu/gorm.DB).Create"), 0)
+            Map.entry(GoMethodDescriptor.of("(*github.com/jinzhu/gorm.DB).Create"), 0),
+            Map.entry(GoMethodDescriptor.of("(*github.com/jinzhu/gorm.DB).Save"), 0)
     );
     GoCallableDescriptorSet OTHER_SCHEMA_CALLABLES_SET = new GoCallableDescriptorSet(OTHER_SCHEMA_CALLABLES.keySet());