forked from lonng/nanoserver
/
params.go
122 lines (107 loc) · 2.2 KB
/
params.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
120
121
122
package algoutil
import (
"bytes"
"encoding/json"
"fmt"
"net/url"
"sort"
"strings"
"github.com/yearnfar/nanoserver/pkg/errutil"
)
const (
argSep = "&"
kvSep = "="
)
func ParseParams(params string) map[string]string {
m := map[string]string{}
parts := strings.Split(params, argSep)
for _, arg := range parts {
// strings.Split(s, "=") will cause error when signature has
// padding(that is something like "==")
i := strings.IndexAny(arg, kvSep)
if i < 0 {
continue
}
k := arg[:i]
v := arg[i+1:]
m[k] = v
}
return m
}
// TODO: WARNING!!!!!!!
// All struct field must be string type
func ParamsToStruct(params string, v interface{}) error {
if params == "" || len(strings.TrimSpace(params)) < 1 {
return errutil.ErrIllegalParameter
}
m := ParseParams(params)
bytes, err := json.Marshal(m)
if err != nil {
return err
}
return json.Unmarshal(bytes, v)
}
func SortParams(m map[string]string) string {
if len(m) == 0 {
return ""
}
var (
i = 0
keys = make([]string, len(m))
buf = &bytes.Buffer{}
)
for k := range m {
keys[i] = k
i++
}
sort.Strings(keys)
for _, k := range keys {
buf.WriteString(k)
buf.WriteString("=")
buf.WriteString(m[k])
buf.WriteString("&")
}
buf.Truncate(buf.Len() - 1)
return buf.String()
}
func ConcatWithURLEncode(params map[string]string) *bytes.Buffer {
if params == nil || len(params) == 0 {
return nil
}
buf := &bytes.Buffer{}
for k, v := range params {
buf.WriteString(k)
buf.WriteString("=")
buf.WriteString(url.QueryEscape(v))
buf.WriteString("&")
}
buf.Truncate(buf.Len() - 1)
return buf
}
// SortAndConcat sort the map by key in ASCII order,
// and concat it in form of "k1=v1&k2=v2"
func SortAndConcat(params map[string]string, extras ...bool) []byte {
if params == nil || len(params) == 0 {
return nil
}
var trimSpace bool = true
if len(extras) > 0 {
trimSpace = extras[0]
}
keys := make([]string, len(params))
i := 0
for k := range params {
keys[i] = k
i++
}
sort.Strings(keys)
buf := &bytes.Buffer{}
for _, k := range keys {
if trimSpace && params[k] == "" {
continue
}
buf.WriteString(fmt.Sprintf("%s=%s&", k, params[k]))
}
buf.Truncate(buf.Len() - 1)
return buf.Bytes()
}