如何动态更改 Golang HTTP 服务器中的 Mux 处理程序？

如何动态更改 Golang HTTP 服务器中的 Mux 处理程序

在 Go 中，http.ServeMux 和 Gorilla 的 mux.Router 提供了多功能的处理工具HTTP 路由。但是，没有内置机制可以在不重新启动应用程序的情况下修改与现有路由关联的处理函数。

可能的解决方案：

选项 1 ：具有启用/禁用标志的自定义处理程序包装器

此方法涉及创建嵌入实际处理程序函数并包含启用标志的自定义处理程序类型。然后可以将自定义处理程序添加到 mux，如果该标志设置为 true，它将为该处理程序提供服务。

<code class="go">type Handler struct {
    http.HandlerFunc
    Enabled bool
}

type Handlers map[string]*Handler

// Implementation of ServeHTTP that checks if the handler is enabled
func (h Handlers) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    path := r.URL.Path
    if handler, ok := h[path]; ok &amp;&amp; handler.Enabled {
        handler.ServeHTTP(w, r)
    } else {
        http.Error(w, "Not Found", http.StatusNotFound)
    }
}</code>

然后，您可以将此自定义处理程序类型与 http.ServeMux 或 gorilla/mux 一起使用动态启用或禁用路由。

http.ServeMux 示例：

<code class="go">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
})

http.Handle("/", mux)</code>

选项 2：带有 URL 黑名单的中间件

或者，您可以实现中间件，根据禁用路由列表检查请求的 URL。如果 URL 与禁用的路由匹配，中间件可以阻止处理程序执行并返回自定义响应。

示例：

<code class="go">func DisableRouteMiddleware(disabledRoutes []string) func(http.Handler) http.Handler {
    return func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            for _, route := range disabledRoutes {
                if r.URL.Path == route {
                    http.Error(w, "Disabled", http.StatusNotFound)
                    return
                }
            }
            next.ServeHTTP(w, r)
        })
    }
}</code>

然后您可以使用此中间件在将所有处理程序添加到 mux 之前对其进行包装。

这两种方法都允许您在应用程序中动态禁用或启用路由，而无需重新启动。选择使用哪个选项取决于您的具体要求以及所需的灵活性级别。

以上是如何动态更改 Golang HTTP 服务器中的 Mux 处理程序？的详细内容。更多信息请关注PHP中文网其他相关文章！