-
Notifications
You must be signed in to change notification settings - Fork 327
/
request.go
119 lines (105 loc) · 2.32 KB
/
request.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
package http
import (
"bytes"
"encoding/json"
"fmt"
"io"
"mime/multipart"
"net/http"
"net/url"
)
type (
Request struct {
*http.Request
Body io.Reader
}
// The BufferedReader behaves exactly like a bytes.Reader, with the exception
// when the last block is read, it automatically rewinds the internal pointer to the start,
// so effectively, the content can be read again without calling Seek() externally.
BufferedReader struct {
s []byte
i int64 // current reading index
prevRune int // index of previous rune; or < 0
}
)
func NewRequest(r *http.Request) (rr *Request, err error) {
rs, err := NewBufferedReader(r.Body)
if err != nil {
return
}
rr = &Request{r, rs}
return
}
// NewBufferedReader copies original data to the
// BufferedReader
func NewBufferedReader(rr io.Reader) (bb *BufferedReader, err error) {
var (
buf = &bytes.Buffer{}
)
bb = &BufferedReader{}
_, err = io.Copy(buf, rr)
if err != nil {
return
}
return &BufferedReader{
s: buf.Bytes(),
i: 0,
prevRune: -1,
}, nil
}
func (r *BufferedReader) Read(b []byte) (n int, err error) {
if r.i >= int64(len(r.s)) {
n = 0
err = io.EOF
r.Seek(0, io.SeekStart)
return
}
r.prevRune = -1
n = copy(b, r.s[r.i:])
r.i += int64(n)
return
}
func (r *BufferedReader) Seek(offset int64, whence int) (int64, error) {
r.prevRune = -1
var abs int64
switch whence {
case io.SeekStart:
abs = offset
case io.SeekCurrent:
abs = r.i + offset
case io.SeekEnd:
abs = int64(len(r.s)) + offset
default:
return 0, fmt.Errorf("bytes.Reader.Seek: invalid whence")
}
if abs < 0 {
return 0, fmt.Errorf("bytes.Reader.Seek: negative position")
}
r.i = abs
return abs, nil
}
func (bb *Request) MarshalJSON() ([]byte, error) {
return json.Marshal(&struct {
Method string
URL *url.URL
Header http.Header
ContentLength int64
Host string
Form url.Values
PostForm url.Values
MultipartForm *multipart.Form
RemoteAddr string
RequestURI string
}{
Method: bb.Method,
URL: bb.URL,
Header: bb.Header,
ContentLength: bb.ContentLength,
Host: bb.Host,
Form: bb.Form,
PostForm: bb.PostForm,
MultipartForm: bb.MultipartForm,
RemoteAddr: bb.RemoteAddr,
RequestURI: bb.RequestURI,
})
}