fix(websocket): 修复 websocket 超时问题

Author: wsshow

js/src/xterm.tsx

@@ -17,8 +17,8 @@ export class GoTTYXterm {
     messageTimeout: number;
     messageTimer: NodeJS.Timeout;
 
-    onResizeHandler: IDisposable;
-    onDataHandler: IDisposable;
+    onResizeHandler: IDisposable | undefined;
+    onDataHandler: IDisposable | undefined;
 
     fitAddOn: FitAddon;
     zmodemAddon: ZModemAddon;
@@ -139,9 +139,11 @@ export class GoTTYXterm {
     deactivate(): void {
         if (this.onDataHandler) {
             this.onDataHandler.dispose();
+            this.onDataHandler = undefined;
         }
         if (this.onResizeHandler) {
             this.onResizeHandler.dispose();
+            this.onResizeHandler = undefined;
         }
         this.term.blur();
     }

server/handlers.go

@@ -11,6 +11,7 @@ import (
 	"net/url"
 	"strings"
 	"sync/atomic"
+	"time"
 
 	"github.com/gorilla/websocket"
 	"github.com/pkg/errors"
@@ -134,7 +135,40 @@ func (server *Server) generateHandleWS(ctx context.Context, cancel context.Cance
 	}
 }
 
+const (
+	wsPingInterval = 30 * time.Second
+	wsPingTimeout  = 10 * time.Second
+	wsPongWait     = 60 * time.Second
+)
+
 func (server *Server) processWSConn(ctx context.Context, conn *websocket.Conn, headers map[string][]string) error {
+	// Set up WebSocket protocol-level ping/pong to keep the connection alive
+	// through proxies, load balancers, and firewalls that have idle timeouts.
+	conn.SetReadDeadline(time.Now().Add(wsPongWait))
+	conn.SetPongHandler(func(string) error {
+		conn.SetReadDeadline(time.Now().Add(wsPongWait))
+		return nil
+	})
+
+	wrapper := &wsWrapper{Conn: conn}
+
+	pingCtx, pingCancel := context.WithCancel(ctx)
+	defer pingCancel()
+	go func() {
+		ticker := time.NewTicker(wsPingInterval)
+		defer ticker.Stop()
+		for {
+			select {
+			case <-ticker.C:
+				if err := wrapper.WritePing(); err != nil {
+					return
+				}
+			case <-pingCtx.Done():
+				return
+			}
+		}
+	}()
+
 	typ, initLine, err := conn.ReadMessage()
 	if err != nil {
 		return errors.Wrapf(err, "failed to authenticate websocket connection")
@@ -209,7 +243,7 @@ func (server *Server) processWSConn(ctx context.Context, conn *websocket.Conn, h
 	if server.options.Height > 0 {
 		opts = append(opts, webtty.WithFixedRows(server.options.Height))
 	}
-	tty, err := webtty.New(&wsWrapper{conn}, slave, opts...)
+	tty, err := webtty.New(wrapper, slave, opts...)
 	if err != nil {
 		return errors.Wrapf(err, "failed to create webtty")
 	}

server/ws_wrapper.go

@@ -2,16 +2,22 @@ package server
 
 import (
 	"io"
+	"sync"
+	"time"
 
 	"github.com/gorilla/websocket"
 	"github.com/pkg/errors"
 )
 
 type wsWrapper struct {
 	*websocket.Conn
+	writeMu sync.Mutex
 }
 
 func (wsw *wsWrapper) Write(p []byte) (n int, err error) {
+	wsw.writeMu.Lock()
+	defer wsw.writeMu.Unlock()
+	wsw.Conn.SetWriteDeadline(time.Now().Add(30 * time.Second))
 	writer, err := wsw.Conn.NextWriter(websocket.TextMessage)
 	if err != nil {
 		return 0, err
@@ -20,6 +26,13 @@ func (wsw *wsWrapper) Write(p []byte) (n int, err error) {
 	return writer.Write(p)
 }
 
+func (wsw *wsWrapper) WritePing() error {
+	wsw.writeMu.Lock()
+	defer wsw.writeMu.Unlock()
+	wsw.Conn.SetWriteDeadline(time.Now().Add(wsPingTimeout))
+	return wsw.Conn.WriteMessage(websocket.PingMessage, nil)
+}
+
 func (wsw *wsWrapper) Read(p []byte) (n int, err error) {
 	for {
 		msgType, reader, err := wsw.Conn.NextReader()