静态swagger

api/etc.tpl

@@ -1,6 +1,7 @@
 Name: {{.serviceName}}
 Host: {{.host}}
 Port: {{.port}}
+Mode: dev
 Mysql: 
     Host: 127.0.0.1
     Port: 3306

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 {

api/main.tpl

@@ -3,6 +3,7 @@ package main
 import (
 	"flag"
 	"fmt"
+	"net/http"
 
 	{{.importPackages}}
 )
@@ -14,9 +15,16 @@ func main() {
 
 	var c config.Config
 	conf.MustLoad(*configFile, &c)
-		
-	server := rest.MustNewServer(c.RestConf)
+	server := *rest.Server
+	// dev 模式下开启swagger
+	if c.Mode == "dev" {
+		server = rest.MustNewServer(c.RestConf, rest.WithCors(), rest.WithCorsHeaders("*"), rest.WithFileServer("/swagger", http.Dir("./swagger")))
+	} else {
+		server = rest.MustNewServer(c.RestConf, rest.WithCors())
+	}
 	defer server.Stop()
+	// rest.MustNewServer(c.RestConf)
+	// defer server.Stop()
 
 	ctx := svc.NewServiceContext(c)
 	handler.RegisterHandlers(server, ctx)