From b2e88a8beabf04adc28947cf82996e8692a68b71 Mon Sep 17 00:00:00 2001 From: metamuffin Date: Sat, 24 Jan 2026 04:31:48 +0100 Subject: move things around; kv crate --- database/src/backends/rocksdb.rs | 78 ---------------------------------------- 1 file changed, 78 deletions(-) delete mode 100644 database/src/backends/rocksdb.rs (limited to 'database/src/backends/rocksdb.rs') diff --git a/database/src/backends/rocksdb.rs b/database/src/backends/rocksdb.rs deleted file mode 100644 index 056af9e..0000000 --- a/database/src/backends/rocksdb.rs +++ /dev/null @@ -1,78 +0,0 @@ -/* - 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) 2026 metamuffin -*/ - -use crate::backends::{Database, ReadTransaction, WriteTransaction}; -use anyhow::Result; -use rocksdb::{Direction, ErrorKind, IteratorMode, OptimisticTransactionDB}; -use std::path::Path; - -pub fn new(path: &Path) -> Result { - Ok(OptimisticTransactionDB::open_default(path)?) -} - -impl Database for OptimisticTransactionDB { - fn write_transaction( - &self, - f: &mut dyn FnMut(&mut dyn WriteTransaction) -> Result<()>, - ) -> Result<()> { - loop { - let mut txn = self.transaction(); - f(&mut txn)?; - match txn.commit() { - Ok(()) => break Ok(()), - Err(e) if e.kind() == ErrorKind::Busy => continue, - Err(e) => return Err(e.into()), - } - } - } - fn read_transaction( - &self, - f: &mut dyn FnMut(&dyn ReadTransaction) -> Result<()>, - ) -> Result<()> { - loop { - let txn = self.transaction(); - f(&txn)?; - match txn.commit() { - Ok(()) => break Ok(()), - Err(e) if e.kind() == ErrorKind::Busy => continue, - Err(e) => return Err(e.into()), - } - } - } -} -impl WriteTransaction for rocksdb::Transaction<'_, OptimisticTransactionDB> { - fn set(&mut self, key: &[u8], value: &[u8]) -> Result<()> { - Ok(self.put(key, value)?) - } - - fn del(&mut self, key: &[u8]) -> Result<()> { - Ok(self.delete(key)?) - } -} -impl ReadTransaction for rocksdb::Transaction<'_, OptimisticTransactionDB> { - fn get(&self, key: &[u8]) -> Result>> { - Ok(self.get(key)?) - } - - fn iter<'a>( - &'a self, - key: &[u8], - reverse: bool, - ) -> Result>> + 'a>> { - let mut iter = self.iterator(IteratorMode::Start); - iter.set_mode(IteratorMode::From( - key, - if reverse { - Direction::Reverse - } else { - Direction::Forward - }, - )); - Ok(Box::new(iter.map(|e| { - e.map(|(k, _)| k.into_vec()).map_err(|e| e.into()) - }))) - } -} -- cgit v1.3