iobroker.opendtu/lib/websocketController.js

Adapter for the OpenDTU project

const WebSocket = require('ws');
let wsClient;
const wsHeartbeatIntervall = 30000;
const restartTimeout = 2000;
let ping;
let pingTimeout;
let autoRestartTimeout;

class WebsocketController {
    constructor(adapter) {
        this.adapter = adapter;
    }

    initWsClient() {
        try {
            const encodedUsername = encodeURIComponent(this.adapter.config.userName);
            const encodedPassword = encodeURIComponent(this.adapter.config.password);

            const wsURL = `${this.adapter.config.webUIScheme == 'http' ? 'ws' : 'wss'}://${encodedUsername}:${encodedPassword}@${this.adapter.config.webUIServer}:${this.adapter.config.webUIPort}/livedata`;
            //this.adapter.log.debug(`Websocket connect over URL: ${wsURL}`);

            wsClient = new WebSocket(wsURL);

            wsClient.on('open', () => {
                // Send ping to server
                this.sendPingToServer();
                // Start Heartbeat
                this.wsHeartbeat();
            });

            wsClient.on('pong', () => {
                this.wsHeartbeat();
            });

            wsClient.on('close', () => {
                clearTimeout(pingTimeout);
                clearTimeout(ping);

                if (wsClient.readyState === WebSocket.CLOSED) {
                    this.autoRestart();
                }
            });

            wsClient.on('message', () => {});

            wsClient.on('error', err => {
                this.adapter.log.debug(`Websocket error: ${err}`);
            });

            return wsClient;
        } catch (err) {
            this.adapter.log.error(err);
        }
    }

    sendPingToServer() {
        //this.logDebug('Send ping to server');
        wsClient.ping();
        ping = setTimeout(() => {
            this.sendPingToServer();
        }, wsHeartbeatIntervall);
    }

    wsHeartbeat() {
        clearTimeout(pingTimeout);
        pingTimeout = setTimeout(() => {
            this.adapter.log.debug('Websocked connection timed out');
            wsClient.terminate();
        }, wsHeartbeatIntervall + 3000);
    }

    autoRestart() {
        this.adapter.log.debug(`Start try again in ${restartTimeout / 1000} seconds...`);
        clearTimeout(autoRestartTimeout);
        autoRestartTimeout = setTimeout(() => {
            this.adapter.startWebsocket();
        }, restartTimeout);
    }

    closeConnection() {
        if (wsClient && wsClient.readyState !== WebSocket.CLOSED) {
            wsClient.close();
        }
    }

    async allTimerClear() {
        clearTimeout(pingTimeout);
        clearTimeout(ping);
        clearTimeout(autoRestartTimeout);
    }
}

module.exports = {
    WebsocketController,
};