ethstats-server/app.js
2015-06-09 04:03:31 +03:00

420 lines
7.9 KiB
JavaScript

var _ = require('lodash');
var logger = require('./lib/utils/logger');
var chalk = require('chalk');
var askedForHistoryTime = 0;
var Primus = require('primus'),
api,
client;
var WS_SECRET = process.env.WS_SECRET || "eth-net-stats-has-a-secret";
var Collection = require('./lib/collection');
var Nodes = new Collection();
var server;
var env = 'production';
if( process.env.NODE_ENV !== 'production' )
{
var express = require('express');
var app = express();
var path = require('path');
var bodyParser = require('body-parser');
// view engine setup
app.set('views', path.join(__dirname, 'src/views'));
app.set('view engine', 'jade');
app.use(bodyParser.json());
app.use(bodyParser.urlencoded({ extended: false }));
app.use(express.static(path.join(__dirname, 'dist')));
app.get('/', function(req, res) {
res.render('index');
});
// catch 404 and forward to error handler
app.use(function(req, res, next) {
var err = new Error('Not Found');
err.status = 404;
next(err);
});
// error handlers
app.use(function(err, req, res, next) {
res.status(err.status || 500);
res.render('error', {
message: err.message,
error: err
});
});
// production error handler
app.use(function(err, req, res, next) {
res.status(err.status || 500);
res.render('error', {
message: err.message,
error: {}
});
});
server = require('http').createServer(app);
}
else
{
server = require('http').createServer();
}
api = new Primus(server, {
transformer: 'websockets',
pathname: '/api',
parser: 'JSON'
});
api.use('emit', require('primus-emit'));
api.use('spark-latency', require('primus-spark-latency'));
var client = new Primus(server, {
transformer: 'websockets',
pathname: '/primus',
parser: 'JSON'
});
var clientLatency = 0;
Nodes.setChartsCallback(function (err, charts)
{
if(err !== null)
{
console.error('COL', 'CHR', 'Charts error:', err);
}
else
{
client.write({
action: 'charts',
data: charts
});
}
});
client.use('emit', require('primus-emit'));
api.on('connection', function (spark)
{
console.info('API', 'CON', 'Open:', spark.address.ip);
spark.on('hello', function (data)
{
console.info('API', 'CON', 'Hello', data['id']);
if( _.isUndefined(data.secret) || data.secret !== WS_SECRET )
{
spark.end(undefined, { reconnect: false });
console.error('API', 'CON', 'Closed - wrong auth', data);
return false;
}
if( !_.isUndefined(data.id) && !_.isUndefined(data.info) )
{
data.ip = spark.address.ip;
data.spark = spark.id;
data.latency = spark.latency || 0;
Nodes.add( data, function (err, info)
{
if(err !== null)
{
console.error('API', 'CON', 'Connection error:', err);
return false;
}
if(info !== null)
{
spark.emit('ready');
console.success('API', 'CON', 'Connected', data.id);
client.write({
action: 'add',
data: info
});
}
});
}
});
spark.on('update', function (data)
{
if( !_.isUndefined(data.id) && !_.isUndefined(data.stats) )
{
Nodes.update(data.id, data.stats, function (err, stats)
{
if(err !== null)
{
console.error('API', 'UPD', 'Update error:', err);
}
else
{
if(stats !== null)
{
client.write({
action: 'update',
data: stats
});
console.info('API', 'UPD', 'Update from:', data.id, 'for:', stats);
Nodes.getCharts();
}
}
});
}
else
{
console.error('API', 'UPD', 'Update error:', data);
}
});
spark.on('block', function (data)
{
if( !_.isUndefined(data.id) && !_.isUndefined(data.block) )
{
Nodes.addBlock(data.id, data.block, function (err, stats)
{
if(err !== null)
{
console.error('API', 'BLK', 'Block error:', err);
}
else
{
if(stats !== null)
{
client.write({
action: 'block',
data: stats
});
console.success('API', 'BLK', 'Block:', data.block['number'], 'from:', data.id);
Nodes.getCharts();
}
}
});
}
else
{
console.error('API', 'BLK', 'Block error:', data);
}
});
spark.on('pending', function (data)
{
if( !_.isUndefined(data.id) && !_.isUndefined(data.stats) )
{
Nodes.updatePending(data.id, data.stats, function (err, stats) {
if(err !== null)
{
console.error('API', 'TXS', 'Pending error:', err);
}
if(stats !== null)
{
client.write({
action: 'pending',
data: stats
});
console.success('API', 'TXS', 'Pending:', data.stats['pending'], 'from:', data.id);
}
});
}
else
{
console.error('API', 'TXS', 'Pending error:', data);
}
});
spark.on('stats', function (data)
{
if( !_.isUndefined(data.id) && !_.isUndefined(data.stats) )
{
Nodes.updateStats(data.id, data.stats, function (err, stats)
{
if(err !== null)
{
console.error('API', 'STA', 'Stats error:', err);
}
else
{
if(stats !== null)
{
client.write({
action: 'stats',
data: stats
});
console.success('API', 'STA', 'Stats from:', data.id);
}
}
});
}
else
{
console.error('API', 'STA', 'Stats error:', data);
}
});
spark.on('history', function (data)
{
console.success('API', 'HIS', 'Got history from:', data.id);
var time = chalk.reset.cyan((new Date()).toJSON()) + " ";
console.time(time, 'COL', 'CHR', 'Got charts in');
Nodes.addHistory(data.id, data.history, function (err, history)
{
console.timeEnd(time, 'COL', 'CHR', 'Got charts in');
if(err !== null)
{
console.error('COL', 'CHR', 'History error:', err);
}
else
{
client.write({
action: 'charts',
data: history
});
}
});
});
spark.on('node-ping', function (data)
{
var start = (!_.isUndefined(data) && !_.isUndefined(data.clientTime) ? data.clientTime : null);
spark.emit('node-pong', {
clientTime: start,
serverTime: _.now()
});
console.info('API', 'PIN', 'Ping from:', data['id']);
});
spark.on('latency', function (data)
{
if( !_.isUndefined(data.id) )
{
Nodes.updateLatency(data.id, data.latency, function (err, latency)
{
if(err !== null)
{
console.error('API', 'PIN', 'Latency error:', err);
}
if(latency !== null)
{
client.write({
action: 'latency',
data: latency
});
console.info('API', 'PIN', 'Latency:', latency, 'from:', data.id);
}
});
if( Nodes.requiresUpdate(data.id) && (!Nodes.askedForHistory() || _.now() - askedForHistoryTime > 2*60*1000) )
{
var range = Nodes.getHistory().getHistoryRequestRange();
spark.emit('history', range);
console.info('API', 'HIS', 'Asked:', data.id, 'for history:', range.min, '-', range.max);
Nodes.askedForHistory(true);
askedForHistoryTime = _.now();
}
}
});
spark.on('end', function (data)
{
Nodes.inactive(spark.id, function (err, stats)
{
if(err !== null)
{
console.error('API', 'CON', 'Connection end error:', err);
}
else
{
client.write({
action: 'inactive',
data: stats
});
console.warn('API', 'CON', 'Connection with:', spark.id, 'ended:', data);
}
});
});
});
client.on('connection', function (clientSpark)
{
clientSpark.on('ready', function (data)
{
clientSpark.emit('init', { nodes: Nodes.all() });
Nodes.getCharts();
});
clientSpark.on('client-pong', function (data)
{
var start = (!_.isUndefined(data) && !_.isUndefined(data.serverTime) ? data.serverTime : clientLatency);
var latency = Math.ceil( (_.now() - start) / 2 );
clientSpark.emit('client-latency', { latency: latency });
});
});
var latencyTimeout = setInterval( function ()
{
clientLatency = _.now();
client.write({
action: 'client-ping',
data: {
serverTime: clientLatency
}
});
}, 5000);
// Cleanup old inactive nodes
var nodeCleanupTimeout = setInterval( function ()
{
client.write({
action: 'init',
data: Nodes.all()
});
Nodes.getCharts();
}, 1000*60*60);
server.listen(process.env.PORT || 3000);
module.exports = server;