Switch to iterators

This commit is contained in:
2020-07-06 09:16:19 +02:00
parent 5b5d3a9b64
commit 2b89a4169b
2 changed files with 18 additions and 19 deletions

View File

@@ -84,29 +84,27 @@ impl<F, K> IndexedDestructured<F, K> for Index<F, K>
where
K: Clone + Eq + Hash + PartialEq + Ord,
{
fn find(&self, key: &K) -> Vec<&F> {
fn find<'i>(&'i self, key: &K) -> Box<dyn Iterator<Item = &F> + 'i> {
let mut values = vec![];
if let Some(fields) = self.hashmap.get(key) {
values.push(fields);
}
values
Box::new(values.into_iter())
}
fn find_range(&self, start: &K, end: &K) -> Vec<(K, &F)> {
fn find_range<'i>(&'i self, start: &K, end: &K) -> Box<dyn Iterator<Item = (K, &F)> + 'i> {
let start = self.index(start);
let end = self.index(end);
(start..end)
.filter_map(|i| {
let key = &self.keys[i];
if let Some(fields) = self.hashmap.get(key) {
Some((key.clone(), fields))
} else {
None
}
})
.collect()
Box::new((start..=end).filter_map(move |i| {
let key = &self.keys[i];
if let Some(fields) = self.hashmap.get(key) {
Some((key.clone(), fields))
} else {
None
}
}))
}
}

View File

@@ -85,22 +85,23 @@ where
R: Record<K>,
K: Clone + Eq + Hash + PartialEq + Ord,
{
fn find(&self, key: &K) -> Vec<&R> {
fn find<'i>(&'i self, key: &K) -> Box<dyn Iterator<Item = &R> + 'i> {
let mut values = vec![];
if let Some(record) = self.hashmap.get(key) {
values.push(record);
}
values
Box::new(values.into_iter())
}
fn find_range(&self, start: &K, end: &K) -> Vec<&R> {
fn find_range<'i>(&'i self, start: &K, end: &K) -> Box<dyn Iterator<Item = &R> + 'i> {
let start = self.index(start);
let end = self.index(end);
(start..end)
.filter_map(|i| self.hashmap.get(&self.keys[i]))
.collect()
Box::new((start..=end).filter_map(move |i| {
let k = &self.keys[i];
self.hashmap.get(k)
}))
}
}