diff --git a/cmd/protoc-gen-drpc/main.go b/cmd/protoc-gen-drpc/main.go index 4d3a980..b7f51eb 100644 --- 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) }