去代理中间件并修改响应

时间:2017-03-25 18:11:17

标签: go reverse-proxy go-gin

我正在尝试将来自Go后端的请求代理到微服务,并在将响应发送到客户端之前修改响应。请求链是:客户端 - >转到后端 - >微服务 - >转到后端 - >客户端

我正在使用Go Gin框架。工作中间件:

func ReverseProxy(target string) gin.HandlerFunc {
    log.Println(target)
    url, err := url.Parse(target)
    if err != nil {
        log.Fatal(err)
    }
    proxy := httputil.NewSingleHostReverseProxy(url)
    return func(c *gin.Context) {
        proxy.ServeHTTP(c.Writer, c.Request)
    }
}

现在我的问题是:我如何接收和修改微服务发送的响应?

1 个答案:

答案 0 :(得分:1)

如何使用ReverseProxy。ModifyResponse?

例如,这会为响应添加自定义标头。

func addCustomHeader(r *http.Response) error {
    r.Header["Hello"] = []string{"World"}
    return nil
}

proxy.ModifyResponse = addCustomHeader