1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798 |
- const WebSocket = require('ws');
- //const _ = require('lodash');
- const BUCServer = require('../core/BookUpdateChecker/BUCServer');
- const log = new (require('../core/AppLogger'))().log;//singleton
- //const utils = require('../core/utils');
- const cleanPeriod = 1*60*1000;//1 минута
- const closeSocketOnIdle = 5*60*1000;//5 минут
- class BookUpdateCheckerController {
- constructor(wss, config) {
- this.config = config;
- this.isDevelopment = (config.branch == 'development');
- //this.readerStorage = new JembaReaderStorage();
- this.bucServer = new BUCServer(config);
- this.bucServer.main(); //no await
- this.wss = wss;
- wss.on('connection', (ws) => {
- ws.on('message', (message) => {
- this.onMessage(ws, message.toString());
- });
- ws.on('error', (err) => {
- log(LM_ERR, err);
- });
- });
- setTimeout(() => { this.periodicClean(); }, cleanPeriod);
- }
- periodicClean() {
- try {
- const now = Date.now();
- this.wss.clients.forEach((ws) => {
- if (!ws.lastActivity || now - ws.lastActivity > closeSocketOnIdle - 50) {
- ws.terminate();
- }
- });
- } finally {
- setTimeout(() => { this.periodicClean(); }, cleanPeriod);
- }
- }
- async onMessage(ws, message) {
- let req = {};
- try {
- if (this.isDevelopment) {
- log(`WebSocket-IN: ${message.substr(0, 4000)}`);
- }
- req = JSON.parse(message);
- ws.lastActivity = Date.now();
-
- //pong for WebSocketConnection
- this.send({_rok: 1}, req, ws);
- switch (req.action) {
- case 'test':
- await this.test(req, ws); break;
- default:
- throw new Error(`Action not found: ${req.action}`);
- }
- } catch (e) {
- this.send({error: e.message}, req, ws);
- }
- }
- send(res, req, ws) {
- if (ws.readyState == WebSocket.OPEN) {
- ws.lastActivity = Date.now();
- let r = res;
- if (req.requestId)
- r = Object.assign({requestId: req.requestId}, r);
- const message = JSON.stringify(r);
- ws.send(message);
- if (this.isDevelopment) {
- log(`WebSocket-OUT: ${message.substr(0, 4000)}`);
- }
- }
- }
- //Actions ------------------------------------------------------------------
- async test(req, ws) {
- this.send({message: 'Liberama project is awesome'}, req, ws);
- }
- }
- module.exports = BookUpdateCheckerController;
|