48 lines
1.2 KiB
JavaScript
48 lines
1.2 KiB
JavaScript
var Logger = require('./logger');
|
|
var path = require('path');
|
|
var counterLog = new Logger.Logger(path.resolve(__dirname, '..', 'counters.log'));
|
|
import os from 'os';
|
|
import io from 'socket.io';
|
|
import Socket from 'socket.io/lib/socket';
|
|
|
|
var counters = {};
|
|
|
|
exports.add = function (counter, value) {
|
|
if (!value) {
|
|
value = 1;
|
|
}
|
|
|
|
if (!counters.hasOwnProperty(counter)) {
|
|
counters[counter] = value;
|
|
} else {
|
|
counters[counter] += value;
|
|
}
|
|
};
|
|
|
|
Socket.prototype._packet = Socket.prototype.packet;
|
|
Socket.prototype.packet = function () {
|
|
this._packet.apply(this, arguments);
|
|
exports.add('socket.io:packet');
|
|
};
|
|
|
|
function getConnectedSockets() {
|
|
var sockets = io.instance.sockets.sockets;
|
|
if (typeof sockets.length === 'number') {
|
|
return sockets.length;
|
|
} else {
|
|
return Object.keys(sockets).length;
|
|
}
|
|
}
|
|
|
|
setInterval(function () {
|
|
try {
|
|
counters['memory:rss'] = process.memoryUsage().rss / 1048576;
|
|
counters['load:1min'] = os.loadavg()[0];
|
|
counters['socket.io:count'] = getConnectedSockets();
|
|
counterLog.log(JSON.stringify(counters));
|
|
} catch (e) {
|
|
Logger.errlog.log(e.stack);
|
|
}
|
|
counters = {};
|
|
}, 60000);
|