fix(kitex tool): import unused protection when -module is not set (#1412

)
cloudwego · Jun 24, 2024 · 1b9e269 · 1b9e269
1 parent 22991e2
commit 1b9e269
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tool/internal_pkg/pluginmode/thriftgo/plugin.go b/tool/internal_pkg/pluginmode/thriftgo/plugin.go
@@ -128,6 +128,10 @@ func Run() int {
 		protocol:              conv.Config.Protocol,
 		handlerReturnKeepResp: conv.Config.HandlerReturnKeepResp,
 	}
+	// for cmd without setting -module
+	if p.module == "" {
+		p.module = conv.Config.PackagePrefix
+	}
 	patches, err := p.patch(req)
 	if err != nil {
 		return conv.fail(fmt.Errorf("patch: %w", err))