RanchiMall/fcoin
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

migrate: update tx map.

Browse Source
This commit is contained in:
Christopher Jeffrey 2016-10-04 15:18:23 -07:00
parent 5c3ed7dfb8
commit be34f8c7bf
No known key found for this signature in database
GPG Key ID: 8962AB9DE6666BBD
1 changed files with 40 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
migrate/walletdb2to3.js
View File

@ -28,6 +28,7 @@ db = bcoin.ldb({
}); });
var updateVersion = co(function* updateVersion() { var updateVersion = co(function* updateVersion() {
var bak = process.env.HOME + '/walletdb-bak-' + Date.now() + '.ldb';
var data, ver; var data, ver;
console.log('Checking version.'); console.log('Checking version.');
@ -42,9 +43,9 @@ var updateVersion = co(function* updateVersion() {
if (ver !== 2) if (ver !== 2)
throw Error('DB is version ' + ver + '.'); throw Error('DB is version ' + ver + '.');
console.log('Backing up DB.'); console.log('Backing up DB to: %s.', bak);
yield db.backup(process.env.HOME + '/walletdb-bak-' + Date.now() + '.ldb'); yield db.backup(bak);
ver = new Buffer(4); ver = new Buffer(4);
ver.writeUInt32LE(3, 0, true); ver.writeUInt32LE(3, 0, true);
@ -94,7 +95,7 @@ var updatePathMap = co(function* updatePathMap() {
batch.put(layout.P(key, hash), path.toRaw()); batch.put(layout.P(key, hash), path.toRaw());
} }
batch.put(layout.p(hash), serializeWallets(keys.sort())); batch.put(item.key, serializeWallets(keys.sort()));
} }
console.log('Migrated %d paths.', total); console.log('Migrated %d paths.', total);
@ -121,7 +122,7 @@ var updateAccounts = co(function* updateAccounts() {
total++; total++;
account = accountFromRaw(item.value, item.key); account = accountFromRaw(item.value, item.key);
account = new Account({ network: account.network, options: {} }, account); account = new Account({ network: account.network, options: {} }, account);
batch.put(layout.a(account.wid, account.accountIndex), account.toRaw()); batch.put(item.key, account.toRaw());
if (account._old) { if (account._old) {
batch.del(layout.i(account.wid, account._old)); batch.del(layout.i(account.wid, account._old));
@ -155,7 +156,7 @@ var updateWallets = co(function* updateWallets() {
total++; total++;
wallet = walletFromRaw(item.value); wallet = walletFromRaw(item.value);
wallet = new Wallet({ network: wallet.network }, wallet); wallet = new Wallet({ network: wallet.network }, wallet);
batch.put(layout.w(wallet.wid), wallet.toRaw()); batch.put(item.key, wallet.toRaw());
if (wallet._old) { if (wallet._old) {
batch.del(layout.l(wallet._old)); batch.del(layout.l(wallet._old));
@ -168,6 +169,31 @@ var updateWallets = co(function* updateWallets() {
console.log('Migrated %d wallets.', total); console.log('Migrated %d wallets.', total);
}); });
var updateTXMap = co(function* updateTXMap() {
var total = 0;
var iter, item, wallets;
iter = db.iterator({
gte: layout.e(constants.NULL_HASH),
lte: layout.e(constants.HIGH_HASH),
values: true
});
console.log('Migrating tx map.');
for (;;) {
item = yield iter.next();
if (!item)
break;
wallets = parseWallets(item.value);
batch.put(item.key, serializeWallets(wallets.sort()));
}
console.log('Migrated %d tx map.', total);
});
function pathFromRaw(data) { function pathFromRaw(data) {
var path = {}; var path = {};
var p = new BufferReader(data); var p = new BufferReader(data);
@ -217,6 +243,14 @@ function parsePaths(data, hash) {
return out; return out;
} }
function parseWallets(data) {
var p = new BufferReader(data);
var wallets = [];
while (p.left())
wallets.push(p.readU32());
return wallets;
}
function serializeWallets(wallets) { function serializeWallets(wallets) {
var p = new BufferWriter(); var p = new BufferWriter();
var i, wid; var i, wid;
@ -336,6 +370,7 @@ co.spawn(function* () {
yield updatePathMap(); yield updatePathMap();
yield updateAccounts(); yield updateAccounts();
yield updateWallets(); yield updateWallets();
yield updateTXMap();
yield batch.write(); yield batch.write();
}).then(function() { }).then(function() {
console.log('Migration complete.'); console.log('Migration complete.');