前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >golang 上传文件到第三方接口

golang 上传文件到第三方接口

原创
作者头像
IT工作者
发布2022-04-05 16:40:55
4960
发布2022-04-05 16:40:55
举报
文章被收录于专栏:程序技术知识程序技术知识

定义连接超时及传输超时

代码语言:javascript
复制
//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 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档