From 7342952ec48748d2e5496e233842bc4a7d91edfc Mon Sep 17 00:00:00 2001
From: Javed Khan <tuxcanfly@gmail.com>
Date: Wed, 27 Mar 2019 18:58:27 +0530
Subject: [PATCH] migrate: update chaindb5to6 migration

---
 migrate/chaindb5to6.js | 56 +++++++++++++++++++++++++++++++-----------
 1 file changed, 41 insertions(+), 15 deletions(-)

diff --git a/migrate/chaindb5to6.js b/migrate/chaindb5to6.js
index 29d2642a..8a85b607 100644
--- a/migrate/chaindb5to6.js
+++ b/migrate/chaindb5to6.js
@@ -17,6 +17,34 @@ const db = bdb.create({
   createIfMissing: false
 });
 
+async function updateVersion() {
+  const ver = await checkVersion();
+
+  console.log('Updating version to %d.', ver + 1);
+
+  const buf = Buffer.allocUnsafe(5 + 4);
+  buf.write('chain', 0, 'ascii');
+  buf.writeUInt32LE(6, 5, true);
+
+  const parent = db.batch();
+  parent.put(layout.V.encode(), buf);
+  await parent.write();
+}
+
+async function checkVersion() {
+  console.log('Checking version.');
+
+  const data = await db.get(layout.V.encode());
+  assert(data, 'No version.');
+
+  const ver = data.readUInt32LE(5, true);
+
+  if (ver !== 5)
+    throw Error(`DB is version ${ver}.`);
+
+  return ver;
+}
+
 async function removeKey(name, key) {
   const iter = db.iterator({
     gte: key.min(),
@@ -43,6 +71,16 @@ async function removeKey(name, key) {
   console.log('Cleaned up %d %s index records.', total, name);
 }
 
+async function migrateIndexes() {
+  const t = bdb.key('t', ['hash256']);
+  const T = bdb.key('T', ['hash', 'hash256']);
+  const C = bdb.key('C', ['hash', 'hash256', 'uint32']);
+
+  await removeKey('hash -> tx', t);
+  await removeKey('addr -> tx', T);
+  await removeKey('addr -> coin', C);
+}
+
 /*
  * Execute
  */
@@ -51,24 +89,12 @@ async function removeKey(name, key) {
   await db.open();
 
   console.log('Opened %s.', process.argv[2]);
-  console.log('Checking version.');
-  await db.verify(layout.V.build(), 'chain', 5);
 
-  const t = bdb.key('t', ['hash256']);
-  const T = bdb.key('T', ['hash', 'hash256']);
-  const C = bdb.key('C', ['hash', 'hash256', 'uint32']);
+  await checkVersion();
+  await migrateIndexes();
+  await updateVersion();
 
-  await removeKey('hash -> tx', t);
-  await removeKey('addr -> tx', T);
-  await removeKey('addr -> coin', C);
-
-  console.log('Compacting database...');
   await db.compactRange();
-
-  console.log('Updating version to %d.', 6);
-  await db.del(layout.V.build());
-  await db.verify(layout.V.build(), 'chain', 6);
-
   await db.close();
 })().then(() => {
   console.log('Migration complete.');