add autoIncrement

skynet2 · Jul 6, 2022 · 3dea08f · 3dea08f
1 parent fc112a7
commit 3dea08f
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 1 deletion.
diff --git a/data/view/model/def.go b/data/view/model/def.go
@@ -39,6 +39,7 @@ type TabInfo struct {
 type ColumnsInfo struct {
 	BaseInfo
 	IsNull         bool         // null if db is set null
+	Extra          string       // Extra (AUTO_INCREMENT 自增加)
 	Type           string       // Type.类型标记
 	Gormt          string       // 默认值
 	Index          []KList      // index list.index列表
@@ -117,7 +118,7 @@ type funDef struct {
 	Em          []EmInfo      // index 列表
 	Primary     []FList       // primary unique
 	Index       []FList       // index
-	IsOutPage   bool		  // 是否开启分页
+	IsOutPage   bool          // 是否开启分页
 }
 
 //
diff --git a/data/view/model/genmysql/def.go b/data/view/model/genmysql/def.go
@@ -16,6 +16,7 @@ type genColumns struct {
 	Key     string  `gorm:"column:Key"`
 	Desc    string  `gorm:"column:Comment"`
 	Null    string  `gorm:"column:Null"`
+	Extra   string  `gorm:"Extra"`
 	Default *string `gorm:"column:Default"`
 }
 

diff --git a/data/view/model/genmysql/genmysql.go b/data/view/model/genmysql/genmysql.go
@@ -153,6 +153,7 @@ func (m *mysqlModel) getTableElement(orm *mysqldb.MySqlDB, tab string) (el []mod
 		var tmp model.ColumnsInfo
 		tmp.Name = v.Field
 		tmp.Type = v.Type
+		tmp.Extra = v.Extra
 		FixNotes(&tmp, v.Desc) // 分析表注释
 
 		if v.Default != nil {

diff --git a/data/view/model/model.go b/data/view/model/model.go
@@ -140,6 +140,9 @@ func (m *_Model) genTableElement(cols []ColumnsInfo) (el []genstruct.GenElement)
 			if len(_tagGorm) > 0 {
 				// not simple output. 默认只输出gorm主键和字段标签
 				if !config.GetSimple() {
+					if strings.EqualFold(v.Extra, "auto_increment") {
+						tmp.AddTag(_tagGorm, "autoIncrement:true")
+					}
 					for _, v1 := range v.Index {
 						switch v1.Key {
 						// case ColumnsKeyDefault: