mirror of
https://github.com/status-im/nim-rocksdb.git
synced 2025-02-16 17:07:40 +00:00
5f6282e8d4
* Column families support (#34) * Update library to support column families. If not specified, uses the 'default' column family. * Added tests for column family changes. * Update library version and readme. * Updated the librocksdb c library to the latest stable version. * Started rewrite of library. * Commit library rewrite progress. * Completed initial rewrite and refactored tests. * Completed implementation of backup engine. * Added tests for new types. * Completed tests for existing features. * Remove features not supported by older versions of RocksDB to fix CI (temporary fix). * Remove flush before backup support from BackupEngine to fix CI. * Transactions support (#36) * Update library to support column families. If not specified, uses the 'default' column family. * Added tests for column family changes. * Update library version and readme. * Updated the librocksdb c library to the latest stable version. * Started rewrite of library. * Commit library rewrite progress. * Completed initial rewrite and refactored tests. * Completed implementation of backup engine. * Added tests for new types. * Completed tests for existing features. * Remove features not supported by older versions of RocksDB to fix CI (temporary fix). * Remove flush before backup support from BackupEngine to fix CI. * Implemented RocksDB iterator. * Implemented pairs iterator. * Completed implementation of WriteBatch with tests. * Fix example code. * Completed implementation of TransactionDb. * Support setting default column family. * Remove unneeded usage of var for ref type parameters. * Completed transactiondb tests. * Improve and refactor rocksdb test. * Added support for ingesting sst files using the SstFileWriter. (#37) * Create ColFamilyReadOnly and ColFamilyReadWrite types for using a specific column family. * Use inline pragma for small procs and add lock to RocksDbRef type close to prevent double free. * Added documentation for the public API. * Initial implementation of sst filewriter. * Added tests for sstfilewriter. * Documentation minor improvements.
92 lines
2.4 KiB
Nim
92 lines
2.4 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/os,
|
|
tempfile,
|
|
unittest2,
|
|
../rocksdb/[rocksdb, sstfilewriter],
|
|
./test_helper
|
|
|
|
suite "SstFileWriterRef Tests":
|
|
|
|
const
|
|
CF_DEFAULT = "default"
|
|
CF_OTHER = "other"
|
|
|
|
let
|
|
key1 = @[byte(1)]
|
|
val1 = @[byte(1)]
|
|
key2 = @[byte(2)]
|
|
val2 = @[byte(2)]
|
|
key3 = @[byte(3)]
|
|
val3 = @[byte(3)]
|
|
|
|
setup:
|
|
let
|
|
dbPath = mkdtemp() / "data"
|
|
sstFilePath = mkdtemp() / "sst"
|
|
db = initReadWriteDb(dbPath,
|
|
columnFamilyNames = @[CF_DEFAULT, CF_OTHER])
|
|
|
|
teardown:
|
|
db.close()
|
|
removeDir($dbPath)
|
|
|
|
test "Write to sst file then load into db using default column family":
|
|
let res = openSstFileWriter(sstFilePath)
|
|
check res.isOk()
|
|
let writer = res.get()
|
|
defer: writer.close()
|
|
|
|
check:
|
|
writer.put(key1, val1).isOk()
|
|
writer.put(key2, val2).isOk()
|
|
writer.put(key3, val3).isOk()
|
|
writer.delete(@[byte(4)]).isOk()
|
|
writer.finish().isOk()
|
|
|
|
db.ingestExternalFile(sstFilePath).isOk()
|
|
db.get(key1).get() == val1
|
|
db.get(key2).get() == val2
|
|
db.get(key3).get() == val3
|
|
|
|
test "Write to sst file then load into db using specific column family":
|
|
let res = openSstFileWriter(sstFilePath)
|
|
check res.isOk()
|
|
let writer = res.get()
|
|
defer: writer.close()
|
|
|
|
check:
|
|
writer.put(key1, val1).isOk()
|
|
writer.put(key2, val2).isOk()
|
|
writer.put(key3, val3).isOk()
|
|
writer.finish().isOk()
|
|
|
|
db.ingestExternalFile(sstFilePath, CF_OTHER).isOk()
|
|
db.keyExists(key1, CF_DEFAULT).get() == false
|
|
db.keyExists(key2, CF_DEFAULT).get() == false
|
|
db.keyExists(key3, CF_DEFAULT).get() == false
|
|
db.get(key1, CF_OTHER).get() == val1
|
|
db.get(key2, CF_OTHER).get() == val2
|
|
db.get(key3, CF_OTHER).get() == val3
|
|
|
|
test "Test close":
|
|
let res = openSstFileWriter(sstFilePath)
|
|
check res.isOk()
|
|
let writer = res.get()
|
|
|
|
check not writer.isClosed()
|
|
writer.close()
|
|
check writer.isClosed()
|
|
writer.close()
|
|
check writer.isClosed()
|