173 lines
3.9 KiB
JavaScript
173 lines
3.9 KiB
JavaScript
'use strict';
|
|
|
|
require('classtool');
|
|
|
|
|
|
function spec() {
|
|
|
|
var TIMESTAMP_ROOT = 'b-ts-'; // b-ts-<ts> => <hash>
|
|
var PREV_ROOT = 'b-prev-'; // b-prev-<hash> => <prev_hash> (0 if orphan)
|
|
|
|
|
|
/**
|
|
* Module dependencies.
|
|
*/
|
|
var RpcClient = require('bitcore/RpcClient').class(),
|
|
util = require('bitcore/util/util'),
|
|
levelup = require('levelup'),
|
|
BitcoreBlock= require('bitcore/Block').class(),
|
|
config = require('../config/config');
|
|
|
|
var BlockDb = function(db) {
|
|
this.db = db || levelup(config.leveldb + '/blocks');
|
|
};
|
|
|
|
BlockDb.prototype.close = function(cb) {
|
|
this.db.close(cb);
|
|
};
|
|
|
|
BlockDb.prototype.drop = function(cb) {
|
|
var self = this;
|
|
var path = config.leveldb + '/blocks';
|
|
self.db.close(function() {
|
|
require('leveldown').destroy(path, function () {
|
|
self.db = levelup(path);
|
|
return cb();
|
|
});
|
|
});
|
|
};
|
|
|
|
BlockDb.prototype.add = function(b, cb) {
|
|
var self = this;
|
|
|
|
if (!b.hash) return cb(new Error('no Hash at Block.save'));
|
|
|
|
|
|
var time_key = TIMESTAMP_ROOT +
|
|
( b.time || Math.round(new Date().getTime() / 1000) );
|
|
|
|
self.db.batch()
|
|
.put(time_key, b.hash)
|
|
.put(PREV_ROOT + b.hash, b.previousblockhash)
|
|
.write(cb);
|
|
};
|
|
|
|
|
|
|
|
BlockDb.prototype.setOrphan = function(hash, cb) {
|
|
var self = this;
|
|
|
|
var k = PREV_ROOT + hash;
|
|
|
|
self.db.get(k, function (err,oldPrevHash) {
|
|
if (err || !oldPrevHash) return cb(err);
|
|
self.db.put(PREV_ROOT + hash, 0, function() {
|
|
return cb(err, oldPrevHash);
|
|
});
|
|
});
|
|
// We keep the block in TIMESTAMP_ROOT
|
|
};
|
|
|
|
//mainly for testing
|
|
BlockDb.prototype.setPrev = function(hash, prevHash, cb) {
|
|
this.db.put(PREV_ROOT + hash, prevHash, function(err) {
|
|
return cb(err);
|
|
});
|
|
};
|
|
|
|
//mainly for testing
|
|
BlockDb.prototype.getPrev = function(hash, cb) {
|
|
this.db.get(PREV_ROOT + hash, function(err,val) {
|
|
return cb(err,val);
|
|
});
|
|
};
|
|
|
|
|
|
|
|
BlockDb.prototype.countNotOrphan = function(cb) {
|
|
var c = 0;
|
|
console.log('Counting connected blocks. This could take some minutes');
|
|
this.db.createReadStream({start: PREV_ROOT, end: PREV_ROOT + '~' })
|
|
.on('data', function (data) {
|
|
if (data.value !== 0) c++;
|
|
})
|
|
.on('error', function (err) {
|
|
return cb(err);
|
|
})
|
|
.on('end', function () {
|
|
return cb(null, c);
|
|
});
|
|
};
|
|
|
|
BlockDb.prototype.has = function(hash, cb) {
|
|
var self = this;
|
|
|
|
var k = PREV_ROOT + hash;
|
|
self.db.get(k, function (err,val) {
|
|
var ret;
|
|
if (err && err.notFound) {
|
|
err = null;
|
|
ret = false;
|
|
}
|
|
if (typeof val !== 'undefined') {
|
|
ret = true;
|
|
}
|
|
return cb(err, ret);
|
|
});
|
|
};
|
|
|
|
|
|
BlockDb.prototype.fromHashWithInfo = function(hash, cb) {
|
|
var rpc = new RpcClient(config.bitcoind);
|
|
|
|
rpc.getBlock(hash, function(err, info) {
|
|
// Not found?
|
|
if (err && err.code === -5) return cb();
|
|
if (err) return cb(err);
|
|
|
|
info.reward = BitcoreBlock.getBlockValue(info.height) / util.COIN ;
|
|
|
|
return cb(null, {
|
|
hash: hash,
|
|
info: info.result,
|
|
});
|
|
});
|
|
};
|
|
|
|
BlockDb.prototype.getBlocksByDate = function(start_ts, end_ts, cb) {
|
|
var self = this;
|
|
var list = [];
|
|
self.db.createReadStream({
|
|
start: TIMESTAMP_ROOT + start_ts,
|
|
end: TIMESTAMP_ROOT + end_ts
|
|
})
|
|
.on('data', function (data) {
|
|
list.push({
|
|
ts: data.key.replace(TIMESTAMP_ROOT, ''),
|
|
hash: data.value,
|
|
info: {}
|
|
});
|
|
})
|
|
.on('error', function (err) {
|
|
return cb(err);
|
|
})
|
|
.on('end', function () {
|
|
return cb(null, list);
|
|
});
|
|
};
|
|
|
|
BlockDb.blockIndex = function(height, cb) {
|
|
var rpc = new RpcClient(config.bitcoind);
|
|
rpc.getBlockHash(height, function(err, bh){
|
|
if (err) return cb(err);
|
|
|
|
cb(null, { blockHash: bh.result });
|
|
});
|
|
};
|
|
|
|
return BlockDb;
|
|
}
|
|
module.defineClass(spec);
|
|
|
|
|