Question

我正在尝试编写代码以流式传输有关某个主题的数据，例如广播电台（一个广播员，多个听众）。我一直在坚持如何处理新的WebSocket连接请求，而不必为每个打开的WebSocket使用goroutine（对于开始访问同一“站”的许多“监听器”来说，资源开始变得很消耗资源了。）

此刻，我有一个dataStream结构图，如下所示：

struct dataStream {
  data chan byte[]
  conns []*websocket.Connection
}

这是将请求升级到WebSocket，然后尝试将WebSocket连接添加到dataStreams conns的伪代码：

func process_request(w http.ResponseWriter, r *http.Request) {
  // hundred lines of business logic...
  c := upgrade websocket connection
  defer c.Close()
  if dataStream exists {
    append the new connection c to the dataStream.conns slice
  } else {
    create new dataStream
    append the new connection c to the dataStream.conns slice
    stream(dataStream)
  }
}

然后是上面的代码块中提到的stream函数。其中一个针对每个dataStream（而不是针对每个WebSocket连接）在后台运行。

func stream(ds *dataStream) {
  ticker := time.NewTicker(poll every ~10 seconds)
  go func() { // this is to poll and remove closed connections
  for _ = range ticker.C {
    for traverse ds.conns {
      ping all connections, remove any closed ones and free memory
      if len(ds.conns == 0){ // no more connections are listening to this dataStream
        delete the ds dataStream and free the memory
        stop ticker
        return // kill goroutine and free the memory
      }
    }
  }}()
  while len(ds.conns) != 0 { // while there are open connections
    fetch any available <-ds.data from channel
    write the data as websocket message to each connection
  }
}

此方法的问题在于，在process_request函数中，当新连接附加到{{1 }}分片，该函数终止关闭WebSocket连接！。结果，if statement在后台运行，并轮询已向dataStream.conns分片添加了关闭的连接，然后删除它。

我的问题：

即使在stream()处理程序函数返回后仍要保持WebSocket连接打开（最好不为每个连接运行单独的goroutine），我应该采取什么方法？

Answer 1

应用程序必须显式关闭Gorilla连接。 HTTP处理函数返回时，连接不会自动关闭。

在这种情况下，应用程序使用defer语句在从处理程序返回时关闭连接。删除defer语句，以避免关闭连接。

HTTP处理函数返回后，如何保持WebSocket连接打开？

1 个答案: