package httpapi

import (

"context"

"time"

"nhooyr.io/websocket"

"cdr.dev/slog"

)

// 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

}

}

}

// Heartbeat loops to ping a WebSocket to keep it alive. It calls `exit` on ping

// failure.

func HeartbeatClose(ctx context.Context, logger slog.Logger, exit func(), 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 {

_ = conn.Close(websocket.StatusGoingAway, "Ping failed")

logger.Info(ctx, "failed to heartbeat ping", slog.Error(err))

exit()

return

}

}

}