为什么Golang http.ResponseWriter执行被延迟？

Z时代
2024-01-10
分类：问答

我试图在收到请求后立即发送页面响应，然后进行处理，但是我发现响应即使按代码顺序排列也没有“首先”发送。在现实生活中，我有一个页面可供上传一个Excel工作表，该工作表保存到数据库中需要花费时间（50000+行），并且希望更新用户进度。这是一个简化的示例；（取决于您有多少RAM，您可能需要添加几个零来计数才能看到结果）

package main
import (
    "fmt"
    "net/http"
)
func writeAndCount(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("Starting to count"))
    for i := 0; i < 1000000; i++ {
        if i%1000 == 0 {
            fmt.Println(i)
        }
    }
    w.Write([]byte("Finished counting"))
}
func main() {
    http.HandleFunc("/", writeAndCount)
    http.ListenAndServe(":8080", nil)
}

回答：

HTTP协议的原始概念是一个简单的请求-

响应服务器-客户端计算模型。没有流或“连续”客户端更新支持。总是（总是）首先联系服务器的客户端需要它的某种信息。

同样，由于大多数Web服务器都会缓存响应，直到响应完全就绪（或达到一定的限制，通常是缓冲区大小）为止，因此您写入（发送）到客户端的数据将不会立即传输。

为了避免这种“局限性”，已经开发出了几种技术，以便服务器能够通知客户端有关更改或进度的信息，例如HTTP Long轮询，HTTP流，HTTP / 2

Server

Push或Websockets。

因此，要实现您想要的目标，您必须绕过HTTP协议的原始“边界”。

如果要定期发送数据或将数据流传输到客户端，则必须将此信息告知服务器。最简单的方法是检查http.ResponseWriter传递给您的http.Flusher接口是否实现了接口（使用类型断言），如果接口实现了，则调用其Flusher.Flush()方法会将所有缓冲的数据发送到客户端。

使用http.Flusher仅是解决方案的一半。由于这是HTTP协议的非标准用法，因此通常也需要客户端支持才能正确处理。

首先，您必须通过设置ContentType=text/event-stream响应头来让客户端了解响应的“流”性质。

接下来，为避免客户端缓存响应，请务必同时设置Cache-Control=no-cache。

最后，要让客户端知道您可能不会以单个单元（而是以定期更新或流的形式）发送响应，并且让客户端保持连接活动并等待更多数据，请设置Connection=keep-

alive响应头。

将响应标头设置为上述值后，就可以开始漫长的工作，并且每当您要向客户端更新进度时，都可以编写一些数据并调用Flusher.Flush()。

让我们看一个简单的示例，它“正确”地执行所有操作：

func longHandler(w http.ResponseWriter, r *http.Request) {
    flusher, ok := w.(http.Flusher)
    if !ok {
        http.Error(w, "Server does not support Flusher!",
            http.StatusInternalServerError)
        return
    }
    w.Header().Set("Content-Type", "text/event-stream")
    w.Header().Set("Cache-Control", "no-cache")
    w.Header().Set("Connection", "keep-alive")
    start := time.Now()
    for rows, max := 0, 50*1000; rows < max; {
        time.Sleep(time.Second) // Simulating work...
        rows += 10 * 1000
        fmt.Fprintf(w, "Rows done: %d (%d%%), elapsed: %v\n",
            rows, rows*100/max, time.Since(start).Truncate(time.Millisecond))
        flusher.Flush()
    }
}
func main() {
    http.HandleFunc("/long", longHandler)
    panic(http.ListenAndServe("localhost:8080", nil))
}

现在，如果您http://localhost:8080/long在浏览器中打开，您将看到输出每秒“增长”：

Rows done: 10000 (20%), elapsed: 1s
Rows done: 20000 (40%), elapsed: 2s
Rows done: 30000 (60%), elapsed: 3s
Rows done: 40000 (80%), elapsed: 4.001s
Rows done: 50000 (100%), elapsed: 5.001s

另请注意，使用SSE时，应将更新“打包”到SSE框架中，即应以"data:"前缀开头，并以2个换行符结束每个框架"\n\n"。