x

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) {

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"
 )
 

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))