From 55a171fff303f371e5f2b6f60b91acb40c51c507 Mon Sep 17 00:00:00 2001 From: zhouyonggao <1971162852@qq.com> Date: Fri, 26 Dec 2025 16:29:26 +0800 Subject: [PATCH] =?UTF-8?q?refactor(router):=20=E9=87=8D=E6=9E=84exports?= =?UTF-8?q?=E8=B7=AF=E7=94=B1=E5=A4=84=E7=90=86=E9=80=BB=E8=BE=91=E4=BB=A5?= =?UTF-8?q?=E6=8F=90=E9=AB=98=E5=8F=AF=E8=AF=BB=E6=80=A7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 将直接调用的ServeHTTP改为先创建handler变量再调用，使代码结构更清晰 --- server/internal/api/router.go | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/server/internal/api/router.go b/server/internal/api/router.go index 10e03ce..79fa301 100644 --- a/server/internal/api/router.go +++ b/server/internal/api/router.go @@ -21,12 +21,14 @@ func NewRouter(metaDB *sql.DB, marketingDB *sql.DB, marketingAuthDB *sql.DB, res // exports 路由处理（特殊处理下载接口） mux.HandleFunc("/api/exports/", func(w http.ResponseWriter, r *http.Request) { path := r.URL.Path - // 如果是下载接口，不需要认证 if strings.HasSuffix(path, "/download") { - exportsDownloadHandler(metaDB, marketingDB, ymtDB).ServeHTTP(w, r) + // 下载接口，不需要认证 + handler := exportsDownloadHandler(metaDB, marketingDB, ymtDB) + handler.ServeHTTP(w, r) } else { // 其他 exports 路由需要认证 - withAccess(withTrace(authMiddleware(ExportsHandler(metaDB, marketingDB, ymtDB)))).ServeHTTP(w, r) + handler := withAccess(withTrace(authMiddleware(ExportsHandler(metaDB, marketingDB, ymtDB)))) + handler.ServeHTTP(w, r) } })