如何终止来自golang子函数的请求

对于一个Golang开发者来说，牢固扎实的基础是十分重要的，golang学习网就来带大家一点点的掌握基础知识点。今天本篇文章带大家了解《如何终止来自golang子函数的请求》，主要介绍了，希望对大家的知识积累有所帮助，快点收藏起来吧，否则需要时就找不到了！

我想返回并终止 checksomething 函数中的请求。但问题是该过程会继续，并且不会返回响应，除非到达 main() 的末尾。这是我的代码：

package main

import (
    "fmt"
    "net/http"
)

func main() {
    http.handlefunc("/", handler)
    // ...

    http.listenandserve(":8080", nil)
}

func handler(w http.responsewriter, r *http.request) {
    checksomething(w, r)

    http.error(w, "operation completed!", http.statusok)
    fmt.println("end of handler.")
}

func checksomething(w http.responsewriter, r *http.request) {

    http.error(w, "bad request!", http.statusbadrequest)
    return
}

运行程序后，输出如下：

// Output:    
< HTTP/1.1 400 Bad Request
< Content-Type: text/plain; charset=utf-8
< X-Content-Type-Options: nosniff
< Date: Sun, 12 Sep 2021 12:38:49 GMT
< Content-Length: 34
< 
Bad Request!
Operation completed!

正确答案

始终尝试在 api 级别处理错误 - 在调用堆栈中冒泡错误。但是，在某些情况下这是不可能的 - 也许您的处理程序是中间件处理程序链中的众多处理程序之一，并且您不希望其他层完成。所以...

如果您想立即中止来自处理程序的请求，标准库支持 panic 和恢复。

来自http.Handler docs：

如果 servehttp 发生恐慌，服务器（servehttp 的调用者）会认为 恐慌的影响与主动请求无关。它 恢复恐慌，将堆栈跟踪记录到服务器错误日志中，以及 关闭网络连接或发送 http/2 rst_stream， 取决于 http 协议。 中止处理程序以便客户端看到 响应中断，但服务器没有记录错误，恐慌 值为 erraborthandler。

和http.erraborthandler：

... 是用于中止处理程序的哨兵恐慌值。尽管 servehttp 的任何恐慌都会中止对客户端的响应，恐慌 使用 erraborthandler 还可以抑制将堆栈跟踪记录到 服务器的错误日志。

因此要在您的处理程序中执行此操作：

func checksomething(w http.responsewriter, r *http.request) {
    http.error(w, "bad request!", http.statusbadrequest)
    panic(http.erraborthandler) // terminate request (and any more handlers in the chain)
}

有一个问题...

对客户端的写入将被缓冲 - 并且 panicing 可能会导致这些缓冲的写入丢失。

至 forcibly flush 缓冲区：

if f, ok := w.(http.flusher); ok {
    f.flush()
}

panic(http.erraborthandler)

或者您可以添加自己的 panic 恢复。只要确保在恢复过程中“重新加注”任何 panics - 如果它们不是由您引起的 - 这样它们就不会丢失：

func handler(w http.responsewriter, r *http.request) {

    defer func() {
        r := recover()

        switch r {

        case nil: // no panic

        // (2) goes here...
        case http.erraborthandler:
            log.println("recovered in handler")

        // (2) or here...
        default:
            panic(r) // re-raise any unexpected panic
        }
    }()
    checksomething(w, r)  // (1) panic here ...

    // (3) ... and this never runs
}

在上面的恐慌恢复代码中，http.servehttp永远不会知道发生了恐慌 - 并自然地刷新请求缓冲区。

根据 http.error 文档

因此，在执行 checksomething 函数后，它只是将错误字符串写入 responsewriter 并继续处理进一步的操作。

下面给出了代码的工作版本：

package main

import (
    "errors"
    "fmt"
    "net/http"
)

func main() {
    http.HandleFunc("/", handler)
    // ...

    http.ListenAndServe(":8080", nil)
}

func handler(w http.ResponseWriter, r *http.Request){
    err := checkSomeThing(w, r)
    if err != nil {
        return
    }

    http.Error(w, "Operation completed!", http.StatusOK)
    fmt.Println("End of Handler.")
    return
}

func checkSomeThing(w http.ResponseWriter, r *http.Request) error{

    http.Error(w, "Bad Request!", http.StatusBadRequest)
    return errors.New("bad request")
}

今天关于《如何终止来自golang子函数的请求》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！