diff --git a/main.go b/main.go index f7b9b6e..5545fc1 100644 --- a/main.go +++ b/main.go @@ -181,6 +181,7 @@ func generateLogicFile(gen *protogen.Plugin, file *protogen.File, service *proto if strings.ToLower(method.Output.GoIdent.GoName) == "statusreply" { imports = append(imports, "\"time\"") + imports = append(imports, "\"git.apinb.com/bsm-sdk/core/vars\"") code = strings.ReplaceAll(code, "{return}", tpl.StatusReplyCode) } else { code = strings.ReplaceAll(code, "{return}", "return ") diff --git a/tpl/logic.go b/tpl/logic.go index e79e894..c1b634b 100644 --- a/tpl/logic.go +++ b/tpl/logic.go @@ -4,9 +4,9 @@ var LogicFile = `package {methodName} import ( "context" + {import} "git.apinb.com/bsm-sdk/core/service" - "git.apinb.com/bsm-sdk/core/vars" ) {comment}