fix: fix get gorm tag column name error

Author: afumu

cmd/gplus/gen.go

@@ -3,16 +3,16 @@ package main
 import (
 	"bytes"
 	"fmt"
+	"github.com/dave/jennifer/jen"
 	"go/ast"
 	"go/format"
 	"go/types"
 	"gorm.io/gorm/schema"
 	"io"
-
-	"github.com/dave/jennifer/jen"
 	"sigs.k8s.io/controller-tools/pkg/genall"
 	"sigs.k8s.io/controller-tools/pkg/loader"
 	"sigs.k8s.io/controller-tools/pkg/markers"
+	"strings"
 )
 
 type StructInfo struct {
@@ -89,7 +89,10 @@ func buildGenFile(root *loader.Package, allStructInfos []StructInfo) *jen.File {
 		genFile.Id("}").Id("{")
 		for i, field := range s.Fields {
 			tag := s.FieldTags[i]
-			tagSetting := schema.ParseTagSetting(tag, ";")
+			// todo Need to optimize
+			newTag := strings.ReplaceAll(tag, "gorm:", "")
+			newTag = strings.ReplaceAll(newTag, "\"", "")
+			tagSetting := schema.ParseTagSetting(newTag, ";")
 			columnName := tagSetting["COLUMN"]
 			if columnName == "" {
 				// Use NamingStrategy by default for now

example/user.go

@@ -2,7 +2,7 @@ package example
 
 import "time"
 
-//  +gplus:column=true
+// +gplus:column=true
 
 type User struct {
 	ID        int64  `gorm:"primaryKey"`
@@ -20,3 +20,15 @@ type User struct {
 func (User) TableName() string {
 	return "Users"
 }
+
+// +gplus:column=true
+
+type Clazz struct {
+	ID          int64 `gorm:"primaryKey"`
+	Name        string
+	Count       string
+	TeacherName string
+	HouseNumber int `gorm:"column:house"`
+	CreatedAt   time.Time
+	UpdatedAt   time.Time
+}

example/zz_gen.column.go

@@ -23,3 +23,20 @@ var UserColumn = struct {
 	CreatedAt: "created_at",
 	UpdatedAt: "updated_at",
 }
+var ClazzColumn = struct {
+	ID          string
+	Name        string
+	Count       string
+	TeacherName string
+	HouseNumber string
+	CreatedAt   string
+	UpdatedAt   string
+}{
+	ID:          "id",
+	Name:        "name",
+	Count:       "count",
+	TeacherName: "teacher_name",
+	HouseNumber: "house",
+	CreatedAt:   "created_at",
+	UpdatedAt:   "updated_at",
+}