Question

我想将RecoverWrap连接到马提尼路线的所有处理程序，以便panic内的代码完成任何RecoverWrap。

我尝试像m.Use(RecoverWrap)那样做，但不知道如何完成它，它在编译时失败。

package main

import (
    "errors"
    "github.com/go-martini/martini"
    "net/http"
)

func main() {
    m := martini.Classic()
    //m.Use(RecoverWrap)
    m.Get("/", func() {
        panic("some panic")
    })

    m.Run()
}

func RecoverWrap(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
        var err error
        defer func() {
            r := recover()
            if r != nil {
                switch t := r.(type) {
                case string:
                    err = errors.New(t)
                case error:
                    err = t
                default:
                    err = errors.New("Unknown error")
                }

                http.Error(w, "Something goes wrong", http.StatusInternalServerError)
            }
        }()
        h.ServeHTTP(w, req)
    })
}

Answer 1

Martini中的中间件处理程序没有进入＆＃34; wrap＆＃34;其他处理程序调用，因此注入器找不到http.Handler。

您可以使用context.Next()：

package main

import (
"errors"
"github.com/go-martini/martini"
"net/http"
)

func main() {
    m := martini.Classic()
    m.Use(RecoverWrap)
    m.Get("/", func() {
        panic("some panic")
        })

    m.Run()
}

func RecoverWrap(c martini.Context, w http.ResponseWriter) {
    var err error
    defer func(w http.ResponseWriter) {
        r := recover()
        if r != nil {
            switch t := r.(type) {
            case string:
                err = errors.New(t)
            case error:
                err = t
            default:
                err = errors.New("Unknown error")
            }
            http.Error(w, "Something goes wrong", http.StatusInternalServerError)
        }
        }(w)
    c.Next()
}

您必须确保您的错误处理程序是第一个注册的中间件，或者之前运行的那些处理程序不会被捕获。

实际上，martini.Recovery中实现了相同的方法：

https://github.com/go-martini/martini/blob/6241001738f6e1b1ea7c4a4089195e1b0681609a/recovery.go#L115

马丁尼可以恢复任何恐慌

1 个答案: