nim-rocksdb/tests/test_helper.nim
web3-developer cf1267e845
Add support for optimistic transaction db, write batch with index, keyMayExist and empty keys. (#63)
* Add support for optimistic transaction db.

* Add keyMayExist to RocksDbRef.

* Add support for write batch with index.

* Allow empty keys to be used in API.
2024-07-08 22:18:30 +08:00

74 lines
2.0 KiB
Nim

# Nim-RocksDB
# Copyright 2024 Status Research & Development GmbH
# Licensed under either of
#
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or http://www.apache.org/licenses/LICENSE-2.0)
# * GPL license, version 2.0, ([LICENSE-GPLv2](LICENSE-GPLv2) or https://www.gnu.org/licenses/old-licenses/gpl-2.0.en.html)
#
# at your option. This file may not be copied, modified, or distributed except according to those terms.
{.used.}
import std/sequtils, ../rocksdb/[backup, rocksdb, transactiondb, optimistictxdb]
proc initReadWriteDb*(
path: string, columnFamilyNames: openArray[string] = @[]
): RocksDbReadWriteRef =
let res = openRocksDb(
path,
columnFamilies = columnFamilyNames.mapIt(
initColFamilyDescriptor(it, defaultColFamilyOptions(autoClose = true))
),
)
if res.isErr():
echo res.error()
doAssert res.isOk()
res.value()
proc initReadOnlyDb*(
path: string, columnFamilyNames: openArray[string] = @[]
): RocksDbReadOnlyRef =
let res = openRocksDbReadOnly(
path,
columnFamilies = columnFamilyNames.mapIt(
initColFamilyDescriptor(it, defaultColFamilyOptions(autoClose = true))
),
)
if res.isErr():
echo res.error()
doAssert res.isOk()
res.value()
proc initBackupEngine*(path: string): BackupEngineRef =
let res = openBackupEngine(path)
doAssert res.isOk()
res.value()
proc initTransactionDb*(
path: string, columnFamilyNames: openArray[string] = @[]
): TransactionDbRef =
let res = openTransactionDb(
path,
columnFamilies = columnFamilyNames.mapIt(
initColFamilyDescriptor(it, defaultColFamilyOptions(autoClose = true))
),
)
if res.isErr():
echo res.error()
doAssert res.isOk()
res.value()
proc initOptimisticTxDb*(
path: string, columnFamilyNames: openArray[string] = @[]
): OptimisticTxDbRef =
let res = openOptimisticTxDb(
path,
columnFamilies = columnFamilyNames.mapIt(
initColFamilyDescriptor(it, defaultColFamilyOptions(autoClose = true))
),
)
if res.isErr():
echo res.error()
doAssert res.isOk()
res.value()