golang 生成对应的数据表struct定义操作


Posted in Golang onApril 28, 2021

在开发过程中,常常需要将数据库表对应到golang的一个struct,特别是使用一些ORM工具,sqlx库等,我是个懒人,即使数据表的字段不多,我也懒得去一个个对应的敲入代码,更别提数据表字段比较多的情况了,码农的时间,不能浪费在这啊,对吧?所以我在想,是不是有办法可以自动生成。

我在工作时,用得最多的是mysql了,因此

本文针对mysql的数据表来自动生成golang 的struct定义

mysql有个自带的数据库information_schema,里面的信息量比较多,朋友们可以去百度下,我这里用到了表COLUMNS,它的字段包含数据库名、表名、字段名、字段类型等,利用这个表的数据,把对应的表的字段信息读取出来,然后再根据golang的语法规则,生成文件就可以了。

大致思路确定了,开始动手。

我采用sqlx进行数据库的访问,首先定义一个struct来表示COLUMNS的数据,这里我只需要几个字段,因此,没有把表COLUMNS的所有字段都对应到struct:

type FieldInfo struct {
 ColName    string `db:"COLUMN_NAME"`
 DataType   string `db:"DATA_TYPE"`
 ColComment string `db:"COLUMN_COMMENT"`
 IsNullable string `db:"IS_NULLABLE"`
}

需要指定生成的struct对应的是哪个库、哪个表,最终的golang文件保存地址

这里利用命令行参数来输入:

var dbname= flag.String("db", "", "the database name")
var tblname = flag.String("tbl", "", "the table name to export")
var savepath = flag.String("path", "./", "the path to save file")

另外,我们项目习惯使用下划线“_”来分割单词,比如info_user,表示user表,而生成的struct名称为InfoUser,字段名也是类似规则

因此定义了如下函数来处理这种情况:

func fmtFieldDefine(src string) string {
 temp := strings.Split(src, "_") // 有下划线的,需要拆分
 var str string
 for i := 0; i < len(temp); i++ {
  b := []rune(temp[i])
  for j := 0; j < len(b); j++ {
   if j == 0 {
    // 首字母大写转换
    b[j] -= 32
    str += string(b[j])
   } else {
    str += string(b[j])
   }
  }
 } 
 return str
}

即把下划线去掉,且将单词的首字母改为大写。

有些字段,在设计数据库时,是可空的,information_schema->COLUMNS中有个字段IS_NULLABLE专门表示,而golang的sql有几个类型对应:sql.NullString、sql.NullBool、sql.NullFloat64、sql.NullInt64,基本上是可以满足使用要求的了。

有人可能会有疑问,假如字段类型为date、timestamp等,该对应哪种呢?通常第三方的类库会转为string类型,那么就对应sql.NullString好了。不过我这里没有进行这方面的处理。

前期工作做好了,开始编码:

func main() {
 flag.Parse()
 fmt.Println("table name -->", *tblname) 
 dns := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8", dbuser, dbpwd, dbhost, "information_schema") 
 db := sqlx.MustConnect("mysql", dns)
 
 var fs []FieldInfo
 err := db.Select(&fs, "SELECT COLUMN_NAME, DATA_TYPE, COLUMN_COMMENT, IS_NULLABLE FROM COLUMNS WHERE TABLE_NAME=? and table_schema=?", *tblname, *dbname)
 if err != nil {
  fmt.Println(err)
  panic(err)
 }
 
 if len(fs) > 0 {
  var buffer bytes.Buffer
  buffer.WriteString("package models\n")
  buffer.WriteString("type " + fmtFieldDefine(*tblname) + " struct {\n")
  for _, v := range fs {
   buffer.WriteString("" + fmtFieldDefine(v.ColName) + " ")
   switch v.DataType {
   case "int", "tinyint", "smallint":
    if v.IsNullable == "YES" {
     buffer.WriteString("sql.NullInt64 ")
    } else {
     buffer.WriteString("int ")
    }
   case "bigint":
    if v.IsNullable == "YES" {
     buffer.WriteString("sql.NullInt64 ")
    } else {
     buffer.WriteString("int64 ")
    }
   case "char", "varchar", "longtext", "text", "tinytext":
    if v.IsNullable == "YES" {
     buffer.WriteString("sql.NullString ")
    } else {
     buffer.WriteString("string ")
    }
   case "date", "datetime", "timestamp":
    buffer.WriteString("time.Time ")
   case "double", "float":
    if v.IsNullable == "YES" {
     buffer.WriteString("sql.NullFloat64 ")
    } else {
     buffer.WriteString("float64 ")
    }
   default:
    // 其他类型当成string处理
    if v.IsNullable == "YES" {
     buffer.WriteString("sql.NullString ")
    } else {
     buffer.WriteString("string ")
    }
   }
 
   buffer.WriteString(fmt.Sprintf("`db:\"%s\" json:\"%s\"`\n", v.ColName, v.ColName))
 
  }
  buffer.WriteString(`}`) 
  fmt.Println(buffer.String()) 
  filename := *savepath + "\\" + *tblname + ".go"
  f, _ := os.Create(filename)
  f.Write([]byte(buffer.String()))
  f.Close()
 
  cmd := exec.Command("goimports", "-w", filename)
  cmd.Run()
 } else {
  fmt.Println("查询不到数据")
 }
}

我把每个字段的tag,包括db和json的都加了了,在代码最后,使用goimport工具添加需要import的package,它连format的工作都做了,实在不错。

以下是我生成的一个用户购物概要表的struct定义:

package models 
import (
 "database/sql"
 "time"
)
 
type InfoUserShoppingSummary struct {
 Id            int            `db:"id" json:"id"`
 TransactionId sql.NullString `db:"transaction_id" json:"transaction_id"`
 OutTradeNo    sql.NullString `db:"out_trade_no" json:"out_trade_no"`
 WuId          int            `db:"wu_id" json:"wu_id"`
 WdId          int            `db:"wd_id" json:"wd_id"`
 TotalFee      float64        `db:"total_fee" json:"total_fee"`
 PayStaus      sql.NullInt64  `db:"pay_staus" json:"pay_staus"`
 CreateTime    time.Time      `db:"create_time" json:"create_time"`
 UpdateTime    time.Time      `db:"update_time" json:"update_time"`
 Address       sql.NullString `db:"address" json:"address"`
}

补充:Golang之方法(自定义类型,struct)

方法的使用,请看本天师的代码

//Golang的方法定义
//Golang中的方法是作用在特定类型的变量上,因此自定义类型,都可以有方法,不仅仅是struct
//定义:func (recevier type) methodName(参数列表)(返回值列表){}
//方法和函数的区别
/*
1,函数调用:function(variable,参数列表)
2, 方法,variable.function(参数列表)
方法的控制,通过大小写空格控制
 */
package main
//Golang的方法定义
//Golang中的方法是作用在特定类型的变量上,因此自定义类型,都可以有方法,不仅仅是struct
//定义:func (recevier type) methodName(参数列表)(返回值列表){}
import "fmt"
type integer int
func (p integer) print() {
    fmt.Println("p is:", p)
}
//这里传递的是副本,想改变p的值,需要传递指针
func (p *integer) set(b integer) {
    *p = b
}
type Student struct {
    Name  string
    Age   int
    Score int
    sex   int
}
//这里需要接受指针 *Student(接收者),否则修改不了值
func (p *Student) init(name string, age int, score int) {
    p.Name = name
    p.Age = age
    p.Score = score
    fmt.Println(p)
}
func (p Student) get() Student {
    return p
}
func main() {
    var stu Student
    //修改地址的写法(&stu).init
    //但是go可以自动知道,接受者是指针,这里stu就传递地址
    stu.init("stu", 18, 99)
    stu1 := stu.get()
    fmt.Println(stu1)
    //type integer方法
    var a integer
    a = 100
    a.print()
    a.set(1000)
    a.print()
}

以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。如有错误或未考虑完全的地方,望不吝赐教。

Golang 相关文章推荐
Golang二维切片初始化的实现
Apr 08 Golang
go 原生http web 服务跨域restful api的写法介绍
Apr 27 Golang
golang 生成对应的数据表struct定义操作
Apr 28 Golang
解决go在函数退出后子协程的退出问题
Apr 30 Golang
golang gopm get -g -v 无法获取第三方库的解决方案
May 05 Golang
Go 在 MongoDB 中常用查询与修改的操作
May 07 Golang
go使用Gin框架利用阿里云实现短信验证码功能
Aug 04 Golang
如何解决goland,idea全局搜索快捷键失效问题
Apr 03 Golang
golang定时器
Apr 14 Golang
Golang 实现WebSockets
Apr 24 Golang
Go gRPC进阶教程gRPC转换HTTP
Jun 16 Golang
基于Python实现西西成语接龙小助手
Aug 05 Golang
golang 如何通过反射创建新对象
Apr 28 #Golang
golang 实现两个结构体复制字段
Apr 28 #Golang
go结构体嵌套的切片数组操作
Apr 28 #Golang
golang json数组拼接的实例
Apr 28 #Golang
golang 实现对Map进行键值自定义排序
Apr 28 #Golang
go语言中json数据的读取和写出操作
Apr 28 #Golang
golang 实现菜单树的生成方式
Apr 28 #Golang
You might like
PHP 中dirname(_file_)讲解
2007/03/18 PHP
php学习 函数 课件
2008/06/15 PHP
常见的PHP五种设计模式小结
2011/03/23 PHP
仿Aspnetpager的一个PHP分页类代码 附源码下载
2012/10/08 PHP
利用“多说”制作留言板、评论系统
2015/07/14 PHP
php+ajax实现无刷新数据分页的办法
2015/11/02 PHP
PHP QRCODE生成彩色二维码的方法
2016/05/19 PHP
IE8 下的Js错误HTML Parsing Error...
2009/08/14 Javascript
js获取单元格自定义属性值的代码(IE/Firefox)
2010/04/05 Javascript
js中更短的 Array 类型转换
2011/10/30 Javascript
JavaScript数字和字符串转换示例
2014/03/26 Javascript
Javascript iframe交互并兼容各种浏览器的解决方法
2016/07/12 Javascript
微信小程序 常用工具类详解及实例
2017/02/15 Javascript
angular使用post、get向后台传参的问题实例
2017/05/27 Javascript
Vue代码整洁之去重方法整理
2019/08/06 Javascript
jQuery实现判断滚动条滚动到document底部的方法分析
2019/08/27 jQuery
vue-router 中 meta的用法详解
2019/11/01 Javascript
jQuery实现放大镜案例
2020/10/19 jQuery
vue 判断两个时间插件结束时间必选大于开始时间的代码
2020/11/04 Javascript
[01:27:30]LGD vs Newbee 2019国际邀请赛小组赛 BO2 第二场 8.16
2019/08/19 DOTA
Python操作MongoDB数据库PyMongo库使用方法
2015/04/27 Python
Python抽象类的新写法
2015/06/18 Python
利用Django提供的ModelForm增删改数据的方法
2019/01/06 Python
python实现切割url得到域名、协议、主机名等各个字段的例子
2019/07/25 Python
html+css3实现的登录界面
2020/12/09 HTML / CSS
基于 HTML5 WebGL 实现的医疗物流系统
2019/10/08 HTML / CSS
Bugatchi官方网站:男士服装在线
2019/04/10 全球购物
Saks Fifth Avenue澳洲/亚太地区:萨克斯第五大道精品百货店
2019/06/09 全球购物
幼儿园英语教学反思
2014/01/30 职场文书
市场营销专业毕业生求职信
2014/03/26 职场文书
优秀护士先进事迹
2014/05/08 职场文书
企业挂职心得体会
2014/09/10 职场文书
部队反四风对照检查材料
2014/09/26 职场文书
办公室年度工作总结2015
2015/05/21 职场文书
大学毕业典礼致辞
2015/07/29 职场文书
JVM入门之类加载与字节码技术(类加载与类的加载器)
2021/06/15 Java/Android