aboutsummaryrefslogtreecommitdiff
path: root/database/src/kv
diff options
context:
space:
mode:
Diffstat (limited to 'database/src/kv')
-rw-r--r--database/src/kv/binning.rs36
-rw-r--r--database/src/kv/mod.rs3
-rw-r--r--database/src/kv/tests.rs10
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(())