好冷,早知道不写GO了
嗯,就是开个玩笑,冬天有点冷,特别是寒潮来了,各位注意保暖。
为什么写这个生成器
最近要写GO项目,然后发现orm着实难用,一个model要去手动写,更坑的是,`号里面的内容,没有自动打印。天好冷吗,手好抖,南方的冬天,你懂的。
像JAVA、PHP等语言,都有成熟的模型生成器,然而Go我并没有找到,可能是我没有和百度(当然还有墙外的哥)达成深度合作吧?为此,懒人李只能造个轮子,为了提高效率(ps:就是想偷懒、摸鱼)。
过程分析
那我们要将数据库如何转换成go的代码呢?开始我想的是,直接拿create sql进行解析
比如表cate
1CREATE TABLE `cate` ( 2 `id` int(11) unsigned NOT NULL AUTO_INCREMENT, 3 `name` varchar(50) NOT NULL DEFAULT '', 4 `create_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP, 5 `update_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, 6 PRIMARY KEY (`id`) 7) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
emm,正则写起来好像有点复杂(ps:好吧,我真不擅长正则,别拆穿,我知道你们有大佬能写出来)
后面遨游各大搜索引擎,终于找到了便捷的方式。
1SELECT COLUMN_NAME,DATA_TYPE,IS_NULLABLE,TABLE_NAME,COLUMN_COMMENT 2 FROM information_schema.COLUMNS 3 WHERE table_schema = DATABASE() and table_name='cate'
哇,看结果: 
我想要的东西好像都出来了,神奇动物原来在这里(ps:他还好没在车底,阿杜对不起)
结构定义
既然基础结构有了,那不如就直接定义结构把。
1type column struct { 2 ColumnName string //字段名 3 Type string //字段类型 4 Nullable string //允许为空 5 TableName string //表名 6 ColumnComment string //字段备注 7 Tag string //go渲染用的tag 8}
然后分析下,mysql的类型和go的不一样啊,这时候搞个mapping就好了。
1var TypeMappingMysqlToGo = map[string]string{ 2 "int": "int", 3 "double": "float64", 4 "varbinary": "string", 5 // 好多啊 6}
然后的套路就是内容的生成了
无非是下划线转驼峰,首字符提升大写等等。
1prefix, ok := convert.TablePrefix[tableRealName] 2 if ok { 3 tableRealName = tableRealName[len(prefix):] 4 } 5 tableName := tableRealName 6 7 switch len(tableName) { 8 case 0: 9 continue 10 case 1: 11 tableName = strings.ToUpper(tableName[0:1]) 12 default: 13 tableName = strings.ToUpper(tableName[0:1]) + tableName[1:] 14 } 15 depth := 1 16 var content string 17 content += "package " + convert.PackageName + "\n\n" //写包名 18 content += "type " + tableName + " struct {\n" 19 columns, ok := convert.TableColumn[tableRealName] 20 for _, v := range columns { 21 var comment string 22 if v.ColumnComment != "" { 23 comment = fmt.Sprintf(" // %s", v.ColumnComment) 24 } 25 content += fmt.Sprintf("%s%s %s %s%s\n", 26 Tab(depth), v.GetGoColumn(prefix, true), v.GetGoType(), v.GetTag("orm"), comment) 27 } 28 content += Tab(depth-1) + "}\n\n" 29 30 content += fmt.Sprintf("func (%s *%s) %s() string {\n", 31 LcFirst(tableName), tableName, "GetTableName") 32 content += fmt.Sprintf("%sreturn \"%s\"\n", 33 Tab(depth), tableRealName) 34 content += "}\n\n" 35 convert.writeModel(tableRealName, content) //写文件
当当当当,华丽转身
使用demo
代码,哟西。https://github.com/go-libraries/model/blob/master/mysql_test.go
1 Mysql := GetMysqlToGo() 2 Mysql.SetDsn(dsn) 3 Mysql.SetModelPath("/tmp") 4 Mysql.GetTables() 5 Mysql.ReadTablesColumns() 6 Mysql.SetPackageName("models") 7 Mysql.Run()
执行结果:
1ll /tmp 2 3 4total * 5-rw-r--r-- 1 limars wheel 297 12 18 17:59 cate.go 6-rw-r--r-- 1 limars wheel 597 12 18 17:59 comment.go 7-rw-r--r-- 1 limars wheel 826 12 18 17:59 content.go 8...... 9 10 11cat cate.go 12 13 14package models 15 16type Cate struct { 17 Id int `orm:"id" json:"id"` 18 Name string `orm:"name" json:"name"` 19 CreateTime string `orm:"create_time" json:"create_time"` 20 UpdateTime string `orm:"update_time" json:"update_time"` 21} 22 23func (cate *Cate) GetTableName() string { 24 return "cate" 25}
广告广告
诶,像我这么不出名的人,当然是打个人广告了。
给我来个STAR吧,我会写pgsql和Mariadb的。
也会逐步支持bee、gin等框架哒
https://github.com/go-libraries/gmodel
好冷,早知道不写GO了,真的冷。