zishuo/sponge
1
0
Fork 0
mirror of https://github.com/zhufuyi/sponge.git synced 2025-10-05 16:57:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: mongodb model tag

Browse Source
This commit is contained in:
zhuyasen
2025-03-25 23:58:35 +08:00
parent 5ccb334c21
commit f1de101d75
1 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
pkg/sql2code/parser/tableInfo.go
View File

@@ -3,6 +3,7 @@ package parser
import (
"encoding/json"
"fmt"
"strings"
"github.com/huandu/xstrings"
"github.com/jinzhu/inflection"
@@ -68,7 +69,7 @@ func newTableInfo(data tmplData) TableInfo {
TableNamePluralCamelFCL: customFirstLetterToLower(customEndOfLetterToLower(data.TableName, pluralName)),
TableNameSnake: xstrings.ToSnakeCase(data.TName),
TableComment: data.Comment,
Columns: getColumns(data.Fields),
Columns: getColumns(data.DBDriver, data.Fields),
PrimaryKey: getPrimaryKeyInfo(data.CrudInfo),
DBDriver: data.DBDriver,
ColumnSubStructure: data.SubStructs,
@@ -84,7 +85,7 @@ func (table TableInfo) getCode() []byte {
return code
}
func getColumns(fields []tmplField) []Field {
func getColumns(dbDriver string, fields []tmplField) []Field {
var columns []Field
for _, field := range fields {
@@ -95,13 +96,23 @@ func getColumns(fields []tmplField) []Field {
ColumnComment: field.Comment,
IsPrimaryKey: field.IsPrimaryKey,
GoType: field.GoType,
Tag: field.Tag,
Tag: handleTag(dbDriver, field.Tag),
})
}
return columns
}
func handleTag(dbDriver string, tag string) string {
if dbDriver == DBDriverMongodb {
tag = strings.ReplaceAll(tag, `bson:"column:`, `bson:"`)
tag = strings.ReplaceAll(tag, `;type:"`, `"`)
tag = strings.ReplaceAll(tag, `;type:;primary_key`, ``)
tag = strings.ReplaceAll(tag, `bson:"id" json:"id"`, `bson:"_id" json:"id"`)
}
return tag
}
func getPrimaryKeyInfo(info *CrudInfo) *PrimaryKey {
if info == nil {
return nil