在Go語言的Web框架go-chi中，為子路由設(shè)置不同的中間件是一種常見的需求。中間件是一種處理HTTP請(qǐng)求和響應(yīng)的函數(shù)，它可以在請(qǐng)求到達(dá)目標(biāo)處理器之前和之后執(zhí)行一些額外的操作。通過為不同的子路由設(shè)置不同的中間件，我們可以根據(jù)業(yè)務(wù)需求對(duì)每個(gè)子路由進(jìn)行個(gè)性化的處理。在本文中，我們將介紹如何在go-chi中靈活地為子路由設(shè)置不同的中間件，以實(shí)現(xiàn)更加靈活的路由處理。
問題內(nèi)容
我有一個(gè)可以通過配置停用的代理服務(wù)器，使用 mustBeSecure。我希望這個(gè)代理在某個(gè)子域中表現(xiàn)不同：“/application/health”，使其始終不安全。到目前為止我嘗試過的所有更改都失敗了。有沒有辦法為“/application/health”配置不同的組，該組仍然使用代理但從不需要身份驗(yàn)證？
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)
登錄后復(fù)制
根據(jù)要求，這里是 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)
})
}
登錄后復(fù)制
解決方法
我相信你正在嘗試這個(gè)。
對(duì)所有 /application 路由使用路由器
聲明 GET /health
為所有其他路由創(chuàng)建一個(gè)組（在 /application 中）
應(yīng)用條件中間件
為了便于理解，我稍微簡化了代碼。您仍然可以使用 setProxy 函數(shù)。
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)
}
}
登錄后復(fù)制
