44"context"
55"database/sql"
66"encoding/json"
7- "errors"
87"fmt"
98"io"
109"net"
@@ -521,23 +520,23 @@ func convertWorkspaceAgent(dbAgent database.WorkspaceAgent, agentUpdateFrequency
521520}
522521
523522// wsNetConn wraps net.Conn created by websocket.NetConn(). Cancel func
524- // is called ifio.EOF is encountered.
523+ // is called ifa read or write error is encountered.
525524type wsNetConn struct {
526525cancel context.CancelFunc
527526net.Conn
528527}
529528
530529func (c * wsNetConn )Read (b []byte ) (n int ,err error ) {
531530n ,err = c .Conn .Read (b )
532- if errors . Is ( err , io . EOF ) {
531+ if err != nil {
533532c .cancel ()
534533}
535534return n ,err
536535}
537536
538537func (c * wsNetConn )Write (b []byte ) (n int ,err error ) {
539538n ,err = c .Conn .Write (b )
540- if errors . Is ( err , io . EOF ) {
539+ if err != nil {
541540c .cancel ()
542541}
543542return n ,err
@@ -549,8 +548,8 @@ func (c *wsNetConn) Close() error {
549548}
550549
551550// websocketNetConn wraps websocket.NetConn and returns a context that
552- // is tied to the parent context and the lifetime of the conn.A io.EOF
553- //error during read or write will cancel the context, but not close the
551+ // is tied to the parent context and the lifetime of the conn.Any error
552+ // during read or write will cancel the context, but not close the
554553// conn. Close should be called to release context resources.
555554func websocketNetConn (ctx context.Context ,conn * websocket.Conn ,msgType websocket.MessageType ) (context.Context , net.Conn ) {
556555ctx ,cancel := context .WithCancel (ctx )