tuwunel_database/map/
keys_prefix.rs

1use std::{convert::AsRef, fmt::Debug, sync::Arc};
2
3use futures::{Stream, StreamExt, TryStreamExt, future};
4use serde::{Deserialize, Serialize};
5use tuwunel_core::{Result, implement};
6
7use crate::keyval::{Key, result_deserialize_key, serialize_key};
8
9#[implement(super::Map)]
10pub fn keys_prefix<'a, K, P>(
11	self: &'a Arc<Self>,
12	prefix: &P,
13) -> impl Stream<Item = Result<Key<'_, K>>> + Send + use<'a, K, P>
14where
15	P: Serialize + ?Sized + Debug,
16	K: Deserialize<'a> + Send,
17{
18	self.keys_prefix_raw(prefix)
19		.map(result_deserialize_key::<K>)
20}
21
22#[implement(super::Map)]
23#[tracing::instrument(skip(self), level = "trace")]
24pub fn keys_prefix_raw<P>(
25	self: &Arc<Self>,
26	prefix: &P,
27) -> impl Stream<Item = Result<Key<'_>>> + Send + use<'_, P>
28where
29	P: Serialize + ?Sized + Debug,
30{
31	let key = serialize_key(prefix).expect("failed to serialize query key");
32	self.raw_keys_from(&key)
33		.try_take_while(move |k: &Key<'_>| future::ok(k.starts_with(&key)))
34}
35
36#[implement(super::Map)]
37pub fn keys_raw_prefix<'a, K, P>(
38	self: &'a Arc<Self>,
39	prefix: &'a P,
40) -> impl Stream<Item = Result<Key<'_, K>>> + Send + 'a
41where
42	P: AsRef<[u8]> + ?Sized + Debug + Sync + 'a,
43	K: Deserialize<'a> + Send + 'a,
44{
45	self.raw_keys_prefix(prefix)
46		.map(result_deserialize_key::<K>)
47}
48
49#[implement(super::Map)]
50pub fn raw_keys_prefix<'a, P>(
51	self: &'a Arc<Self>,
52	prefix: &'a P,
53) -> impl Stream<Item = Result<Key<'_>>> + Send + 'a
54where
55	P: AsRef<[u8]> + ?Sized + Debug + Sync + 'a,
56{
57	self.raw_keys_from(prefix)
58		.try_take_while(|k: &Key<'_>| future::ok(k.starts_with(prefix.as_ref())))
59}
tuwunel_database/map/keys_prefix.rs

tuwunel_database/map/
keys_prefix.rs
