dqve65954 2018-08-27 15:02
浏览 69
已采纳

Golang频道串流

I am relatively new to Golang and don't fully understand streams. I have a function (circuit breaker function) that is making Rest calls. I have it working but it is only streaming the "responseBody" back. I would actually like to stream back the entire request of stream back both the Body and the Header together.

When I try to use a similar approach on the "header" then I get an error that the header is not streamable.

Is there a good/best way to accomplish this? Thank you. Below is my function.

func CallWithRetries(req *http.Request, output chan []byte) error {
    r := retrier.New(retrier.ConstantBackoff(RETRIES, 100 * time.Millisecond), nil)
    attempt := 0
    err := r.Run(func() error {
        attempt++
        resp, err := Client.Do(req)
        if err == nil && resp.StatusCode < 299 {
            responseBody, err := ioutil.ReadAll(resp.Body)
            if err == nil {
                output <- responseBody
                return err
            }
            return err
        } else if err == nil {
            customLogger.LogDebug("Status code was: " , transactionId)
            err = fmt.Errorf("Status was %v", resp.StatusCode)
        }
        return err
    })
    return err
}
  • 写回答

1条回答 默认 最新

  • dongzhang6544 2018-08-27 17:17
    关注

    You are looking for the httputil.DumpResponse function.

    The code might be changed to something similar to

    func CallWithRetries(req *http.Request, output chan []byte) error {
        r := retrier.New(retrier.ConstantBackoff(RETRIES, 100*time.Millisecond), nil)
        attempt := 0
        err := r.Run(func() error {
            attempt++
            resp, err := Client.Do(req)
            if err == nil && resp.StatusCode < 299 {
                dump, err := httputil.DumpResponse(resp, true)
                if err == nil {
                    output <- dump
                    return err
                }
                return err
            } else if err == nil {
                customLogger.LogDebug("Status code was: ", transactionId)
                err = fmt.Errorf("Status was %v", resp.StatusCode)
            }
            return err
        })
        return err
    }
    

    Side notes,

    • you might want to consider to close the response body as mentioned in the documentation https://golang.org/pkg/net/http/#Client.Get

    • Looks likes the err variable is shadowed, this should be modified to avoid any surprises.

    This version of the code attempts to return errors early, and to close the response body. It was not tested, only written on the fly, to use with care.

    func CallWithRetries(req *http.Request, output chan []byte) error {
        r := retrier.New(retrier.ConstantBackoff(RETRIES, 100*time.Millisecond), nil)
        attempt := 0
        return r.Run(func() error {
            attempt++
            var resp *http.Response
            {
                r, err := Client.Do(req)
                if err != nil {
                    return err
                }
                defer r.Body.Close()
                if resp.StatusCode > 299 {
                    customLogger.LogDebug("Status code was: ", transactionId)
                    return fmt.Errorf("Status was %v", resp.StatusCode)
                }
                resp = r
            }
            var out []byte
            {
                x, err := httputil.DumpResponse(resp, true)
                if err != nil {
                    return err
                }
                out = x
            }
            output <- out
            return nil
        })
    }
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

悬赏问题

  • ¥15 关于#cesium#的问题:cesium 根据经纬度、高度、方位角、俯仰角和距离求另一点高度
  • ¥15 word脚注设置不起空格符号
  • ¥15 为什么apriori的运行时间会比fp growth的运行时间短呢
  • ¥15 求解密json或者美化
  • ¥15 帮我解决一下膳食平衡的线性规划模型的数据实例
  • ¥40 万年历缺少农历,需要和阳历同时显示
  • ¥250 雷电模拟器内存穿透、寻基址和特征码的教学
  • ¥200 比特币ord程序wallet_constructor.rs文件支持一次性铸造1000个代币,并将它们分配到40个UTXO上(每个UTXO上分配25个代币),并设置找零地址
  • ¥15 关于Java的学习问题
  • ¥15 如何使用chatgpt完成文本分类任务?