protoc-gen增加导出ServiceDesc方法插件desc

Author: hb-chen

go.mod

@@ -5,7 +5,7 @@ go 1.12
 require (
 github.com/golang/protobuf v1.3.2
 github.com/hashicorp/golang-lru v0.5.3 // indirect
-github.com/micro/go-micro v1.7.1-0.20190724203029-7ca8f8f0ab98
+github.com/micro/go-micro v1.8.0
 golang.org/x/net v0.0.0-20190724013045-ca1201d0de80 // indirect
 golang.org/x/sys v0.0.0-20190726091711-fc99dfbffb4e // indirect
 golang.org/x/time v0.0.0-20190308202827-9d24e82272b4

go.sum

@@ -171,10 +171,10 @@ github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hd
 github.com/mattn/go-runewidth v0.0.4/go.mod h1:LwmH8dsx7+W8Uxz3IHJYH5QSwggIsqBzpuz5H//U1FU=
 github.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0=
 github.com/micro/cli v0.2.0/go.mod h1:jRT9gmfVKWSS6pkKcXQ8YhUyj6bzwxK8Fp5b0Y7qNnk=
-github.com/micro/go-micro v1.7.1-0.20190724203029-7ca8f8f0ab98 h1:n99uuhM5kq8UlwpkD4ZCpVz7QB+VkzlEq/5pSPUuJfQ=
-github.com/micro/go-micro v1.7.1-0.20190724203029-7ca8f8f0ab98/go.mod h1:wyjw4MrxhEdQ3kVLcyXklmYKV0HjYm9F2E1YNuORRl4=
-github.com/micro/mdns v0.1.0 h1:fuLybUsfynbigJmCot/54i+gwe0hpc/vtCMvWt2WfDI=
-github.com/micro/mdns v0.1.0/go.mod h1:KJ0dW7KmicXU2BV++qkLlmHYcVv7/hHnbtguSWt9Aoc=
+github.com/micro/go-micro v1.8.0 h1:4etDwf3DqP61cM0xrI+3OP+FlAsRUnM2rwBXFRTrl+Y=
+github.com/micro/go-micro v1.8.0/go.mod h1:RUJTXVb+qUnGx6jBK2yMCOzuB/xEKh/vdZc3uagM5TI=
+github.com/micro/mdns v0.1.1-0.20190729112526-ef68c9635478 h1:L6jnZZ763dMLlvst8P0dWHa1WbUu7ppUY1q3AY2hhIU=
+github.com/micro/mdns v0.1.1-0.20190729112526-ef68c9635478/go.mod h1:KJ0dW7KmicXU2BV++qkLlmHYcVv7/hHnbtguSWt9Aoc=
 github.com/miekg/dns v1.0.14/go.mod h1:W1PPwlIAgtquWBMBEV9nkV9Cazfe8ScdGz/Lj7v3Nrg=
 github.com/miekg/dns v1.1.3/go.mod h1:W1PPwlIAgtquWBMBEV9nkV9Cazfe8ScdGz/Lj7v3Nrg=
 github.com/miekg/dns v1.1.15 h1:CSSIDtllwGLMoA6zjdKnaE6Tx6eVUxQ29LUgGetiDCI=

proto/service.hb.grpc.go

@@ -55,6 +55,7 @@ import (
 "github.com/golang/protobuf/proto"
 
 "github.com/hb-go/grpc-contrib/protoc-gen-hb-grpc/generator"
+_ "github.com/hb-go/grpc-contrib/protoc-gen-hb-grpc/plugin/desc"
 _ "github.com/hb-go/grpc-contrib/protoc-gen-hb-grpc/plugin/registry"
 )
 

protoc-gen-hb-grpc/main.go

@@ -0,0 +1,60 @@
+package desc
+
+import (
+pb "github.com/golang/protobuf/protoc-gen-go/descriptor"
+"github.com/hb-go/grpc-contrib/protoc-gen-hb-grpc/generator"
+)
+
+func init() {
+generator.RegisterPlugin(new(exportDesc))
+}
+
+type exportDesc struct {
+gen *generator.Generator
+}
+
+// Name returns the name of this plugin, "desc".
+func (g *exportDesc) Name() string {
+return "desc"
+}
+
+// The names for packages imported in the generated code.
+// They may vary from the final path component of the import path
+// if the name is used by other packages.
+var (
+pkgImports map[generator.GoPackageName]bool
+)
+
+// Init initializes the plugin.
+func (g *exportDesc) Init(gen *generator.Generator) {
+g.gen = gen
+}
+
+// P forwards to g.gen.P.
+func (g *exportDesc) P(args ...interface{}) { g.gen.P(args...) }
+
+// Generate generates code for the services in the given file.
+func (g *exportDesc) Generate(file *generator.FileDescriptor) {
+if len(file.FileDescriptorProto.Service) == 0 {
+return
+}
+
+g.P()
+g.P("// Export service desc")
+
+for i, service := range file.FileDescriptorProto.Service {
+g.generateService(file, service, i)
+}
+}
+
+func (g *exportDesc) GenerateImports(file *generator.FileDescriptor) {
+
+}
+
+func (g *exportDesc) generateService(file *generator.FileDescriptor, service *pb.ServiceDescriptorProto, index int) {
+g.P()
+g.P("// " + service.GetName() + " desc")
+g.P("func ServiceDesc" + service.GetName() + "() *grpc.ServiceDesc {")
+g.P("return &_" + service.GetName() + "_serviceDesc")
+g.P("}")
+}

protoc-gen-hb-grpc/plugin/desc/desc.go

@@ -20,7 +20,7 @@ type grpcRegistry struct {
 gen *generator.Generator
 }
 
-// Name returns the name of this plugin, "hb-grpc".
+// Name returns the name of this plugin, "registry".
 func (g *grpcRegistry) Name() string {
 return "registry"
 }
@@ -30,7 +30,6 @@ func (g *grpcRegistry) Name() string {
 // if the name is used by other packages.
 var (
 registryPkg string
-pkgImports map[generator.GoPackageName]bool
 )
 
 // Init initializes the plugin.
@@ -39,18 +38,6 @@ func (g *grpcRegistry) Init(gen *generator.Generator) {
 registryPkg = generator.RegisterUniquePackageName("registry", nil)
 }
 
-// Given a type name defined in a .proto, return its object.
-// Also record that we're using it, to guarantee the associated import.
-func (g *grpcRegistry) objectNamed(name string) generator.Object {
-g.gen.RecordTypeUse(name)
-return g.gen.ObjectNamed(name)
-}
-
-// Given a type name defined in a .proto, return its name as we will print it.
-func (g *grpcRegistry) typeName(str string) string {
-return g.gen.TypeName(g.objectNamed(str))
-}
-
 // P forwards to g.gen.P.
 func (g *grpcRegistry) P(args ...interface{}) { g.gen.P(args...) }