feat(cmd/pgh): add processing when generating HTTP code (#1306)

* feat: add processing when generating HTTP code
pull/1311/head
包子 3 years ago committed by GitHub
parent e5ae8dc3f1
commit ff78611766
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 7
      cmd/protoc-gen-go-http/http.go

@ -2,6 +2,7 @@ package main
import ( import (
"fmt" "fmt"
"os"
"strings" "strings"
"github.com/go-kratos/kratos/v2" "github.com/go-kratos/kratos/v2"
@ -129,7 +130,10 @@ func buildHTTPRule(g *protogen.GeneratedFile, m *protogen.Method, rule *annotati
body = rule.Body body = rule.Body
responseBody = rule.ResponseBody responseBody = rule.ResponseBody
md := buildMethodDesc(g, m, method, path) md := buildMethodDesc(g, m, method, path)
if method == "GET" { if method == "GET" || method == "DELETE" {
if body != "" {
_, _ = fmt.Fprintf(os.Stderr, "\u001B[31mWARN\u001B[m: %s %s body should not be declared.\n", method, path)
}
md.HasBody = false md.HasBody = false
} else if body == "*" { } else if body == "*" {
md.HasBody = true md.HasBody = true
@ -139,6 +143,7 @@ func buildHTTPRule(g *protogen.GeneratedFile, m *protogen.Method, rule *annotati
md.Body = "." + camelCaseVars(body) md.Body = "." + camelCaseVars(body)
} else { } else {
md.HasBody = false md.HasBody = false
_, _ = fmt.Fprintf(os.Stderr, "\u001B[31mWARN\u001B[m: %s %s is does not declare a body.\n", method, path)
} }
if responseBody == "*" { if responseBody == "*" {
md.ResponseBody = "" md.ResponseBody = ""

Loading…
Cancel
Save