feat: allow full control on rocksdb db opening

cosmos · Mar 15, 2024 · 447969c · 447969c
1 parent fab0ae0
commit 447969c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Changelog
 
+## UNRELEASED
+
+* Allow full control in rocksdb opening
+
 ## [v1.0.2] - 2024-02-26
 
 * Downgrade Go version in go.mod to 1.19

diff --git a/rocksdb.go b/rocksdb.go
@@ -71,7 +71,12 @@ func NewRocksDBWithOptions(name string, dir string, opts *grocksdb.Options) (*Ro
 	wo := grocksdb.NewDefaultWriteOptions()
 	woSync := grocksdb.NewDefaultWriteOptions()
 	woSync.SetSync(true)
-	return NewRocksDBWithRaw(db, ro, wo, woSync), nil
+	return NewRocksDBWithRawDB(db, ro, wo, woSync)
+}
+
+// NewRocksDBWithRawDB lets caller has full control on how the db instance is constructed
+func NewRocksDBWithRawDB(db *grocksdb.DB, ro *grocksdb.ReadOptions, wo *grocksdb.WriteOptions, woSync *grocksdb.WriteOptions) *RocksDB {
+	return NewRocksDBWithRaw(db, ro, wo, woSync)
 }
 
 // NewRocksDBWithRaw is useful if user want to create the db in read-only or seconday-standby mode,