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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
use rand::{self, Rng, ThreadRng};
#[derive(Debug, Clone)]
pub struct NodeOptions<R> {
pub rng: R,
pub max_active_view_size: u8,
pub max_passive_view_size: u8,
pub shuffle_active_view_size: u8,
pub shuffle_passive_view_size: u8,
pub active_random_walk_len: u8,
pub passive_random_walk_len: u8,
}
impl NodeOptions<ThreadRng> {
pub fn new() -> Self {
Self::default()
}
}
impl<R: Rng> NodeOptions<R> {
pub const DEFAULT_MAX_ACTIVE_VIEW_SIZE: u8 = 4;
pub const DEFAULT_MAX_PASSIVE_VIEW_SIZE: u8 = 24;
pub const DEFAULT_SHUFFLE_ACTIVE_VIEW_SIZE: u8 = 2;
pub const DEFAULT_SHUFFLE_PASSIVE_VIEW_SIZE: u8 = 2;
pub const DEFAULT_ACTIVE_RANDOM_WALK_LEN: u8 = 5;
pub const DEFAULT_PASSIVE_RANDOM_WALK_LEN: u8 = 2;
pub fn set_rng<S: Rng>(self, rng: S) -> NodeOptions<S> {
NodeOptions {
rng,
max_active_view_size: self.max_active_view_size,
max_passive_view_size: self.max_passive_view_size,
shuffle_active_view_size: self.shuffle_active_view_size,
shuffle_passive_view_size: self.shuffle_passive_view_size,
active_random_walk_len: self.active_random_walk_len,
passive_random_walk_len: self.passive_random_walk_len,
}
}
}
impl Default for NodeOptions<ThreadRng> {
fn default() -> Self {
NodeOptions {
rng: rand::thread_rng(),
max_active_view_size: Self::DEFAULT_MAX_ACTIVE_VIEW_SIZE,
max_passive_view_size: Self::DEFAULT_MAX_PASSIVE_VIEW_SIZE,
shuffle_active_view_size: Self::DEFAULT_SHUFFLE_ACTIVE_VIEW_SIZE,
shuffle_passive_view_size: Self::DEFAULT_SHUFFLE_PASSIVE_VIEW_SIZE,
active_random_walk_len: Self::DEFAULT_ACTIVE_RANDOM_WALK_LEN,
passive_random_walk_len: Self::DEFAULT_PASSIVE_RANDOM_WALK_LEN,
}
}
}