1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
use crate::bucket_strainer::{Layer, Rule};
use crate::Scalar;
use std::mem;
#[derive(Clone)]
pub struct Bucket<T>
where
T: Clone,
{
id: String,
items: Vec<T>,
rule: Box<dyn Rule<T>>,
pub acceptable_score_treshold: Scalar,
}
impl<T> Bucket<T>
where
T: Clone,
{
pub fn new(id: String, rule: Box<dyn Rule<T>>) -> Self {
Self {
id,
items: vec![],
rule,
acceptable_score_treshold: 0.0,
}
}
pub fn id(&self) -> &str {
&self.id
}
pub fn items(&self) -> &[T] {
&self.items
}
pub fn rule(&self) -> &Box<dyn Rule<T>> {
&self.rule
}
pub fn replace_rule(&mut self, rule: Box<dyn Rule<T>>) -> Box<dyn Rule<T>> {
mem::replace(&mut self.rule, rule)
}
pub(crate) fn score(&self, item: &T) -> Option<Scalar> {
let score = self.rule.score(item, self);
if score > self.acceptable_score_treshold {
Some(score)
} else {
None
}
}
pub(crate) fn clear(&mut self) {
self.items.clear();
}
pub(crate) fn insert(&mut self, item: T) {
self.items.push(item);
}
}
impl<T> Into<Layer<T>> for Bucket<T>
where
T: Clone,
{
fn into(self) -> Layer<T> {
Layer::new(vec![self])
}
}