Golang HTTP 路由处理程序通用包装器实现

Question

我正在尝试转换返回响应正文和错​​误的路由处理程序，而不是直接将其写入响应编写器。然后我想从包装函数发送成功/错误响应。

它将帮助我在所有路由的公共(public)位置添加跟踪和指标。

为了实现这一点，我尝试了这个:

router.HandleFunc(app, "/login", WrapHandler(Login)).Methods("POST")

func WrapHandler(handler func(http.ResponseWriter, *http.Request) (interface{}, *types.HandlerErrorResponse)) func(http.ResponseWriter, *http.Request) {
    return func(w http.ResponseWriter, r *http.Request) {
        response, errResponse := handler(w, r)

        sendResponse(r.Context(), w, response, errResponse)
    }
}

登录是一个带有签名的接口(interface):

Login(w http.ResponseWriter, r *http.Request) (*AuthenticationResponse, *types.HandlerErrorResponse)

现在，使用这段代码，错误出现在编译中:

cannot use Login (type func(http.ResponseWriter, *http.Request) (*AuthenticationResponse, *types.HandlerErrorResponse)) as type func(http.ResponseWriter, *http.Request) (interface {}, *types.HandlerErrorResponse) in argument to WrapHandlerNew

但是，为了制作通用的 Wrapper，我必须将响应主体作为接口(interface)。

请让我知道，我该如何实现。

Answer 1

让它编译Login(w http.ResponseWriter, r *http.Request) (*AuthenticationResponse, *types.HandlerErrorResponse) 需要是 Login(w http.ResponseWriter, r *http.Request) (interface {}, *types.HandlerErrorResponse)

但是我建议您定义一个接口(interface)(或使用现有接口(interface):io.Reader 可能是您想要的)

定义您需要从响应中访问的方法，然后将其返回。

type GenericBody interface{
    Bytes() []byte
}

现在只要你的返回对象实现了这个方法，就可以返回了。但是，函数签名必须包含这个新类型。