Go-zero框架修改模版进行handler统一响应封装

使用go-zero快速生成接口的时候，发现还是有一些情况不太好处理，比如说，想要自定义响应封装等等。

最开始第一版写api文件的时候，写法是这样的。

type LoginRequest {
   
   
  UserName string `json:"userName"`
  Password string `json:"password"`
}

type Response {
   
   
  Code int    `json:"code"`
  Data string `json:"data"`
  Msg  string `json:"msg"`
}

type UserInfo {
   
   
  UserName string `json:"userName"`
  Addr     string `json:"addr"`
  Id       uint   `json:"id"`
}

type UserInfoResponse {
   
   
  Code int      `json:"code"`
  Data UserInfo `json:"data"`
  Msg  string   `json:"msg"`
}

service users {
   
   
  @handler login
  post /api/users/login (LoginRequest) returns (Response)
  
  @handler userInfo
  get /api/users/info returns (UserInfoResponse)
}

// goctl api go -api v1.api -dir .

后面发现可以不把code、data、msg这三个重要信息写在api里边，而是通过统一封装，在统一响应中去加上code、data、msg，使之成为我们一个公共的库供我们使用。

首先我们封装好response文件：

package response

import (
	"github.com/zeromicro/go-zero/rest/httpx"
	"net/http"
)

type Body struct {
   
   
	Code uint32      `json:"code"`
	Msg  string      `json:"msg"`
	Data interface{
   
   } `json:"data"`
}

// Response http返回
func Response(r *http.Request, w http.ResponseWriter, resp interface{
   
   }, err error) {
   
   
	if err == nil {
   
   
		//成功返回
		r := &Body{
   
   
			Code: 0,
			Msg:  "成功",
			Data: resp,
		}
		httpx.WriteJson(w, http.StatusOK, r)
		return
	}
	//错误返回
	errCode := uint32(7)
	// 可以根据错误码，返回具体错误信息
	//errMsg := "服务器错误"

	httpx.WriteJson(w, http