filip/Stalwart
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Delete in-memory keys by prefix

Browse Source
This commit is contained in:
mdecimus
2024-12-26 19:01:49 +01:00
parent 1ba229aac7
commit e5ebc5ce33
13 changed files with 566 additions and 215 deletions
Download Patch File Download Diff File Expand all files Collapse all files

147
tests/src/store/lookup.rs
View File

@@ -28,6 +28,7 @@ pub async fn lookup_tests() {
};
for (store_id, store) in stores.in_memory_stores {
let is_mysql = store_id == "mysql";
println!("Testing lookup store {}...", store_id);
if let InMemoryStore::Store(store) = &store {
store.destroy().await;
@@ -80,10 +81,15 @@ pub async fn lookup_tests() {
.await
.unwrap();
assert_eq!(3, store.counter_get(key.clone()).await.unwrap());
store
.counter_incr(KeyValue::new(key.clone(), -3))
.await
.unwrap();
if !is_mysql {
store
.counter_incr(KeyValue::new(key.clone(), -3))
.await
.unwrap();
} else {
// TODO: Detect mySQL version and use RETURNING
store.counter_delete(key.clone()).await.unwrap();
}
assert_eq!(0, store.counter_get(key.clone()).await.unwrap());
// Test counter expiry
@@ -119,5 +125,138 @@ pub async fn lookup_tests() {
if let InMemoryStore::Store(store) = &store {
store.assert_is_empty(store.clone().into()).await;
}
// Test prefix delete
store
.key_set(KeyValue::with_prefix(
1,
[0],
"hello".to_string().into_bytes(),
))
.await
.unwrap();
for v in 0u32..2020u32 {
store
.key_set(KeyValue::with_prefix(
0,
pack_u32(0, v),
"world".to_string().into_bytes(),
))
.await
.unwrap();
store
.counter_incr(KeyValue::with_prefix(0, pack_u32(1, v), 123).expires(3600))
.await
.unwrap();
}
// Make sure the keys are there
assert_eq!(
Some("hello"),
store
.key_get::<String>(KeyValue::<()>::build_key(1, [0]))
.await
.unwrap()
.as_deref()
);
for v in [0, 1000, 1001, 2000, 2001] {
assert_eq!(
Some("world"),
store
.key_get::<String>(KeyValue::<()>::build_key(0, pack_u32(0, v)))
.await
.unwrap()
.as_deref()
);
}
for v in [0, 1000, 1001, 2000, 2001] {
assert_ne!(
0,
store
.counter_get(KeyValue::<()>::build_key(0, pack_u32(1, v)))
.await
.unwrap()
);
}
// Delete [0, 0, 0, 0, 1] prefix and make sure only the keys with that prefix are gone
store
.key_delete_prefix(&KeyValue::<()>::build_key(0, 1u32.to_be_bytes()))
.await
.unwrap();
assert_eq!(
Some("hello"),
store
.key_get::<String>(KeyValue::<()>::build_key(1, [0]))
.await
.unwrap()
.as_deref()
);
for v in [0, 1000, 1001, 2000, 2001] {
assert_eq!(
Some("world"),
store
.key_get::<String>(KeyValue::<()>::build_key(0, pack_u32(0, v)))
.await
.unwrap()
.as_deref()
);
}
for v in [0, 1000, 1001, 2000, 2001] {
assert_eq!(
0,
store
.counter_get(KeyValue::<()>::build_key(0, pack_u32(1, v)))
.await
.unwrap()
);
}
// Delete [0, 0, 0, 0, 0] prefix and make sure only the keys with that prefix are gone
store
.key_delete_prefix(&KeyValue::<()>::build_key(0, 0u32.to_be_bytes()))
.await
.unwrap();
assert_eq!(
Some("hello"),
store
.key_get::<String>(KeyValue::<()>::build_key(1, [0]))
.await
.unwrap()
.as_deref()
);
for v in [0, 1000, 1001, 2000, 2001] {
assert_eq!(
None,
store
.key_get::<String>(KeyValue::<()>::build_key(0, pack_u32(0, v)))
.await
.unwrap()
.as_deref()
);
}
// Delete [1, ...] prefix and make sure it's all gone
store.key_delete_prefix(&[1u8]).await.unwrap();
assert_eq!(
None,
store
.key_get::<String>(KeyValue::<()>::build_key(1, [0]))
.await
.unwrap()
.as_deref()
);
if let InMemoryStore::Store(store) = &store {
store.assert_is_empty(store.clone().into()).await;
}
}
}
fn pack_u32(a: u32, b: u32) -> Vec<u8> {
((a as u64) << 32 | b as u64).to_be_bytes().to_vec()
}

5
tests/src/store/mod.rs
View File

@@ -67,6 +67,11 @@ type = "redis"
urls = "redis://127.0.0.1"
redis-type = "single"
[storage]
lookup = "mysql"
data = "postgresql"
blob = "sqlite"
"#;
#[tokio::test(flavor = "multi_thread")]