coco
/
protoc-gen-coco
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

fix: autogen router map

This commit is contained in:
Young Xu 2023-03-22 00:29:59 +08:00
parent 0a81c9ad68
commit 881445d337
Signed by: xuthus5
GPG Key ID: A23CF9620CBB55F9
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
main.go
View File

@ -54,19 +54,6 @@ func (c *Coco) generateRouterMap(plugin *protogen.Plugin) {
if len(pbFile.Services) == 0 {
continue
}
// write file header
filename := fmt.Sprintf("autogen_router_%s.go", pbFile.GeneratedFilenamePrefix)
g := plugin.NewGeneratedFile(filename, pbFile.GoImportPath)
g.P("// Code generated by protoc-gen-coco. DO NOT EDIT.")
g.P("// source: ", pbFile.GeneratedFilenamePrefix, ".proto")
g.P("// generate at: ", time.Now().Format("2006-01-02 15:04:05"))
g.P()
g.P("package ", pbFile.GoPackageName)
g.P()
g.P(`import (
"gitter.top/coco/coco/core"
)`)
g.P()
for _, service := range pbFile.Services {
// is service router group
@ -76,6 +63,19 @@ func (c *Coco) generateRouterMap(plugin *protogen.Plugin) {
if len(service.Methods) == 0 {
continue
}
// write file header
filename := fmt.Sprintf("%s/autogen_router_%s.go", pbFile.GeneratedFilenamePrefix, service.GoName)
g := plugin.NewGeneratedFile(filename, pbFile.GoImportPath)
g.P("// Code generated by protoc-gen-coco. DO NOT EDIT.")
g.P("// source: ", pbFile.GeneratedFilenamePrefix, ".proto")
g.P("// generate at: ", time.Now().Format("2006-01-02 15:04:05"))
g.P()
g.P("package ", pbFile.GoPackageName)
g.P()
g.P(`import (
"gitter.top/coco/coco/core"
)`)
g.P()
values, err := GenerateRouterMap(service)
if err != nil {
_, _ = fmt.Fprintf(os.Stderr, "generate router map failed: %v", err)