105 lines
2.4 KiB
Plaintext
105 lines
2.4 KiB
Plaintext
|
package webUtil
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"crypto/tls"
|
|||
|
|
"fmt"
|
|||
|
|
"net"
|
|||
|
|
"net/http"
|
|||
|
|
"sort"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"goutil/netUtil"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 获取请求的IP(obsolete, 直接调用netUtil.GetHttpAddr)
|
|||
|
|
// 返回值：
|
|||
|
|
// 请求的IP
|
|||
|
|
func GetRequestIP(r *http.Request) string {
|
|||
|
|
/*
|
|||
|
|
http中调用JoinHostPort来给RemoteAddr赋值；它的规则如下：
|
|||
|
|
JoinHostPort combines host and port into a network address of the
|
|||
|
|
form "host:port" or, if host contains a colon or a percent sign,
|
|||
|
|
"[host]:port".
|
|||
|
|
|
|||
|
|
所以现在要将RemoteAddr解析成host和port，则需要找到最后一个:，前面的部分则是host；
|
|||
|
|
如果host包含[]，则需要去除
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
return netUtil.GetHttpAddr(r).Host
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewTransport() *http.Transport {
|
|||
|
|
return &http.Transport{}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// timeout 超时时间
|
|||
|
|
func GetTimeoutTransport(transport *http.Transport, timeout int) *http.Transport {
|
|||
|
|
transport.Dial = func(netw, addr string) (net.Conn, error) {
|
|||
|
|
deadline := time.Now().Add(time.Duration(timeout) * time.Second)
|
|||
|
|
c, err := net.DialTimeout(netw, addr, time.Second*time.Duration(timeout))
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
c.SetDeadline(deadline)
|
|||
|
|
return c, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return transport
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// b 表示是否需要验证http证书
|
|||
|
|
func GetTLSTransport(transport *http.Transport, b bool) *http.Transport {
|
|||
|
|
transport.TLSClientConfig = &tls.Config{InsecureSkipVerify: b}
|
|||
|
|
|
|||
|
|
return transport
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 组装请求字符串
|
|||
|
|
// requestMap:请求数据集合
|
|||
|
|
func AssembleRequestParam(requestMap map[string]string) (dataStr string) {
|
|||
|
|
count := 0
|
|||
|
|
for k, v := range requestMap {
|
|||
|
|
if count == len(requestMap)-1 {
|
|||
|
|
dataStr += fmt.Sprintf("%s=%s", k, v)
|
|||
|
|
} else {
|
|||
|
|
dataStr += fmt.Sprintf("%s=%s&", k, v)
|
|||
|
|
}
|
|||
|
|
count += 1
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 按照参数名称升序组装请求字符串
|
|||
|
|
// requestMap:请求数据集合
|
|||
|
|
// asc:是否升序
|
|||
|
|
func AssembleRequestParamSort(requestMap map[string]string, asc bool) (dataStr string) {
|
|||
|
|
finalKeys := make([]string, 0, len(requestMap))
|
|||
|
|
|
|||
|
|
// 按key进行排序
|
|||
|
|
sortKeys := make([]string, 0, len(requestMap))
|
|||
|
|
for k, _ := range requestMap {
|
|||
|
|
sortKeys = append(sortKeys, k)
|
|||
|
|
}
|
|||
|
|
sort.Strings(sortKeys)
|
|||
|
|
|
|||
|
|
// 排序是升序还是降序
|
|||
|
|
if asc {
|
|||
|
|
finalKeys = sortKeys
|
|||
|
|
} else {
|
|||
|
|
for i := len(sortKeys) - 1; i >= 0; i-- {
|
|||
|
|
finalKeys = append(finalKeys, sortKeys[i])
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for index, key := range finalKeys {
|
|||
|
|
if index == len(finalKeys)-1 {
|
|||
|
|
dataStr += fmt.Sprintf("%s=%s", key, requestMap[key])
|
|||
|
|
} else {
|
|||
|
|
dataStr += fmt.Sprintf("%s=%s&", key, requestMap[key])
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|