diff options
Diffstat (limited to 'database/src/kv')
| -rw-r--r-- | database/src/kv/binning.rs | 36 | ||||
| -rw-r--r-- | database/src/kv/mod.rs | 3 | ||||
| -rw-r--r-- | database/src/kv/tests.rs | 10 |
3 files changed, 39 insertions, 10 deletions
diff --git a/database/src/kv/binning.rs b/database/src/kv/binning.rs index 42e58fc..c41d6e9 100644 --- a/database/src/kv/binning.rs +++ b/database/src/kv/binning.rs @@ -4,7 +4,7 @@ Copyright (C) 2026 metamuffin <metamuffin.org> */ -use crate::Filter; +use crate::{Filter, Value}; use jellyobject::{Object, Path}; /// Sorted list of components to bin objects by filtered values. @@ -88,8 +88,7 @@ impl Filter { Filter::Match(path, value) => { vec![vec![(BinningComponent::Match(path.to_owned()), { let mut co = Vec::new(); - co.extend((value.len() as u32).to_be_bytes()); - co.extend(value); + value.write_with_len(&mut co); co })]] } @@ -100,6 +99,37 @@ impl Filter { } } +impl Value { + pub fn write_with_len(&self, out: &mut Vec<u8>) { + match self { + Value::Tag(tag) => { + out.extend(4u32.to_be_bytes()); + out.extend(tag.0.to_be_bytes()); + } + Value::U32(x) => { + out.extend(4u32.to_be_bytes()); + out.extend(x.to_be_bytes()); + } + Value::U64(x) => { + out.extend(8u32.to_be_bytes()); + out.extend(x.to_be_bytes()); + } + Value::I64(x) => { + out.extend(8u32.to_be_bytes()); + out.extend(x.to_be_bytes()); + } + Value::String(s) => { + out.extend((s.len() as u32).to_be_bytes()); + out.extend(s.as_bytes()); + } + Value::Binary(s) => { + out.extend((s.len() as u32).to_be_bytes()); + out.extend(s); + } + } + } +} + #[cfg(test)] mod test { use jellyobject::{Path, Tag}; diff --git a/database/src/kv/mod.rs b/database/src/kv/mod.rs index f5dcc72..f90f07e 100644 --- a/database/src/kv/mod.rs +++ b/database/src/kv/mod.rs @@ -14,8 +14,6 @@ pub mod sort; #[cfg(test)] pub mod tests; -use std::borrow::Cow; - use crate::{ Database, Query, RowNum, Transaction, kv::{ @@ -29,6 +27,7 @@ use crate::{ use anyhow::{Result, anyhow}; use jellyobject::ObjectBuffer; use log::{debug, info}; +use std::borrow::Cow; pub type SubtreeNum = u32; diff --git a/database/src/kv/tests.rs b/database/src/kv/tests.rs index 5b587ed..85ee5d7 100644 --- a/database/src/kv/tests.rs +++ b/database/src/kv/tests.rs @@ -1,5 +1,5 @@ use crate::{ - Database, Filter, Query, Sort, + Database, Filter, Query, Sort, Value, test_shared::{AGE, NAME, new_alice, new_bob, new_charlie}, }; use anyhow::Result; @@ -67,7 +67,7 @@ pub fn query_match_int() -> Result<()> { })?; db.transaction(&mut |txn| { result = txn.query_single(Query { - filter: Filter::Match(Path(vec![AGE.0]), 35_u32.to_be_bytes().to_vec()), + filter: Filter::Match(Path(vec![AGE.0]), Value::U32(35)), sort: Sort::None, })?; Ok(()) @@ -94,7 +94,7 @@ pub fn query_match_str() -> Result<()> { })?; db.transaction(&mut |txn| { result = txn.query_single(Query { - filter: Filter::Match(Path(vec![NAME.0]), "Alice".as_bytes().to_vec()), + filter: Filter::Match(Path(vec![NAME.0]), "Alice".into()), sort: Sort::None, })?; Ok(()) @@ -113,7 +113,7 @@ pub fn query_match_str_after() -> Result<()> { db.transaction(&mut |txn| { result = txn.query_single(Query { - filter: Filter::Match(Path(vec![NAME.0]), "Alice".as_bytes().to_vec()), + filter: Filter::Match(Path(vec![NAME.0]), "Alice".into()), sort: Sort::None, })?; Ok(()) @@ -131,7 +131,7 @@ pub fn query_match_str_after() -> Result<()> { db.transaction(&mut |txn| { result = txn.query_single(Query { - filter: Filter::Match(Path(vec![NAME.0]), "Alice".as_bytes().to_vec()), + filter: Filter::Match(Path(vec![NAME.0]), "Alice".into()), sort: Sort::None, })?; Ok(()) |