go语言 | 快速生成数据库表的 model 和 queryset

news2024/10/6 12:33:46

就是生成 model 目录的 xxx.go 和 xxx_gen.go 文件

使用的工具：
快速生成 model：gentool：https://github.com/go-gorm/gen/tree/master/tools/gentool
根据 model 生成 queryset：go-queryset：https://github.com/jirfag/go-queryset/

概叙

Golang 操作数据库表时，可能经常需要创建像下图中所示的与数据表相对应的 struct 对象，常被称为 model，有同学可能会说，model 也没几个字段，手写不就好了，没必要用工具来生成。当数据库表比较少，且表的字段少时用手敲确实可以完成，但是如果需要生成model的表数量比较多，每个model的字段也比较多，这时是用工具来生成可能是个更合适的选择，也更不容易出错。所以这里介绍一个能快速生成 model 文件和 queryset 的方法。

在这里插入图片描述
先安装下载 gentool 工具

go install gorm.io/gen/tools/gentool@latest

执行下面这一条命令即可生成与数据表相对应的 model 文件，命令的用法参考下面的“案例演示”。

gentool -dsn "user:pwd@tcp(127.0.0.1:3306)/database?charset=utf8mb4&parseTime=True&loc=Local" -tables "student"  -onlyModel

案例演示

1、生成 model 文件
比如我想给student表生成 model ，student 表有下面这些字段。

CREATE TABLE `student`(
    `id` BIGINT(20) UNSIGNED not NULL AUTO_INCREMENT COMMENT '自增id',
    `school_id` BIGINT(20) NOT NULL DEFAULT '0' COMMENT '学号',
                `name` VARCHAR(64) NOT NULL DEFAULT '0' COMMENT '姓名',
                `male` TINYINT(4) NOT NULL DEFAULT '0' COMMENT '性别',
                `phone_number` BIGINT(20) NOT NULL DEFAULT '0' COMMENT '手机号',
                `addr` VARCHAR(255) NOT NULL DEFAULT '' COMMENT '家庭住址',
                `class` VARCHAR(255) NOT NULL DEFAULT '0' COMMENT '班级',
                `major` INT NOT NULL DEFAULT '0' COMMENT '专业id',
                `grade` INT NOT NULL DEFAULT '0' COMMENT '入学年份/年级',
    `extra` text NOT NULL DEFAULT '' COMMENT '其他',
    `create_time` TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
    `modify_time` TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',
    PRIMARY KEY (`id`)
) ENGINE=INNODB DEFAULT CHARSET=utf8 COMMENT='学生基本信息表';

使用如下命令生成model，model文件会生成在当前路径上一级的名为model的文件夹中，比如在demo/aaa下执行上面的命令，那么 model 会生成在 demo/model 路径下，如果 model 路径不存在会自动创建。

gentool -dsn "root:123456@tcp(127.0.0.1:3306)/my_test_db?charset=utf8mb4&parseTime=True&loc=Local" -tables "student"  -onlyModel

命令执行完毕后，会发现在 demo 路径下生成了一个名为 model 的文件夹，文件夹里有个 student.gen.go 文件，这个就是我们刚才生成的 model 文件了

在这里插入图片描述
如果想指定 model 文件的路径，需要添加 -modelPkgName 参数，完整命令如下所示，执行完毕后，model 文件将生成在 aaa 路径下，需要注意 -modelPkgName 的路径需要是绝对路径，不能是相对路径。

gentool -dsn "root:123456@tcp(127.0.0.1:3306)/my_test_db?charset=utf8mb4&parseTime=True&loc=Local" -tables "student"  -onlyModel -modelPkgName="/Users/luochunhui/go/src/demo/aaa"

修改文件名

我们一般习惯于把 student.gen.go这个model 文件中的 ‘.gen’ 字符串去除，也就是重命名为 student.go

model 文件添加必要注释

接下来需要用 model 文件生成 queryset 文件，也就是我们经常看到的以 ‘_gen.go’ 结尾的文件，生成 queryset 文件前，需要在 model 文件中添加 2 行注释，虽然是注释，但是这两行注释必不可少。

student.go 是当前model 文件的文件名，student_gen.go 是将要生成的 queryset 文件的文件名

//go:generate goqueryset -in student.go -out student_gen.go

下面这行日志必须放在model struct正上方第一行。

// gen:qs

安装 goqueryset

执行下面的命令进行 goqueryset 工具安装

go get -u github.com/jirfag/go-queryset/cmd/goqueryset
# 如果上面的 go get 命令安装 goqueryset 失败，可以尝试下面这条命令，如果成功了就不用尝试了
go install github.com/jirfag/go-queryset/cmd/goqueryset@latest

生成 queryset 文件

注释添加完毕后，进入到 model 文件所在路径，执行下面的命令来生成 queryset 文件。

go generate ./...

执行完毕可以看到与 model 文件同级目录下生成了一个 student_gen.go 文件，这个文件就是 queryset 文件。

最后

在生成的queryset 可以看到 github.com/jinzhu/gorm这个依赖还没安装，执行下面的命令拉取即可，拉取完毕应该就不报错了。

go get github.com/jinzhu/gorm
go mod tidy

在这里插入图片描述

@ sqllite方式

脚本

创建makefile

# 命令
gen:
	gentool -dsn "root:123456@tcp(192.168.1.13:3306)/bk?charset=utf8mb4&parseTime=True&loc=Local" -outPath "./bkmodel/dao/query"

make gen运行展示

root@tdev:/home/code/tapi# make gen
gentool -dsn "root:123456@tcp(192.168.1.13:3306)/bk?charset=utf8mb4&parseTime=True&loc=Local" -outPath "./bkmodel/dao/query"
2023/02/11 07:19:53 got 6 columns from table <user>
2023/02/11 07:19:53 Start generating code.
2023/02/11 07:19:53 generate model file(table <user> -> {model.User}): /home/code/tapi/bkmodel/dao/model/user.gen.go
2023/02/11 07:19:53 generate query file: /home/code/tapi/bkmodel/dao/query/user.gen.go
2023/02/11 07:19:53 generate query file: /home/code/tapi/bkmodel/dao/query/gen.go
2023/02/11 07:19:53 Generate code done.

时区有问题改一下

sudo timedatectl set-timezone Asia/Shanghai

// 检查
root@tdev:/home/code/tapi# sudo timedatectl set-timezone Asia/Shanghai
root@tdev:/home/code/tapi# timedatectl
               Local time: Sat 2023-02-11 15:31:07 CST
           Universal time: Sat 2023-02-11 07:31:07 UTC
                 RTC time: Sat 2023-02-11 07:31:07
                Time zone: Asia/Shanghai (CST, +0800)
System clock synchronized: yes
              NTP service: active
          RTC in local TZ: no