golang - 从任何地方退出 http 处理程序

Question

我正在使用 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
    }

    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")
}

理想情况下，我想从 checkSomeThing 函数中退出处理程序，而不必返回，然后再次返回一个级别，随着应用程序的增长，情况会变得更糟。这纯粹是为了代码的可读性。

Answer 1

惯用的方法是检查调用链上的错误返回。

要从任何地方退出处理程序，请按照encoding/json包中的模式使用panic和recover 。

为恐慌定义一个独特的类型：

type httpError struct {
    status  int
    message string
}

编写一个要在 defer 语句中使用的函数。该函数检查类型并适当处理错误。否则，该函数将继续出现恐慌。

func handleExit(w http.ResponseWriter) {
    if r := recover(); r != nil {
        if he, ok := r.(httpError); ok {
            http.Error(w, he.message, he.status)
        } else {
            panic(r)
        }
    }
}

编写一个辅助函数来调用panic：

func exit(status int, message string) {
    panic(httpError{status: status, message: message})
}

使用这样的函数：

func example() {
   exit(http.StatusBadRequest, "Bad!")
}

func someHandler(w http.ResponseWriter, r *http.Request) {
   defer handleExit(w)
   example()
}