49 lines
963 B
Nim
49 lines
963 B
Nim
|
import rocksdb, os
|
||
|
|
|
||
|
|
type
|
||
|
|
MyDB = object
|
||
|
|
rocksdb: RocksDBInstance
|
||
|
|
|
||
|
|
proc initMyDb(path: string): MyDB =
|
||
|
|
let
|
||
|
|
dataDir = path / "data"
|
||
|
|
backupsDir = path / "backups"
|
||
|
|
|
||
|
|
createDir(dataDir)
|
||
|
|
createDir(backupsDir)
|
||
|
|
|
||
|
|
var s = result.rocksdb.init(dataDir, backupsDir)
|
||
|
|
doAssert s.ok
|
||
|
|
|
||
|
|
proc main =
|
||
|
|
var db = initMyDb("/tmp/mydb")
|
||
|
|
defer: close(db.rocksdb)
|
||
|
|
|
||
|
|
let key = @[byte(1), 2, 3, 4, 5]
|
||
|
|
let otherKey = @[byte(1), 2, 3, 4, 5, 6]
|
||
|
|
let val = @[byte(1), 2, 3, 4, 5]
|
||
|
|
|
||
|
|
var s = db.rocksdb.put(key, val)
|
||
|
|
doAssert s.ok
|
||
|
|
|
||
|
|
var r1 = db.rocksdb.getBytes(key)
|
||
|
|
doAssert r1.ok and r1.value == val
|
||
|
|
|
||
|
|
var r2 = db.rocksdb.getBytes(otherKey)
|
||
|
|
doAssert r2.ok and r2.value.len == 0
|
||
|
|
|
||
|
|
var e1 = db.rocksdb.contains(key)
|
||
|
|
doAssert e1.ok and e1.value == true
|
||
|
|
|
||
|
|
var e2 = db.rocksdb.contains(otherKey)
|
||
|
|
doAssert e2.ok and e2.value == false
|
||
|
|
|
||
|
|
s = db.rocksdb.del(key)
|
||
|
|
doAssert s.ok
|
||
|
|
|
||
|
|
e1 = db.rocksdb.contains(key)
|
||
|
|
doAssert e1.ok and e1.value == false
|
||
|
|
|
||
|
|
main()
|
||
|
|
|