jager/gormt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'master' of https://github.com/xxjwxc/gormt

Browse Source
This commit is contained in:
xxj
2021-08-26 20:51:06 +08:00
parent 575a045275 f47a533ecb
commit a235233956
4 changed files with 65 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
data/view/model/model.go
View File

@@ -123,44 +123,61 @@ func (m *_Model) genTableElement(cols []ColumnsInfo) (el []genstruct.GenElement)
tmp.SetName(getCamelName(v.Name))
tmp.SetNotes(v.Notes)
tmp.SetType(getTypeName(v.Type, v.IsNull))
// not simple output. 默认不输出gorm标签
if !config.GetSimple() {
for _, v1 := range v.Index {
switch v1.Key {
// case ColumnsKeyDefault:
case ColumnsKeyPrimary: // primary key.主键
tmp.AddTag(_tagGorm, "primaryKey")
isPK = true
case ColumnsKeyUnique: // unique key.唯一索引
tmp.AddTag(_tagGorm, "unique")
case ColumnsKeyIndex: // index key.复合索引
uninStr := getUninStr("index", ":", v1.KeyName)
// 兼容 gorm 本身 sort 标签
if v1.KeyName == "sort" {
uninStr = "index"
// is_db_tag. 是否输出gorm标签
if config.GetIsDbTag() {
// not simple output. 默认只输出gorm主键和字段标签
if !config.GetSimple() {
for _, v1 := range v.Index {
switch v1.Key {
// case ColumnsKeyDefault:
case ColumnsKeyPrimary: // primary key.主键
tmp.AddTag(_tagGorm, "primaryKey")
isPK = true
case ColumnsKeyUnique: // unique key.唯一索引
tmp.AddTag(_tagGorm, "unique")
case ColumnsKeyIndex: // index key.复合索引
uninStr := getUninStr("index", ":", v1.KeyName)
// 兼容 gorm 本身 sort 标签
if v1.KeyName == "sort" {
uninStr = "index"
}
if v1.KeyType == "FULLTEXT" {
uninStr += ",class:FULLTEXT"
}
tmp.AddTag(_tagGorm, uninStr)
case ColumnsKeyUniqueIndex: // unique index key.唯一复合索引
tmp.AddTag(_tagGorm, getUninStr("uniqueIndex", ":", v1.KeyName))
}
if v1.KeyType == "FULLTEXT" {
uninStr += ",class:FULLTEXT"
}
} else {
for _, v1 := range v.Index {
switch v1.Key {
// case ColumnsKeyDefault:
case ColumnsKeyPrimary: // primary key.主键
tmp.AddTag(_tagGorm, "primaryKey")
isPK = true
}
tmp.AddTag(_tagGorm, uninStr)
case ColumnsKeyUniqueIndex: // unique index key.唯一复合索引
tmp.AddTag(_tagGorm, getUninStr("uniqueIndex", ":", v1.KeyName))
}
}
}
}
if len(v.Name) > 0 {
// not simple output
if !config.GetSimple() {
tmp.AddTag(_tagGorm, "column:"+v.Name)
tmp.AddTag(_tagGorm, "type:"+v.Type)
if !v.IsNull {
tmp.AddTag(_tagGorm, "not null")
}
// default tag
if len(v.Gormt) > 0 {
tmp.AddTag(_tagGorm, v.Gormt)
// is_db_tag. 是否输出gorm标签
if config.GetIsDbTag() {
// not simple output
if !config.GetSimple() {
tmp.AddTag(_tagGorm, "column:"+v.Name)
tmp.AddTag(_tagGorm, "type:"+v.Type)
if !v.IsNull {
tmp.AddTag(_tagGorm, "not null")
}
// default tag
if len(v.Gormt) > 0 {
tmp.AddTag(_tagGorm, v.Gormt)
}
} else {
tmp.AddTag(_tagGorm, "column:"+v.Name)
}
}