From b8978f22aba000c781ae83ab1710ae6fa9e2640d Mon Sep 17 00:00:00 2001
From: dark <nndaa@qq.com>
Date: Mon, 14 Jul 2025 04:20:22 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=A0=E5=85=A5token?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/handler.tpl | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/api/handler.tpl b/api/handler.tpl
index fce5acc..4b62e9b 100644
--- a/api/handler.tpl
+++ b/api/handler.tpl
@@ -2,7 +2,7 @@ package {{.PkgName}}
 
 import (
 	"net/http"
-
+	"context"
 	"github.com/zeromicro/go-zero/rest/httpx"
 	{{.ImportPackages}}
 )
@@ -15,7 +15,13 @@ func {{.HandlerName}}(svcCtx *svc.ServiceContext) http.HandlerFunc {
 			httpx.ErrorCtx(r.Context(), w, err)
 			return
 		}
-
+		// 获取请求头里的token
+		token := r.Header.Get("Authorization")
+		// 如果有token, 使用context传值到logic中
+		if token != "" {
+			ctx := context.WithValue(r.Context(), "token", token)
+			r = r.WithContext(ctx)
+		}
 		{{end}}l := {{.LogicName}}.New{{.LogicType}}(r.Context(), svcCtx)
 		{{if .HasResp}}resp, {{end}}err := l.{{.Call}}({{if .HasRequest}}&req{{end}})
 		if err != nil {