2014-11-24 170 views
1

我正在用SSE编写测试应用程序,但我的问题是 ReadTimeout和WriteTimeout正在每隔10秒关闭客户端连接,因此主页面正在丢失数据。Http服务器读写超时和服务器端事件

我该如何管理这个问题,一起服务SSE和网页,而没有goroutines泄漏风险和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 
     } 
    } 
} 
+3

为什么不增加/禁用超时?你需要每个处理程序单独的超时? – tomasz 2014-11-24 14:07:46

+0

@tomasz可能很棒有一个不同的超时处理程序,我不知道是否有可能,也许有类似的情况下,我的一些最佳做法,我真的不知道它呢... – Mmeyer 2014-11-24 15:04:03

回答

2

两个ReadTimeoutWriteTimeout限定在其内整体请求必须从被读或写回客户端的持续时间。这些超时适用于底层连接(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.ResponseWriterhttp.TimeoutHandler传递没有实现http.CloseNotifier。如果需要,您可以深入探索implementation,看看他们如何解决超时问题,以便提出类似的增强型解决方案。