chore: new GetLatestVersion api (#961)

(cherry picked from commit a514883)
cosmos · Jul 9, 2024 · 7cbae69 · 7cbae69
1 parent a955aea
commit 7cbae69
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Changelog
 
+### Improvements
+
+- [#961](https://github.com/cosmos/iavl/pull/961) Add new `GetLatestVersion` API to get the latest version.
+
 ## v1.2.0 May 13, 2024
 
 ### Improvements

diff --git a/mutable_tree.go b/mutable_tree.go
@@ -72,6 +72,11 @@ func (tree *MutableTree) IsEmpty() bool {
 	return tree.ImmutableTree.Size() == 0
 }
 
+// GetLatestVersion returns the latest version of the tree.
+func (tree *MutableTree) GetLatestVersion() (int64, error) {
+	return tree.ndb.getLatestVersion()
+}
+
 // VersionExists returns whether or not a version exists.
 func (tree *MutableTree) VersionExists(version int64) bool {
 	legacyLatestVersion, err := tree.ndb.getLegacyLatestVersion()

diff --git a/mutable_tree_test.go b/mutable_tree_test.go
@@ -347,11 +347,15 @@ func prepareTree(t *testing.T) *MutableTree {
 	return newTree
 }
 
-func TestMutableTree_VersionExists(t *testing.T) {
+func TestMutableTree_Version(t *testing.T) {
 	tree := prepareTree(t)
 	require.True(t, tree.VersionExists(1))
 	require.True(t, tree.VersionExists(2))
 	require.False(t, tree.VersionExists(3))
+
+	v, err := tree.GetLatestVersion()
+	require.NoError(t, err)
+	require.Equal(t, int64(2), v)
 }
 
 func checkGetVersioned(t *testing.T, tree *MutableTree, version int64, key, value []byte) {