如何在不使用 DefaultServeMux 的情况下实现 HandlerFunc

Question

如果我要使用 DefaultServeMux（我通过将它nil作为第二个参数传递给 ListenAndServe 来指定它），那么我可以访问http.HandleFunc，您在下面的 Go wiki 示例中看到了它：func handler(w http.ResponseWriter, r *http.Request) {    fmt.Fprintf(w, "Hi there, I love %s!", r.URL.Path[1:])}func main() {    http.HandleFunc("/", handler)    http.ListenAndServe(":8080", nil)}在我当前的代码中，我无法使用 DefaultServeMux，即我将自定义处理程序传递给 ListenAndServe    h := &mypackage.Handler{        Database: mydb    }    http.ListenAndServe(":8080", h)所以我没有http.HandleFunc内置。但是，我必须将一些授权代码调整到我的代码库中，这些代码需要类似http.HandleFunc. 例如，如果我一直在使用 DefaultServeMux，当我点击"/protected"路由时，我会想去Protected处理程序，但只有在通过h.AuthorizationHandlerFunc这样的   h.AuthorizationHandlerFunc(Protected)但是，由于我没有使用 DefaultServeMux，因此它不起作用，即我无法将Protected函数（并调用它）传递给AuthorizationHandlerFunc. 这是下面 AuthorizationHandlerFunc 的实现。你可以在下面看到Protected永远不会被调用的。问题：HandlerFunc在这种情况下如何实现（不使用 DefaultServeMux）？func (h *Handler) AuthorizationHandlerFunc(next http.HandlerFunc) http.Handler{     return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request){         h.AuthorizationMiddleWare(w, r, next)   })}func (h *Handler) AuthorizationMiddleWare(w http.ResponseWriter, r *http.Request, next http.HandlerFunc){     //other stuff happens     log.Println("this is never getting called")     next(w,r)}func (h *Handler)Protected(w http.ResponseWriter, r *http.Request){      log.Println("this is never getting called")}更新 ServeHTTP 在 mypackage.Handler 上实现。为什么 Protected 函数没有被调用，或者，AuthorizationMiddleWare 中的相关代码没有被调用？

Answer 1

将您的授权中间件重新实现为http.Handler：

type auth struct {

   DB *sql.DB

   UnauthorizedHandler http.Handler

}

func NewAuth(db *sql.DB, unauthorized http.Handler) *auth {

    return auth{db, unauthorized}

}

func (a *auth) Protected(h http.Handler) http.Handler {

    fn := func(w http.ResponseWriter, r *http.Request) {

        // Check whether the request is valid

        // If it's invalid, call your error func and make sure to *return* early!

        if !valid {

            a.UnauthorizedHandler.ServeHTTP(w, r)

            return

        }

        // Call the next handler on success

        h.ServeHTTP(w, r)

        return

    }

    return http.HandlerFunc(fn)

}

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

    io.WriteString(w, "Hello!\n")

}

func main() {

    auth := NewAuth(db, errorHandler)

    r := http.NewServeMux()

    // We have a http.Handler implementation that wraps a http.HandlerFunc

    // ... so we call r.Handle on our ServeMux and type-cast the wrapped func

    r.Handle("/protected", auth.Protected(http.HandlerFunc(someHandler)))

    // Just a simple http.HandlerFunc here

    r.HandleFunc("/public", someOtherHandler)

    log.Fatal(http.ListenAndServe(":8000", r))

}

看看我为一个不同的例子编写的httpauth libServeHTTP方法。以上和ServeHTTP在您的类型上显式创建方法都是有效的方法。