mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
fix: avoid agent runLoop exiting due to ws ping (#8852)
This commit is contained in:
@@ -204,7 +204,7 @@ func (c *Client) DERPMapUpdates(ctx context.Context) (<-chan DERPMapUpdate, io.C
|
||||
defer close(updates)
|
||||
defer close(updatesClosed)
|
||||
defer cancelFunc()
|
||||
defer conn.Close(websocket.StatusGoingAway, "Listen closed")
|
||||
defer conn.Close(websocket.StatusGoingAway, "DERPMapUpdates closed")
|
||||
for {
|
||||
var update DERPMapUpdate
|
||||
err := dec.Decode(&update.DERPMap)
|
||||
@@ -240,7 +240,7 @@ func (c *Client) DERPMapUpdates(ctx context.Context) (<-chan DERPMapUpdate, io.C
|
||||
closeFunc: func() error {
|
||||
cancelFunc()
|
||||
<-pingClosed
|
||||
_ = conn.Close(websocket.StatusGoingAway, "Listen closed")
|
||||
_ = conn.Close(websocket.StatusGoingAway, "DERPMapUpdates closed")
|
||||
<-updatesClosed
|
||||
return nil
|
||||
},
|
||||
|
||||
Reference in New Issue
Block a user