前言

上一篇文章带你实现了Go-Zero和goctl：解锁微服务开发的神器，快速上手指南，本文将继续深入探讨Go-Zero的强大之处，并介绍如何使用goctl工具实现模板定制化，并根据实际项目业务需求进行模板定制化实现。

通过本文的教程，你能够亲自实践并完成goctl模板的定制化，进一步提升你的Go-Zero开发技能。

概述

goctl 代码生成是基于 go 的模板去实现数据驱动的，默认情况会选择内存中的模板进行生成，当开发需要修改模板时，就需要定制化模板，goctl为我们实现了这一功能。

实战前准备

首先需要你在本地安装goctl、protoc、go-zero，下载教学和地址点击这里，按照教程操作即可，非常简单。

下面按顺序和我操作吧，对整体开发流程不清楚的同学务必先看我前篇文章：GoZero的开发技巧 & 整体开发流程

本文重在实战，如果对goctl毫不了解的话，建议先看我前一篇文章：Go-Zero和goctl：解锁微服务开发的神器，快速上手指南

以下均以我的商业项目举例，应该对你有启发：

（后面我会把商业项目脱敏开源出来，欢迎关注我）

数据表生成Model方法脚本

首先在deploy下新增script目录，结构如下图所示。

脚本内容如下：

#!/usr/bin/env bash

# 使用方法：
# ./genModel.sh lottery lottery
# ./genModel.sh lottery prize

模板定制化使用方法

相关命令使用详情，参考：官网文档
具体使用方法网上有很多文章介绍，官网也有详细步骤。这里更加注重商业项目对于模板定制化的实战，对相关操作不进行赘述，快速过一遍流程即可。

初始化模板到本地

依据前文所介绍的项目目录结构，我们将自定义模板放在deploy下面即可，并且采用的版本号为1.6.1（目录路径根据自己实际情况修改）

goctl template init --home $HOME/Desktop/lottery-backend/deploy/goctl/1.6.1

注意：如果不指定–home 他会初始化到$HOME/.goctl

这样就生成好自己版本的goctl模板啦，可以根据自己的实际需求进行模板的修改。

接下来分享我们项目中关于自定义goctl的实战。

自定义goctl实战

实战1：Model层方法定制化

很多时候我们需要对数据进行分页查询。这个方法是一个通用的方法，可以在很多地方复用，所以放入模板去生成，这样可以减少重复代码，提高开发效率。

步骤一：在model/update.tpl下面新增一个方法FindPageListByPage

方法具体实现如下

func (m *default{{.upperStartCamelObject}}Model) FindPageListByPage(ctx context.Context,builder squirrel.SelectBuilder,page ,pageSize int64,orderBy string) ([]*{{.upperStartCamelObject}},error) {

    builder = builder.Columns({{.lowerStartCamelObject}}Rows)

	if orderBy == ""{
		builder = builder.OrderBy("id DESC")
	}else{
		builder = builder.OrderBy(orderBy)
	}

	if page < 1{
		page = 1
	}
	offset := (page - 1) * pageSize

	query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql()
	if err != nil {
		return nil, err
	}

	var resp []*{{.upperStartCamelObject}}
	{{if .withCache}}err = m.QueryRowsNoCacheCtx(ctx,&resp, query, values...){{else}}
	err = m.conn.QueryRowsCtx(ctx,&resp, query, values...)
	{{end}}
	switch err {
	case nil:
		return resp, nil
	default:
		return nil, err
	}
}

步骤二：使用之前做好的脚本生成代码

使用GitBash打开deploy/script/mysql目录，执行脚本

此时genModel目录下面就会生成相关代码

步骤三：将生成的代码剪切到项目目录的对应位置

效果

默认模板生成的Model层方法

自定义模板生成的Model层方法

生成的FindPageListByPage方法

func (m *defaultLotteryModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Lottery, error) {

	builder = builder.Columns(lotteryRows)

	if orderBy == "" {
		builder = builder.OrderBy("id DESC")
	} else {
		builder = builder.OrderBy(orderBy)
	}

	if page < 1 {
		page = 1
	}
	offset := (page - 1) * pageSize

	query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql()
	if err != nil {
		return nil, err
	}

	var resp []*Lottery
	err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...)
	switch err {
	case nil:
		return resp, nil
	default:
		return nil, err
	}
}

实战2：api自定义响应返回以及集成validator库校验参数

当我们希望自定义统一返回响应体以及希望每个api接口都进行参数校验时，我们可以在模板中修改handler层的代码，从而实现这些效果。

步骤一：实现自定义统一返回响应

在common目录下新建result目录和httpResult.go文件，如下图所示

具体实现代码不是本文重点，下面是提供的代码

package result

import (
	"fmt"
	"net/http"

	"looklook/common/xerr"

	"github.com/pkg/errors"
	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/rest/httpx"
	"google.golang.org/grpc/status"
)

步骤二：在handler下面引入定制的validator包

关于定制validator也不是本文重点，感兴趣的同学可以关注我，留言。

package translator

import (
	"errors"
	"github.com/go-playground/locales/zh"
	ut "github.com/go-playground/universal-translator"
	"github.com/go-playground/validator/v10"
	zh_translations "github.com/go-playground/validator/v10/translations/zh"
	"looklook/app/lottery/cmd/api/internal/logic/lottery"
	"looklook/app/lottery/cmd/api/internal/types"
	"reflect"
	"strings"
)

func Validate(dataStruct interface{}) error {
	zh_ch := zh.New()
	validate := validator.New()
	

步骤三：修改handler.tpl模板代码

将模板替换为以下内容

package {{.PkgName}}

import (
	"net/http"

	"looklook/common/result"

	"github.com/zeromicro/go-zero/rest/httpx"
	{{.ImportPackages}}
)

func {{.HandlerName}}(svcCtx *svc.ServiceContext) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		{{if .HasRequest}}var req types.{{.RequestType}}
		if err := httpx.Parse(r, &req); err != nil {
			httpx.ErrorCtx(r.Context(), w, err)
			return
		}

		validateErr := translator.Validate(&req)
        if validateErr != nil {
            result.ParamErrorResult(r, w, validateErr)
            return
        }

		{{end}}l := {{.LogicName}}.New{{.LogicType}}(r.Context(), svcCtx)
		{{if .HasResp}}resp, {{end}}err := l.{{.Call}}({{if .HasRequest}}&req{{end}})

		result.HttpResult(r, w, {{if .HasResp}}resp{{else}}nil{{end}}, err)
	}
}

步骤四：生成对应的代码

注意生成handler后需要手动点开生成的handler文件，导入translator包，否则服务会报错！！！

# 使用自定义的goctl 生成api
goctl api go -api main.api -dir ../  --style=goZero --home=../../../../../deploy/goctl/1.6.1

修改后的响应体

{
  "code": 200,
  "msg": "OK",
  "data": {
    "message": ""
  }
}

模板自定义规则

1. 在 goctl 提供的有效数据范围内修改，即不支持外部变量
2. 不支持新增模板文件
3. 不支持变量修改

总结

本文介绍了如何使用Go-Zero的goctl工具进行自定义模板的实战，并提供了一个具体的案例来演示定制化模板的过程。

如果你需要详细的命令使用详情，可以参考官方文档中的相关内容。模板定制化 | go-zero Documentation

我将继续更新Go-Zero系列文章，如果你对Go语言或者微服务感兴趣，欢迎关注我，也欢迎直接私信我。

gozero&微服务交流群

我将继续更新Go-Zero系列文章，如果你对Go语言或者微服务感兴趣，欢迎关注我，也欢迎直接私信我。

微信：wangzhongyang1993