support sqlite

支持sqlite

Makefile

@@ -1,17 +1,17 @@
 all: # 构建
 	make tar 
 windows:
-	CGO_ENABLED=0 GOOS=windows GOARCH=amd64 go build -o gormt.exe main.go 
+	CGO_ENABLED=1 GOOS=windows GOARCH=amd64 go build -o gormt.exe main.go 
 mac:
-	CGO_ENABLED=0 GOOS=darwin GOARCH=amd64 go build -o gormt main.go 
+	CGO_ENABLED=1 GOOS=darwin GOARCH=amd64 go build -o gormt main.go 
 linux:
-	CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o gormt main.go 
+	CGO_ENABLED=1 GOOS=linux GOARCH=amd64 go build -o gormt main.go 
 tar: # 打包
-	CGO_ENABLED=0 GOOS=windows GOARCH=amd64 go build -o gormt.exe main.go 
+	CGO_ENABLED=1 GOOS=windows GOARCH=amd64 go build -o gormt.exe main.go 
 	tar czvf gormt_windows.zip gormt.exe config.yml
-	CGO_ENABLED=0 GOOS=darwin GOARCH=amd64 go build -o gormt main.go 
+	CGO_ENABLED=1 GOOS=darwin GOARCH=amd64 go build -o gormt main.go 
 	tar czvf gormt_mac.zip gormt config.yml
-	CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o gormt main.go 
+	CGO_ENABLED=1 GOOS=linux GOARCH=amd64 go build -o gormt main.go 
 	tar czvf gormt_linux.zip gormt config.yml
 clear:
 	test ! -d model/ || rm -rf  model/*

config.yml

@@ -14,10 +14,11 @@ is_foreign_key : true # 是否导出外键关联
 is_gui : false # 是否ui模式显示
 is_table_name : false # 是否直接生成表名函数
 is_null_to_point : false # 数据库默认 'DEFAULT NULL' 时设置结构为指针类型
-mysql_info:
+db_info:
-    host : 127.0.0.1
+    host : 127.0.0.1 # type=1的时候，host为yml文件全路径
     port : 3306
     username : root
     password : 123456
     database : oauth_db
+    type: 0 # 数据库类型:0:mysql , 1:sqlite , 2:mssql
 

data/cmd/cmd.go

@@ -13,7 +13,7 @@ import (
 	"gopkg.in/go-playground/validator.v9"
 )
 
-var mysqlInfo config.MysqlDbInfo
+var mysqlInfo config.DBInfo
 var outDir string
 var singularTable bool
 var foreignKey bool
@@ -78,20 +78,20 @@ func init() {
 func initConfig() {
 	MergeMysqlDbInfo()
 	validate := validator.New()
-	err := validate.Struct(config.GetMysqlDbInfo())
+	err := validate.Struct(config.GetDbInfo())
 	if err != nil {
 		mylog.Info("Can't read cmd: using （-h, --help) to get more info")
 		mylog.Error(err)
 		os.Exit(1)
 	} else {
 		mylog.Info("using database info:")
-		mylog.JSON(config.GetMysqlDbInfo())
+		mylog.JSON(config.GetDbInfo())
 	}
 }
 
 // MergeMysqlDbInfo merge parm
 func MergeMysqlDbInfo() {
-	var tmp = config.GetMysqlDbInfo()
+	var tmp = config.GetDbInfo()
 	if len(mysqlInfo.Database) > 0 {
 		tmp.Database = mysqlInfo.Database
 	}

data/config/MyIni.go

@@ -9,50 +9,51 @@ import (
 // Config custom config struct
 type Config struct {
 	CfgBase          `yaml:"base"`
-	MySQLInfo        MysqlDbInfo `yaml:"mysql_info"`
+	DBInfo           DBInfo `yaml:"db_info"`
-	OutDir           string      `yaml:"out_dir"`
+	OutDir           string `yaml:"out_dir"`
-	URLTag           string      `yaml:"url_tag"`  // url tag
+	URLTag           string `yaml:"url_tag"`  // url tag
-	Language         string      `yaml:"language"` // language
+	Language         string `yaml:"language"` // language
-	DbTag            string      `yaml:"db_tag"`   // 数据库标签（gormt,db）
+	DbTag            string `yaml:"db_tag"`   // 数据库标签（gormt,db）
-	Simple           bool        `yaml:"simple"`
+	Simple           bool   `yaml:"simple"`
-	IsWEBTag         bool        `yaml:"is_web_tag"`
+	IsWEBTag         bool   `yaml:"is_web_tag"`
-	IsWebTagPkHidden bool        `yaml:"is_web_tag_pk_hidden"` // web标记是否隐藏主键
+	IsWebTagPkHidden bool   `yaml:"is_web_tag_pk_hidden"` // web标记是否隐藏主键
-	SingularTable    bool        `yaml:"singular_table"`
+	SingularTable    bool   `yaml:"singular_table"`
-	IsForeignKey     bool        `yaml:"is_foreign_key"`
+	IsForeignKey     bool   `yaml:"is_foreign_key"`
-	IsOutSQL         bool        `yaml:"is_out_sql"`
+	IsOutSQL         bool   `yaml:"is_out_sql"`
-	IsOutFunc        bool        `yaml:"is_out_func"`
+	IsOutFunc        bool   `yaml:"is_out_func"`
-	IsGUI            bool        `yaml:"is_gui"` //
+	IsGUI            bool   `yaml:"is_gui"` //
-	IsTableName      bool        `yaml:"is_table_name"`
+	IsTableName      bool   `yaml:"is_table_name"`
-	IsNullToPoint    bool        `yaml:"is_null_to_point"` // null to porint
+	IsNullToPoint    bool   `yaml:"is_null_to_point"` // null to porint
 }
 
-// MysqlDbInfo mysql database information. mysql 数据库信息
+// DBInfo mysql database information. mysql 数据库信息
-type MysqlDbInfo struct {
+type DBInfo struct {
 	Host     string `validate:"required"` // Host. 地址
-	Port     int    `validate:"required"` // Port 端口号
+	Port     int    // Port 端口号
-	Username string `validate:"required"` // Username 用户名
+	Username string // Username 用户名
 	Password string // Password 密码
-	Database string `validate:"required"` // Database 数据库名
+	Database string // Database 数据库名
+	Type     int    // 数据库类型:0:mysql 1:配置
 }
 
 // SetMysqlDbInfo Update MySQL configuration information
-func SetMysqlDbInfo(info *MysqlDbInfo) {
+func SetMysqlDbInfo(info *DBInfo) {
-	_map.MySQLInfo = *info
+	_map.DBInfo = *info
 }
 
-// GetMysqlDbInfo Get MySQL configuration information .获取mysql配置信息
+// GetDbInfo Get configuration information .获取数据配置信息
-func GetMysqlDbInfo() MysqlDbInfo {
+func GetDbInfo() DBInfo {
-	return _map.MySQLInfo
+	return _map.DBInfo
 }
 
 // GetMysqlConStr Get MySQL connection string.获取mysql 连接字符串
 func GetMysqlConStr() string {
 	return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8&parseTime=True&loc=Local&interpolateParams=True",
-		_map.MySQLInfo.Username,
+		_map.DBInfo.Username,
-		_map.MySQLInfo.Password,
+		_map.DBInfo.Password,
-		_map.MySQLInfo.Host,
+		_map.DBInfo.Host,
-		_map.MySQLInfo.Port,
+		_map.DBInfo.Port,
-		_map.MySQLInfo.Database,
+		_map.DBInfo.Database,
 	)
 }
 

data/config/common.go

@@ -24,7 +24,7 @@ var _map = Config{
 	CfgBase: CfgBase{
 		IsDev: false,
 	},
-	MySQLInfo: MysqlDbInfo{
+	DBInfo: DBInfo{
 		Host:     "127.0.0.1",
 		Port:     3306,
 		Username: "root",
@@ -65,11 +65,12 @@ func InitFile(filename string) error {
 	if _, e := os.Stat(filename); e != nil {
 		fmt.Println("init default config file: ", filename)
 		if err := SaveToFile(); err == nil {
-			fmt.Println("done,please restart.")
+			InitFile(filename)
+			return nil
 		} else {
 			fmt.Println("shit,fail", err)
 		}
-		os.Exit(0)
+		// os.Exit(0)
 	}
 	bs, err := ioutil.ReadFile(filename)
 	if err != nil {

data/dlg/common.go

@@ -1,14 +1,17 @@
 package dlg
 
 import (
+	"fmt"
 	"os/exec"
 
 	"github.com/jroimartin/gocui"
 	"github.com/xxjwxc/gormt/data/view/model"
 	"github.com/xxjwxc/gormt/data/view/model/genmysql"
+	"github.com/xxjwxc/gormt/data/view/model/gensqlite"
 
 	"github.com/xxjwxc/gormt/data/config"
 
+	"github.com/xxjwxc/public/mylog"
 	"github.com/xxjwxc/public/tools"
 )
 
@@ -72,7 +75,18 @@ func getBool(bstr string) bool {
 }
 
 func generate(g *gocui.Gui, v *gocui.View) {
-	modeldb := genmysql.GetMysqlModel()
+	var modeldb model.IModel
+	switch config.GetDbInfo().Type {
+	case 0:
+		modeldb = genmysql.GetModel()
+	case 1:
+		modeldb = gensqlite.GetModel()
+	}
+	if modeldb == nil {
+		mylog.Error(fmt.Errorf("modeldb not fund : please check db_info.type (0:mysql , 1:sqlite , 2:mssql) "))
+		return
+	}
+
 	pkg := modeldb.GenModel()
 	// just for test
 	// out, _ := json.Marshal(pkg)

data/dlg/cui.go

@@ -203,16 +203,18 @@ func enterSet(g *gocui.Gui, v *gocui.View) error {
 	// add input field
 	form.AddInputField("out_dir", SLocalize("out_dir"), formPart[0], formPart[1]).SetText(config.GetOutDir()).
 		AddValidate("required input", requireValidator)
-	form.AddInputField("db_host", SLocalize("db_host"), formPart[0], formPart[1]).SetText(config.GetMysqlDbInfo().Host).
+	form.AddInputField("db_host", SLocalize("db_host"), formPart[0], formPart[1]).SetText(config.GetDbInfo().Host).
 		AddValidate("required input", requireValidator)
-	form.AddInputField("db_port", SLocalize("db_port"), formPart[0], formPart[1]).SetText(tools.AsString(config.GetMysqlDbInfo().Port)).
+	form.AddInputField("db_port", SLocalize("db_port"), formPart[0], formPart[1]).SetText(tools.AsString(config.GetDbInfo().Port)).
 		AddValidate("required input", requireValidator)
-	form.AddInputField("db_usename", SLocalize("db_usename"), formPart[0], formPart[1]).SetText(config.GetMysqlDbInfo().Username).
+	form.AddInputField("db_usename", SLocalize("db_usename"), formPart[0], formPart[1]).SetText(config.GetDbInfo().Username).
 		AddValidate("required input", requireValidator)
-	form.AddInputField("db_pwd", SLocalize("db_pwd"), formPart[0], formPart[1]).SetText(config.GetMysqlDbInfo().Password).
+	form.AddInputField("db_pwd", SLocalize("db_pwd"), formPart[0], formPart[1]).SetText(config.GetDbInfo().Password).
 		SetMask().SetMaskKeybinding(gocui.KeyCtrlA).
 		AddValidate("required input", requireValidator)
-	form.AddInputField("db_name", SLocalize("db_name"), formPart[0], formPart[1]).SetText(config.GetMysqlDbInfo().Database).
+	form.AddInputField("db_name", SLocalize("db_name"), formPart[0], formPart[1]).SetText(config.GetDbInfo().Database).
+		AddValidate("required input", requireValidator)
+	form.AddInputField("db_type", SLocalize("db_type"), formPart[0], formPart[1]).SetText(tools.AsString(config.GetDbInfo().Type)).
 		AddValidate("required input", requireValidator)
 
 	// add select
@@ -264,7 +266,7 @@ func buttonSave(g *gocui.Gui, v *gocui.View) error {
 	mp := form.GetFieldTexts()
 	config.SetOutDir(mp["out_dir"])
 
-	var dbInfo config.MysqlDbInfo
+	var dbInfo config.DBInfo
 	dbInfo.Host = mp["db_host"]
 	port, err := strconv.Atoi(mp["db_port"])
 	if err != nil {

data/view/cnf/def.go

@@ -40,6 +40,7 @@ var TypeMysqlDicMp = map[string]string{
 	"blob":                "[]byte",
 	"mediumblob":          "[]byte",
 	"longblob":            "[]byte",
+	"integer":             "int64",
 }
 
 // TypeMysqlMatchMp Fuzzy Matching Types.模糊匹配类型
@@ -58,6 +59,7 @@ var TypeMysqlMatchMp = map[string]string{
 	`^(enum)[(](.)+[)]`:              "string",
 	`^(varchar)[(]\d+[)]`:            "string",
 	`^(varbinary)[(]\d+[)]`:          "[]byte",
+	`^(blob)[(]\d+[)]`:               "[]byte",
 	`^(binary)[(]\d+[)]`:             "[]byte",
 	`^(decimal)[(]\d+,\d+[)]`:        "float64",
 	`^(mediumint)[(]\d+[)]`:          "string",
@@ -66,4 +68,6 @@ var TypeMysqlMatchMp = map[string]string{
 	`^(float)[(]\d+,\d+[)] unsigned`: "float64",
 	`^(datetime)[(]\d+[)]`:           "time.Time",
 	`^(bit)[(]\d+[)]`:                "[]uint8",
+	`^(text)[(]\d+[)]`:               "string",
+	`^(integer)[(]\d+[)]`:            "int",
 }

data/view/gtools/gtools.go

@@ -1,6 +1,7 @@
 package gtools
 
 import (
+	"fmt"
 	"os/exec"
 
 	"github.com/xxjwxc/public/mylog"
@@ -11,6 +12,7 @@ import (
 	"github.com/xxjwxc/gormt/data/config"
 
 	"github.com/xxjwxc/gormt/data/view/model/genmysql"
+	"github.com/xxjwxc/gormt/data/view/model/gensqlite"
 	"github.com/xxjwxc/public/tools"
 )
 
@@ -28,8 +30,20 @@ func showCmd() {
 	// tt.Nickname = "ticket_001"
 	// orm.Where("nickname = ?", "ticket_001").Find(&tt)
 	// fmt.Println(tt)
-	modeldb := genmysql.GetMysqlModel()
+	var modeldb model.IModel
+	switch config.GetDbInfo().Type {
+	case 0:
+		modeldb = genmysql.GetModel()
+	case 1:
+		modeldb = gensqlite.GetModel()
+	}
+	if modeldb == nil {
+		mylog.Error(fmt.Errorf("modeldb not fund : please check db_info.type (0:mysql , 1:sqlite , 2:mssql) "))
+		return
+	}
+
 	pkg := modeldb.GenModel()
+	// gencnf.GenOutPut(&pkg)
 	// just for test
 	// out, _ := json.Marshal(pkg)
 	// tools.WriteFile("test.txt", []string{string(out)}, true)

data/view/model/gencnf/gencnf.go

@@ -0,0 +1,103 @@
+package gencnf
+
+import (
+	"fmt"
+	"os"
+	"path"
+	"strings"
+
+	"gopkg.in/yaml.v3"
+
+	"github.com/xxjwxc/gormt/data/config"
+	"github.com/xxjwxc/gormt/data/view/model"
+	"github.com/xxjwxc/public/mylog"
+	"github.com/xxjwxc/public/tools"
+)
+
+// GetCnfModel get model interface. 获取model接口
+func GetCnfModel() model.IModel {
+	//now just support mysql
+	return &CnfModel
+}
+
+// GenOutPut 输出
+func GenOutPut(info *model.DBInfo) {
+	path := path.Join(config.GetOutDir(), info.DbName+".yml")
+	out, _ := yaml.Marshal(info)
+
+	flag := os.O_CREATE | os.O_WRONLY | os.O_TRUNC
+	f, err := os.OpenFile(path, flag, 0666)
+	if err != nil {
+		mylog.Error(err)
+		return
+	}
+	defer f.Close()
+	f.Write(out)
+}
+
+// CnfModel yaml model from IModel
+var CnfModel cnfModel
+
+type cnfModel struct {
+}
+
+// GenModel get model.DBInfo info.获取数据库相关属性
+func (m *cnfModel) GenModel() model.DBInfo {
+	var dbInfo model.DBInfo
+	// getPackageInfo(orm, &dbInfo)
+	// 添加逻辑
+	dbInfo.PackageName = m.GetPkgName()
+	dbInfo.DbName = m.GetDbName()
+	return dbInfo
+}
+
+// GetDbName get database name.获取数据库名字
+func (m *cnfModel) GetDbName() string {
+	dir := config.GetDbInfo().Host
+	dir = strings.Replace(dir, "\\", "/", -1)
+	if len(dir) > 0 {
+		if dir[len(dir)-1] == '/' {
+			dir = dir[:(len(dir) - 1)]
+		}
+	}
+	var dbName string
+	list := strings.Split(dir, "/")
+	if len(list) > 0 {
+		dbName = list[len(list)-1]
+	}
+	list = strings.Split(dbName, ".")
+	if len(list) > 0 {
+		dbName = list[0]
+	}
+
+	if len(dbName) == 0 || dbName == "." {
+		panic(fmt.Sprintf("%v : db host config err.must file dir", dbName))
+	}
+
+	return dbName
+}
+
+// GetPkgName package names through config outdir configuration.通过config outdir 配置获取包名
+func (m *cnfModel) GetPkgName() string {
+	dir := config.GetOutDir()
+	dir = strings.Replace(dir, "\\", "/", -1)
+	if len(dir) > 0 {
+		if dir[len(dir)-1] == '/' {
+			dir = dir[:(len(dir) - 1)]
+		}
+	}
+	var pkgName string
+	list := strings.Split(dir, "/")
+	if len(list) > 0 {
+		pkgName = list[len(list)-1]
+	}
+
+	if len(pkgName) == 0 || pkgName == "." {
+		list = strings.Split(tools.GetModelPath(), "/")
+		if len(list) > 0 {
+			pkgName = list[len(list)-1]
+		}
+	}
+
+	return pkgName
+}

data/view/model/genmysql/common.go

@@ -47,8 +47,8 @@ func fixForeignKey(list []genForeignKey, columuName string, result *[]model.Fore
 	}
 }
 
-// GetMysqlModel get model interface. 获取model接口
+// GetModel get model interface. 获取model接口
-func GetMysqlModel() model.IModel {
+func GetModel() model.IModel {
 	//now just support mysql
 	return &MySQLModel
 }

data/view/model/genmysql/genmysql.go

@@ -23,7 +23,7 @@ func (m *mysqlModel) GenModel() model.DBInfo {
 	defer orm.OnDestoryDB()
 
 	var dbInfo model.DBInfo
-	getPackageInfo(orm, &dbInfo)
+	m.getPackageInfo(orm, &dbInfo)
 	dbInfo.PackageName = m.GetPkgName()
 	dbInfo.DbName = m.GetDbName()
 	return dbInfo
@@ -31,7 +31,7 @@ func (m *mysqlModel) GenModel() model.DBInfo {
 
 // GetDbName get database name.获取数据库名字
 func (m *mysqlModel) GetDbName() string {
-	return config.GetMysqlDbInfo().Database
+	return config.GetDbInfo().Database
 }
 
 // GetPkgName package names through config outdir configuration.通过config outdir 配置获取包名
@@ -59,8 +59,8 @@ func (m *mysqlModel) GetPkgName() string {
 	return pkgName
 }
 
-func getPackageInfo(orm *mysqldb.MySqlDB, info *model.DBInfo) {
+func (m *mysqlModel) getPackageInfo(orm *mysqldb.MySqlDB, info *model.DBInfo) {
-	tabls := getTables(orm) // get table and notes
+	tabls := m.getTables(orm) // get table and notes
 	// if m := config.GetTableList(); len(m) > 0 {
 	// 	// 制定了表之后
 	// 	newTabls := make(map[string]string)
@@ -94,7 +94,7 @@ func getPackageInfo(orm *mysqldb.MySqlDB, info *model.DBInfo) {
 		}
 
 		// build element.构造元素
-		tab.Em = getTableElement(orm, tabName)
+		tab.Em = m.getTableElement(orm, tabName)
 		// --------end
 
 		info.TabList = append(info.TabList, tab)
@@ -106,7 +106,7 @@ func getPackageInfo(orm *mysqldb.MySqlDB, info *model.DBInfo) {
 }
 
 // getTableElement Get table columns and comments.获取表列及注释
-func getTableElement(orm *mysqldb.MySqlDB, tab string) (el []model.ColumnsInfo) {
+func (m *mysqlModel) getTableElement(orm *mysqldb.MySqlDB, tab string) (el []model.ColumnsInfo) {
 	keyNameCount := make(map[string]int)
 	KeyColumnMp := make(map[string][]keys)
 	// get keys
@@ -133,7 +133,7 @@ func getTableElement(orm *mysqldb.MySqlDB, tab string) (el []model.ColumnsInfo)
 	var foreignKeyList []genForeignKey
 	if config.GetIsForeignKey() {
 		sql := fmt.Sprintf(`select table_schema as table_schema,table_name as table_name,column_name as column_name,referenced_table_schema as referenced_table_schema,referenced_table_name as referenced_table_name,referenced_column_name as referenced_column_name
-		from INFORMATION_SCHEMA.KEY_COLUMN_USAGE where table_schema = '%v' AND REFERENCED_TABLE_NAME IS NOT NULL AND TABLE_NAME = '%v'`, config.GetMysqlDbInfo().Database, tab)
+		from INFORMATION_SCHEMA.KEY_COLUMN_USAGE where table_schema = '%v' AND REFERENCED_TABLE_NAME IS NOT NULL AND TABLE_NAME = '%v'`, m.GetDbName(), tab)
 		orm.Raw(sql).Scan(&foreignKeyList)
 	}
 	// ------------------end
@@ -189,7 +189,7 @@ func getTableElement(orm *mysqldb.MySqlDB, tab string) (el []model.ColumnsInfo)
 }
 
 // getTables Get columns and comments.获取表列及注释
-func getTables(orm *mysqldb.MySqlDB) map[string]string {
+func (m *mysqlModel) getTables(orm *mysqldb.MySqlDB) map[string]string {
 	tbDesc := make(map[string]string)
 
 	// Get column names.获取列名
@@ -212,7 +212,7 @@ func getTables(orm *mysqldb.MySqlDB) map[string]string {
 	rows.Close()
 
 	// Get table annotations.获取表注释
-	rows1, err := orm.Raw("SELECT TABLE_NAME,TABLE_COMMENT FROM information_schema.TABLES WHERE table_schema= '" + config.GetMysqlDbInfo().Database + "'").Rows()
+	rows1, err := orm.Raw("SELECT TABLE_NAME,TABLE_COMMENT FROM information_schema.TABLES WHERE table_schema= '" + m.GetDbName() + "'").Rows()
 	if err != nil {
 		if !config.GetIsGUI() {
 			fmt.Println(err)

data/view/model/gensqlite/common.go

@@ -0,0 +1,74 @@
+package gensqlite
+
+import (
+	"strings"
+
+	"github.com/xxjwxc/gormt/data/config"
+
+	"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.Name, "id") ||
+			strings.EqualFold(v.Name, "created_at") ||
+			strings.EqualFold(v.Name, "updated_at") ||
+			strings.EqualFold(v.Name, "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 &SQLiteModel
+}
+
+// FixElementNote 分析元素表注释
+func FixElementNote(em *model.ColumnsInfo, note string) {
+	matches := noteRegex.FindStringSubmatch(note)
+	if len(matches) < 2 {
+		em.Notes = note
+		return
+	}
+	em.Notes = note[len(matches[0]):]
+
+	list := strings.Split(matches[1], ";")
+	for _, v := range list {
+		tmp := strings.Split(v, ":")
+		if len(tmp) == 2 {
+			if strings.EqualFold(tmp[0], "default") { // 默认值
+				em.Default = tmp[1]
+			}
+		}
+	}
+}

data/view/model/gensqlite/def.go

@@ -0,0 +1,31 @@
+package gensqlite
+
+import "regexp"
+
+type keys struct {
+	NonUnique  int    `gorm:"column:Non_unique"`
+	KeyName    string `gorm:"column:Key_name"`
+	ColumnName string `gorm:"column:Column_name"`
+}
+
+// genColumns show full columns
+type genColumns struct {
+	Name    string `gorm:"column:name"`
+	Type    string `gorm:"column:type"`
+	Pk      int    `gorm:"column:pk"`
+	NotNull int    `gorm:"column:notnull"`
+}
+
+//select table_schema,table_name,column_name,referenced_table_schema,referenced_table_name,referenced_column_name from INFORMATION_SCHEMA.KEY_COLUMN_USAGE
+// where table_schema ='matrix' AND REFERENCED_TABLE_NAME IS NOT NULL AND TABLE_NAME = 'credit_card' ;
+// genForeignKey Foreign key of db info . 表的外键信息
+type genForeignKey struct {
+	TableSchema           string `gorm:"column:table_schema"`            // Database of columns.列所在的数据库
+	TableName             string `gorm:"column:table_name"`              // Data table of column.列所在的数据表
+	ColumnName            string `gorm:"column:column_name"`             // Column names.列名
+	ReferencedTableSchema string `gorm:"column:referenced_table_schema"` // The database where the index is located.该索引所在的数据库
+	ReferencedTableName   string `gorm:"column:referenced_table_name"`   // Affected tables . 该索引受影响的表
+	ReferencedColumnName  string `gorm:"column:referenced_column_name"`  // Which column of the affected table.该索引受影响的表的哪一列
+}
+
+var noteRegex = regexp.MustCompile(`^\[@gormt\s(\S+)+\]`)

data/view/model/gensqlite/gensqlite.go

@@ -0,0 +1,197 @@
+package gensqlite
+
+import (
+	"fmt"
+	"sort"
+	"strings"
+
+	"github.com/xxjwxc/public/mylog"
+
+	"github.com/jinzhu/gorm"
+	_ "github.com/mattn/go-sqlite3"
+	"github.com/xxjwxc/gormt/data/config"
+	"github.com/xxjwxc/gormt/data/view/model"
+	"github.com/xxjwxc/public/tools"
+)
+
+// SQLiteModel mysql model from IModel
+var SQLiteModel sqliteModel
+
+type sqliteModel struct {
+}
+
+// GenModel get model.DBInfo info.获取数据库相关属性
+func (m *sqliteModel) GenModel() model.DBInfo {
+	db, err := gorm.Open("sqlite3", config.GetDbInfo().Host)
+	if err != nil {
+		mylog.Error(err)
+		return model.DBInfo{}
+	}
+	defer db.Close()
+
+	var dbInfo model.DBInfo
+	m.getPackageInfo(db, &dbInfo)
+	dbInfo.PackageName = m.GetPkgName()
+	dbInfo.DbName = m.GetDbName()
+	return dbInfo
+}
+
+// GetDbName get database name.获取数据库名字
+func (m *sqliteModel) GetDbName() string {
+	dir := config.GetDbInfo().Host
+	dir = strings.Replace(dir, "\\", "/", -1)
+	if len(dir) > 0 {
+		if dir[len(dir)-1] == '/' {
+			dir = dir[:(len(dir) - 1)]
+		}
+	}
+	var dbName string
+	list := strings.Split(dir, "/")
+	if len(list) > 0 {
+		dbName = list[len(list)-1]
+	}
+	list = strings.Split(dbName, ".")
+	if len(list) > 0 {
+		dbName = list[0]
+	}
+
+	if len(dbName) == 0 || dbName == "." {
+		panic(fmt.Sprintf("%v : db host config err.must file dir", dbName))
+	}
+
+	return dbName
+}
+
+// GetPkgName package names through config outdir configuration.通过config outdir 配置获取包名
+func (m *sqliteModel) GetPkgName() string {
+	dir := config.GetOutDir()
+	dir = strings.Replace(dir, "\\", "/", -1)
+	if len(dir) > 0 {
+		if dir[len(dir)-1] == '/' {
+			dir = dir[:(len(dir) - 1)]
+		}
+	}
+	var pkgName string
+	list := strings.Split(dir, "/")
+	if len(list) > 0 {
+		pkgName = list[len(list)-1]
+	}
+
+	if len(pkgName) == 0 || pkgName == "." {
+		list = strings.Split(tools.GetModelPath(), "/")
+		if len(list) > 0 {
+			pkgName = list[len(list)-1]
+		}
+	}
+
+	return pkgName
+}
+
+func (m *sqliteModel) getPackageInfo(orm *gorm.DB, info *model.DBInfo) {
+	tabls := m.getTables(orm) // get table and notes
+	for tabName, notes := range tabls {
+		var tab model.TabInfo
+		tab.Name = tabName
+		tab.Notes = notes
+
+		if config.GetIsOutSQL() {
+			// Get create SQL statements.获取创建sql语句
+			rows, err := orm.Raw("SELECT tbl_name,sql FROM sqlite_master WHERE type='table' AND name = " + assemblyTable(tabName)).Rows()
+			//defer rows.Close()
+			if err == nil {
+				if rows.Next() {
+					var table, CreateTable string
+					rows.Scan(&table, &CreateTable)
+					tab.SQLBuildStr = CreateTable
+				}
+			}
+			rows.Close()
+			// ----------end
+		}
+
+		// build element.构造元素
+		tab.Em = m.getTableElement(orm, tabName)
+		// --------end
+
+		info.TabList = append(info.TabList, tab)
+	}
+	// sort tables
+	sort.Slice(info.TabList, func(i, j int) bool {
+		return info.TabList[i].Name < info.TabList[j].Name
+	})
+}
+
+// getTableElement Get table columns and comments.获取表列及注释
+func (m *sqliteModel) getTableElement(orm *gorm.DB, tab string) (el []model.ColumnsInfo) {
+	var list []genColumns
+	// Get table annotations.获取表注释
+	orm.Raw(fmt.Sprintf("PRAGMA table_info(%v)", assemblyTable(tab))).Scan(&list)
+	// filter gorm.Model.过滤 gorm.Model
+	if filterModel(&list) {
+		el = append(el, model.ColumnsInfo{
+			Type: "gorm.Model",
+		})
+	}
+	// -----------------end
+
+	// ForeignKey
+	var foreignKeyList []genForeignKey
+	if config.GetIsForeignKey() {
+	}
+	// ------------------end
+
+	for _, v := range list {
+		var tmp model.ColumnsInfo
+		tmp.Name = v.Name
+		tmp.Type = v.Type
+		FixElementNote(&tmp, "")
+		if v.Pk == 1 { // 主键
+			tmp.Index = append(tmp.Index, model.KList{
+				Key:   model.ColumnsKeyPrimary,
+				Multi: false,
+			})
+		}
+
+		tmp.IsNull = (v.NotNull != 1)
+
+		// ForeignKey
+		fixForeignKey(foreignKeyList, tmp.Name, &tmp.ForeignKeyList)
+		// -----------------end
+		el = append(el, tmp)
+	}
+	return
+}
+
+// getTables Get columns and comments.获取表列及注释
+func (m *sqliteModel) getTables(orm *gorm.DB) map[string]string {
+	tbDesc := make(map[string]string)
+
+	// Get column names.获取列名
+	var tables []string
+
+	rows, err := orm.Raw("SELECT name FROM sqlite_master WHERE type='table'").Rows()
+	if err != nil {
+		if !config.GetIsGUI() {
+			fmt.Println(err)
+		}
+		return tbDesc
+	}
+
+	for rows.Next() {
+		var table string
+		rows.Scan(&table)
+		if !strings.EqualFold(table, "sqlite_sequence") { // 剔除系统默认
+			tables = append(tables, table)
+			tbDesc[table] = ""
+		}
+	}
+	rows.Close()
+
+	// TODO.获取表注释
+
+	return tbDesc
+}
+
+func assemblyTable(name string) string {
+	return "'" + name + "'"
+}

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/jinzhu/gorm v1.9.12
 	github.com/jroimartin/gocui v0.4.0
 	github.com/leodido/go-urn v1.2.0 // indirect
+	github.com/mattn/go-sqlite3 v2.0.1+incompatible
 	github.com/nicksnyder/go-i18n/v2 v2.0.3
 	github.com/spf13/cobra v1.0.0
 	github.com/xxjwxc/public v0.0.0-20200605161710-72fc28675475
@@ -17,4 +18,4 @@ require (
 	gopkg.in/yaml.v3 v3.0.0-20191120175047-4206685974f2
 )
 
-// replace github.com/xxjwxc/public => ../public
+// replace github.com/xxjwxc/public => ../public

go.sum

@@ -207,6 +207,7 @@ github.com/xxjwxc/public v0.0.0-20200604090416-5afd146414d5 h1:8AYUDpnj279r9Gf2J
 github.com/xxjwxc/public v0.0.0-20200604090416-5afd146414d5/go.mod h1:0BFWVHqt7nKW8MtIx7R7bOkoGQFFnKsaJeeVbkzY88E=
 github.com/xxjwxc/public v0.0.0-20200605161710-72fc28675475 h1:VDny2T9+fQwq9WZkwrnD7hfANzX5lUOx53l31NPA9T8=
 github.com/xxjwxc/public v0.0.0-20200605161710-72fc28675475/go.mod h1:0BFWVHqt7nKW8MtIx7R7bOkoGQFFnKsaJeeVbkzY88E=
+github.com/xxjwxc/public v0.0.0-20200921104903-7ca9b2863090 h1:cmzvPQvZ94OAA6SY2JWvFr8BKJxFd6pa/TEC7ZZkATs=
 go.etcd.io/bbolt v1.3.2/go.mod h1:IbVyRI1SCnLcuJnV2u8VeU0CEYM7e686BmAb1XKL+uU=
 go.opencensus.io v0.20.1/go.mod h1:6WKK9ahsWS3RSO+PY9ZHZUfv2irvY6gN279GOPZjmmk=
 go.uber.org/atomic v1.4.0 h1:cxzIVoETapQEqDhQu3QfnvXAV4AlzcvUCxkVUFw3+EU=

sql.yml

@@ -0,0 +1,42 @@
+dbname: caoguo_db  # 数据库名
+comment: 潮果 # 数据库描述
+dbs: [mysql] # 数据库类型
+tables: # 数据库表
+    name: product_tbl # 表名
+    comment: 递增Id # 表注释
+    columns: # 表列信息
+        [
+            name: id , # 列名
+            comment: 递增Id,  # 列注释
+            isnull: false , # 是否为空
+            type: bigint(20) , # 类型
+            default: 10, # 默认值
+            klist:
+                [
+                    key:  ColumnsKeyPrimary,
+                    multi: true, # 是否多个(复合组建)
+                    keyname: test, 
+                ],
+            foreignkey: # 外键
+                [
+                    tablename: product_tbl, # 该索引受影响的表
+                    columnname: id # 该索引受影响的表的哪一列
+                ]
+        ],[
+            name: id , # 列名
+            comment: 递增Id,  # 列注释
+            isnull: false , # 是否为空
+            type: bigint(20) , # 类型
+            default: 10, # 默认值
+            klist:
+                [
+                    key:  ColumnsKeyPrimary,
+                    multi: true, # 是否多个(复合组建)
+                    keyname: test, 
+                ]
+            foreignkey: # 外键
+                [
+                    tablename: product_tbl, # 该索引受影响的表
+                    columnname: id # 该索引受影响的表的哪一列
+                ]
+        ]