157 lines
2.7 KiB
JavaScript
157 lines
2.7 KiB
JavaScript
var _ = require('lodash');
|
|
var Blockchain = require('./history');
|
|
var Node = require('./node');
|
|
|
|
var Collection = function Collection()
|
|
{
|
|
this._list = [];
|
|
this._history = new Blockchain();
|
|
this._bestBlock = null;
|
|
|
|
return this;
|
|
}
|
|
|
|
Collection.prototype.add = function(data)
|
|
{
|
|
var node = this.getNodeOrNew({ id : data.id }, data);
|
|
node.setInfo(data);
|
|
|
|
return node.getInfo();
|
|
}
|
|
|
|
Collection.prototype.update = function(id, stats)
|
|
{
|
|
var node = this.getNode({ id: id });
|
|
|
|
if(!node)
|
|
return false;
|
|
|
|
var block = this._history.add(stats.block, id);
|
|
|
|
if(! block)
|
|
return false;
|
|
|
|
var propagationHistory = this._history.getNodePropagation(id);
|
|
|
|
stats.block.arrived = block.arrived;
|
|
stats.block.received = block.received;
|
|
stats.block.propagation = block.propagation;
|
|
|
|
return node.setStats(stats, propagationHistory);
|
|
}
|
|
|
|
Collection.prototype.addHistory = function(id, blocks)
|
|
{
|
|
var node = this.getNode({ id: id });
|
|
|
|
if(!node)
|
|
return false;
|
|
|
|
for (var i = blocks.length - 1; i >= 0; i--) {
|
|
this._history.add(blocks[i], id);
|
|
};
|
|
|
|
return this.getCharts();
|
|
}
|
|
|
|
Collection.prototype.updateLatency = function(id, latency)
|
|
{
|
|
var node = this.getNode({ id: id });
|
|
|
|
if(!node)
|
|
return false;
|
|
|
|
return node.setLatency(latency);
|
|
}
|
|
|
|
Collection.prototype.inactive = function(id)
|
|
{
|
|
var node = this.getNode({ spark: id });
|
|
|
|
if(!node)
|
|
return false;
|
|
|
|
node.setState(false);
|
|
|
|
return node.getStats();
|
|
}
|
|
|
|
Collection.prototype.getIndex = function(search)
|
|
{
|
|
return _.findIndex(this._list, search);
|
|
}
|
|
|
|
Collection.prototype.getNode = function(search)
|
|
{
|
|
var index = this.getIndex(search);
|
|
|
|
if(index >= 0)
|
|
return this._list[index];
|
|
|
|
return false;
|
|
}
|
|
|
|
Collection.prototype.getNodeByIndex = function(index)
|
|
{
|
|
if(this._list[index])
|
|
return this._list[index];
|
|
|
|
return false;
|
|
}
|
|
|
|
Collection.prototype.getIndexOrNew = function(search, data)
|
|
{
|
|
var index = this.getIndex(search);
|
|
|
|
return (index >= 0 ? index : this._list.push(new Node(data)) - 1);
|
|
}
|
|
|
|
Collection.prototype.getNodeOrNew = function(search, data)
|
|
{
|
|
return this.getNodeByIndex(this.getIndexOrNew(search, data));
|
|
}
|
|
|
|
Collection.prototype.all = function()
|
|
{
|
|
return this._list;
|
|
}
|
|
|
|
Collection.prototype.blockPropagationChart = function()
|
|
{
|
|
return this._history.getBlockPropagation();
|
|
}
|
|
|
|
Collection.prototype.getUncleCount = function()
|
|
{
|
|
return this._history.getUncleCount();
|
|
}
|
|
|
|
Collection.prototype.getCharts = function()
|
|
{
|
|
return this._history.getCharts();
|
|
}
|
|
|
|
Collection.prototype.getHistory = function()
|
|
{
|
|
return this._history;
|
|
}
|
|
|
|
Collection.prototype.canNodeUpdate = function(id)
|
|
{
|
|
var node = this.getNode({id: id});
|
|
|
|
if(!node)
|
|
return false;
|
|
|
|
if(node.canUpdate())
|
|
{
|
|
var diff = this._history.bestBlockNumber() - node.getBlockNumber();
|
|
|
|
if(diff <= 0)
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
module.exports = Collection; |