Browse Source

传入token

master
dark 7 months ago
parent
commit
b8978f22ab
  1. 10
      api/handler.tpl

10
api/handler.tpl

@ -2,7 +2,7 @@ package {{.PkgName}}
import ( import (
"net/http" "net/http"
"context"
"github.com/zeromicro/go-zero/rest/httpx" "github.com/zeromicro/go-zero/rest/httpx"
{{.ImportPackages}} {{.ImportPackages}}
) )
@ -15,7 +15,13 @@ func {{.HandlerName}}(svcCtx *svc.ServiceContext) http.HandlerFunc {
httpx.ErrorCtx(r.Context(), w, err) httpx.ErrorCtx(r.Context(), w, err)
return 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) {{end}}l := {{.LogicName}}.New{{.LogicType}}(r.Context(), svcCtx)
{{if .HasResp}}resp, {{end}}err := l.{{.Call}}({{if .HasRequest}}&req{{end}}) {{if .HasResp}}resp, {{end}}err := l.{{.Call}}({{if .HasRequest}}&req{{end}})
if err != nil { if err != nil {

Loading…
Cancel
Save