Feat: add grpc provider reference in codes generated by protoc-gen-du…

…bbo3 (#1240) * fix: change 3.0 triple provider reference * fix: change 3.0 triple provider reference * Fix: add liscence
apache · Jun 7, 2021 · 3e881f6 · 3e881f6
1 parent b5cd555
commit 3e881f6
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/protocol/dubbo3/protoc-gen-dubbo3/examples/helloworld.pb.go b/protocol/dubbo3/protoc-gen-dubbo3/examples/helloworld.pb.go
diff --git a/protocol/dubbo3/protoc-gen-dubbo3/plugin/dubbo3/dubbo3.go b/protocol/dubbo3/protoc-gen-dubbo3/plugin/dubbo3/dubbo3.go
@@ -240,6 +240,12 @@ func (g *dubboGrpc) generateService(file *generator.FileDescriptor, service *pb.
 	g.P("}")
 	g.P()
 
+	// return reference
+	g.P("func (c *", serverType, ") ", " Reference() string ", "{")
+	g.P(`return "`, unexport(servName), `Impl"`)
+	g.P("}")
+	g.P()
+
 	// add handler
 	var handlerNames []string
 	for _, method := range service.Method {

diff --git a/protocol/grpc/protoc-gen-dubbo/plugin/dubbo/dubbo.go b/protocol/grpc/protoc-gen-dubbo/plugin/dubbo/dubbo.go
@@ -188,6 +188,12 @@ func (g *dubboGrpc) generateService(file *generator.FileDescriptor, service *pb.
 	g.P("}")
 	g.P()
 
+	// return reference
+	g.P("func (c *", serverType, ") ", " Reference() string ", "{")
+	g.P(`return "`, unexport(servName), `Impl"`)
+	g.P("}")
+	g.P()
+
 	// add handler
 	var handlerNames []string
 	for _, method := range service.Method {