我正在用SSE编写测试应用程序,但我的问题是 ReadTimeout和WriteTimeout每10秒钟关闭一次客户端连接,因此主页面正在丢失数据。
我如何管理此问题,在没有泄漏风险和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
}
}
}
答案 0 :(得分:4)
ReadTimeout
和WriteTimeout
都定义了必须从客户端读取或写回整个请求的持续时间。这些超时应用于基础连接(http://golang.org/src/pkg/net/http/server.go?s=15704:15902),这是在收到任何标头之前,因此您无法为单独的处理程序设置不同的限制 - 服务器内的所有连接将共享相同的超时限制。
这样说,如果您需要为每个请求定制超时,则需要在处理程序中实现它们。在你的代码中你已经在为你的工作使用超时了,所以这将是在处理程序的开头创建一个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.ResponseWriter
传递的http.TimeoutHandler
没有实现http.CloseNotifier
。如果需要,您可以深入了解implementation,看看他们如何解决超时问题,以便提出类似但增强的解决方案。