go的数据库模型生成器

Wesley13
• 阅读 648

好冷,早知道不写GO了

嗯,就是开个玩笑,冬天有点冷,特别是寒潮来了,各位注意保暖。

为什么写这个生成器

最近要写GO项目,然后发现orm着实难用,一个model要去手动写,更坑的是,`号里面的内容,没有自动打印。天好冷吗,手好抖,南方的冬天,你懂的。

像JAVA、PHP等语言,都有成熟的模型生成器,然而Go我并没有找到,可能是我没有和百度(当然还有墙外的哥)达成深度合作吧?为此,懒人李只能造个轮子,为了提高效率(ps:就是想偷懒、摸鱼)。

过程分析

那我们要将数据库如何转换成go的代码呢?开始我想的是,直接拿create sql进行解析

比如表cate

CREATE TABLE `cate` (
  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
  `name` varchar(50) NOT NULL DEFAULT '',
  `create_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
  `update_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

emm,正则写起来好像有点复杂(ps:好吧,我真不擅长正则,别拆穿,我知道你们有大佬能写出来)

后面遨游各大搜索引擎,终于找到了便捷的方式。

SELECT COLUMN_NAME,DATA_TYPE,IS_NULLABLE,TABLE_NAME,COLUMN_COMMENT
    FROM information_schema.COLUMNS 
    WHERE table_schema = DATABASE() and table_name='cate'

哇,看结果: go的数据库模型生成器

我想要的东西好像都出来了,神奇动物原来在这里(ps:他还好没在车底,阿杜对不起)

结构定义

既然基础结构有了,那不如就直接定义结构把。

type column struct {
    ColumnName    string  //字段名
    Type          string   //字段类型
    Nullable      string  //允许为空
    TableName     string  //表名
    ColumnComment string  //字段备注
    Tag           string    //go渲染用的tag
}

然后分析下,mysql的类型和go的不一样啊,这时候搞个mapping就好了。

var TypeMappingMysqlToGo = map[string]string{
    "int":                "int",
    "double":             "float64",
    "varbinary":          "string",
    // 好多啊
}

然后的套路就是内容的生成了

无非是下划线转驼峰,首字符提升大写等等。

prefix, ok := convert.TablePrefix[tableRealName]
        if ok {
            tableRealName = tableRealName[len(prefix):]
        }
        tableName := tableRealName

        switch len(tableName) {
        case 0:
            continue
        case 1:
            tableName = strings.ToUpper(tableName[0:1])
        default:
            tableName = strings.ToUpper(tableName[0:1]) + tableName[1:]
        }
        depth := 1
        var content string
        content += "package " + convert.PackageName + "\n\n" //写包名
        content += "type " + tableName + " struct {\n"
        columns, ok := convert.TableColumn[tableRealName]
        for _, v := range columns {
            var comment string
            if v.ColumnComment != "" {
                comment = fmt.Sprintf(" // %s", v.ColumnComment)
            }
            content += fmt.Sprintf("%s%s %s %s%s\n",
                Tab(depth), v.GetGoColumn(prefix, true), v.GetGoType(), v.GetTag("orm"), comment)
        }
        content += Tab(depth-1) + "}\n\n"

        content += fmt.Sprintf("func (%s *%s) %s() string {\n",
            LcFirst(tableName), tableName, "GetTableName")
        content += fmt.Sprintf("%sreturn \"%s\"\n",
            Tab(depth), tableRealName)
        content += "}\n\n"
        convert.writeModel(tableRealName, content) //写文件

当当当当,华丽转身

使用demo

代码,哟西。https://github.com/go-libraries/model/blob/master/mysql_test.go

    Mysql := GetMysqlToGo()
    Mysql.SetDsn(dsn)
    Mysql.SetModelPath("/tmp")
    Mysql.GetTables()
    Mysql.ReadTablesColumns()
    Mysql.SetPackageName("models")
    Mysql.Run()

执行结果:

ll /tmp


total *
-rw-r--r--  1 limars  wheel  297 12 18 17:59 cate.go
-rw-r--r--  1 limars  wheel  597 12 18 17:59 comment.go
-rw-r--r--  1 limars  wheel  826 12 18 17:59 content.go
......


cat cate.go


package models

type Cate struct {
    Id         int    `orm:"id" json:"id"`
    Name       string `orm:"name" json:"name"`
    CreateTime string `orm:"create_time" json:"create_time"`
    UpdateTime string `orm:"update_time" json:"update_time"`
}

func (cate *Cate) GetTableName() string {
    return "cate"
}

广告广告

诶,像我这么不出名的人,当然是打个人广告了。

给我来个STAR吧,我会写pgsql和Mariadb的。

也会逐步支持bee、gin等框架哒

https://github.com/go-libraries/gmodel

好冷,早知道不写GO了,真的冷。

点赞
收藏
评论区
推荐文章
blmius blmius
3年前
MySQL:[Err] 1292 - Incorrect datetime value: ‘0000-00-00 00:00:00‘ for column ‘CREATE_TIME‘ at row 1
文章目录问题用navicat导入数据时,报错:原因这是因为当前的MySQL不支持datetime为0的情况。解决修改sql\mode:sql\mode:SQLMode定义了MySQL应支持的SQL语法、数据校验等,这样可以更容易地在不同的环境中使用MySQL。全局s
Karen110 Karen110
3年前
一篇文章带你了解JavaScript日期
日期对象允许您使用日期(年、月、日、小时、分钟、秒和毫秒)。一、JavaScript的日期格式一个JavaScript日期可以写为一个字符串:ThuFeb02201909:59:51GMT0800(中国标准时间)或者是一个数字:1486000791164写数字的日期,指定的毫秒数自1970年1月1日00:00:00到现在。1\.显示日期使用
皕杰报表之UUID
​在我们用皕杰报表工具设计填报报表时,如何在新增行里自动增加id呢?能新增整数排序id吗?目前可以在新增行里自动增加id,但只能用uuid函数增加UUID编码,不能新增整数排序id。uuid函数说明:获取一个UUID,可以在填报表中用来创建数据ID语法:uuid()或uuid(sep)参数说明:sep布尔值,生成的uuid中是否包含分隔符'',缺省为
待兔 待兔
3个月前
手写Java HashMap源码
HashMap的使用教程HashMap的使用教程HashMap的使用教程HashMap的使用教程HashMap的使用教程22
Jacquelyn38 Jacquelyn38
3年前
2020年前端实用代码段,为你的工作保驾护航
有空的时候,自己总结了几个代码段,在开发中也经常使用,谢谢。1、使用解构获取json数据let jsonData  id: 1,status: "OK",data: 'a', 'b';let  id, status, data: number   jsonData;console.log(id, status, number )
Wesley13 Wesley13
3年前
00:Java简单了解
浅谈Java之概述Java是SUN(StanfordUniversityNetwork),斯坦福大学网络公司)1995年推出的一门高级编程语言。Java是一种面向Internet的编程语言。随着Java技术在web方面的不断成熟,已经成为Web应用程序的首选开发语言。Java是简单易学,完全面向对象,安全可靠,与平台无关的编程语言。
Stella981 Stella981
3年前
Django中Admin中的一些参数配置
设置在列表中显示的字段,id为django模型默认的主键list_display('id','name','sex','profession','email','qq','phone','status','create_time')设置在列表可编辑字段list_editable
Wesley13 Wesley13
3年前
MySQL部分从库上面因为大量的临时表tmp_table造成慢查询
背景描述Time:20190124T00:08:14.70572408:00User@Host:@Id:Schema:sentrymetaLast_errno:0Killed:0Query_time:0.315758Lock_
为什么mysql不推荐使用雪花ID作为主键
作者:毛辰飞背景在mysql中设计表的时候,mysql官方推荐不要使用uuid或者不连续不重复的雪花id(long形且唯一),而是推荐连续自增的主键id,官方的推荐是auto_increment,那么为什么不建议采用uuid,使用uuid究
Python进阶者 Python进阶者
9个月前
Excel中这日期老是出来00:00:00,怎么用Pandas把这个去除
大家好,我是皮皮。一、前言前几天在Python白银交流群【上海新年人】问了一个Pandas数据筛选的问题。问题如下:这日期老是出来00:00:00,怎么把这个去除。二、实现过程后来【论草莓如何成为冻干莓】给了一个思路和代码如下:pd.toexcel之前把这