[wallet] Add FlushWallets() function to wallet/init.cpp

jnewbery · jnewbery · commit 2da5eafa47cd · 2017-09-07T16:21:26.000-07:00
diff --git a/src/init.cpp b/src/init.cpp
@@ -189,9 +189,7 @@ void Shutdown()
     StopRPC();
     StopHTTPServer();
 #ifdef ENABLE_WALLET
-    for (CWalletRef pwallet : vpwallets) {
-        pwallet->Flush(false);
-    }
+    FlushWallets(false);
 #endif
     MapPort(false);
     UnregisterValidationInterface(peerLogic.get());
@@ -246,9 +244,7 @@ void Shutdown()
         pblocktree = nullptr;
     }
 #ifdef ENABLE_WALLET
-    for (CWalletRef pwallet : vpwallets) {
-        pwallet->Flush(true);
-    }
+    FlushWallets(true);
 #endif
 
 #if ENABLE_ZMQ
diff --git a/src/wallet/init.cpp b/src/wallet/init.cpp
@@ -245,3 +245,9 @@ bool OpenWallets()
 
     return true;
 }
+
+void FlushWallets(bool shutdown) {
+    for (CWalletRef pwallet : vpwallets) {
+        pwallet->Flush(shutdown);
+    }
+}
diff --git a/src/wallet/init.h b/src/wallet/init.h
@@ -22,4 +22,7 @@ bool VerifyWallets();
 //! Load wallet databases.
 bool OpenWallets();
 
+//! Flush all wallets in preparation for shutdown.
+//! Call with shutdown = true to actually shutdown the wallet.
+void FlushWallets(bool shutdown);
 #endif // BITCOIN_WALLET_INIT_H

Original file line number	Diff line number	Diff line change
`@@ -245,3 +245,9 @@ bool OpenWallets()`
`245`	`245`
`246`	`246`	`return true;`
`247`	`247`	`}`
	`248`	`+`
	`249`	`+void FlushWallets(bool shutdown) {`
	`250`	`+ for (CWalletRef pwallet : vpwallets) {`
	`251`	`+ pwallet->Flush(shutdown);`
	`252`	`+ }`
	`253`	`+}`