starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-03-22 17:19:06 +01:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

small refactor

Browse Source
This commit is contained in:
yusing
2025-02-20 17:45:03 +08:00
parent 3ce3520c45
commit c66b17583f
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
internal/net/gphttp/reverseproxy/reverse_proxy_mod.go
View File

@@ -410,7 +410,7 @@ func (p *ReverseProxy) handler(rw http.ResponseWriter, req *http.Request) {
rw.WriteHeader(res.StatusCode)
err = U.CopyClose(U.NewContextWriter(ctx, rw), U.NewContextReader(ctx, res.Body)) // close now, instead of defer, to populate res.Trailer
err = U.CopyCloseWithContext(ctx, rw, res.Body) // close now, instead of defer, to populate res.Trailer
if err != nil {
if !errors.Is(err, context.Canceled) {
p.errorHandler(rw, req, err, false)

4
internal/utils/io.go
View File

@@ -223,3 +223,7 @@ func CopyClose(dst *ContextWriter, src *ContextReader) (err error) {
}
}
}
func CopyCloseWithContext(ctx context.Context, dst io.Writer, src io.Reader) (err error) {
return CopyClose(NewContextWriter(ctx, dst), NewContextReader(ctx, src))
}