cmd: gracefully close stores on shutdown

cmd/walletd/node.go

@@ -123,13 +123,23 @@ func (db *boltDB) Close() error {
 }
 
 type node struct {
-	cm *chain.Manager
-	s  *syncer.Syncer
-	wm *wallet.Manager
+	chainStore *boltDB
+	cm         *chain.Manager
+
+	s *syncer.Syncer
+
+	walletStore *sqlite.Store
+	wm          *wallet.Manager
 
 	Start func() (stop func())
 }
 
+// Close shuts down the node and closes its database.
+func (n *node) Close() error {
+	n.chainStore.Close()
+	return n.walletStore.Close()
+}
+
 func newNode(addr, dir string, chainNetwork string, useUPNP bool, log *zap.Logger) (*node, error) {
 	var network *consensus.Network
 	var genesisBlock types.Block
@@ -215,9 +225,11 @@ func newNode(addr, dir string, chainNetwork string, useUPNP bool, log *zap.Logge
 	}
 
 	return &node{
-		cm: cm,
-		s:  s,
-		wm: wm,
+		chainStore:  db,
+		cm:          cm,
+		s:           s,
+		walletStore: walletDB,
+		wm:          wm,
 		Start: func() func() {
 			ch := make(chan struct{})
 			go func() {