From 3eb28d2d37c149323cbfc73441fe37db52255c21 Mon Sep 17 00:00:00 2001
From: Alex <alex@akselrod.org>
Date: Wed, 12 Jul 2017 18:04:37 -0600
Subject: [PATCH] wallet: batch initial block hash catch-up process every 10K
 blocks

---
 wallet/wallet.go | 36 ++++++++++++++++++++++++++++++------
 1 file changed, 30 insertions(+), 6 deletions(-)

diff --git a/wallet/wallet.go b/wallet/wallet.go
index f584fb0..4a5c6e3 100644
--- a/wallet/wallet.go
+++ b/wallet/wallet.go
@@ -356,21 +356,45 @@ func (w *Wallet) syncWithChain() error {
 		if err != nil {
 			return err
 		}
+		// Initialize the first database transaction.
+		tx, err := w.db.BeginReadWriteTx()
+		if err != nil {
+			return err
+		}
+		ns := tx.ReadWriteBucket(waddrmgrNamespaceKey)
 		for height := int32(1); height <= bestHeight; height++ {
 			hash, err := chainClient.GetBlockHash(int64(height))
 			if err != nil {
+				tx.Rollback()
 				return err
 			}
-			err = walletdb.Update(w.db, func(tx walletdb.ReadWriteTx) error {
-				ns := tx.ReadWriteBucket(waddrmgrNamespaceKey)
-				return w.Manager.SetSyncedTo(ns, &waddrmgr.BlockStamp{
-					Hash:   *hash,
-					Height: height,
-				})
+			err = w.Manager.SetSyncedTo(ns, &waddrmgr.BlockStamp{
+				Hash:   *hash,
+				Height: height,
 			})
 			if err != nil {
+				tx.Rollback()
 				return err
 			}
+			// Every 10K blocks, commit and start a new database TX.
+			if height%10000 == 0 {
+				err = tx.Commit()
+				if err != nil {
+					tx.Rollback()
+					return err
+				}
+				tx, err = w.db.BeginReadWriteTx()
+				if err != nil {
+					return err
+				}
+				ns = tx.ReadWriteBucket(waddrmgrNamespaceKey)
+			}
+		}
+		// Commit (or roll back) the final database transaction.
+		err = tx.Commit()
+		if err != nil {
+			tx.Rollback()
+			return err
 		}
 	}