mirror of
https://github.com/eiblog/eiblog.git
synced 2026-02-23 06:32:28 +08:00
update vendor
This commit is contained in:
63
vendor/github.com/qiniu/x/bytes.v7/seekable/seekable.go
generated
vendored
Normal file
63
vendor/github.com/qiniu/x/bytes.v7/seekable/seekable.go
generated
vendored
Normal file
@@ -0,0 +1,63 @@
|
||||
// This package provide a method to read and replace http.Request's body.
|
||||
package seekable
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
|
||||
"qiniupkg.com/x/bytes.v7"
|
||||
)
|
||||
|
||||
// ---------------------------------------------------
|
||||
|
||||
type Seekabler interface {
|
||||
Bytes() []byte
|
||||
Read(val []byte) (n int, err error)
|
||||
SeekToBegin() error
|
||||
}
|
||||
|
||||
type SeekableCloser interface {
|
||||
Seekabler
|
||||
io.Closer
|
||||
}
|
||||
|
||||
// ---------------------------------------------------
|
||||
|
||||
type readCloser struct {
|
||||
Seekabler
|
||||
io.Closer
|
||||
}
|
||||
|
||||
var ErrNoBody = errors.New("no body")
|
||||
|
||||
func New(req *http.Request) (r SeekableCloser, err error) {
|
||||
if req.Body == nil {
|
||||
return nil, ErrNoBody
|
||||
}
|
||||
var ok bool
|
||||
if r, ok = req.Body.(SeekableCloser); ok {
|
||||
return
|
||||
}
|
||||
b, err2 := ReadAll(req)
|
||||
if err2 != nil {
|
||||
return nil, err2
|
||||
}
|
||||
r = bytes.NewReader(b)
|
||||
req.Body = readCloser{r, req.Body}
|
||||
return
|
||||
}
|
||||
|
||||
func ReadAll(req *http.Request) (b []byte, err error) {
|
||||
if req.ContentLength > 0 {
|
||||
b = make([]byte, int(req.ContentLength))
|
||||
_, err = io.ReadFull(req.Body, b)
|
||||
return
|
||||
} else if req.ContentLength == 0 {
|
||||
return nil, ErrNoBody
|
||||
}
|
||||
return ioutil.ReadAll(req.Body)
|
||||
}
|
||||
|
||||
// ---------------------------------------------------
|
||||
43
vendor/github.com/qiniu/x/bytes.v7/seekable/seekable_test.go
generated
vendored
Normal file
43
vendor/github.com/qiniu/x/bytes.v7/seekable/seekable_test.go
generated
vendored
Normal file
@@ -0,0 +1,43 @@
|
||||
package seekable
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"net/http"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestSeekable_EOFIfReqAlreadyParsed(t *testing.T) {
|
||||
body := "a=1"
|
||||
req, err := http.NewRequest("POST", "/a", bytes.NewBufferString(body))
|
||||
assert.NoError(t, err)
|
||||
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||
req.Header.Set("Content-Length", "3")
|
||||
req.ParseForm()
|
||||
_, err = New(req)
|
||||
assert.Equal(t, err.Error(), "EOF")
|
||||
}
|
||||
|
||||
func TestSeekable_WorkaroundForEOF(t *testing.T) {
|
||||
body := "a=1"
|
||||
req, err := http.NewRequest("POST", "/a", bytes.NewBufferString(body))
|
||||
assert.NoError(t, err)
|
||||
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||
req.Header.Set("Content-Length", "3")
|
||||
_, _ = New(req)
|
||||
req.ParseForm()
|
||||
assert.Equal(t, req.FormValue("a"), "1")
|
||||
_, err = New(req)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestSeekable(t *testing.T) {
|
||||
body := "a=1"
|
||||
req, err := http.NewRequest("POST", "/a", bytes.NewBufferString(body))
|
||||
assert.NoError(t, err)
|
||||
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||
req.Header.Set("Content-Length", "3")
|
||||
_, err = New(req)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
Reference in New Issue
Block a user