分享两个常用的代码生成工具：

• gormgen
• handlergen

gormgen

基于 MySQL 数据表结构进行生成 3 个文件：

生成表的 struct 结构体

生成表的 Markdown 文档

生成表的 CURD 方法

场景

在进行业务需求开发时，创建完数据表后，执行代码生成工具，常用的 CURD 操作全部生成完毕，使用的时候只需要 . 后面的方法即可，这样大大提高了业务开发效率。

示例

表结构：

CREATETABLE`user_demo`(`id`int(11)unsignedNOTNULLAUTO_INCREMENTCOMMENT'主键',`user_name`varchar(32)NOTNULLDEFAULT''COMMENT'用户名',`nick_name`varchar(100)NOTNULLDEFAULT''COMMENT'昵称',`mobile`varchar(20)NOTNULLDEFAULT''COMMENT'手机号',`is_deleted`tinyint(1)NOTNULLDEFAULT'-1'COMMENT'是否删除1:是-1:否',`created_at`timestampNOTNULLDEFAULTCURRENT_TIMESTAMPCOMMENT'创建时间',`updated_at`timestampNOTNULLDEFAULTCURRENT_TIMESTAMPONUPDATECURRENT_TIMESTAMPCOMMENT'更新时间',PRIMARYKEY(`id`))ENGINE=InnoDBDEFAULTCHARSET=utf8mb4COMMENT='用户Demo表';

1、在 config 中设置需要自动生成的表，参数为 cmd.genTables，如果设置为空，表示生成当前数据库中的所有的表，如果设置多张表还可以用 “,” 分割。

[cmd]genTables = 'user_demo'

2、在根目录下执行脚本

./scripts/gormgen.sh

执行完毕后，会在 /internal/api/repository/db_repo 中生成 user_demo_repo 目录，同时也会生成 3 个文件：

• gen_model.go
• gen_table.md
• gen_user_demo.go

gen_model.go 内容如下：

packageuser_demo_repo import"time" //用户Demo表//go:generategormgen-structsUserDemo-input.typeUserDemostruct{Idint32//主键UserNamestring//用户名NickNamestring//昵称Mobilestring//手机号IsDeletedint32//是否删除1:是-1:否CreatedAttime.Time`gorm:"time"`//创建时间UpdatedAttime.Time`gorm:"time"`//更新时间}

gen_table.md Markdown 内容如下：

gen_user_demo.go 内容如下：

funcNewModel()*UserDemo{...} funcNewQueryBuilder()*userDemoRepoQueryBuilder{...} func(t*UserDemo)Create(db*gorm.DB)(idint32,errerror){...} func(t*UserDemo)Delete(db*gorm.DB)(errerror){...} func(t*UserDemo)Updates(db*gorm.DB,mmap[string]interface{})(errerror){...} typeuserDemoRepoQueryBuilderstruct{...} func(qb*userDemoRepoQueryBuilder)buildQuery(db*gorm.DB)*gorm.DB{...} func(qb*userDemoRepoQueryBuilder)Count(db*gorm.DB)(int64,error){...} func(qb*userDemoRepoQueryBuilder)First(db*gorm.DB)(*UserDemo,error){...} func(qb*userDemoRepoQueryBuilder)QueryOne(db*gorm.DB)(*UserDemo,error){...} func(qb*userDemoRepoQueryBuilder)QueryAll(db*gorm.DB)([]*UserDemo,error){...} func(qb*userDemoRepoQueryBuilder)Limit(limitint)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)Offset(offsetint)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)WhereId(pdb_repo.Predicate,valueint32)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)OrderById(ascbool)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)WhereUserName(pdb_repo.Predicate,valuestring)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)OrderByUserName(ascbool)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)WhereNickName(pdb_repo.Predicate,valuestring)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)OrderByNickName(ascbool)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)WhereMobile(pdb_repo.Predicate,valuestring)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)OrderByMobile(ascbool)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)WhereIsDeleted(pdb_repo.Predicate,valueint32)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)OrderByIsDeleted(ascbool)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)WhereCreatedAt(pdb_repo.Predicate,valuetime.Time)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)OrderByCreatedAt(ascbool)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)WhereUpdatedAt(pdb_repo.Predicate,valuetime.Time)*userDemoRepoQueryBuilder{...} func(qb*userDemoRepoQueryBuilder)OrderByUpdatedAt(ascbool)*userDemoRepoQueryBuilder{...}

使用

这样使用生成后的方法：

//查询示例：user_demo_repo.NewQueryBuilder().WhereUserName(db_repo.EqualPredicate,"tom").OrderById(true).QueryOne(u.db.GetDbR().WithContext(ctx.RequestContext()))

handlergen

基于定义的 Handler 文件中 type interface 中接口方法，进行生成文件。

场景

本次需求的研发负责人通过定义 type interface 的方式，定义出需要开发的方法，执行代码生成工具，每个方法的空实现都会生成在一个单独的文件中，开发人员只需去实现各自方法即可，便于进行分工和代码管理。

示例

比如 test_handler 中定义的 type interface 如下：

var_Handler=(*handler)(nil) typeHandlerinterface{//i为了避免被其他包实现i()//Create创建用户Create()core.HandlerFunc//Update编辑用户Update()core.HandlerFunc//Delete删除用户Delete()core.HandlerFunc//Detail用户详情Detail()core.HandlerFunc} typehandlerstruct{logger*zap.Loggercachecache.RepouserServiceuser_service.UserService} funcNew(logger*zap.Logger,dbdb.Repo,cachecache.Repo)Handler{return&handler{logger:logger,cache:cache,userService:user_service.NewUserService(db,cache),}} func(h*handler)i(){}

在根目录下执行脚本：

./scripts/handlergen.sh test_handler // test_handler 为需要生成 handler 的包名

执行完毕后，会在 /internal/api/controller/test_handler 中生成 4 个文件：

• func_create.go
• func_update.go
• func_detail.go
• func_delete.go

func_create.go 内容如下：

type createRequest struct{} type createResponse struct{} func (h *handler) Create() core.HandlerFunc { return func(c core.Context) { }}

其中 createRequest 为入参结构体，createResponse 为出参结构体。

func_update.go 内容如下：

type updateRequest struct{} type updateResponse struct{} func (h *handler) Update() core.HandlerFunc { return func(c core.Context) { }}

func_detail.go 内容如下：

type detailRequest struct{} type detailResponse struct{} func (h *handler) Detail() core.HandlerFunc { return func(c core.Context) { }}

func_delete.go 内容如下：

type deleteRequest struct{} type deleteResponse struct{} func (h *handler) Delete() core.HandlerFunc { return func(c core.Context) { }}

以上代码都在 go-gin-api 项目中，地址：https://github.com/xinliangnote/go-gin-api

到此这篇关于Go 代码生成工具的文章就介绍到这了,更多相关Go 代码生成工具内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持！