mirror of
https://github.com/gospider007/requests.git
synced 2025-12-24 13:57:52 +08:00
sync
This commit is contained in:
25
response.go
25
response.go
@@ -292,16 +292,23 @@ func (obj *Response) ReadBody() (err error) {
|
||||
}
|
||||
obj.readBody = true
|
||||
bBody := bytes.NewBuffer(nil)
|
||||
|
||||
if obj.requestOption.Bar && obj.ContentLength() > 0 {
|
||||
_, err = io.Copy(&barBody{
|
||||
bar: bar.NewClient(obj.response.ContentLength),
|
||||
body: bBody,
|
||||
}, obj.Body())
|
||||
} else {
|
||||
_, err = io.Copy(bBody, obj.Body())
|
||||
done := make(chan struct{})
|
||||
go func() {
|
||||
if obj.requestOption.Bar && obj.ContentLength() > 0 {
|
||||
_, err = io.Copy(&barBody{
|
||||
bar: bar.NewClient(obj.response.ContentLength),
|
||||
body: bBody,
|
||||
}, obj.Body())
|
||||
} else {
|
||||
_, err = io.Copy(bBody, obj.Body())
|
||||
}
|
||||
close(done)
|
||||
}()
|
||||
select {
|
||||
case <-obj.ctx.Done():
|
||||
err = obj.ctx.Err()
|
||||
case <-done:
|
||||
}
|
||||
|
||||
if obj.requestOption.Logger != nil {
|
||||
obj.requestOption.Logger(Log{
|
||||
Id: obj.requestOption.requestId,
|
||||
|
||||
Reference in New Issue
Block a user