127 lines
3.5 KiB
JavaScript
127 lines
3.5 KiB
JavaScript
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.accessToken = config.accessToken;
|
|
this.bucServer = new BUCServer(config);
|
|
|
|
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);
|
|
|
|
if (req.accessToken !== this.accessToken)
|
|
throw new Error('Access denied');
|
|
|
|
switch (req.action) {
|
|
case 'test':
|
|
await this.test(req, ws); break;
|
|
case 'get-buc':
|
|
await this.getBuc(req, ws); break;
|
|
case 'update-buc':
|
|
await this.updateBuc(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);
|
|
}
|
|
|
|
async getBuc(req, ws) {
|
|
if (!req.fromCheckTime)
|
|
throw new Error(`key 'fromCheckTime' is empty`);
|
|
|
|
await this.bucServer.getBuc(req.fromCheckTime, (rows) => {
|
|
this.send({state: 'get', rows}, req, ws);
|
|
});
|
|
|
|
this.send({state: 'finish'}, req, ws);
|
|
}
|
|
|
|
async updateBuc(req, ws) {
|
|
if (!req.bookUrls)
|
|
throw new Error(`key 'bookUrls' is empty`);
|
|
|
|
if (!Array.isArray(req.bookUrls))
|
|
throw new Error(`key 'bookUrls' must be array`);
|
|
|
|
await this.bucServer.updateBuc(req.bookUrls);
|
|
|
|
this.send({state: 'success'}, req, ws);
|
|
}
|
|
}
|
|
|
|
module.exports = BookUpdateCheckerController;
|