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
#![allow(clippy::assign_op_pattern)]
use fluvio_types::SpuId;
use dataplane::core::{Encoder, Decoder};
#[derive(Decoder, Encoder, Debug, Clone, PartialEq)]
#[cfg_attr(
feature = "use_serde",
derive(serde::Serialize, serde::Deserialize),
serde(rename_all = "camelCase")
)]
pub struct PartitionSpec {
pub leader: SpuId,
pub replicas: Vec<SpuId>,
}
impl std::default::Default for PartitionSpec {
fn default() -> Self {
PartitionSpec {
leader: 0,
replicas: Vec::default(),
}
}
}
impl PartitionSpec {
pub fn new(leader: SpuId, replicas: Vec<SpuId>) -> Self {
Self { leader, replicas }
}
pub fn has_spu(&self, spu: &SpuId) -> bool {
self.replicas.contains(spu)
}
pub fn followers(&self) -> Vec<SpuId> {
self.replicas
.iter()
.filter_map(|r| if r == &self.leader { None } else { Some(*r) })
.collect()
}
}
impl From<Vec<i32>> for PartitionSpec {
fn from(replicas: Vec<i32>) -> Self {
if !replicas.is_empty() {
Self::new(replicas[0], replicas)
} else {
Self::new(0, replicas)
}
}
}