/
data.go
74 lines (62 loc) · 1.41 KB
/
data.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
package lotus
import (
"errors"
"fmt"
"github.com/valyala/fasthttp"
"net/url"
"reflect"
)
type DataType int
const (
Binary DataType = 1
JSON = 2
Form = 3
)
type Payload interface{} // This will be much better with generics
type ServiceRequest struct {
RouteParams map[string]string
QueryParams map[string]string
Body Payload
DataType DataType
}
type Context struct {
*fasthttp.RequestCtx
ServiceClients []ServiceClient
}
func (ctx *Context) Payload() (Payload, error) {
if payload, ok := ctx.UserValue(DefaultKey).(interface{}); ok {
return payload, nil
}
return nil, errors.New("fail to convert payload")
}
func (ctx *Context) ServiceClient(sub ServiceContract) *ServiceClient {
for _, c := range ctx.ServiceClients {
if c.Label == sub.Label {
return &c
}
}
return nil
}
type RequestHandler func(ctx *Context)
func dataToUrlValues(data interface{}) (form url.Values, err error) {
form = map[string][]string{}
iValue := reflect.ValueOf(data)
for i := 0; i < iValue.NumField(); i++ {
k := iValue.Type().Field(i).Name
v := iValue.Field(i)
if form[k] == nil {
form[k] = []string{}
}
switch v.Kind() {
case reflect.Slice:
v2 := reflect.ValueOf(v.Interface())
for j := 0; j < v2.Len(); j++ {
f2 := v2.Index(j)
form[k] = append(form[k], fmt.Sprint(f2))
}
default:
form[k] = append(form[k], fmt.Sprint(v))
}
}
return form, err
}