OpenFlare/backend/openflare/plugins/server/domain/fleet/websocket/read_pump.go

// Copyright 2026 Arctel.net
// SPDX-License-Identifier: Apache-2.0

package websocket

import (
	"encoding/json"
	"log/slog"
	"time"

	"github.com/gorilla/websocket"
)

func runReadPump(
	nodeID string,
	conn *websocket.Conn,
	closeFn func(),
	logLabel string,
	sendPong func(string) bool,
	clientPongType string,
) {
	defer closeFn()
	_ = conn.SetReadDeadline(time.Now().Add(wsReadDeadline))
	conn.SetPongHandler(func(string) error {
		return conn.SetReadDeadline(time.Now().Add(wsReadDeadline))
	})

	for {
		_, data, err := conn.ReadMessage()
		if err != nil {
			slog.Debug(logLabel+" read closed", "node_id", nodeID, "error", err)
			return
		}

		var message Message
		if err = json.Unmarshal(data, &message); err != nil {
			slog.Debug(logLabel+" invalid message", "node_id", nodeID, "error", err)
			continue
		}

		switch message.Type {
		case messageTypePing:
			_ = sendPong(nodeID)
		case clientPongType:
			// Refresh read deadline when the client replies with a JSON pong.
			// This keeps the connection alive when the WebSocket is proxied
			// through Cloudflare, which enforces a 100-second idle timeout on
			// the TCP stream. Without this refresh, the server's 90-second read
			// deadline expires and terminates the connection even though the
			// client is actively responding to pings.
			_ = conn.SetReadDeadline(time.Now().Add(wsReadDeadline))
		default:
			slog.Debug(logLabel+" unsupported message", "node_id", nodeID, "type", message.Type)
		}
	}
}