You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
jd/params.go

98 lines
1.7 KiB

package jd
import (
"encoding/json"
"net/url"
"sort"
"strconv"
"time"
)
// Params 请求参数
type Params map[string]interface{}
func NewParams() Params {
p := make(Params)
return p
}
func NewParamsWith(params ...Params) Params {
p := make(Params)
for _, v := range params {
p.SetParams(v)
}
return p
}
func NewParamsWithType(_method string, params ...Params) Params {
p := make(Params)
p["method"] = _method
hh, _ := time.ParseDuration("8h")
loc := time.Now().UTC().Add(hh)
p["timestamp"] = loc.Format("2006-01-02 15:04:05")
p["format"] = "json"
p["v"] = "1.0"
p["sign_method"] = "md5"
for _, v := range params {
p.SetParams(v)
}
return p
}
func (c *Client) Sign(p Params) {
p["app_key"] = c.GetAppKey()
// 排序所有的 key
var keys []string
for key := range p {
keys = append(keys, key)
}
sort.Strings(keys)
signStr := c.GetSecretKey()
for _, key := range keys {
signStr += key + getString(p[key])
}
signStr += c.GetSecretKey()
p["sign"] = createSign(signStr)
}
func (p Params) Set(key string, value interface{}) {
p[key] = value
}
func (p Params) SetParams(params Params) {
for key, value := range params {
p[key] = value
}
}
func (p Params) SetCustomParameters(uid string, sid string) {
p["custom_parameters"] = map[string]interface{}{
"uid": uid,
"sid": sid,
}
}
func (p Params) GetQuery() string {
u := url.Values{}
for k, v := range p {
u.Set(k, getString(v))
}
return u.Encode()
}
func getString(i interface{}) string {
switch v := i.(type) {
case string:
return v
case []byte:
return string(v)
case int:
return strconv.Itoa(v)
case bool:
return strconv.FormatBool(v)
default:
bytes, _ := json.Marshal(v)
return string(bytes)
}
}