feat(store): Add snapshot manager Close method (#17294)

Co-authored-by: Julien Robert <[email protected]>
cosmos · Aug 6, 2023 · 6a76448 · 6a76448
1 parent 213ddf7
commit 6a76448
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/store/CHANGELOG.md b/store/CHANGELOG.md
@@ -25,6 +25,10 @@ Ref: https://keepachangelog.com/en/1.0.0/
 
 ## [Unreleased]
 
+### Features
+
+* [#17294](https://github.com/cosmos/cosmos-sdk/pull/17294) Add snapshot manager Close method.
+
 ### Improvements
 
 * [#17158](https://github.com/cosmos/cosmos-sdk/pull/17158) Start the goroutine after need to create a snapshot.

diff --git a/store/snapshots/manager.go b/store/snapshots/manager.go
@@ -551,3 +551,9 @@ func (m *Manager) snapshot(height int64) {
 		m.logger.Debug("pruned state snapshots", "pruned", pruned)
 	}
 }
+
+// Close the snapshot database.
+func (m *Manager) Close() error {
+	m.logger.Info("snapshotManager Close Database")
+	return m.store.db.Close()
+}
diff --git a/store/snapshots/manager_test.go b/store/snapshots/manager_test.go
@@ -35,6 +35,8 @@ func TestManager_List(t *testing.T) {
 	list, err := manager.List()
 	require.NoError(t, err)
 	assert.Equal(t, []*types.Snapshot{}, list)
+
+	require.NoError(t, manager.Close())
 }
 
 func TestManager_LoadChunk(t *testing.T) {