42 lines
1.0 KiB
Nim
42 lines
1.0 KiB
Nim
|
import os, rocksdb, ranges
|
||
|
|
import ../storage_types
|
||
|
|
|
||
|
|
type
|
||
|
|
RocksChainDB* = object
|
||
|
|
store: RocksDBInstance
|
||
|
|
|
||
|
|
ChainDB* = RocksChainDB
|
||
|
|
|
||
|
|
proc initChainDB*(basePath: string): ChainDB =
|
||
|
|
let
|
||
|
|
dataDir = basePath / "data"
|
||
|
|
backupsDir = basePath / "backups"
|
||
|
|
|
||
|
|
createDir(dataDir)
|
||
|
|
createDir(backupsDir)
|
||
|
|
|
||
|
|
let s = result.store.init(dataDir, backupsDir)
|
||
|
|
if not s.ok: raiseStorageInitError()
|
||
|
|
|
||
|
|
proc get*(db: ChainDB, key: DbKey): ByteRange =
|
||
|
|
let s = db.store.getBytes(key.toOpenArray)
|
||
|
|
if not s.ok: raiseKeyReadError(key)
|
||
|
|
return s.value.toRange
|
||
|
|
|
||
|
|
proc put*(db: var ChainDB, key: DbKey, value: ByteRange) =
|
||
|
|
let s = db.store.put(key.toOpenArray, value.toOpenArray)
|
||
|
|
if not s.ok: raiseKeyWriteError(key)
|
||
|
|
|
||
|
|
proc contains*(db: ChainDB, key: DbKey): bool =
|
||
|
|
let s = db.store.contains(key.toOpenArray)
|
||
|
|
if not s.ok: raiseKeySearchError(key)
|
||
|
|
return s.value
|
||
|
|
|
||
|
|
proc del*(db: var ChainDB, key: DbKey) =
|
||
|
|
let s = db.store.del(key.toOpenArray)
|
||
|
|
if not s.ok: raiseKeyDeletionError(key)
|
||
|
|
|
||
|
|
proc close*(db: var ChainDB) =
|
||
|
|
db.store.close
|
||
|
|
|