go-zero 自定义模板

2022-04-22 10:53 更新

场景

实现统一格式的body响应,格式如下:

{
  "code": 0,
  "msg": "OK",
  "data": {} // ①
}

① 实际响应数据

go-zero生成的代码没有对其进行处理

准备工作

我们提前在module为greet的工程下的response包中写一个Response方法,目录树类似如下:

greet
├── response
│   └── response.go
└── xxx...

代码如下

package response

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)
}

修改handler模板

$ vim ~/.goctl/api/handler.tpl

将模板替换为以下内容

package handler

import (
    "net/http"
    "greet/response"// ①
    {% raw %}
    {{.ImportPackages}}
    {% endraw %}
)

{% raw %}
func {{.HandlerName}}(ctx *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 := logic.New{{.LogicType}}(r.Context(), ctx)
        {{if .HasResp}}resp, {{end}}err := l.{{.Call}}({{if .HasRequest}}req{{end}})
        {{if .HasResp}}response.Response(w, resp, err){{else}}response.Response(w, nil, err){{end}}//②

    }
}
{% endraw %}

① 替换为你真实的response包名,仅供参考

② 自定义模板内容

如果本地没有~/.goctl/api/handler.tpl文件,可以通过模板初始化命令goctl template init进行初始化

修改模板前后对比

  • 修改前

func GreetHandler(ctx *svc.ServiceContext) http.HandlerFunc {
  return func(w http.ResponseWriter, r *http.Request) {
      var req types.Request
      if err := httpx.Parse(r, &req); err != nil {
          httpx.Error(w, err)
          return
      }

      l := logic.NewGreetLogic(r.Context(), ctx)
      resp, err := l.Greet(req)
      // 以下内容将被自定义模板替换
      if err != nil {
          httpx.Error(w, err)
      } else {
          httpx.OkJson(w, resp)
      }
  }
}

  • 修改后

func GreetHandler(ctx *svc.ServiceContext) http.HandlerFunc {
  return func(w http.ResponseWriter, r *http.Request) {
      var req types.Request
      if err := httpx.Parse(r, &req); err != nil {
          httpx.Error(w, err)
          return
      }

      l := logic.NewGreetLogic(r.Context(), ctx)
      resp, err := l.Greet(req)
      response.Response(w, resp, err)
  }
}

修改模板前后响应体对比

  • 修改前

{
  "message": "Hello go-zero!"
}

  • 修改后

{
  "code": 0,
  "msg": "OK",
  "data": {
      "message": "Hello go-zero!"
  }
}

总结

本文档仅对http相应为例讲述了自定义模板的流程,除此之外,自定义模板的场景还有:

  • model 层添加kmq
  • model 层生成待有效期option的model实例
  • http自定义相应格式 ...


以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号