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

下面是详细操作步骤。

概述

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
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='学生基本信息表';
modeldemo/aaademo/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 文件了

-modelPkgName-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" 

2、修改文件名

student.gen.gostudent.go

3、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

4、安装 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

5、生成 queryset 文件

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

go generate ./...

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

6、最后

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