tun2: connection: do things better
This commit is contained in:
parent
223b816f88
commit
64524eefc1
|
@ -4,7 +4,6 @@ import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"context"
|
"context"
|
||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"time"
|
"time"
|
||||||
|
@ -139,20 +138,12 @@ func (c *Connection) RoundTrip(req *http.Request) (*http.Response, error) {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, errors.Wrap(err, ErrCantReadResponse.Error())
|
return nil, errors.Wrap(err, ErrCantReadResponse.Error())
|
||||||
}
|
}
|
||||||
defer resp.Body.Close()
|
|
||||||
|
|
||||||
body, err := ioutil.ReadAll(resp.Body)
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.Wrap(err, "can't read response body")
|
|
||||||
}
|
|
||||||
|
|
||||||
resp.Body = &mixedReadCloser{
|
resp.Body = &mixedReadCloser{
|
||||||
Reader: buf,
|
Reader: buf,
|
||||||
Closer: stream,
|
Closer: stream,
|
||||||
}
|
}
|
||||||
|
|
||||||
resp.ContentLength = int64(len(body))
|
|
||||||
|
|
||||||
return resp, nil
|
return resp, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue