aboutsummaryrefslogtreecommitdiff
path: root/kv/src
diff options
context:
space:
mode:
Diffstat (limited to 'kv/src')
-rw-r--r--kv/src/bin/fs_to_rocksdb.rs38
-rw-r--r--kv/src/bin/rocksdb_remove_prefix.rs25
2 files changed, 63 insertions, 0 deletions
diff --git a/kv/src/bin/fs_to_rocksdb.rs b/kv/src/bin/fs_to_rocksdb.rs
new file mode 100644
index 0000000..d283dcb
--- /dev/null
+++ b/kv/src/bin/fs_to_rocksdb.rs
@@ -0,0 +1,38 @@
+/*
+ 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 <metamuffin.org>
+*/
+
+use anyhow::{Result, bail};
+use rocksdb::DB;
+use std::{env::args, fs::File, io::Read, path::Path};
+
+fn main() -> Result<()> {
+ let in_path = args().nth(1).unwrap();
+ let out_path = args().nth(2).unwrap();
+ let db = DB::open_default(out_path)?;
+ if !in_path.ends_with("/") {
+ bail!("path needs to end with /")
+ }
+ traverse(&db, &in_path, in_path.as_ref())?;
+ db.flush()?;
+ Ok(())
+}
+
+fn traverse(db: &DB, prefix: &str, path: &Path) -> Result<()> {
+ if path.is_dir() {
+ for e in path.read_dir()? {
+ traverse(db, prefix, &e?.path())?;
+ }
+ }
+ if path.is_file() {
+ let key = path.to_string_lossy();
+ let key = key.strip_prefix(prefix).unwrap();
+ let mut value = Vec::new();
+ File::open(path)?.read_to_end(&mut value)?;
+ println!("{key}");
+ db.put(key, value)?;
+ }
+ Ok(())
+}
diff --git a/kv/src/bin/rocksdb_remove_prefix.rs b/kv/src/bin/rocksdb_remove_prefix.rs
new file mode 100644
index 0000000..e09ce61
--- /dev/null
+++ b/kv/src/bin/rocksdb_remove_prefix.rs
@@ -0,0 +1,25 @@
+/*
+ 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 <metamuffin.org>
+*/
+
+use anyhow::Result;
+use rocksdb::DB;
+use std::env::args;
+
+fn main() -> Result<()> {
+ let db = DB::open_default(args().nth(1).unwrap())?;
+ let prefix = args().nth(2).unwrap();
+ for r in db.prefix_iterator(&prefix) {
+ let key = r?.0;
+ let key_s = String::from_utf8_lossy(&key);
+ if !key_s.starts_with(&prefix) {
+ break;
+ }
+ println!("{key_s}");
+ db.delete(key)?;
+ }
+ db.flush()?;
+ Ok(())
+}