crabka_protocol/opt/rustwide/workdir/generated/
BrokerHeartbeatRequest.borrowed.rs

1// AUTO-GENERATED by crabka-protocol-codegen against a9ce3221537b8653448750697915607dc7936cf3. Do not edit.
2
3use bytes::BufMut;
4
5use crate::primitives::fixed::{get_bool, get_i32, get_i64, put_bool, put_i32, put_i64};
6use crate::tagged_fields::{
7    WriteTaggedFields, encode_to_bytes, read_tagged_fields, tagged_fields_len,
8};
9use crate::{DecodeBorrow, Encode, ProtocolError, UnknownTaggedFields};
10
11pub const API_KEY: i16 = 63;
12pub const MIN_VERSION: i16 = 0;
13pub const MAX_VERSION: i16 = 2;
14pub const FLEXIBLE_MIN: i16 = 0;
15
16#[inline]
17fn is_flexible(version: i16) -> bool {
18    version >= FLEXIBLE_MIN
19}
20
21#[derive(Debug, Clone, PartialEq, Eq)]
22pub struct BrokerHeartbeatRequest {
23    pub broker_id: i32,
24    pub broker_epoch: i64,
25    pub current_metadata_offset: i64,
26    pub want_fence: bool,
27    pub want_shut_down: bool,
28    pub offline_log_dirs: Vec<crate::primitives::uuid::Uuid>,
29    pub cordoned_log_dirs: Option<Vec<crate::primitives::uuid::Uuid>>,
30    pub unknown_tagged_fields: UnknownTaggedFields,
31}
32impl Default for BrokerHeartbeatRequest {
33    fn default() -> Self {
34        Self {
35            broker_id: 0i32,
36            broker_epoch: -1i64,
37            current_metadata_offset: 0i64,
38            want_fence: false,
39            want_shut_down: false,
40            offline_log_dirs: Vec::new(),
41            cordoned_log_dirs: None,
42            unknown_tagged_fields: Default::default(),
43        }
44    }
45}
46impl BrokerHeartbeatRequest {
47    pub fn to_owned(&self) -> crate::owned::broker_heartbeat_request::BrokerHeartbeatRequest {
48        crate::owned::broker_heartbeat_request::BrokerHeartbeatRequest {
49            broker_id: (self.broker_id),
50            broker_epoch: (self.broker_epoch),
51            current_metadata_offset: (self.current_metadata_offset),
52            want_fence: (self.want_fence),
53            want_shut_down: (self.want_shut_down),
54            offline_log_dirs: (self.offline_log_dirs).clone(),
55            cordoned_log_dirs: (self.cordoned_log_dirs).clone(),
56            unknown_tagged_fields: self.unknown_tagged_fields.clone(),
57        }
58    }
59}
60impl Encode for BrokerHeartbeatRequest {
61    fn encode<B: BufMut>(&self, buf: &mut B, version: i16) -> Result<(), ProtocolError> {
62        if !(MIN_VERSION..=MAX_VERSION).contains(&version) {
63            return Err(ProtocolError::UnsupportedVersion {
64                api_key: API_KEY,
65                version,
66            });
67        }
68        let flex = is_flexible(version);
69        if version >= 0 {
70            put_i32(buf, self.broker_id);
71        }
72        if version >= 0 {
73            put_i64(buf, self.broker_epoch);
74        }
75        if version >= 0 {
76            put_i64(buf, self.current_metadata_offset);
77        }
78        if version >= 0 {
79            put_bool(buf, self.want_fence);
80        }
81        if version >= 0 {
82            put_bool(buf, self.want_shut_down);
83        }
84        if flex {
85            let mut tagged = WriteTaggedFields::new();
86            if !(crate::codegen_helpers::is_default(&self.offline_log_dirs)) {
87                let payload = encode_to_bytes(
88                    {
89                        let prefix = crate::primitives::array::array_len_prefix_len(
90                            (self.offline_log_dirs).len(),
91                            flex,
92                        );
93                        let body: usize = (self.offline_log_dirs).iter().map(|_| 16).sum();
94                        prefix + body
95                    },
96                    |b| {
97                        {
98                            crate::primitives::array::put_array_len(
99                                b,
100                                (self.offline_log_dirs).len(),
101                                flex,
102                            );
103                            for it in &self.offline_log_dirs {
104                                crate::primitives::uuid::put_uuid(b, *it);
105                            }
106                        };
107                        Ok(())
108                    },
109                );
110                tagged.add(0, payload);
111            }
112            if !(self.cordoned_log_dirs.is_none()) {
113                let payload = encode_to_bytes(
114                    {
115                        let opt: Option<&Vec<_>> = (self.cordoned_log_dirs).as_ref();
116                        let prefix = crate::primitives::array::nullable_array_len_prefix_len(
117                            opt.map(std::vec::Vec::len),
118                            flex,
119                        );
120                        let body: usize = opt.map_or(0, |v| v.iter().map(|_| 16).sum());
121                        prefix + body
122                    },
123                    |b| {
124                        {
125                            let len = (self.cordoned_log_dirs).as_ref().map(Vec::len);
126                            crate::primitives::array::put_nullable_array_len(b, len, flex);
127                            if let Some(v) = &self.cordoned_log_dirs {
128                                for it in v {
129                                    crate::primitives::uuid::put_uuid(b, *it);
130                                }
131                            }
132                        };
133                        Ok(())
134                    },
135                );
136                tagged.add(1, payload);
137            }
138            tagged.write(buf, &self.unknown_tagged_fields);
139        }
140        Ok(())
141    }
142    fn encoded_len(&self, version: i16) -> usize {
143        let flex = is_flexible(version);
144        let mut n: usize = 0;
145        if version >= 0 {
146            n += 4;
147        }
148        if version >= 0 {
149            n += 8;
150        }
151        if version >= 0 {
152            n += 8;
153        }
154        if version >= 0 {
155            n += 1;
156        }
157        if version >= 0 {
158            n += 1;
159        }
160        if flex {
161            let mut known_pairs: Vec<(u32, usize)> = Vec::new();
162            if !(crate::codegen_helpers::is_default(&self.offline_log_dirs)) {
163                known_pairs.push((0, {
164                    let prefix = crate::primitives::array::array_len_prefix_len(
165                        (self.offline_log_dirs).len(),
166                        flex,
167                    );
168                    let body: usize = (self.offline_log_dirs).iter().map(|_| 16).sum();
169                    prefix + body
170                }));
171            }
172            if !(self.cordoned_log_dirs.is_none()) {
173                known_pairs.push((1, {
174                    let opt: Option<&Vec<_>> = (self.cordoned_log_dirs).as_ref();
175                    let prefix = crate::primitives::array::nullable_array_len_prefix_len(
176                        opt.map(std::vec::Vec::len),
177                        flex,
178                    );
179                    let body: usize = opt.map_or(0, |v| v.iter().map(|_| 16).sum());
180                    prefix + body
181                }));
182            }
183            n += tagged_fields_len(&known_pairs, &self.unknown_tagged_fields);
184        }
185        n
186    }
187}
188impl<'de> DecodeBorrow<'de> for BrokerHeartbeatRequest {
189    fn decode_borrow(buf: &mut &'de [u8], version: i16) -> Result<Self, ProtocolError> {
190        if !(MIN_VERSION..=MAX_VERSION).contains(&version) {
191            return Err(ProtocolError::UnsupportedVersion {
192                api_key: API_KEY,
193                version,
194            });
195        }
196        let flex = is_flexible(version);
197        let mut out = Self::default();
198        if version >= 0 {
199            out.broker_id = get_i32(buf)?;
200        }
201        if version >= 0 {
202            out.broker_epoch = get_i64(buf)?;
203        }
204        if version >= 0 {
205            out.current_metadata_offset = get_i64(buf)?;
206        }
207        if version >= 0 {
208            out.want_fence = get_bool(buf)?;
209        }
210        if version >= 0 {
211            out.want_shut_down = get_bool(buf)?;
212        }
213        if flex {
214            let mut tag_offline_log_dirs = None;
215            let mut tag_cordoned_log_dirs = None;
216            out.unknown_tagged_fields = read_tagged_fields(buf, |tag, payload| match tag {
217                0 => {
218                    tag_offline_log_dirs = Some({
219                        let b: &mut &[u8] = payload;
220                        {
221                            let n = crate::primitives::array::get_array_len(b, flex)?;
222                            let mut v = Vec::with_capacity(n);
223                            for _ in 0..n {
224                                v.push(crate::primitives::uuid::get_uuid(b)?);
225                            }
226                            v
227                        }
228                    });
229                    Ok(true)
230                }
231                1 => {
232                    tag_cordoned_log_dirs = Some({
233                        let b: &mut &[u8] = payload;
234                        {
235                            let opt = crate::primitives::array::get_nullable_array_len(b, flex)?;
236                            match opt {
237                                None => None,
238                                Some(n) => {
239                                    let mut v = Vec::with_capacity(n);
240                                    for _ in 0..n {
241                                        v.push(crate::primitives::uuid::get_uuid(b)?);
242                                    }
243                                    Some(v)
244                                }
245                            }
246                        }
247                    });
248                    Ok(true)
249                }
250                _ => Ok(false),
251            })?;
252            if let Some(v) = tag_offline_log_dirs {
253                out.offline_log_dirs = v;
254            }
255            if let Some(v) = tag_cordoned_log_dirs {
256                out.cordoned_log_dirs = v;
257            }
258        }
259        Ok(out)
260    }
261}
262#[cfg(test)]
263impl BrokerHeartbeatRequest {
264    #[must_use]
265    pub fn populated(version: i16) -> Self {
266        let mut m = Self::default();
267        if version >= 0 {
268            m.broker_id = 1i32;
269        }
270        if version >= 0 {
271            m.broker_epoch = 1i64;
272        }
273        if version >= 0 {
274            m.current_metadata_offset = 1i64;
275        }
276        if version >= 0 {
277            m.want_fence = true;
278        }
279        if version >= 0 {
280            m.want_shut_down = true;
281        }
282        if version >= 1 {
283            m.offline_log_dirs = vec![crate::primitives::uuid::Uuid([1u8; 16])];
284        }
285        if version >= 2 {
286            m.cordoned_log_dirs = Some(vec![crate::primitives::uuid::Uuid([1u8; 16])]);
287        }
288        m
289    }
290}
crabka_protocol/opt/rustwide/workdir/generated/BrokerHeartbeatRequest.borrowed.rs

crabka_protocol/opt/rustwide/workdir/generated/
BrokerHeartbeatRequest.borrowed.rs
