go - Http 服务器读写超时和服务器端事件

Question

我正在使用 SSE 编写测试应用程序，但我的问题是 ReadTimeout 和 WriteTimeout 每 10 秒关闭一次客户端连接，因此主页正在丢失数据。

我如何管理这个问题，在没有 goroutines 泄漏风险和完成 SSE 工作的情况下同时提供 SSE 和网页？

服务器:

server := &http.Server{Addr: addr,
    ReadTimeout:  10 * time.Second,
    WriteTimeout: 10 * time.Second,
    Handler:      s.mainHandler(),
}

处理程序:

func sseHandler(w http.ResponseWriter, r *http.Requests) {
    f, ok := w.(http.Flusher)
    if !ok {
        http.Error(w, "Streaming not supported!", http.StatusInternalServerError)
        log.Println("Streaming not supported")
        return
    }
    messageChannel := make(chan string)
    hub.addClient <- messageChannel
    notify := w.(http.CloseNotifier).CloseNotify()

    w.Header().Set("Content-Type", "text/event-stream")
    w.Header().Set("Cache-Control", "no-cache")
    w.Header().Set("Connection", "keep-alive")
    for i := 0; i < 1440; {
        select {
        case msg := <-messageChannel:
            jsonData, _ := json.Marshal(msg)
            str := string(jsonData)
            fmt.Fprintf(w, "data: {\"str\": %s, \"time\": \"%v\"}\n\n", str, time.Now())
            f.Flush()

        case <-time.After(time.Second * 60):
            fmt.Fprintf(w, "data: {\"str\": \"No Data\"}\n\n")

            f.Flush()
            i++

        case <-notify:
            f.Flush()
            i = 1440
            hub.removeClient <- messageChannel
        }
    }
}

Answer 1

ReadTimeout 和WriteTimeout 都定义了必须从客户端读取整个请求或将其写回客户端的持续时间。这些超时应用于底层连接 (http://golang.org/src/pkg/net/http/server.go?s=15704:15902)，这是在收到任何 header 之前，因此您不能为单独的处理程序设置不同的限制——服务器内的所有连接将共享相同的超时限制。

也就是说，如果您需要为每个请求自定义超时，则需要在您的处理程序中实现它们。在你的代码中你已经为你的工作使用了超时，所以这将是在处理程序的开头创建一个 time.After 的问题，继续检查(在处理程序本身甚至传递它周围)并在必要时停止请求。这实际上可以让您更精确地控制超时，因为 WriteTimeout 只会在尝试写入响应时触发(例如，如果超时设置为 10 秒，并且在任何响应之前准备响应需要一分钟写，在工作完成之前你不会得到任何错误，所以你会浪费资源 50 秒)。从这个角度来看，我认为 WriteTimeout 本身只有在您的响应准备得相当快时才有用，但您希望在网络变得非常慢(或客户端故意停止接收数据)时断开连接。

库里有个小辅助函数，http.TimeoutHandler ，其行为类似于 WriteTimeout，但如果响应花费的时间超过预定义的时间，则会返回 503 错误。您可以使用它为每个处理程序设置类似于 WriteTimeout 的行为，例如:

package main

import (
    "log"   
    "net/http"
    "time"
)

type Handler struct {
}

func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    time.Sleep(3*time.Second)
    // This will return http.ErrHandlerTimeout
    log.Println(w.Write([]byte("body")))
}

func main() {
    h := &Handler{}
    http.Handle("/t1", http.TimeoutHandler(h, 1*time.Second, ""))
    http.Handle("/t2", http.TimeoutHandler(h, 2*time.Second, ""))   
    http.ListenAndServe(":8080", nil)
}

这看起来很方便，但我发现了一个会影响您的代码的缺点:从 http.TimeoutHandler 传递的 http.ResponseWriter 没有实现 http。关闭通知程序。如果需要，您可以深入了解 implementation并了解他们如何解决超时问题，以便提出类似但增强的解决方案。