feat(gen): 生成 mq 进行校验，如果没有使用 mq 的 proto 不生成 mq

Author: Ccheers

protoc-gen-mq/main.go

@@ -18,9 +18,8 @@ func main() {
 		return
 	}
 
-	var flags flag.FlagSet
 	options := protogen.Options{
-		ParamFunc: flags.Set,
+		ParamFunc: flag.CommandLine.Set,
 	}
 	options.Run(func(gen *protogen.Plugin) error {
 		gen.SupportedFeatures = uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL)

protoc-gen-mq/template.go

@@ -3,7 +3,9 @@ package main
 import (
 	"fmt"
 
+	"github.com/Ccheers/kratos-mq/protoc-gen-mq/proto/kmq/v1/options"
 	"google.golang.org/protobuf/compiler/protogen"
+	"google.golang.org/protobuf/proto"
 	"google.golang.org/protobuf/types/descriptorpb"
 )
 
@@ -22,6 +24,9 @@ var methodSets = make(map[string]int)
 
 // generateFile generates a _gin.pb.go file.
 func generateFile(gen *protogen.Plugin, file *protogen.File) *protogen.GeneratedFile {
+	if len(file.Services) == 0 || !hasKMQRule(file.Services) {
+		return nil
+	}
 	if len(file.Services) == 0 {
 		return nil
 	}
@@ -197,3 +202,18 @@ func generateClientMethodList(g *protogen.GeneratedFile, s *service) {
 		g.P("}")
 	}
 }
+
+func hasKMQRule(services []*protogen.Service) bool {
+	for _, service := range services {
+		for _, method := range service.Methods {
+			if method.Desc.IsStreamingClient() || method.Desc.IsStreamingServer() {
+				continue
+			}
+			rule, ok := proto.GetExtension(method.Desc.Options(), options.E_Mq).(*options.MQ)
+			if rule != nil && ok {
+				return true
+			}
+		}
+	}
+	return false
+}