定义连接超时及传输超时
//TimeoutDialer 连接超时和传输超时
func TimeoutDialer(cTimeout time.Duration, rwTimeout time.Duration) func(net,addr string)
(c net.Conn, err error) {
return func(netw, addr string) (net.Conn, error) {
conn, err := net.DialTimeout(netw, addr, cTimeout)
if err != nil {
return nil, er
}
conn.SetDeadline(time.Now().Add(rwTimeout))
return conn, nil
}
}
//封装文件上传函数(本实例采用put上传到第三方接口)
//UploadFile 上传文件 rURL为第三方接口url,b为文件内容,header为自定义header头
func UploadFile(rURL string, b []byte, header map[string]string) error {
req, err := http.NewRequest("PUT", rURL, bytes.NewBuffer(b))
if err != nil {
logging.Info(fmt.Sprintf("http newrequest error %s", err))
return er
}
for h, v := range header {
req.Header.Set(h, v)
}
connectTimeout := 120 * time.Second
readWriteTimeout := 5184000 * time.Millisecond
client := http.Client{
//忽略证书验证
Transport: &http.Transport{
TLSClientConfig: &tls.Config{
InsecureSkipVerify: false,
},
Dial: TimeoutDialer(connectTimeout, readWriteTimeout),
},
}
resp, err := client.Do(req)
if err != nil {
logging.Info(fmt.Sprintf("http client error %s", err))
return er
}
defer resp.Body.Close()
if resp != nil {
// 判断请求状态
if resp.StatusCode == http.StatusOK {
respData, err := ioutil.ReadAll(resp.Body)
if err != nil {
logging.Info(err)
return er
}
logging.Info(fmt.Sprintf("\n请求地址: %s \n请求参数: %s \n请求头: %s \n返回 :%s\n",
rURL, "上传文件", header, string(respData)))
fmt.Println(string(respData))
return nil
} else if resp.StatusCode != http.StatusOK {
respData, err := ioutil.ReadAll(resp.Body)
if err != nil {
logging.Info(err)
return er
}
logging.Info(fmt.Sprintf("\n请求地址: %s \n请求参数: %s \n请求头: %s \n返回 :%s\n",
rURL, "上传文件", header, string(respData)))
return errors.New("上传文件请求成功,上传失败")
}
return errors.New("请求失败")
}
return errors.New("请求失败")
}
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。