zhouyonggao
/
MarketingSystemDataExportTool
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor(router): 重构exports路由处理逻辑以提高可读性 

将直接调用的ServeHTTP改为先创建handler变量再调用，使代码结构更清晰
This commit is contained in:
zhouyonggao 2025-12-26 16:29:26 +08:00
parent 20deec5879
commit 55a171fff3
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
server/internal/api/router.go
View File

@ -21,12 +21,14 @@ func NewRouter(metaDB *sql.DB, marketingDB *sql.DB, marketingAuthDB *sql.DB, res
// exports 路由处理（特殊处理下载接口） // exports 路由处理（特殊处理下载接口）
mux.HandleFunc("/api/exports/", func(w http.ResponseWriter, r *http.Request) { mux.HandleFunc("/api/exports/", func(w http.ResponseWriter, r *http.Request) {
path := r.URL.Path path := r.URL.Path
// 如果是下载接口，不需要认证
if strings.HasSuffix(path, "/download") { if strings.HasSuffix(path, "/download") {
exportsDownloadHandler(metaDB, marketingDB, ymtDB).ServeHTTP(w, r) // 下载接口，不需要认证
handler := exportsDownloadHandler(metaDB, marketingDB, ymtDB)
handler.ServeHTTP(w, r)
} else { } else {
// 其他 exports 路由需要认证 // 其他 exports 路由需要认证
withAccess(withTrace(authMiddleware(ExportsHandler(metaDB, marketingDB, ymtDB)))).ServeHTTP(w, r) handler := withAccess(withTrace(authMiddleware(ExportsHandler(metaDB, marketingDB, ymtDB))))
handler.ServeHTTP(w, r)
} }
}) })