add comments to distinguish drpc generated code

egonelbre · egonelbre · commit 368f26ca5b7f · 2020-02-26T13:03:05.000Z
Change-Id: I938d4ebec515dafd0834a05532de51cc513c8c6c
diff --git a/cmd/protoc-gen-drpc/main.go b/cmd/protoc-gen-drpc/main.go
@@ -59,6 +59,9 @@ func (d *drpc) Generate(file *generator.FileDescriptor) {
 	d.drpcServer = d.drpcPkg + ".Server"
 	d.contextPkg = string(d.AddImport("context"))
 
+	d.P("// --- DRPC BEGIN ---\n")
+	defer d.P("// --- DRPC END ---\n")
+
 	for i, service := range file.FileDescriptorProto.Service {
 		d.generateService(file, service, i)
 	}

Original file line number	Diff line number	Diff line change
`@@ -59,6 +59,9 @@ func (d drpc) Generate(file generator.FileDescriptor) {`
`59`	`59`	`d.drpcServer = d.drpcPkg + ".Server"`
`60`	`60`	`d.contextPkg = string(d.AddImport("context"))`
`61`	`61`
	`62`	`+ d.P("// --- DRPC BEGIN ---\n")`
	`63`	`+ defer d.P("// --- DRPC END ---\n")`
	`64`	`+`
`62`	`65`	`for i, service := range file.FileDescriptorProto.Service {`
`63`	`66`	`d.generateService(file, service, i)`
`64`	`67`	`}`