aboutsummaryrefslogtreecommitdiff
path: root/database/src/backends/redb.rs
diff options
context:
space:
mode:
Diffstat (limited to 'database/src/backends/redb.rs')
-rw-r--r--database/src/backends/redb.rs49
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),
+ }
+ }
+}