yutingNie
/
hertz
forked from cloudwego/hertz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(hz): handle error in client call's template (#516)

This commit is contained in:
kinggo 2023-01-06 17:16:36 +08:00 committed by GitHub
parent 4d7af18988
commit 6e1b64174e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
cmd/hz/generator/package_tpl.go
View File

@ -289,9 +289,12 @@ type use interface {
// Definition of global data and types.
type ResponseResultDecider func(statusCode int, rawResponse *protocol.Response) (isError bool)
type bindRequestBodyFunc func(c *cli, r *request) (contentType string, body io.Reader, err error)
type beforeRequestFunc func(*cli, *request) error
type afterResponseFunc func(*cli, *response) error
type (
bindRequestBodyFunc func(c *cli, r *request) (contentType string, body io.Reader, err error)
beforeRequestFunc func(*cli, *request) error
afterResponseFunc func(*cli, *response) error
)
var (
hdrContentTypeKey = http.CanonicalHeaderKey("Content-Type")
@ -906,11 +909,13 @@ func (s *{{$.ServiceName}}Client) {{$MethodInfo.Name}}(context context.Context,
setRequestOption(reqOpt...).
setResult(httpResp).
execute("{{$MethodInfo.HTTPMethod}}", "{{$MethodInfo.Path}}")
if err == nil {
resp = httpResp
if err != nil {
return nil, nil, err
}
resp = httpResp
rawResponse = ret.rawResponse
return resp, rawResponse, err
return resp, rawResponse, nil
}
{{end}}