feat: handle comparisons =, >, <, >=, <= for unsigned int condition

This commit is contained in:
2025-10-19 10:51:38 +02:00
parent 01a6e2ce48
commit 2e42afbb04
5 changed files with 112 additions and 27 deletions

View File

@@ -8,6 +8,7 @@ pub fn run<'a>(filter: &'a Filter) -> Result<Vec<Box<dyn ToSql + 'a>>, FailedTo>
match condition { match condition {
Condition::Bool(value) => params.push(Box::new(value)), Condition::Bool(value) => params.push(Box::new(value)),
Condition::SignedInt(value) => params.push(Box::new(value)), Condition::SignedInt(value) => params.push(Box::new(value)),
Condition::UnsignedInt(value) => params.push(Box::new(value)),
} }
} }
Ok(params) Ok(params)

View File

@@ -20,9 +20,11 @@ pub fn run(filter: &Filter) -> Result<String, FailedTo> {
let mut statement = String::from(BASE_SELECT); let mut statement = String::from(BASE_SELECT);
for (i, (name, comparison, condition)) in filter.conditions().enumerate() { for (i, (name, comparison, condition)) in filter.conditions().enumerate() {
let fragment = match (comparison, condition) { let fragment = match (comparison, condition) {
// BOOL
(Comparison::Equal, Condition::Bool(_)) => { (Comparison::Equal, Condition::Bool(_)) => {
compose_fragment(name, "value_bool", "=", i + 1) compose_fragment(name, "value_bool", "=", i + 1)
} }
// SIGNED INT
(Comparison::Equal, Condition::SignedInt(_)) => { (Comparison::Equal, Condition::SignedInt(_)) => {
compose_fragment(name, "value_int", "=", i + 1) compose_fragment(name, "value_int", "=", i + 1)
} }
@@ -38,6 +40,22 @@ pub fn run(filter: &Filter) -> Result<String, FailedTo> {
(Comparison::LesserOrEqual, Condition::SignedInt(_)) => { (Comparison::LesserOrEqual, Condition::SignedInt(_)) => {
compose_fragment(name, "value_int", "<=", i + 1) compose_fragment(name, "value_int", "<=", i + 1)
} }
// UNSIGNED INT
(Comparison::Equal, Condition::UnsignedInt(_)) => {
compose_fragment(name, "value_uint", "=", i + 1)
}
(Comparison::Greater, Condition::UnsignedInt(_)) => {
compose_fragment(name, "value_uint", ">", i + 1)
}
(Comparison::Lesser, Condition::UnsignedInt(_)) => {
compose_fragment(name, "value_uint", "<", i + 1)
}
(Comparison::GreaterOrEqual, Condition::UnsignedInt(_)) => {
compose_fragment(name, "value_uint", ">=", i + 1)
}
(Comparison::LesserOrEqual, Condition::UnsignedInt(_)) => {
compose_fragment(name, "value_uint", "<=", i + 1)
}
_ => todo!(), _ => todo!(),
}; };
statement.push_str(&fragment); statement.push_str(&fragment);

View File

@@ -1702,7 +1702,6 @@ mod tests {
if path.exists() { if path.exists() {
std::fs::remove_file(path).unwrap(); std::fs::remove_file(path).unwrap();
} }
let mut catalog_setup = Catalog::new(db_path); let mut catalog_setup = Catalog::new(db_path);
catalog_setup.init().unwrap(); catalog_setup.init().unwrap();
let items = vec![ let items = vec![
@@ -1730,20 +1729,17 @@ mod tests {
]; ];
catalog_setup.insert_many(items).unwrap(); catalog_setup.insert_many(items).unwrap();
catalog_setup.persist().unwrap(); catalog_setup.persist().unwrap();
// Edge Case 1: Value equal to filter value should not be included. // Edge Case 1: Value equal to filter value should not be included.
let mut catalog1 = Catalog::new(db_path); let mut catalog1 = Catalog::new(db_path);
let filter1 = Filter::new().with_signed_int("sell_trend", Comparison::Lesser, 10); let filter1 = Filter::new().with_signed_int("sell_trend", Comparison::Lesser, 10);
assert!(catalog1.load_by_filter(&filter1).is_ok()); assert!(catalog1.load_by_filter(&filter1).is_ok());
assert_eq!(catalog1.items.len(), 1); assert_eq!(catalog1.items.len(), 1);
assert!(catalog1.items.contains_key("item-2")); assert!(catalog1.items.contains_key("item-2"));
// Edge Case 2: No values lesser than filter value. // Edge Case 2: No values lesser than filter value.
let mut catalog2 = Catalog::new(db_path); let mut catalog2 = Catalog::new(db_path);
let filter2 = Filter::new().with_signed_int("sell_trend", Comparison::Lesser, -5); let filter2 = Filter::new().with_signed_int("sell_trend", Comparison::Lesser, -5);
assert!(catalog2.load_by_filter(&filter2).is_ok()); assert!(catalog2.load_by_filter(&filter2).is_ok());
assert!(catalog2.items.is_empty()); assert!(catalog2.items.is_empty());
// Edge Case 3: All values lesser than filter value. // Edge Case 3: All values lesser than filter value.
let mut catalog3 = Catalog::new(db_path); let mut catalog3 = Catalog::new(db_path);
let filter3 = Filter::new().with_signed_int("sell_trend", Comparison::Lesser, 25); let filter3 = Filter::new().with_signed_int("sell_trend", Comparison::Lesser, 25);
@@ -1752,10 +1748,8 @@ mod tests {
assert!(catalog3.items.contains_key("item-1")); assert!(catalog3.items.contains_key("item-1"));
assert!(catalog3.items.contains_key("item-2")); assert!(catalog3.items.contains_key("item-2"));
assert!(catalog3.items.contains_key("item-3")); assert!(catalog3.items.contains_key("item-3"));
std::fs::remove_file(path).unwrap(); std::fs::remove_file(path).unwrap();
} }
#[test] #[test]
fn load_by_filter_should_load_matching_greater_or_equal_sell_trend() { fn load_by_filter_should_load_matching_greater_or_equal_sell_trend() {
let db_path = "target/test_dbs/lbf_matching_greater_or_equal_sell_trend.db"; let db_path = "target/test_dbs/lbf_matching_greater_or_equal_sell_trend.db";
@@ -1764,7 +1758,6 @@ mod tests {
if path.exists() { if path.exists() {
std::fs::remove_file(path).unwrap(); std::fs::remove_file(path).unwrap();
} }
let mut catalog_setup = Catalog::new(db_path); let mut catalog_setup = Catalog::new(db_path);
catalog_setup.init().unwrap(); catalog_setup.init().unwrap();
let items = vec![ let items = vec![
@@ -1792,19 +1785,15 @@ mod tests {
]; ];
catalog_setup.insert_many(items).unwrap(); catalog_setup.insert_many(items).unwrap();
catalog_setup.persist().unwrap(); catalog_setup.persist().unwrap();
let mut catalog = Catalog::new(db_path); let mut catalog = Catalog::new(db_path);
let filter = Filter::new().with_signed_int("sell_trend", Comparison::GreaterOrEqual, 10); let filter = Filter::new().with_signed_int("sell_trend", Comparison::GreaterOrEqual, 10);
assert!(catalog.load_by_filter(&filter).is_ok()); assert!(catalog.load_by_filter(&filter).is_ok());
assert_eq!(catalog.items.len(), 2); assert_eq!(catalog.items.len(), 2);
assert!(catalog.items.contains_key("item-1")); assert!(catalog.items.contains_key("item-1"));
assert!(!catalog.items.contains_key("item-2")); assert!(!catalog.items.contains_key("item-2"));
assert!(catalog.items.contains_key("item-3")); assert!(catalog.items.contains_key("item-3"));
std::fs::remove_file(path).unwrap(); std::fs::remove_file(path).unwrap();
} }
#[test] #[test]
fn load_by_filter_should_load_matching_greater_or_equal_sell_trend_edge_cases() { fn load_by_filter_should_load_matching_greater_or_equal_sell_trend_edge_cases() {
let db_path = "target/test_dbs/lbf_greater_or_equal_sell_trend_edge_cases.db"; let db_path = "target/test_dbs/lbf_greater_or_equal_sell_trend_edge_cases.db";
@@ -1813,7 +1802,6 @@ mod tests {
if path.exists() { if path.exists() {
std::fs::remove_file(path).unwrap(); std::fs::remove_file(path).unwrap();
} }
let mut catalog_setup = Catalog::new(db_path); let mut catalog_setup = Catalog::new(db_path);
catalog_setup.init().unwrap(); catalog_setup.init().unwrap();
let items = vec![ let items = vec![
@@ -1841,20 +1829,17 @@ mod tests {
]; ];
catalog_setup.insert_many(items).unwrap(); catalog_setup.insert_many(items).unwrap();
catalog_setup.persist().unwrap(); catalog_setup.persist().unwrap();
// Edge Case 1: Value equal to filter value should be included. // Edge Case 1: Value equal to filter value should be included.
let mut catalog1 = Catalog::new(db_path); let mut catalog1 = Catalog::new(db_path);
let filter1 = Filter::new().with_signed_int("sell_trend", Comparison::GreaterOrEqual, 20); let filter1 = Filter::new().with_signed_int("sell_trend", Comparison::GreaterOrEqual, 20);
assert!(catalog1.load_by_filter(&filter1).is_ok()); assert!(catalog1.load_by_filter(&filter1).is_ok());
assert_eq!(catalog1.items.len(), 1); assert_eq!(catalog1.items.len(), 1);
assert!(catalog1.items.contains_key("item-3")); assert!(catalog1.items.contains_key("item-3"));
// Edge Case 2: No values greater than or equal to filter value. // Edge Case 2: No values greater than or equal to filter value.
let mut catalog2 = Catalog::new(db_path); let mut catalog2 = Catalog::new(db_path);
let filter2 = Filter::new().with_signed_int("sell_trend", Comparison::GreaterOrEqual, 21); let filter2 = Filter::new().with_signed_int("sell_trend", Comparison::GreaterOrEqual, 21);
assert!(catalog2.load_by_filter(&filter2).is_ok()); assert!(catalog2.load_by_filter(&filter2).is_ok());
assert!(catalog2.items.is_empty()); assert!(catalog2.items.is_empty());
// Edge Case 3: All values greater than or equal to filter value. // Edge Case 3: All values greater than or equal to filter value.
let mut catalog3 = Catalog::new(db_path); let mut catalog3 = Catalog::new(db_path);
let filter3 = Filter::new().with_signed_int("sell_trend", Comparison::GreaterOrEqual, -5); let filter3 = Filter::new().with_signed_int("sell_trend", Comparison::GreaterOrEqual, -5);
@@ -1863,10 +1848,8 @@ mod tests {
assert!(catalog3.items.contains_key("item-1")); assert!(catalog3.items.contains_key("item-1"));
assert!(catalog3.items.contains_key("item-2")); assert!(catalog3.items.contains_key("item-2"));
assert!(catalog3.items.contains_key("item-3")); assert!(catalog3.items.contains_key("item-3"));
std::fs::remove_file(path).unwrap(); std::fs::remove_file(path).unwrap();
} }
#[test] #[test]
fn load_by_filter_should_load_matching_lesser_or_equal_sell_trend() { fn load_by_filter_should_load_matching_lesser_or_equal_sell_trend() {
let db_path = "target/test_dbs/lbf_matching_lesser_or_equal_sell_trend.db"; let db_path = "target/test_dbs/lbf_matching_lesser_or_equal_sell_trend.db";
@@ -1875,7 +1858,6 @@ mod tests {
if path.exists() { if path.exists() {
std::fs::remove_file(path).unwrap(); std::fs::remove_file(path).unwrap();
} }
let mut catalog_setup = Catalog::new(db_path); let mut catalog_setup = Catalog::new(db_path);
catalog_setup.init().unwrap(); catalog_setup.init().unwrap();
let items = vec![ let items = vec![
@@ -1903,19 +1885,15 @@ mod tests {
]; ];
catalog_setup.insert_many(items).unwrap(); catalog_setup.insert_many(items).unwrap();
catalog_setup.persist().unwrap(); catalog_setup.persist().unwrap();
let mut catalog = Catalog::new(db_path); let mut catalog = Catalog::new(db_path);
let filter = Filter::new().with_signed_int("sell_trend", Comparison::LesserOrEqual, 10); let filter = Filter::new().with_signed_int("sell_trend", Comparison::LesserOrEqual, 10);
assert!(catalog.load_by_filter(&filter).is_ok()); assert!(catalog.load_by_filter(&filter).is_ok());
assert_eq!(catalog.items.len(), 2); assert_eq!(catalog.items.len(), 2);
assert!(catalog.items.contains_key("item-1")); assert!(catalog.items.contains_key("item-1"));
assert!(catalog.items.contains_key("item-2")); assert!(catalog.items.contains_key("item-2"));
assert!(!catalog.items.contains_key("item-3")); assert!(!catalog.items.contains_key("item-3"));
std::fs::remove_file(path).unwrap(); std::fs::remove_file(path).unwrap();
} }
#[test] #[test]
fn load_by_filter_should_load_matching_lesser_or_equal_sell_trend_edge_cases() { fn load_by_filter_should_load_matching_lesser_or_equal_sell_trend_edge_cases() {
let db_path = "target/test_dbs/lbf_lesser_or_equal_sell_trend_edge_cases.db"; let db_path = "target/test_dbs/lbf_lesser_or_equal_sell_trend_edge_cases.db";
@@ -1924,7 +1902,6 @@ mod tests {
if path.exists() { if path.exists() {
std::fs::remove_file(path).unwrap(); std::fs::remove_file(path).unwrap();
} }
let mut catalog_setup = Catalog::new(db_path); let mut catalog_setup = Catalog::new(db_path);
catalog_setup.init().unwrap(); catalog_setup.init().unwrap();
let items = vec![ let items = vec![
@@ -1952,20 +1929,17 @@ mod tests {
]; ];
catalog_setup.insert_many(items).unwrap(); catalog_setup.insert_many(items).unwrap();
catalog_setup.persist().unwrap(); catalog_setup.persist().unwrap();
// Edge Case 1: Value equal to filter value should be included. // Edge Case 1: Value equal to filter value should be included.
let mut catalog1 = Catalog::new(db_path); let mut catalog1 = Catalog::new(db_path);
let filter1 = Filter::new().with_signed_int("sell_trend", Comparison::LesserOrEqual, -5); let filter1 = Filter::new().with_signed_int("sell_trend", Comparison::LesserOrEqual, -5);
assert!(catalog1.load_by_filter(&filter1).is_ok()); assert!(catalog1.load_by_filter(&filter1).is_ok());
assert_eq!(catalog1.items.len(), 1); assert_eq!(catalog1.items.len(), 1);
assert!(catalog1.items.contains_key("item-2")); assert!(catalog1.items.contains_key("item-2"));
// Edge Case 2: No values lesser than or equal to filter value. // Edge Case 2: No values lesser than or equal to filter value.
let mut catalog2 = Catalog::new(db_path); let mut catalog2 = Catalog::new(db_path);
let filter2 = Filter::new().with_signed_int("sell_trend", Comparison::LesserOrEqual, -6); let filter2 = Filter::new().with_signed_int("sell_trend", Comparison::LesserOrEqual, -6);
assert!(catalog2.load_by_filter(&filter2).is_ok()); assert!(catalog2.load_by_filter(&filter2).is_ok());
assert!(catalog2.items.is_empty()); assert!(catalog2.items.is_empty());
// Edge Case 3: All values lesser than or equal to filter value. // Edge Case 3: All values lesser than or equal to filter value.
let mut catalog3 = Catalog::new(db_path); let mut catalog3 = Catalog::new(db_path);
let filter3 = Filter::new().with_signed_int("sell_trend", Comparison::LesserOrEqual, 20); let filter3 = Filter::new().with_signed_int("sell_trend", Comparison::LesserOrEqual, 20);
@@ -1974,7 +1948,6 @@ mod tests {
assert!(catalog3.items.contains_key("item-1")); assert!(catalog3.items.contains_key("item-1"));
assert!(catalog3.items.contains_key("item-2")); assert!(catalog3.items.contains_key("item-2"));
assert!(catalog3.items.contains_key("item-3")); assert!(catalog3.items.contains_key("item-3"));
std::fs::remove_file(path).unwrap(); std::fs::remove_file(path).unwrap();
} }
// ## Integration Tests // ## Integration Tests
@@ -2223,4 +2196,83 @@ mod tests {
// Clean up // Clean up
std::fs::remove_file(path).unwrap(); std::fs::remove_file(path).unwrap();
} }
#[test]
fn load_by_filter_should_load_unsigned_int_comparisons() {
let db_path = "target/test_dbs/lbf_unsigned_int_comparisons.db";
let path = std::path::Path::new(db_path);
std::fs::create_dir_all(path.parent().unwrap()).unwrap();
if path.exists() {
std::fs::remove_file(path).unwrap();
}
let mut catalog_setup = Catalog::new(db_path);
catalog_setup.init().unwrap();
let items = vec![
Item {
id: "item-1".to_string(),
name: "Item One".to_string(),
price: 100,
sell_trend: 0,
in_stock: true,
},
Item {
id: "item-2".to_string(),
name: "Item Two".to_string(),
price: 200,
sell_trend: 0,
in_stock: false,
},
Item {
id: "item-3".to_string(),
name: "Item Three".to_string(),
price: 300,
sell_trend: 0,
in_stock: true,
},
];
catalog_setup.insert_many(items).unwrap();
catalog_setup.persist().unwrap();
// Equal
let mut catalog = Catalog::new(db_path);
let filter = Filter::new().with_unsigned_int("price", Comparison::Equal, 200);
assert!(catalog.load_by_filter(&filter).is_ok());
assert_eq!(catalog.items.len(), 1);
assert!(catalog.items.contains_key("item-2"));
// Greater
let mut catalog = Catalog::new(db_path);
let filter = Filter::new().with_unsigned_int("price", Comparison::Greater, 200);
assert!(catalog.load_by_filter(&filter).is_ok());
assert_eq!(catalog.items.len(), 1);
assert!(catalog.items.contains_key("item-3"));
// Lesser
let mut catalog = Catalog::new(db_path);
let filter = Filter::new().with_unsigned_int("price", Comparison::Lesser, 200);
assert!(catalog.load_by_filter(&filter).is_ok());
assert_eq!(catalog.items.len(), 1);
assert!(catalog.items.contains_key("item-1"));
// GreaterOrEqual
let mut catalog = Catalog::new(db_path);
let filter = Filter::new().with_unsigned_int("price", Comparison::GreaterOrEqual, 200);
assert!(catalog.load_by_filter(&filter).is_ok());
assert_eq!(catalog.items.len(), 2);
assert!(catalog.items.contains_key("item-2"));
assert!(catalog.items.contains_key("item-3"));
// LesserOrEqual
let mut catalog = Catalog::new(db_path);
let filter = Filter::new().with_unsigned_int("price", Comparison::LesserOrEqual, 200);
assert!(catalog.load_by_filter(&filter).is_ok());
assert_eq!(catalog.items.len(), 2);
assert!(catalog.items.contains_key("item-1"));
assert!(catalog.items.contains_key("item-2"));
// Edge case: No match
let mut catalog = Catalog::new(db_path);
let filter = Filter::new().with_unsigned_int("price", Comparison::Equal, 400);
assert!(catalog.load_by_filter(&filter).is_ok());
assert!(catalog.items.is_empty());
// Edge case: All match
let mut catalog = Catalog::new(db_path);
let filter = Filter::new().with_unsigned_int("price", Comparison::GreaterOrEqual, 100);
assert!(catalog.load_by_filter(&filter).is_ok());
assert_eq!(catalog.items.len(), 3);
std::fs::remove_file(path).unwrap();
}
} }

View File

@@ -2,4 +2,5 @@
pub enum E { pub enum E {
Bool(bool), Bool(bool),
SignedInt(i64), SignedInt(i64),
UnsignedInt(i64),
} }

View File

@@ -32,6 +32,19 @@ impl Filter {
)); ));
self self
} }
pub fn with_unsigned_int(
mut self,
attribute_name: &str,
comparison: Comparison,
value: i64,
) -> Self {
self.conditions.push((
attribute_name.to_string(),
comparison,
Condition::UnsignedInt(value),
));
self
}
pub(crate) fn conditions(&self) -> impl Iterator<Item = &(String, Comparison, Condition)> { pub(crate) fn conditions(&self) -> impl Iterator<Item = &(String, Comparison, Condition)> {
self.conditions.iter() self.conditions.iter()
} }