feat: add Comment2

fynntang · Aug 29, 2023 · 6ddd4c6 · 6ddd4c6
1 parent 59aa60a
commit 6ddd4c6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/protoc-gen-go-http/http.go b/protoc-gen-go-http/http.go
@@ -213,6 +213,7 @@ func buildMethodDesc(g *protogen.GeneratedFile, m *protogen.Method, method, path
 		Request:      g.QualifiedGoIdent(m.Input.GoIdent),
 		Reply:        g.QualifiedGoIdent(m.Output.GoIdent),
 		Comment:      comment,
+		Comment2:     m.Comments.Trailing.String(),
 		Path:         replacePathToGinPath(path),
 		Method:       method,
 		HasVars:      len(vars) > 0,

diff --git a/protoc-gen-go-http/template.go b/protoc-gen-go-http/template.go
@@ -11,7 +11,7 @@ var httpTemplate = `
 {{$svrName := .ServiceName}}
 
 {{- range .MethodSets}}
-const Route{{$svrType}}{{.OriginalName}} = "{{.Path}}" {{- if ne .Comment ""}} {{.Comment}} {{- end}}
+const Route{{$svrType}}{{.OriginalName}} = "{{.Path}}" {{- if ne .Comment2 ""}} {{.Comment2}} {{- end}}
 {{- end}}
 
 type {{.ServiceType}}HTTPServer interface {
@@ -108,6 +108,7 @@ type methodDesc struct {
 	Request      string
 	Reply        string
 	Comment      string
+	Comment2     string
 	// http_rule
 	Path         string
 	Method       string