登录
首页 >  Golang >  Go问答

接收服务器推送的HTTP响应

来源:stackoverflow

时间:2024-03-14 09:18:27 315浏览 收藏

本篇文章给大家分享《接收服务器推送的HTTP响应》,覆盖了Golang的常见基础知识,其实一个语言的全部知识点一篇文章是不可能说完的,但希望通过这些问题,让读者对自己的掌握程度有一定的认识(B 数),从而弥补自己的不足,更好的掌握它。

问题内容

当使用 go 抛出 http 请求并接收响应时,考虑到 responsebody 很大(1 gb 或更多)的情况,我希望在流式传输时接收响应。

resp, err: = http.Client.Do(req)

在这种情况下,如果正文很大,我无法读取标题,并且不知道响应的状态。 有什么解决办法吗?


解决方案


(编辑:如果您无法从响应中获取“content-length”标头,则您所访问的 web 服务可能不会返回该标头。在这种情况下,没有办法无需完全读取即可了解响应正文的长度。您可以在以下示例中通过删除在响应中设置 content-length 标头的行来模拟这一点。)

标准的 go net/http 包可以很好地处理大型响应。这是一个独立的示例来演示:

// Start a mock HTTP server that returns 2GB of data in the response. Make a
// HTTP request to this server and print the amount of data read from the
// response.
package main

import (
    "fmt"
    "io"
    "log"
    "net/http"
    "strings"
    "time"
)

const oneMB = 1024 * 1024
const oneGB = 1024 * oneMB
const responseSize = 2 * oneGB

const serverAddr = "localhost:9999"

func startServer() {
    // Mock HTTP server that always returns 2GB of data
    go http.ListenAndServe(serverAddr, http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
        w.Header().Set("Content-length", fmt.Sprintf("%d", responseSize))

        // 1MB buffer that'll be copied multiple times to the response
        buf := []byte(strings.Repeat("x", oneMB))

        for i := 0; i < responseSize/len(buf); i++ {
            if _, err := w.Write(buf); err != nil {
                log.Fatal("Failed to write to response. Error: ", err.Error())
            }
        }
    }))

    // Some grace period for the server to start
    time.Sleep(100 * time.Millisecond)
}

func main() {
    startServer()

    // HTTP client
    req, err := http.NewRequest("GET", "http://"+serverAddr, nil)
    if err != nil {
        log.Fatal("Error creating HTTP request: ", err.Error())
    }

    client := http.Client{}
    resp, err := client.Do(req)
    if err != nil {
        log.Fatal("Error making HTTP request: ", err.Error())
    }

    // Read the response header
    fmt.Println("Response: Content-length:", resp.Header.Get("Content-length"))

    bytesRead := 0
    buf := make([]byte, oneMB)

    // Read the response body
    for {
        n, err := resp.Body.Read(buf)
        bytesRead += n

        if err == io.EOF {
            break
        }

        if err != nil {
            log.Fatal("Error reading HTTP response: ", err.Error())
        }
    }

    fmt.Println("Response: Read", bytesRead, "bytes")
}

如果响应太大,您就不会希望读取内存中的整个响应。将其写入临时文件,然后进行处理。

如果您正在寻找在网络不太可靠时可靠地执行此操作的选项,请查找“http 范围请求”,您可以使用它来恢复部分下载的数据。

文中关于的知识介绍,希望对你的学习有所帮助!若是受益匪浅,那就动动鼠标收藏这篇《接收服务器推送的HTTP响应》文章吧,也可关注golang学习网公众号了解相关技术文章。

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>