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

修复fasthttp copy请求体截断的问题

Browse Source
This commit is contained in:
Liujian
2023-03-09 16:40:55 +08:00
parent 9d9e99bcab
commit 102f14daf9
5 changed files with 22 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -190,10 +190,12 @@ func (ctx *HttpContext) Clone() (eoscContext.EoContext, error) {
copyContext.proxyRequests = make([]http_service.IProxy, 0, 2)
req := fasthttp.AcquireRequest()
ctx.fastHttpRequestCtx.Request.CopyTo(req)
// 当body未读取，调用Body方法读出stream中当所有body内容，避免请求体被截断
ctx.proxyRequest.req.Body()
ctx.proxyRequest.req.CopyTo(req)
resp := fasthttp.AcquireResponse()
ctx.fastHttpRequestCtx.Response.CopyTo(resp)
//ctx.fastHttpRequestCtx.Response.CopyTo(resp)
copyContext.proxyRequest.reset(req, ctx.requestReader.remoteAddr)
copyContext.response.reset(resp)
@@ -224,7 +226,12 @@ func NewContext(ctx *fasthttp.RequestCtx, port int) *HttpContext {
httpContext.fastHttpRequestCtx = ctx
httpContext.requestID = uuid.New().String()
httpContext.requestReader.reset(&ctx.Request, remoteAddr)
// 原始请求最大读取body为8k，使用clone request
request := fasthttp.AcquireRequest()
ctx.Request.CopyTo(request)
httpContext.requestReader.reset(request, remoteAddr)
// proxyRequest保留原始请求
httpContext.proxyRequest.reset(&ctx.Request, remoteAddr)
httpContext.proxyRequests = httpContext.proxyRequests[:0]
httpContext.response.reset(&ctx.Response)