diff --git a/cmd/kit/main.go b/cmd/kit/main.go index 6460df3..e0f5e7e 100644 --- a/cmd/kit/main.go +++ b/cmd/kit/main.go @@ -36,6 +36,7 @@ func (u *Kit) Generate(plugin *protogen.Plugin) error { } u.addImports("context") u.addImports("git.diulo.com/mogfee/kit/transport/http") + u.addImports("git.diulo.com/mogfee/kit/middleware/jwt") for _, f := range plugin.Files { if len(f.Services) == 0 { continue diff --git a/protogen/protogen.go b/protogen/protogen.go index 182a84f..a164611 100644 --- a/protogen/protogen.go +++ b/protogen/protogen.go @@ -40,6 +40,9 @@ func GetAuthKey(m *protogen.Method) (authKey string, needAuth bool) { //log.Println(err) } else { if vv, ok := opts.(*string); ok { + if *vv == "none" { + return "", false + } return *vv, true } }