reactjs - 如何使用 Golang Gorilla/mux 托管并发 websocket 连接？

Question

在此先感谢您的任何帮助或建议!

我正在构建一个国际象棋应用程序，前端使用 Reactjs，后端是使用 Golilla mux 库用 Golang 编写的服务器。后端是一个供人类用户下棋的国际象棋引擎。 React 前端在顶级构造函数中创建与服务器的 WebSocket 连接。

该应用程序适用于单个连接。但是，在打开第二个浏览器选项卡时，第一个浏览器选项卡的 Websocket 连接丢失。

服务器报错，

read error: websocket: close 1001 (going away)

然后，

read error: read tcp 127.0.0.1:8081-\u003e127.0.0.1:64146: use of closed network connection

我查看了 websockets 的 Gorrilla/mux 文档，并且正在从同一函数读取/写入所有内容，以避免在同一连接上并发读取或写入。

React 前端构造函数在这里:

class Game extends React.Component {
  constructor(props) {
    super(props);
    this.state = {
      history: [{
        position: this.props.chess.fen(),
        move: "",
      }],
      ply: 0,
      selectedSq: null,
      engineName: "",
      engineAuthor: "",
      uciOK: false,
      isReady: false,
      engineThinking: false,
      playersTurn: true,
      playerColor: WHITE,
    };
    this.ws = websocketConnect("ws://localhost:8081/uci");
    this.ws.onmessage = (event) => {
      const msg = event.data
      this.processEngineMessage(msg)
    }
  }

它引用的是 websocket.js 在这里:

export function websocketConnect(url) {
  const ws = new WebSocket(url);
  ws.addEventListener('open', (event) => {
    ws.send('uci');
  });
  return ws;
}

Goland Websocket 包在这里

package websocket

import (
    "net/http"

    "github.com/namsral/flag"
    log "github.com/sirupsen/logrus"

    "github.com/gorilla/websocket"
)

type WebsocketServer struct {
    upgrader websocket.Upgrader
    addr     *string
    conn     *websocket.Conn
}

func NewWebsocketServer() *WebsocketServer {
    w := new(WebsocketServer)
    w.addr = flag.String("addr", "localhost:8081", "http service address")
    flag.Parse()
    w.upgrader = websocket.Upgrader{} // use default options
    http.HandleFunc("/uci", w.uciHandler)
    return w
}

func (w *WebsocketServer) uciHandler(rw http.ResponseWriter, r *http.Request) {
    var err error
    log.Error("upgrading to websocket connection")
    w.upgrader.CheckOrigin = func(r *http.Request) bool { return true }
    w.conn, err = w.upgrader.Upgrade(rw, r, nil)
    if err != nil {
        log.Print("upgrade:", err)
        return
    }
    go w.UCI(rw, r)
}

func (w *WebsocketServer) Start() {
    log.Info("starting websocket server")
    http.ListenAndServe(*w.addr, nil)
}

func (w *WebsocketServer) CloseConnection() {
    w.conn.Close()
}

func (w *WebsocketServer) StartReader(channel chan string) {
    for {
        _, message, err := w.conn.ReadMessage()
        if err != nil {
            log.Println("read error:", err)
            break
        }
        log.Printf("recv: %s", message)
        channel <- string(message)
    }
}

func (w *WebsocketServer) Write(msg string) {
    err := w.conn.WriteMessage(websocket.TextMessage, []byte(msg))
    if err != nil {
        log.Println("write:", err)
    }
}

完整代码在这里: 服务器:https://github.com/tonyOreglia/glee 前端:https://github.com/tonyOreglia/chess-board-react

我希望能够支持多个并发用户使用该网站。它目前托管在 DigitalOcean 上。

Answer 1

WebServer 结构的设计只允许一个连接。

发生的事情是，在 /uci 上的每个初始 http 请求上，连接都会升级，并且每次将 http 请求升级为 ws 连接时，都会替换 上的先前连接WebServer 由那个构造。

此外，它不是线程安全的，因为每个请求都在不同的 goroutine 中处理。

我建议您将连接传递给您的 UCI 方法，而不是将其附加到服务器。

type WebsocketServer struct {
    upgrader websocket.Upgrader
    addr     *string
}

func (w *WebsocketServer) uciHandler(rw http.ResponseWriter, r *http.Request) {
    var err error
    log.Error("upgrading to websocket connection")
    w.upgrader.CheckOrigin = func(r *http.Request) bool { return true }
    // here is the difference
    conn, err := w.upgrader.Upgrade(rw, r, nil)
    if err != nil {
        log.Print("upgrade:", err)
        return
    }
    // and here
    go w.UCI(rw, r, conn)
}

您还可以保留一个带有互斥量的 slice/映射，您可以在其中存储所有连接。只是不要忘记在连接关闭时将其删除。