forked from gojek/fiber
-
Notifications
You must be signed in to change notification settings - Fork 0
/
request.go
94 lines (75 loc) · 2.14 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
package http
import (
"bytes"
"fmt"
"io"
"io/ioutil"
"net/http"
"net/url"
"github.com/gojek/fiber"
"github.com/gojek/fiber/protocol"
)
// Request wraps a standard http request
type Request struct {
*fiber.CachedPayload
*http.Request
}
func (r *Request) Protocol() protocol.Protocol {
return protocol.HTTP
}
func (r *Request) Header() map[string][]string {
return r.Request.Header
}
// NewHTTPRequest initialize a new client request from incoming server request
func NewHTTPRequest(req *http.Request) (*Request, error) {
// RequestURI can't be set in client requests
req.RequestURI = ""
var payload *fiber.CachedPayload
if req.ContentLength == 0 && req.Body == nil {
payload = new(fiber.CachedPayload)
req.GetBody = func() (io.ReadCloser, error) { return http.NoBody, nil }
} else {
defer req.Body.Close()
data, err := ioutil.ReadAll(req.Body)
if err != nil {
return nil, err
}
payload = fiber.NewCachedPayload(data)
req.GetBody = func() (io.ReadCloser, error) {
r := bytes.NewReader(data)
return ioutil.NopCloser(r), nil
}
}
return &Request{Request: req, CachedPayload: payload}, nil
}
// Copy creates a deep copy of this request
func (r *Request) Clone() (fiber.Request, error) {
bodyReader := bytes.NewReader(r.Payload())
proxyRequest, err := http.NewRequest(r.Method, r.URL.String(), bodyReader)
if err != nil {
return nil, err
}
proxyRequest.GetBody = func() (io.ReadCloser, error) {
return ioutil.NopCloser(bodyReader), nil
}
proxyRequest.Header = r.Header()
return &Request{CachedPayload: r.CachedPayload, Request: proxyRequest}, nil
}
func (r *Request) OperationName() string {
return fmt.Sprintf("%s %s", r.Method, r.URL.Path)
}
func (r *Request) Transform(backend fiber.Backend) (fiber.Request, error) {
reqPath := ""
// If the URL is set, extract the URI. If nothing is set, ignore.
// Calling RequestURI() on the empty url will return a '/' and the
// backend server may or may not handle it, so skip instead.
if r.URL.String() != "" {
reqPath = r.URL.RequestURI()
}
updatedURL, err := url.Parse(backend.URL(reqPath))
if err != nil {
return nil, err
}
r.URL = updatedURL
return r, nil
}