From e15b12decfac68a16522ffce1833ab10d15b0163 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E4=BC=9F=E4=B9=90?= Date: Tue, 14 Mar 2023 13:57:28 +0800 Subject: [PATCH] x --- api/user.http.go | 4 ++-- api/user_http.pb.go | 2 +- cmd/kit/main.go | 15 ++++++++++++++- 3 files changed, 17 insertions(+), 4 deletions(-) diff --git a/api/user.http.go b/api/user.http.go index 34c6523..c31ed74 100644 --- a/api/user.http.go +++ b/api/user.http.go @@ -1,11 +1,11 @@ package user import ( - "git.diulo.com/mogfee/kit/errors" - "git.diulo.com/mogfee/kit/response" "github.com/gin-gonic/gin" "context" "git.diulo.com/mogfee/kit/middleware" + "git.diulo.com/mogfee/kit/errors" + "git.diulo.com/mogfee/kit/response" ) func RegisterUserHandler(app *gin.Engine, srv UserServer, m ...middleware.Middleware) { diff --git a/api/user_http.pb.go b/api/user_http.pb.go index 6e9c5eb..2deaf5c 100644 --- a/api/user_http.pb.go +++ b/api/user_http.pb.go @@ -1,8 +1,8 @@ package user import ( - "git.diulo.com/mogfee/kit/transport/http" "context" + "git.diulo.com/mogfee/kit/transport/http" "git.diulo.com/mogfee/kit/middleware/jwt" ) diff --git a/cmd/kit/main.go b/cmd/kit/main.go index 89592f9..d70b782 100644 --- a/cmd/kit/main.go +++ b/cmd/kit/main.go @@ -35,12 +35,25 @@ func (u *Kit) Generate(plugin *protogen.Plugin) error { return nil } u.addImports("context") - u.addImports("git.diulo.com/mogfee/kit/middleware/jwt") u.addImports("git.diulo.com/mogfee/kit/transport/http") for _, f := range plugin.Files { if len(f.Services) == 0 { continue } + + for _, s := range f.Services { + addAuthKey := false + for _, m := range s.Methods { + autkKey := protogen2.GetAuthKey(m) + if autkKey != "" { + addAuthKey = true + } + } + if addAuthKey { + u.addImports("git.diulo.com/mogfee/kit/middleware/jwt") + } + } + fname := f.GeneratedFilenamePrefix + "_http.pb.go" t := plugin.NewGeneratedFile(fname, f.GoImportPath) t.P("package ", u.packageName(f))