前提
go环境的配置、goctl的安装、go-zero的基本使用默认都会
需求
go-zero框架中，默认使用goctl命令生成的代码并没有统一响应格式，现在使用自定义模板的方式实现统一响应格式：
{
"code": 0,
"msg": "OK",
"data": {}
}
步骤
1、下载模板
goctl template init
下载完成后会有提示edit on your risk，默认下载路径会在C盘
2、移动模板
将下载好的模板文件拷贝到项目目录中，以便后续操作，移动后的项目结构如下：
3、创建统一响应结构体
创建utils包、再创建统一响应结构体reponse.go
package utils
import (
"net/http"
"github.com/zeromicro/go-zero/rest/httpx"
)
type Body struct {
Code int `json:"code"`
Msg string `json:"msg"`
Data interface{} `json:"data,omitempty"`
}
func Response(w http.ResponseWriter, resp interface{}, err error) {
var body Body
if err != nil {
body.Code = -1
body.Msg = err.Error()
} else {
body.Msg = "OK"
body.Data = resp
}
httpx.OkJson(w, body)
}
4、修改handler模板
将刚刚通过goctl template下载的模板中的handler.tpl替换成如下内容：
package {{.PkgName}}
import (
"net/http"
"gozero_gorm/utils"
"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.Error(w, err)
return
}{{end}}
l := {{.LogicName}}.New{{.LogicType}}(r.Context(), svcCtx)
{{if .HasResp}}resp, {{end}}err := l.{{.Call}}({{if .HasRequest}}&req{{end}})
{{if .HasResp}}utils.Response(w, resp, err){{else}}response.Response(w, nil, err){{end}} // 换成自己的结构体路径
}
}
如果你定义的统一相应结构体在其他路径下，只需修改最后一行代码即可
{{if .HasResp}}utils.Response(w, resp, err){{else}}response.Response(w, nil, err){{end}} // 换成自己的结构体路径
5、生成代码
goctl api go --api .\index.api --dir . --home ..\template\
相比普通的api生成该命令指定了使用的模板，也就是 --home ..\template\这一项，由于模板中的handler.tpl已经被修改，所以生成后的代码会返回统一的响应格式
测试
可以看到响应结果已经成功格式化