package/requests
1
0
Fork 0
mirror of https://github.com/gospider007/requests.git synced 2025-12-24 13:57:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

sync

Browse Source
This commit is contained in:
gospider
2025-08-13 10:55:40 +08:00
parent 61790deab6
commit 31ac1d77ca
10 changed files with 202 additions and 211 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
roundTripper.go
View File

@@ -132,17 +132,7 @@ func (obj *roundTripper) ghttp3Dial(ctx *Response, remoteAddress Address, proxyA
if err != nil {
return nil, err
}
cctx, ccnl := context.WithCancelCause(obj.ctx)
conn = http3.NewConn(cctx, netConn, udpConn, func() {
ccnl(errors.New("http3 client close"))
})
if ct, ok := udpConn.(interface {
SetTcpCloseFunc(f func(error))
}); ok {
ct.SetTcpCloseFunc(func(err error) {
ccnl(errors.New("http3 client close with udp"))
})
}
conn = http3.NewConn(obj.ctx, netConn, udpConn)
return
}
@@ -182,17 +172,7 @@ func (obj *roundTripper) uhttp3Dial(ctx *Response, remoteAddress Address, proxyA
if err != nil {
return nil, err
}
cctx, ccnl := context.WithCancelCause(obj.ctx)
conn = http3.NewConn(cctx, netConn, udpConn, func() {
ccnl(errors.New("http3 client close"))
})
if ct, ok := udpConn.(interface {
SetTcpCloseFunc(f func(error))
}); ok {
ct.SetTcpCloseFunc(func(err error) {
ccnl(errors.New("uhttp3 client close with udp"))
})
}
conn = http3.NewConn(obj.ctx, netConn, udpConn)
return
}
@@ -270,19 +250,14 @@ func (obj *roundTripper) dial(ctx *Response) (conn http1.Conn, err error) {
return obj.dialConnecotr(ctx, rawConn, h2)
}
func (obj *roundTripper) dialConnecotr(ctx *Response, rawCon net.Conn, h2 bool) (conn http1.Conn, err error) {
cctx, ccnl := context.WithCancelCause(obj.ctx)
if h2 {
var spec *http2.Spec
if ctx.option.gospiderSpec != nil {
spec = ctx.option.gospiderSpec.H2Spec
}
conn, err = http2.NewConn(cctx, ctx.Context(), rawCon, spec, func(err error) {
ccnl(tools.WrapError(err, "http2 client close"))
})
conn, err = http2.NewConn(obj.ctx, ctx.Context(), rawCon, spec)
} else {
conn = http1.NewConn(cctx, rawCon, func(err error) {
ccnl(tools.WrapError(err, "http1 client close"))
})
conn = http1.NewConn(obj.ctx, rawCon)
}
return
}