mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-06 01:50:26 +08:00
36 lines
639 B
Go
36 lines
639 B
Go
package requests
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
)
|
|
|
|
type Request struct {
|
|
*http.Request
|
|
BodyData []byte
|
|
}
|
|
|
|
func NewRequest(raw *http.Request) *Request {
|
|
return &Request{
|
|
Request: raw,
|
|
}
|
|
}
|
|
|
|
func (this *Request) Raw() *http.Request {
|
|
return this.Request
|
|
}
|
|
|
|
func (this *Request) ReadBody(max int64) (data []byte, err error) {
|
|
data, err = ioutil.ReadAll(io.LimitReader(this.Request.Body, max))
|
|
return
|
|
}
|
|
|
|
func (this *Request) RestoreBody(data []byte) {
|
|
rawReader := bytes.NewBuffer(data)
|
|
buf := make([]byte, 1024)
|
|
io.CopyBuffer(rawReader, this.Request.Body, buf)
|
|
this.Request.Body = ioutil.NopCloser(rawReader)
|
|
}
|