rocksdb cache backend

2 files changed, 40 insertions, 1 deletions

diff --git a/cache/src/backends/mod.rs b/cache/src/backends/mod.rs
index 6b7dac3..5872255 100644
--- a/cache/src/backends/mod.rs
+++ b/cache/src/backends/mod.rs
@@ -4,10 +4,23 @@
     Copyright (C) 2025 metamuffin <metamuffin.org>
 */
 pub mod filesystem;
+pub mod rocksdb;
 
-use anyhow::Result;
+use crate::{
+    CONF,
+    backends::{filesystem::Filesystem, rocksdb::Rocksdb},
+};
+use anyhow::{Result, bail};
 
 pub(crate) trait CacheStorage: Send + Sync + 'static {
     fn store(&self, key: String, value: &[u8]) -> Result<()>;
     fn read(&self, key: &str) -> Result<Option<Vec<u8>>>;
 }
+
+pub fn init_backend() -> Result<Box<dyn CacheStorage>> {
+    Ok(match CONF.driver.as_str() {
+        "filesystem" => Box::new(Filesystem::new(&CONF)),
+        "rocksdb" => Box::new(Rocksdb::new(&CONF)?),
+        _ => bail!("unknown driver"),
+    })
+}
diff --git a/cache/src/backends/rocksdb.rs b/cache/src/backends/rocksdb.rs
new file mode 100644
index 0000000..183664f
--- /dev/null
+++ b/cache/src/backends/rocksdb.rs
@@ -0,0 +1,26 @@
+/*
+    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::{Config, backends::CacheStorage};
+use anyhow::Result;
+use rocksdb::DB;
+
+pub struct Rocksdb(DB);
+
+impl Rocksdb {
+    pub fn new(config: &Config) -> Result<Self> {
+        Ok(Self(rocksdb::DB::open_default(config.path.clone())?))
+    }
+}
+
+impl CacheStorage for Rocksdb {
+    fn store(&self, key: String, value: &[u8]) -> Result<()> {
+        Ok(self.0.put(key, value)?)
+    }
+    fn read(&self, key: &str) -> Result<Option<Vec<u8>>> {
+        Ok(self.0.get(key)?)
+    }
+}