2014-01-10 11:02:33 -08:00
|
|
|
'use strict';
|
2014-01-07 10:21:59 -08:00
|
|
|
|
2014-01-10 11:02:33 -08:00
|
|
|
require('classtool');
|
2014-01-08 09:50:37 -08:00
|
|
|
|
|
|
|
|
2014-01-10 11:02:33 -08:00
|
|
|
function spec() {
|
2014-01-10 16:42:39 -08:00
|
|
|
var mongoose = require('mongoose');
|
|
|
|
var util = require('util');
|
|
|
|
var RpcClient = require('bitcore/RpcClient').class();
|
|
|
|
var networks = require('bitcore/networks');
|
|
|
|
var async = require('async');
|
|
|
|
var config = require('../config/config');
|
|
|
|
var Block = require('../app/models/Block');
|
|
|
|
var Transaction = require('../app/models/Transaction');
|
|
|
|
var TransactionItem = require('../app/models/TransactionItem');
|
2014-01-14 11:56:02 -08:00
|
|
|
var sockets = require('../app/views/sockets/main.js');
|
2014-01-15 12:36:49 -08:00
|
|
|
var CONCURRENCY = 5;
|
|
|
|
|
2014-01-10 11:57:16 -08:00
|
|
|
|
|
|
|
function Sync(config) {
|
2014-01-15 12:36:49 -08:00
|
|
|
this.tx_count = 0;
|
|
|
|
this.block_count= 0;
|
|
|
|
this.block_total= 0;
|
|
|
|
this.network = config.networkName === 'testnet' ? networks.testnet: networks.livenet;
|
2014-01-10 11:57:16 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
var progress_bar = function(string, current, total) {
|
|
|
|
console.log(util.format('\t%s %d/%d [%d%%]', string, current, total, parseInt(100 * current / total)));
|
|
|
|
};
|
|
|
|
|
2014-01-15 12:36:49 -08:00
|
|
|
Sync.prototype.getPrevNextBlock = function(blockHash, blockEnd, opts, cb) {
|
2014-01-10 11:57:16 -08:00
|
|
|
var that = this;
|
2014-01-15 12:36:49 -08:00
|
|
|
|
|
|
|
// recursion end.
|
|
|
|
if (!blockHash || (blockEnd && blockEnd == blockHash) ) {
|
2014-01-10 11:57:16 -08:00
|
|
|
return cb();
|
|
|
|
}
|
|
|
|
|
2014-01-15 12:36:49 -08:00
|
|
|
var existed = 0;
|
|
|
|
var blockInfo;
|
|
|
|
var blockObj;
|
|
|
|
|
|
|
|
async.series([
|
|
|
|
// Already got it?
|
|
|
|
function(c) {
|
|
|
|
Block.findOne({hash:blockHash}, function(err,block){
|
|
|
|
if (err) { console.log(err); return c(err); };
|
|
|
|
if (block) {
|
|
|
|
existed = 1;
|
|
|
|
blockObj = block;
|
|
|
|
}
|
|
|
|
|
|
|
|
return c();
|
|
|
|
});
|
|
|
|
},
|
|
|
|
//show some (inacurate) status
|
|
|
|
function(c) {
|
|
|
|
if (that.block_count++ % 1000 === 0) {
|
|
|
|
progress_bar('Historic sync status:', that.block_count, that.block_total);
|
|
|
|
}
|
|
|
|
return c();
|
|
|
|
},
|
|
|
|
//get Info from RPC
|
|
|
|
function(c) {
|
|
|
|
|
|
|
|
// TODO: if we store prev/next, no need to go to RPC
|
|
|
|
// if (blockObj && blockObj.nextBlockHash) return c();
|
|
|
|
|
|
|
|
that.rpc.getBlock(blockHash, function(err, ret) {
|
|
|
|
if (err) return c(err);
|
|
|
|
|
|
|
|
blockInfo = ret;
|
|
|
|
return c();
|
|
|
|
});
|
|
|
|
},
|
|
|
|
//store it
|
|
|
|
function(c) {
|
|
|
|
if (existed) return c();
|
|
|
|
|
|
|
|
that.storeBlock(blockInfo.result, function(err, block) {
|
|
|
|
existed = err && err.toString().match(/E11000/);
|
|
|
|
if (err && ! existed) return c(err);
|
|
|
|
return c();
|
|
|
|
});
|
|
|
|
},
|
|
|
|
/* TODO: Should Start to sync backwards? (this is for partial syncs)
|
|
|
|
function(c) {
|
|
|
|
|
|
|
|
if (blockInfo.result.prevblockhash != current.blockHash) {
|
|
|
|
console.log("reorg?");
|
|
|
|
opts.prev = 1;
|
2014-01-10 11:57:16 -08:00
|
|
|
}
|
2014-01-15 12:36:49 -08:00
|
|
|
return c();
|
2014-01-14 02:58:01 -08:00
|
|
|
}
|
2014-01-15 12:36:49 -08:00
|
|
|
*/
|
|
|
|
],
|
|
|
|
function (err){
|
|
|
|
|
|
|
|
if (err)
|
|
|
|
console.log("ERROR: @%s: %s [count: block_count: %d]", blockHash, err, that.block_count);
|
|
|
|
|
|
|
|
if (blockInfo && blockInfo.result) {
|
|
|
|
if (opts.prev && blockInfo.result.prevblockhash)
|
|
|
|
return that.getPrevNextBlock(blockInfo.result.prevblockhash, blockEnd, opts, cb);
|
|
|
|
|
|
|
|
if (opts.next && blockInfo.result.nextblockhash)
|
|
|
|
return that.getPrevNextBlock(blockInfo.result.nextblockhash, blockEnd, opts, cb);
|
|
|
|
}
|
|
|
|
return cb(err);
|
2014-01-10 11:57:16 -08:00
|
|
|
});
|
|
|
|
};
|
|
|
|
|
|
|
|
Sync.prototype.storeBlock = function(block, cb) {
|
2014-01-14 14:42:38 -08:00
|
|
|
var that = this;
|
2014-01-15 12:36:49 -08:00
|
|
|
|
2014-01-15 05:32:18 -08:00
|
|
|
Block.createTimestamped(block, function(err, b){
|
2014-01-15 12:36:49 -08:00
|
|
|
|
2014-01-14 14:42:38 -08:00
|
|
|
if (b && that.opts.broadcast_blocks) {
|
|
|
|
sockets.broadcast_block(b);
|
|
|
|
}
|
2014-01-10 11:57:16 -08:00
|
|
|
|
2014-01-15 12:36:49 -08:00
|
|
|
if (that.opts.broadcast_txs) {
|
|
|
|
block.tx.each(function(tx) {
|
2014-01-14 12:25:04 -08:00
|
|
|
sockets.broadcast_tx(new_tx);
|
2014-01-15 12:36:49 -08:00
|
|
|
});
|
|
|
|
}
|
2014-01-14 13:09:45 -08:00
|
|
|
|
2014-01-15 12:36:49 -08:00
|
|
|
that.tx_count += block.tx.length;
|
2014-01-14 13:09:45 -08:00
|
|
|
|
2014-01-15 12:36:49 -08:00
|
|
|
return cb();
|
2014-01-11 20:29:25 -08:00
|
|
|
});
|
2014-01-10 11:57:16 -08:00
|
|
|
};
|
|
|
|
|
2014-01-15 12:36:49 -08:00
|
|
|
Sync.prototype.syncBlocks = function(start, end, cb) {
|
2014-01-10 11:57:16 -08:00
|
|
|
var that = this;
|
|
|
|
|
2014-01-15 12:36:49 -08:00
|
|
|
console.log('Syncing Blocks, starting from: %s end: %s ',start, end);
|
2014-01-10 11:57:16 -08:00
|
|
|
|
2014-01-15 12:36:49 -08:00
|
|
|
return that.getPrevNextBlock(start, end, { next: 1 }, cb);
|
2014-01-10 11:57:16 -08:00
|
|
|
};
|
|
|
|
|
|
|
|
// This is not currently used. Transactions are represented by txid only
|
|
|
|
// in mongodb
|
2014-01-14 02:58:01 -08:00
|
|
|
Sync.prototype.syncTXs = function(cb) {
|
2014-01-10 11:57:16 -08:00
|
|
|
|
|
|
|
var that = this;
|
|
|
|
|
|
|
|
console.log('Syncing TXs...');
|
|
|
|
|
|
|
|
Transaction.find({
|
|
|
|
blockhash: null
|
|
|
|
},
|
|
|
|
function(err, txs) {
|
|
|
|
if (err) return cb(err);
|
|
|
|
|
|
|
|
var read = 0;
|
|
|
|
var pull = 0;
|
|
|
|
var write = 0;
|
|
|
|
var total = txs.length;
|
|
|
|
console.log('\tneed to pull %d txs', total);
|
|
|
|
|
|
|
|
if (!total) return cb();
|
|
|
|
|
|
|
|
async.each(txs, function(tx, next) {
|
|
|
|
if (!tx.txid) {
|
|
|
|
console.log('NO TXID skipping...', tx);
|
|
|
|
return next();
|
|
|
|
}
|
|
|
|
|
|
|
|
if (read++ % 1000 === 0) progress_bar('read', read, total);
|
|
|
|
|
|
|
|
that.rpc.getRawTransaction(tx.txid, 1, function(err, txInfo) {
|
|
|
|
|
|
|
|
if (pull++ % 1000 === 0) progress_bar('\tpull', pull, total);
|
|
|
|
|
|
|
|
if (!err && txInfo) {
|
|
|
|
Transaction.update({
|
|
|
|
txid: tx.txid
|
|
|
|
},
|
|
|
|
txInfo.result, function(err) {
|
|
|
|
if (err) return next(err);
|
|
|
|
|
|
|
|
if (write++ % 1000 === 0) progress_bar('\t\twrite', write, total);
|
|
|
|
|
|
|
|
return next();
|
|
|
|
});
|
|
|
|
}
|
|
|
|
else return next();
|
|
|
|
});
|
|
|
|
},
|
|
|
|
function(err) {
|
|
|
|
if (err) return cb(err);
|
|
|
|
return cb(err);
|
|
|
|
});
|
|
|
|
});
|
|
|
|
};
|
|
|
|
|
2014-01-10 16:42:39 -08:00
|
|
|
|
2014-01-11 22:06:14 -08:00
|
|
|
// Not used
|
2014-01-10 16:42:39 -08:00
|
|
|
Sync.prototype.processTXs = function(reindex, cb) {
|
|
|
|
|
|
|
|
var that = this;
|
|
|
|
|
|
|
|
console.log('Syncing TXs...');
|
|
|
|
|
|
|
|
var filter = reindex ? {} : { processed: false } ;
|
|
|
|
|
2014-01-11 17:58:00 -08:00
|
|
|
Transaction.find(filter, function(err, txs) {
|
|
|
|
if (err) return cb(err);
|
2014-01-10 16:42:39 -08:00
|
|
|
|
2014-01-11 17:58:00 -08:00
|
|
|
var read = 0,
|
|
|
|
pull = 0,
|
|
|
|
proc = 0,
|
|
|
|
total = txs.length;
|
2014-01-10 16:42:39 -08:00
|
|
|
|
2014-01-11 17:58:00 -08:00
|
|
|
console.log('\tneed to pull %d txs', total);
|
2014-01-10 16:42:39 -08:00
|
|
|
|
2014-01-11 17:58:00 -08:00
|
|
|
if (!total) return cb();
|
2014-01-10 16:42:39 -08:00
|
|
|
|
2014-01-14 13:09:45 -08:00
|
|
|
|
|
|
|
async.forEachLimit(txs, CONCURRENCY, function(tx, next) {
|
2014-01-10 16:42:39 -08:00
|
|
|
if (read++ % 1000 === 0) progress_bar('read', read, total);
|
|
|
|
|
|
|
|
if (!tx.txid) {
|
|
|
|
console.log('NO TXID skipping...', tx);
|
|
|
|
return next();
|
|
|
|
}
|
|
|
|
|
|
|
|
// This will trigger an RPC call
|
2014-01-11 17:58:00 -08:00
|
|
|
Transaction.explodeTransactionItems( tx.txid, function(err) {
|
|
|
|
if (proc++ % 1000 === 0) progress_bar('\tproc', pull, total);
|
|
|
|
next(err);
|
2014-01-10 16:42:39 -08:00
|
|
|
});
|
|
|
|
},
|
2014-01-11 17:58:00 -08:00
|
|
|
cb);
|
|
|
|
});
|
2014-01-10 16:42:39 -08:00
|
|
|
};
|
|
|
|
|
2014-01-10 11:57:16 -08:00
|
|
|
Sync.prototype.init = function(opts) {
|
2014-01-14 11:45:25 -08:00
|
|
|
this.rpc = new RpcClient(config.bitcoind);
|
|
|
|
|
|
|
|
|
2014-01-13 10:17:34 -08:00
|
|
|
if (!(opts && opts.skip_db_connection)) {
|
2014-01-14 11:45:25 -08:00
|
|
|
mongoose.connect(config.db, {server: {auto_reconnect: true}} );
|
2014-01-13 10:17:34 -08:00
|
|
|
}
|
2014-01-14 11:56:02 -08:00
|
|
|
this.opts = opts;
|
2014-01-10 11:57:16 -08:00
|
|
|
this.db = mongoose.connection;
|
|
|
|
|
2014-01-14 11:45:25 -08:00
|
|
|
this.db.on('error', function(err) {
|
|
|
|
console.log('connection error:' + err);
|
|
|
|
moogose.disconnect();
|
|
|
|
});
|
|
|
|
|
|
|
|
this.db.on('disconnect', function(err) {
|
|
|
|
console.log('disconnect:' + err);
|
|
|
|
mongoose.connect(config.db, {server: {auto_reconnect: true}} );
|
|
|
|
});
|
|
|
|
|
2014-01-10 11:57:16 -08:00
|
|
|
|
|
|
|
};
|
|
|
|
|
|
|
|
Sync.prototype.import_history = function(opts, next) {
|
|
|
|
|
|
|
|
var that = this;
|
2014-01-14 11:45:25 -08:00
|
|
|
|
|
|
|
var retry_attemps = 100;
|
|
|
|
var retry_secs = 2;
|
|
|
|
|
2014-01-10 11:57:16 -08:00
|
|
|
this.db.once('open', function() {
|
|
|
|
async.series([
|
|
|
|
function(cb) {
|
|
|
|
if (opts.destroy) {
|
|
|
|
console.log('Deleting Blocks...');
|
|
|
|
that.db.collections.blocks.drop(cb);
|
|
|
|
} else {
|
|
|
|
cb();
|
|
|
|
}
|
|
|
|
},
|
|
|
|
function(cb) {
|
|
|
|
if (opts.destroy) {
|
|
|
|
console.log('Deleting TXs...');
|
|
|
|
that.db.collections.transactions.drop(cb);
|
|
|
|
} else {
|
|
|
|
cb();
|
|
|
|
}
|
|
|
|
},
|
2014-01-11 22:06:14 -08:00
|
|
|
function(cb) {
|
|
|
|
if (opts.destroy) {
|
|
|
|
console.log('Deleting TXItems...');
|
|
|
|
that.db.collections.transactionitems.drop(cb);
|
|
|
|
} else {
|
|
|
|
cb();
|
|
|
|
}
|
|
|
|
},
|
2014-01-15 12:36:49 -08:00
|
|
|
function(cb) {
|
|
|
|
that.rpc.getInfo(function(err, res) {
|
|
|
|
if (err) cb(err);
|
|
|
|
|
|
|
|
that.block_total = res.result.blocks;
|
|
|
|
return cb();
|
|
|
|
});
|
|
|
|
},
|
2014-01-10 11:57:16 -08:00
|
|
|
function(cb) {
|
2014-01-14 11:45:25 -08:00
|
|
|
function sync() {
|
|
|
|
|
2014-01-15 12:36:49 -08:00
|
|
|
var startingBlockHash = that.network.genesisBlock.hash.reverse().toString('hex');
|
|
|
|
|
|
|
|
that.syncBlocks( startingBlockHash, null, function(err) {
|
2014-01-14 11:45:25 -08:00
|
|
|
|
|
|
|
if (err && err.message.match(/ECONNREFUSED/) && retry_attemps--){
|
|
|
|
setTimeout(function() {
|
|
|
|
console.log("Retrying in %d secs ", retry_secs);
|
|
|
|
sync();
|
|
|
|
}, retry_secs * 1000);
|
|
|
|
}
|
|
|
|
else
|
2014-01-15 12:36:49 -08:00
|
|
|
return next(err, that.block_count);
|
2014-01-14 11:45:25 -08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2014-01-10 11:57:16 -08:00
|
|
|
if (!opts.skip_blocks) {
|
2014-01-14 11:45:25 -08:00
|
|
|
sync();
|
2014-01-10 11:57:16 -08:00
|
|
|
}
|
|
|
|
},
|
2014-01-10 16:42:39 -08:00
|
|
|
], function(err) {
|
2014-01-15 12:36:49 -08:00
|
|
|
return next(err, that.block_count);
|
2014-01-10 11:57:16 -08:00
|
|
|
});
|
|
|
|
});
|
|
|
|
};
|
|
|
|
|
|
|
|
Sync.prototype.close = function() {
|
|
|
|
console.log("closing connection");
|
|
|
|
this.db.close();
|
|
|
|
};
|
|
|
|
return Sync;
|
2014-01-10 11:02:33 -08:00
|
|
|
}
|
2014-01-07 10:21:59 -08:00
|
|
|
module.defineClass(spec);
|
2014-01-10 11:02:33 -08:00
|
|
|
|