如何在 Go 中动态管理 HTTP 路由处理程序？

在 Go 中动态管理 HTTP 路由处理程序

在 Go 中使用 HTTP 服务器时，动态修改的灵活性可能会很有帮助无需重新启动应用程序即可路由处理程序。本文为原生 http.ServerMux 和流行的 Gorilla Toolkit 的 mux.Router 提供了一个解决方案。

传统上，管理路由的一种方法是通过返回 404 状态代码来处理禁用的功能。然而，更通用的解决方案包括拦截传入请求并检查路由当前是否已启用。

为此，我们引入了 Handlers 类型，它是具有关联启用标志的路由处理程序的集合。 ServeHTTP 方法通过检查标志并调用处理程序或返回 404 错误来处理传入请求。

HandleFunc 方法向底层多路复用器注册路由并将它们添加到 Handlers 映射中。随后调用时，该方法确保仅执行已启用的处理程序。

<code class="go">package main

import (
    "net/http"
    "sync"
)

type HasHandleFunc interface {
    HandleFunc(pattern string, handler func(w http.ResponseWriter, req *http.Request))
}
type Handler struct {
    http.HandlerFunc
    Enabled bool
}
type Handlers map[string]*Handler

func (h Handlers) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    path := r.URL.Path
    if handler, ok := h[path]; ok && handler.Enabled {
        handler.ServeHTTP(w, r)
    } else {
        http.Error(w, "Not Found", http.StatusNotFound)
    }
}

func (h Handlers) HandleFunc(mux HasHandleFunc, pattern string, handler http.HandlerFunc) {
    h[pattern] = &Handler{handler, true}
    mux.HandleFunc(pattern, h.ServeHTTP)
}

func main() {
    mux := http.NewServeMux()
    handlers := Handlers{}
    handlers.HandleFunc(mux, "/", func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("this will show once"))
        handlers["/"].Enabled = false
        // the route to '/' is now disabled
    })
    http.Handle("/", mux)
    http.ListenAndServe(":9020", nil)
}</code>

通过此解决方案，您可以动态禁用或启用路由，甚至设置基于时间的路由模式，满足灵活的需求Go 中的 HTTP 路由管理。

以上是如何在 Go 中动态管理 HTTP 路由处理程序？的详细内容。更多信息请关注PHP中文网其他相关文章！