diff options
Diffstat (limited to 'database/src/backends/redb.rs')
| -rw-r--r-- | database/src/backends/redb.rs | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/database/src/backends/redb.rs b/database/src/backends/redb.rs new file mode 100644 index 0000000..39fe532 --- /dev/null +++ b/database/src/backends/redb.rs @@ -0,0 +1,49 @@ +/* + This file is part of jellything (https://codeberg.org/metamuffin/jellything) + which is licensed under the GNU Affero General Public License (version 3); see /COPYING. + Copyright (C) 2025 metamuffin <metamuffin.org> +*/ + +use crate::backends::DatabaseStorage; +use anyhow::Result; +use redb::{Database, ReadableDatabase, TableDefinition}; + +pub struct Redb { + db: Database, +} + +const TABLE: TableDefinition<&[u8], &[u8]> = TableDefinition::new("kv"); + +impl DatabaseStorage for Redb { + fn set(&self, key: &[u8], value: &[u8]) -> Result<()> { + let txn = self.db.begin_write()?; + txn.open_table(TABLE)?.insert(key, value)?; + txn.commit()?; + Ok(()) + } + fn get<'a>(&'a self, key: &[u8]) -> Result<Option<Vec<u8>>> { + let txn = self.db.begin_read()?; + match txn.open_table(TABLE)?.get(key)? { + Some(v) => Ok(Some(v.value().to_vec())), + None => Ok(None), + } + } + fn next(&self, key: &[u8]) -> Result<Option<Vec<u8>>> { + let txn = self.db.begin_read()?; + let table = txn.open_table(TABLE)?; + let mut iter = table.range(key..)?; + match iter.next() { + Some(k) => Ok(Some(k?.0.value().to_vec())), + None => Ok(None), + } + } + fn prev(&self, key: &[u8]) -> Result<Option<Vec<u8>>> { + let txn = self.db.begin_read()?; + let table = txn.open_table(TABLE)?; + let mut iter = table.range(..key)?; + match iter.next_back() { + Some(k) => Ok(Some(k?.0.value().to_vec())), + None => Ok(None), + } + } +} |