zishuo/apinto
1
0
Fork 0
mirror of https://github.com/eolinker/apinto synced 2025-09-27 05:06:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

tmp commit

Browse Source
This commit is contained in:
Liujian
2025-02-21 17:41:05 +08:00
parent d53b05151a
commit 8a59d592d0
3 changed files with 25 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
node/http-context/context.go
View File

@@ -3,6 +3,7 @@ package http_context
import (
"context"
"fmt"
"io"
"net"
"strconv"
"strings"
@@ -201,17 +202,29 @@ func (ctx *HttpContext) SendTo(scheme string, node eoscContext.INode, timeout ti
response.Header.CopyTo(&ctx.response.Response.Header)
ctx.response.ResponseHeader.refresh()
if response.IsBodyStream() && response.Header.ContentLength() < 0 {
reader := response.BodyStream()
ctx.response.Response.SetStatusCode(response.StatusCode())
bodyStream := NewBodyStream(response.BodyStream())
ctx.response.bodyStream = bodyStream
ctx.response.Response.SetBodyStream(bodyStream, -1)
agent.setResponseLength(ctx.response.Response.Header.ContentLength())
buf := make([]byte, 4096)
for {
n, err := reader.Read(buf)
if err == io.EOF {
return nil
}
if err != nil {
return err
}
agent.responseBody.Write(response.Body()[:n])
_, err = ctx.response.Response.BodyWriter().Write(buf[:n])
if err != nil {
return err
}
}
agent.setResponseLength(-1)
ctx.proxyRequests = append(ctx.proxyRequests, agent)
return nil
}
response.CopyTo(ctx.response.Response)
agent.responseBody.Write(ctx.response.Response.Body())
agent.setResponseLength(ctx.response.Response.Header.ContentLength())
ctx.proxyRequests = append(ctx.proxyRequests, agent)