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

#![allow(clippy::assign_op_pattern)]

use dataplane::api::Request;
use dataplane::derive::Decode;
use dataplane::derive::Encode;
use fluvio_controlplane_metadata::spu::SpuSpec;
use fluvio_controlplane_metadata::message::SpuMsg;

use crate::InternalSpuApi;

/// Changes to Spu specs
#[derive(Decode, Encode, Debug, Default)]
pub struct UpdateSpuRequest {
    pub epoch: i64,
    pub changes: Vec<SpuMsg>,
    pub all: Vec<SpuSpec>,
}

impl Request for UpdateSpuRequest {
    const API_KEY: u16 = InternalSpuApi::UpdateSpu as u16;
    type Response = UpdateSpuResponse;
}

impl UpdateSpuRequest {
    pub fn with_changes(epoch: i64, changes: Vec<SpuMsg>) -> Self {
        Self {
            epoch,
            changes,
            all: vec![],
        }
    }

    pub fn with_all(epoch: i64, all: Vec<SpuSpec>) -> Self {
        Self {
            epoch,
            changes: vec![],
            all,
        }
    }

    pub fn changes(&self) -> &Vec<SpuMsg> {
        &self.changes
    }

    pub fn changes_owned(self) -> Vec<SpuMsg> {
        self.changes
    }

    /*
    pub fn spus_to_map(&self) -> BTreeMap<SpuId, SpuSpec> {
        let mut res = BTreeMap::new();
        for spu in self.spus.iter() {
            res.insert(spu.content.id.clone(), spu.content.clone());
        }
        res
    }
    */

    /*
    pub fn add<S>(mut self, spu: S) -> Self
    where
        S: Into<SpuMsg>,
    {
        self.spus.push(spu.into());
        self
    }
    */
}

#[derive(Decode, Encode, Default, Debug)]
pub struct UpdateSpuResponse {}