首頁 > 後端開發 > Golang > 在go-chi中為子路由設定不同的中介軟體

在go-chi中為子路由設定不同的中介軟體

王林
發布: 2024-02-09 10:06:08
轉載
750 人瀏覽過

在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
最新問題
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板