处理 ReverseProxy Director 函数内 Web 请求的错误

偷偷努力，悄无声息地变强，然后惊艳所有人！哈哈，小伙伴们又来学习啦~今天我将给大家介绍《处理 ReverseProxy Director 函数内 Web 请求的错误》，这篇文章主要会讲到等等知识点，不知道大家对其都有多少了解，下面我们就一起来看一吧！当然，非常希望大家能多多评论，给出合理的建议，我们一起学习，一起进步！

我想知道是否可以在 reverseproxy.director 函数内取消 web 请求或向客户端发送内部响应。

假设我们做了一些会引发错误的事情，或者我们有其他原因不转发请求。

proxy := &httputil.reverseproxy{
    director: func(r *http.request) {
        err := somethingthatthrows()
    },
}

http.handle("/", proxy)

解决此问题的方法可能如下，但它不如上面使用代理的方式那么简洁。我也不确定请求应该修改到什么程度。导演似乎是这样做的地方。

http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
    err := somethingThatThrows()
    if err != nil {
        w.WriteHeader(http.StatusInternalServerError)
        return
    }
    proxy.ServeHTTP(w, r)
})

正确答案

是否可以取消网络请求[...]

您可以取消传递给 director 函数的请求，但是有一些细节需要考虑：

• 取消请求的正确方法是取消其上下文
• 您无法取消自己未设置（deadline|timeout|cancelfunc）的上下文→即您必须有权访问 cancel 函数→即您无法取消创建的父上下文是别人写的。
• 传递给 director 函数的 *http.request 是原始请求的克隆

基于上述几点，您可以将 director 中的请求替换为另一个具有可取消上下文的请求。它可能如下所示：

proxy := &httputil.reverseproxy{
    director: func(req *http.request) {

        // create a cancellable context, and re-set the request
        ctx, cancel := context.withcancel(req.context())
        *req = *req.withcontext(ctx)

        err := somethingthatthrows()
        if err != nil {
            cancel()
            return
        }
    },
}

那么上面的代码本身不会执行任何其他操作。应该发生的是，实现 http.roundtripper 的 httputil.reverseproxy.transport 函数在实际向上游服务发送任何内容之前检查请求上下文是否已取消。

director 的文档指出：

director 必须是一个将请求修改为要使用 transport 发送的新请求的函数。

当未提供 transport 时，它将回退到 http.defaulttransport，这会在上下文取消时中止请求。当前代码（go 1.17.5）如下所示：

select {
        case <-ctx.done():
            req.closebody()
            return nil, ctx.err()
        default:
        }

如果您提供自己的 http.roundtripper 实现，您可能希望自己实现该行为。另请记住，如果上下文完成通道不可取消，则它是 nil，因此您必须设置取消函数并调用 cancel() 以便让 select 运行“完成”案例。

或在 reverseproxy.director 内向客户端发送内部响应

根据上面文档中的相同引用，您不应该从 director 函数内写入 http.responsewriter - 假设您甚至正在关闭它。正如您所看到的，director 本身没有将 http.responsewriter 作为参数，这应该已经是一个不言自明的细节。

如果您想指定一些其他行为，以防请求无法转发，并假设 http.roundtripper 的任何实现在取消请求上下文时返回 error，您可以提供 reverseproxy.errorhandler 函数：

proxy.ErrorHandler = func(writer http.ResponseWriter, request *http.Request, err error) {
    // inspect err
    // write to writer
}

当 transport 返回错误时，包括当错误来自取消的请求时，将调用 errorhandler，并且它确实有 http.responsewriter 作为参数。

好了，本文到此结束，带大家了解了《处理 ReverseProxy Director 函数内 Web 请求的错误》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！