Go语言 Http服务器读写超时和服务器端事件

xesrikrc  于 2023-08-01  发布在  Go
关注(0)|答案(2)|浏览(193)

我正在用SSE写一个测试应用程序,但我的问题是ReadTimeout和WriteTimeout每10秒关闭一次客户端连接,因此主页正在丢失数据。
我如何管理这个问题,同时为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
        }
    }
}

y53ybaqx

y53ybaqx1#

ReadTimeoutWriteTimeout都定义了整个请求必须从客户端读取或写回客户端的持续时间。这些超时应用于底层连接(http://golang.org/src/pkg/net/http/server.go?s=15704:15902),并且这是在接收任何报头之前,因此您不能为单独的处理程序设置不同的限制-服务器中的所有连接将共享相同的超时限制。
也就是说,如果您需要为每个请求定制超时,则需要在处理程序中实现它们。在你的代码中,你已经在你的工作中使用了超时,所以这将是一个在处理程序开始时创建一个time.After的问题,保持检查(在处理程序本身甚至传递它),并在必要时停止请求。这实际上可以让您更精确地控制超时,因为WriteTimeout只会在尝试写入响应时触发(例如:如果超时设置为10秒,并且在任何写入之前需要一分钟来准备响应,则在作业完成之前不会收到任何错误,因此您将浪费50秒的资源)。从这个Angular 来看,我认为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.CloseNotifier。如果需要的话,您可以深入研究implementation,看看他们是如何解决超时问题的,以便提出一个类似但增强的解决方案。

fbcarpbf

fbcarpbf2#

HTTP处理程序超时是按服务器指定的:ReadTimeoutWriteTimeout。在Go 1.20之后,HTTP ResponseController提供了对http.ResponseWriter接口没有处理的扩展的每请求功能的访问。
ResponseController类型为添加per-handler控件提供了一种更清晰、更易发现的方式。在Go 1.20中也添加了两个这样的控件:SetReadDeadlineSetWriteDeadline,它们允许设置每个请求的读写截止日期。
样品

func exampleHandler(w http.ResponseWriter, r *http.Request) {
    rc := http.NewResponseController(w)

    // Set a write deadline in 5 seconds time.
    err := rc.SetWriteDeadline(time.Now().Add(5 * time.Second))
    if err != nil {
 
    }

    // Some job ...

    // Write the normal response.
    w.Write([]byte("something"))
}

字符串

相关问题