Files
requests/rw.go
gospider 8883fd2cac sync
2025-06-10 14:11:15 +08:00

38 lines
749 B
Go

package requests
import (
"errors"
"io"
)
type wrapBody struct {
rawBody io.ReadCloser
conn *connecotr
}
func (obj *wrapBody) connStream() io.ReadWriteCloser {
return obj.conn.Conn.Stream()
}
func (obj *wrapBody) Read(p []byte) (n int, err error) {
return obj.rawBody.Read(p)
}
func (obj *wrapBody) Proxys() []Address {
return obj.conn.proxys
}
func (obj *wrapBody) CloseWithError(err error) error {
if err != nil {
obj.conn.CloseWithError(err)
}
return obj.rawBody.Close() //reuse conn
}
func (obj *wrapBody) Close() error {
return obj.CloseWithError(nil)
}
// safe close conn
func (obj *wrapBody) CloseConn() {
obj.conn.forceCnl(errors.New("readWriterCloser close conn"))
obj.conn.CloseWithError(errConnectionForceClosed)
}