首页 >后端开发 >Golang >在go-chi中为子路由设置不同的中间件

在go-chi中为子路由设置不同的中间件

王林
王林转载
2024-02-09 10:06:08745浏览

在go-chi中为子路由设置不同的中间件

在Go语言的Web框架go-chi中,为子路由设置不同的中间件是一种常见的需求。中间件是一种处理HTTP请求和响应的函数,它可以在请求到达目标处理器之前和之后执行一些额外的操作。通过为不同的子路由设置不同的中间件,我们可以根据业务需求对每个子路由进行个性化的处理。在本文中,我们将介绍如何在go-chi中灵活地为子路由设置不同的中间件,以实现更加灵活的路由处理。

问题内容

我有一个可以通过配置停用的代理服务器,使用 mustBeSecure。我希望这个代理在某个子域中表现不同:“/application/health”,使其始终不安全。到目前为止我尝试过的所有更改都失败了。有没有办法为“/application/health”配置不同的组,该组仍然使用代理但从不需要身份验证?

router := chi.NewRouter()
router.Route("/", func(r chi.Router) {
    r.Use(chimw.Recoverer)
    router.Use(hlog.NewHandler(log.Logger))
    if mustBeSecure() {
        r.Use(keycloak.MustStandardKeycloakAuth("url"))
    }
    setProxy(r)
    r.Group(func(r chi.Router) {
        r.Get("/health", handleHealth())
    })
})
return http.ListenAndServe("0.0.0.0", router)

根据要求,这里是 setProxy() 的示例

func setProxy(r chi.Router) {
    r.Route("/application", func(r chi.Router) {
        r.Use(func(next http.Handler) http.Handler {
            return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
                ctx := context.WithValue(r.Context(), "target", "http:locahost:9999")
                r = r.WithContext(ctx)
                next.ServeHTTP(w, r)
            })
        })
        r.HandleFunc("/*", httputil.ReverseProxy{}.ServeHTTP)
    })
}

解决方法

我相信你正在尝试这个。

  • 对所有 /application 路由使用路由器
  • 声明 GET /health
  • 为所有其他路由创建一个组(在 /application 中)
  • 应用条件中间件

为了便于理解,我稍微简化了代码。您仍然可以使用 setProxy 函数。

func main() {
    proxy, err := NewProxy("http://localhost:4422")
    if err != nil {
        panic(err)
    }

    router := chi.NewRouter()

    router.Get("/health", handleHealth)

    router.Route("/application", func(r chi.Router) {
        r.Get("/health", ProxyRequestHandler(proxy))

        r.Group(func(r chi.Router) {
            if mustBeSecure() {
                r.Use(keycloak.MustStandardKeycloakAuth("url"))
            }
            r.HandleFunc("/*", ProxyRequestHandler(proxy))
        })
    })

    http.ListenAndServe("0.0.0.0:4411", router)
}

func NewProxy(targetHost string) (*httputil.ReverseProxy, error) {
    targetUrl, err := url.Parse(targetHost)
    if err != nil {
        return nil, err
    }

    return httputil.NewSingleHostReverseProxy(targetUrl), nil
}

func ProxyRequestHandler(proxy *httputil.ReverseProxy) func(http.ResponseWriter, *http.Request) {
    return func(w http.ResponseWriter, r *http.Request) {
        proxy.ServeHTTP(w, r)
    }
}

以上是在go-chi中为子路由设置不同的中间件的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文转载于:stackoverflow.com。如有侵权,请联系admin@php.cn删除