feat: close database connections on stop

database/database.go

@@ -32,6 +32,7 @@ import (
 )
 
 type Database interface {
+	Close() error
 	Metadata() *gorm.DB
 	Blob() *badger.DB
 	Transaction(bool) *Txn
@@ -61,6 +62,23 @@ func (b *BaseDatabase) Transaction(readWrite bool) *Txn {
 	return NewTxn(b, readWrite)
 }
 
+// Close cleans up the database connections
+func (b *BaseDatabase) Close() error {
+	var err error
+	// Close metadata
+	sqlDB, sqlDBerr := b.metadata.DB()
+	if sqlDBerr != nil {
+		err = errors.Join(err, sqlDBerr)
+	} else {
+		metadataErr := sqlDB.Close()
+		err = errors.Join(err, metadataErr)
+	}
+	// Close blob
+	blobErr := b.blob.Close()
+	err = errors.Join(err, blobErr)
+	return err
+}
+
 func (b *BaseDatabase) init() error {
 	if b.logger == nil {
 		// Create logger to throw away logs

node.go

@@ -121,6 +121,9 @@ func (n *Node) Stop() error {
 func (n *Node) shutdown() error {
 	ctx := context.TODO()
 	var err error
+	// Shutdown ledger
+	err = errors.Join(err, n.ledgerState.Close())
+	// Call shutdown functions
 	for _, fn := range n.shutdownFuncs {
 		err = errors.Join(err, fn(ctx))
 	}

state/state.go

@@ -109,6 +109,10 @@ func NewLedgerState(cfg LedgerStateConfig) (*LedgerState, error) {
 	return ls, nil
 }
 
+func (ls *LedgerState) Close() error {
+	return ls.db.Close()
+}
+
 func (ls *LedgerState) scheduleCleanupConsumedUtxos() {
 	ls.Lock()
 	defer ls.Unlock()