支持结构体别名

Kotodian · May 28, 2021 · 47195e1 · 47195e1
1 parent c66687d
commit 47195e1
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/examples/dao/users.go b/examples/dao/users.go
diff --git a/examples/models/common/users.go b/examples/models/common/users.go
@@ -1,9 +1,11 @@
 package common
 
+type Status int
 type User struct {
 	Name    string `gorm:"column:name" query:"equal,scope"`
 	Age     int    `gorm:"column:age" query:"scope"`
 	Company Company
+	Status  Status `gorm:"column:status" query:"equal"`
 }
 
 type Company struct {

diff --git a/generator.go b/generator.go
@@ -117,10 +117,17 @@ func (g *Generator) parseFile() ([]*File, error) {
 							continue
 						}
 						tags := extractTag(v.Tag.Value)
+						typIdent := v.Type.(*ast.Ident)
+						var typ string
+						if typIdent.Obj != nil {
+							typ = file.pkg + "." + typIdent.Name
+						} else {
+							typ = typIdent.Name
+						}
 						if tags.Gorm(v.Names[0].Name) {
 							fields = append(fields, &Field{
 								name:   v.Names[0].Name,
-								goType: v.Type.(*ast.Ident).Name,
+								goType: typ,
 								tags:   tags,
 							})
 						}