44"context"
55"database/sql"
66"encoding/json"
7- "errors"
87"fmt"
98"io"
109"net"
@@ -510,23 +509,23 @@ func convertWorkspaceAgent(dbAgent database.WorkspaceAgent, agentUpdateFrequency
510509}
511510
512511// wsNetConn wraps net.Conn created by websocket.NetConn(). Cancel func
513- // is called ifio.EOF is encountered.
512+ // is called ifa read or write error is encountered.
514513type wsNetConn struct {
515514cancel context.CancelFunc
516515net.Conn
517516}
518517
519518func (c * wsNetConn )Read (b []byte ) (n int ,err error ) {
520519n ,err = c .Conn .Read (b )
521- if errors . Is ( err , io . EOF ) {
520+ if err != nil {
522521c .cancel ()
523522}
524523return n ,err
525524}
526525
527526func (c * wsNetConn )Write (b []byte ) (n int ,err error ) {
528527n ,err = c .Conn .Write (b )
529- if errors . Is ( err , io . EOF ) {
528+ if err != nil {
530529c .cancel ()
531530}
532531return n ,err
@@ -538,8 +537,8 @@ func (c *wsNetConn) Close() error {
538537}
539538
540539// websocketNetConn wraps websocket.NetConn and returns a context that
541- // is tied to the parent context and the lifetime of the conn.A io.EOF
542- //error during read or write will cancel the context, but not close the
540+ // is tied to the parent context and the lifetime of the conn.Any error
541+ // during read or write will cancel the context, but not close the
543542// conn. Close should be called to release context resources.
544543func websocketNetConn (ctx context.Context ,conn * websocket.Conn ,msgType websocket.MessageType ) (context.Context , net.Conn ) {
545544ctx ,cancel := context .WithCancel (ctx )