替朋友合并mssql

data/view/model/genmssql/common.go

@@ -0,0 +1,100 @@
+package genmssql
+
+import (
+	"strings"
+
+	"github.com/xxjwxc/gormt/data/config"
+	"github.com/xxjwxc/public/mylog"
+
+	"github.com/xxjwxc/gormt/data/view/model"
+)
+
+// filterModel filter.过滤 gorm.Model
+func filterModel(list *[]genColumns) bool {
+	if config.GetDBTag() != "gorm" {
+		return false
+	}
+
+	var _temp []genColumns
+	num := 0
+	for _, v := range *list {
+		if strings.EqualFold(v.Field, "id") ||
+			strings.EqualFold(v.Field, "created_at") ||
+			strings.EqualFold(v.Field, "updated_at") ||
+			strings.EqualFold(v.Field, "deleted_at") {
+			num++
+		} else {
+			_temp = append(_temp, v)
+		}
+	}
+
+	if num >= 4 {
+		*list = _temp
+		return true
+	}
+
+	return false
+}
+
+// fixForeignKey fix foreign key.过滤外键
+func fixForeignKey(list []genForeignKey, columuName string, result *[]model.ForeignKey) {
+	for _, v := range list {
+		if strings.EqualFold(v.ColumnName, columuName) { // find it .找到了
+			*result = append(*result, model.ForeignKey{
+				TableName:  v.ReferencedTableName,
+				ColumnName: v.ReferencedColumnName,
+			})
+		}
+	}
+}
+
+// GetModel get model interface. 获取model接口
+func GetModel() model.IModel {
+	//now just support mysql
+	return &MssqlModel
+}
+
+// FixNotes 分析元素表注释
+func FixNotes(em *model.ColumnsInfo, note string) {
+	b0 := FixElementTag(em, note)        // gorm
+	b1 := FixForeignKeyTag(em, em.Notes) // 外键
+	if !b0 && b1 {                       // 补偿
+		FixElementTag(em, em.Notes) // gorm
+	}
+}
+
+// FixElementTag 分析元素表注释
+func FixElementTag(em *model.ColumnsInfo, note string) bool {
+	matches := noteRegex.FindStringSubmatch(note)
+	if len(matches) < 2 {
+		em.Notes = note
+		return false
+	}
+
+	mylog.Infof("get one gorm tag:(%v) ==> (%v)", em.BaseInfo.Name, matches[1])
+	em.Notes = note[len(matches[0]):]
+	em.Gormt = matches[1]
+	return true
+}
+
+// FixForeignKeyTag 分析元素表注释(外键)
+func FixForeignKeyTag(em *model.ColumnsInfo, note string) bool {
+	matches := foreignKeyRegex.FindStringSubmatch(note) // foreign key 外键
+	if len(matches) < 2 {
+		em.Notes = note
+		return false
+	}
+	em.Notes = note[len(matches[0]):]
+
+	// foreign key 外键
+	tmp := strings.Split(matches[1], ".")
+	if len(tmp) > 0 {
+		mylog.Infof("get one foreign key:(%v) ==> (%v)", em.BaseInfo.Name, matches[1])
+		em.ForeignKeyList = append(em.ForeignKeyList, model.ForeignKey{
+			TableName:  tmp[0],
+			ColumnName: tmp[1],
+		})
+	}
+
+	return true
+}