feat: get column name from tag

Author: afumu

cmd/gplus/gen.go

@@ -6,6 +6,7 @@ import (
 	"go/ast"
 	"go/format"
 	"go/types"
+	"gorm.io/gorm/schema"
 	"io"
 
 	"github.com/dave/jennifer/jen"
@@ -87,9 +88,16 @@ func buildGenFile(root *loader.Package, allStructInfos []StructInfo) *jen.File {
 		}
 		genFile.Id("}").Id("{")
 		for i, field := range s.Fields {
-			tagName := s.FieldTags[i]
-			filedName := fmt.Sprintf("\"%s\"", tagName)
-			genFile.Id(field).Op(":").Id(filedName).Id(",")
+			tag := s.FieldTags[i]
+			tagSetting := schema.ParseTagSetting(tag, ";")
+			columnName := tagSetting["COLUMN"]
+			if columnName == "" {
+				// Use NamingStrategy by default for now
+				namingStrategy := schema.NamingStrategy{}
+				columnName = namingStrategy.ColumnName("", field)
+			}
+			columnName = fmt.Sprintf("\"%s\"", columnName)
+			genFile.Id(field).Op(":").Id(columnName).Id(",")
 		}
 		genFile.Id("}")
 	}

example/user.go

@@ -5,10 +5,8 @@ import "time"
 //  +gplus:column=true
 
 type User struct {
-	//ID        int64  `gorm:"primaryKey"`
-	//Username  string `gorm:"column:username"`
-	ID        int64  `id`
-	Username  string `username`
+	ID        int64  `gorm:"primaryKey"`
+	Username  string `gorm:"column:username"`
 	Password  string
 	Address   string
 	Age       int