Restructure

重构

data/view/model/genmysql/common.go

@@ -0,0 +1 @@
+package genmysql

data/view/model/genmysql/def.go

@@ -0,0 +1,15 @@
+package genmysql
+
+const (
+	_tagGorm = "gorm"
+	_tagJSON = "json"
+)
+
+// GenColumns show full columns
+type GenColumns struct {
+	Field string `gorm:"column:Field"`
+	Type  string `gorm:"column:Type"`
+	Key   string `gorm:"column:Key"`
+	Desc  string `gorm:"column:Comment"`
+	Null  string `gorm:"column:Null"`
+}

data/view/model/genmysql/genmysql.go

@@ -0,0 +1,228 @@
+package genmysql
+
+import (
+	"fmt"
+	"regexp"
+	"strings"
+
+	"github.com/xxjwxc/gormt/data/config"
+	"github.com/xxjwxc/gormt/data/view/cnf"
+	"github.com/xxjwxc/gormt/data/view/genstruct"
+	"github.com/xxjwxc/public/mybigcamel"
+	"github.com/xxjwxc/public/mysqldb"
+)
+
+// GenMysql 开始mysql解析
+func GenMysql() genstruct.GenPackage {
+	orm := mysqldb.OnInitDBOrm(config.GetMysqlConStr())
+	defer orm.OnDestoryDB()
+
+	return OnGetPackageInfo(orm, OnGetTables(orm))
+}
+
+// OnGetTables Get columns and comments.获取表列及注释
+func OnGetTables(orm *mysqldb.MySqlDB) map[string]string {
+	tbDesc := make(map[string]string)
+
+	// Get column names.获取列名
+	var tables []string
+	rows, err := orm.Raw("show tables").Rows()
+	if err != nil {
+		fmt.Println(err)
+		return tbDesc
+	}
+	defer rows.Close()
+
+	for rows.Next() {
+		var table string
+		rows.Scan(&table)
+		tables = append(tables, table)
+		tbDesc[table] = ""
+	}
+
+	// Get table annotations.获取表注释
+	rows, err = orm.Raw("SELECT TABLE_NAME,TABLE_COMMENT FROM information_schema.TABLES WHERE table_schema=?;",
+		config.GetMysqlDbInfo().Database).Rows()
+	if err != nil {
+		fmt.Println(err)
+		return tbDesc
+	}
+
+	for rows.Next() {
+		var table, desc string
+		rows.Scan(&table, &desc)
+		tbDesc[table] = desc
+	}
+
+	return tbDesc
+}
+
+// OnGetPackageInfo getpackage info.获取包信息
+func OnGetPackageInfo(orm *mysqldb.MySqlDB, tabls map[string]string) genstruct.GenPackage {
+	var pkg genstruct.GenPackage
+	for tab, desc := range tabls {
+		var sct genstruct.GenStruct
+		sct.SetStructName(OnGetCamelName(tab)) // Big hump.大驼峰
+		sct.SetNotes(desc)
+		// build element.构造元素
+		ems := OnGetTableElement(orm, tab)
+		// --------end
+		sct.AddElement(ems...)
+		// Get table annotations.获取表注释
+		rows, err := orm.Raw("show create table " + tab).Rows()
+		defer rows.Close()
+		if err == nil {
+			if rows.Next() {
+				var table, CreateTable string
+				rows.Scan(&table, &CreateTable)
+				sct.SetCreatTableStr(CreateTable)
+			}
+		}
+		// ----------end
+
+		pkg.AddStruct(sct)
+	}
+
+	return pkg
+}
+
+// OnGetTableElement Get table columns and comments.获取表列及注释
+func OnGetTableElement(orm *mysqldb.MySqlDB, tab string) []genstruct.GenElement {
+	var el []genstruct.GenElement
+
+	keyNums := make(map[string]int)
+	// get keys
+	var Keys []struct {
+		NonUnique  int    `gorm:"column:Non_unique"`
+		KeyName    string `gorm:"column:Key_name"`
+		ColumnName string `gorm:"column:Column_name"`
+	}
+	orm.Raw("show keys from " + tab).Find(&Keys)
+	for _, v := range Keys {
+		keyNums[v.KeyName]++
+	}
+	// ----------end
+
+	var list []GenColumns
+	// Get table annotations.获取表注释
+	orm.Raw("show FULL COLUMNS from " + tab).Find(&list)
+	// filter gorm.Model.过滤 gorm.Model
+	if OnHaveModel(&list) {
+		var tmp genstruct.GenElement
+		tmp.SetType("gorm.Model")
+		el = append(el, tmp)
+	}
+	// -----------------end
+
+	for _, v := range list {
+		var tmp genstruct.GenElement
+		tmp.SetName(OnGetCamelName(v.Field))
+		tmp.SetNotes(v.Desc)
+		tmp.SetType(OnGetTypeName(v.Type))
+
+		if strings.EqualFold(v.Key, "PRI") { // Set primary key.设置主键
+			tmp.AddTag(_tagGorm, "primary_key")
+		} else if strings.EqualFold(v.Key, "UNI") { // unique
+			tmp.AddTag(_tagGorm, "unique")
+		} else {
+			// index
+			for _, v1 := range Keys {
+				if strings.EqualFold(v1.ColumnName, v.Field) {
+					_val := ""
+					if v1.NonUnique == 1 { // index
+						_val += "index"
+					} else {
+						_val += "unique_index"
+					}
+					if keyNums[v1.KeyName] > 1 { // Composite index.复合索引
+						_val += ":" + v1.KeyName
+					}
+
+					tmp.AddTag(_tagGorm, _val)
+				}
+			}
+		}
+
+		// simple output
+		if !config.GetSimple() {
+			tmp.AddTag(_tagGorm, "column:"+v.Field)
+			tmp.AddTag(_tagGorm, "type:"+v.Type)
+			if strings.EqualFold(v.Null, "NO") {
+				tmp.AddTag(_tagGorm, "not null")
+			}
+		}
+
+		// json tag
+		if config.GetIsJSONTag() {
+			if strings.EqualFold(v.Field, "id") {
+				tmp.AddTag(_tagJSON, "-")
+			} else {
+				tmp.AddTag(_tagJSON, v.Field)
+			}
+		}
+		el = append(el, tmp)
+	}
+	return el
+}
+
+// OnHaveModel filter.过滤 gorm.Model
+func OnHaveModel(list *[]GenColumns) bool {
+	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
+}
+
+// OnGetTypeName Type acquisition filtering.类型获取过滤
+func OnGetTypeName(name string) string {
+	// Precise matching first.先精确匹配
+	if v, ok := cnf.TypeMysqlDicMp[name]; ok {
+		return v
+	}
+
+	// Fuzzy Regular Matching.模糊正则匹配
+	for k, v := range cnf.TypeMysqlMatchMp {
+		if ok, _ := regexp.MatchString(k, name); ok {
+			return v
+		}
+	}
+
+	panic(fmt.Sprintf("type (%v) not match in any way.maybe need to add on ()", name))
+}
+
+// OnGetCamelName Big Hump or Capital Letter.大驼峰或者首字母大写
+func OnGetCamelName(name string) string {
+	if config.GetSingularTable() { // If the table name plural is globally disabled.如果全局禁用表名复数
+		return TitleCase(name)
+	}
+
+	return mybigcamel.Marshal(name)
+}
+
+// TitleCase title case.首字母大写
+func TitleCase(name string) string {
+	vv := []rune(name)
+	if len(vv) > 0 {
+		if bool(vv[0] >= 'a' && vv[0] <= 'z') { // title case.首字母大写
+			vv[0] -= 32
+		}
+	}
+
+	return string(vv)
+}