zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feature : responses whit context (#2637)

Browse Source
This commit is contained in:
heyehang
2022-12-03 18:48:02 +08:00
committed by GitHub
parent 9941055eaa
commit a644ec7edd
5 changed files with 192 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
gateway/server.go
View File

@@ -122,7 +122,7 @@ func (s *Server) buildHandler(source grpcurl.DescriptorSource, resolver jsonpb.A
return func(w http.ResponseWriter, r *http.Request) {
parser, err := internal.NewRequestParser(r, resolver)
if err != nil {
httpx.Error(w, err)
httpx.ErrorCtx(r.Context(), w, err)
return
}
@@ -134,12 +134,12 @@ func (s *Server) buildHandler(source grpcurl.DescriptorSource, resolver jsonpb.A
handler := internal.NewEventHandler(w, resolver)
if err := grpcurl.InvokeRPC(ctx, source, cli.Conn(), rpcPath, s.prepareMetadata(r.Header),
handler, parser.Next); err != nil {
httpx.Error(w, err)
httpx.ErrorCtx(r.Context(), w, err)
}
st := handler.Status
if st.Code() != codes.OK {
httpx.Error(w, st.Err())
httpx.ErrorCtx(r.Context(), w, st.Err())
}
}
}