go/kratos
1
0
Fork 0
mirror of https://github.com/go-kratos/kratos.git synced 2025-01-30 04:31:03 +02:00
Code Issues Releases Activity

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

Browse Source 
* feat: add processing when generating HTTP code
This commit is contained in:
包子 2021-08-09 10:28:20 +08:00 committed by GitHub
parent e5ae8dc3f1
commit ff78611766
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
cmd/protoc-gen-go-http/http.go
View File

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