fix go model gen column name
This commit is contained in:
@@ -164,6 +164,17 @@ func (s *GenStruct) GenerateColumnName() []string {
|
|||||||
}
|
}
|
||||||
data.StructName = s.Name
|
data.StructName = s.Name
|
||||||
for _, v := range s.Em {
|
for _, v := range s.Em {
|
||||||
|
if strings.EqualFold(v.Type, "gorm.Model") { // gorm model
|
||||||
|
data.Em = append(data.Em, []struct {
|
||||||
|
ColumnName string
|
||||||
|
StructName string
|
||||||
|
}{
|
||||||
|
{ColumnName: "id", StructName: "ID"},
|
||||||
|
{ColumnName: "created_at", StructName: "CreatedAt"},
|
||||||
|
{ColumnName: "updated_at", StructName: "UpdatedAt"},
|
||||||
|
{ColumnName: "deleted_at", StructName: "DeletedAt"},
|
||||||
|
}...)
|
||||||
|
} else {
|
||||||
data.Em = append(data.Em, struct {
|
data.Em = append(data.Em, struct {
|
||||||
ColumnName string
|
ColumnName string
|
||||||
StructName string
|
StructName string
|
||||||
@@ -172,6 +183,8 @@ func (s *GenStruct) GenerateColumnName() []string {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
var buf bytes.Buffer
|
var buf bytes.Buffer
|
||||||
tmpl.Execute(&buf, data)
|
tmpl.Execute(&buf, data)
|
||||||
return []string{buf.String()}
|
return []string{buf.String()}
|
||||||
|
|||||||
Reference in New Issue
Block a user