mirror of
https://github.com/coder/coder.git
synced 2025-07-03 16:13:58 +00:00
28 lines
581 B
Go
28 lines
581 B
Go
package httpapi
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"nhooyr.io/websocket"
|
|
)
|
|
|
|
// Heartbeat loops to ping a WebSocket to keep it alive.
|
|
// Default idle connection timeouts are typically 60 seconds.
|
|
// See: https://docs.aws.amazon.com/elasticloadbalancing/latest/application/application-load-balancers.html#connection-idle-timeout
|
|
func Heartbeat(ctx context.Context, conn *websocket.Conn) {
|
|
ticker := time.NewTicker(15 * time.Second)
|
|
defer ticker.Stop()
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case <-ticker.C:
|
|
}
|
|
err := conn.Ping(ctx)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
}
|