diff options
Diffstat (limited to 'database/src/backends/rocksdb.rs')
| -rw-r--r-- | database/src/backends/rocksdb.rs | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/database/src/backends/rocksdb.rs b/database/src/backends/rocksdb.rs index 7c6f5f3..f4ed55b 100644 --- a/database/src/backends/rocksdb.rs +++ b/database/src/backends/rocksdb.rs @@ -4,7 +4,9 @@ Copyright (C) 2025 metamuffin <metamuffin.org> */ -use crate::backends::DatabaseStorage; +use std::path::Path; + +use crate::backends::KV; use anyhow::Result; use rocksdb::DB; @@ -12,14 +14,23 @@ pub struct Rocksdb { db: DB, } -impl DatabaseStorage for Rocksdb { +impl Rocksdb { + pub fn new(path: &Path) -> Result<Self> { + Ok(Self { + db: DB::open_default(path)?, + }) + } +} +impl KV for Rocksdb { fn set(&self, key: &[u8], value: &[u8]) -> Result<()> { - self.db.put(key, value)?; - Ok(()) + Ok(self.db.put(key, value)?) } fn get<'a>(&'a self, key: &[u8]) -> Result<Option<Vec<u8>>> { Ok(self.db.get(key)?) } + fn del(&self, key: &[u8]) -> Result<()> { + Ok(self.db.delete(key)?) + } fn next(&self, key: &[u8]) -> Result<Option<Vec<u8>>> { let mut it = self.db.raw_iterator(); it.seek_for_prev(key); |