krabka_protocol/opt/rustwide/workdir/generated/
WriteShareGroupStateRequest.borrowed.rs

1// AUTO-GENERATED by krabka-protocol-codegen against a9ce3221537b8653448750697915607dc7936cf3. Do not edit.
2use bytes::BufMut;
3
4use crate::{
5    DecodeBorrow, Encode, ProtocolError, UnknownTaggedFields,
6    primitives::{
7        fixed::{get_i8, get_i16, get_i32, get_i64, put_i8, put_i16, put_i32, put_i64},
8        string_bytes::{compact_string_len, put_compact_string, put_string, string_len},
9        string_bytes_borrowed::{get_compact_string_borrowed, get_string_borrowed},
10    },
11    tagged_fields::{WriteTaggedFields, read_tagged_fields, tagged_fields_len},
12};
13pub const API_KEY: i16 = 85;
14pub const MIN_VERSION: i16 = 0;
15pub const MAX_VERSION: i16 = 1;
16pub const LATEST_STABLE_VERSION: i16 = 1;
17pub const FLEXIBLE_MIN: i16 = 0;
18#[inline]
19#[must_use]
20pub fn is_flexible(version: i16) -> bool {
21    version >= FLEXIBLE_MIN
22}
23#[derive(Debug, Clone, PartialEq, Eq, Default)]
24pub struct WriteShareGroupStateRequest<'a> {
25    pub group_id: &'a str,
26    pub topics: Vec<WriteStateData>,
27    pub unknown_tagged_fields: UnknownTaggedFields,
28}
29impl WriteShareGroupStateRequest<'_> {
30    /// # Panics
31    ///
32    /// Panics if a records field contains an invalid encoded record batch.
33    #[must_use]
34    pub fn to_owned(
35        &self,
36    ) -> crate::owned::write_share_group_state_request::WriteShareGroupStateRequest {
37        crate::owned::write_share_group_state_request::WriteShareGroupStateRequest {
38            group_id: (self.group_id).to_string(),
39            topics: (self.topics).iter().map(WriteStateData::to_owned).collect(),
40            unknown_tagged_fields: self.unknown_tagged_fields.clone(),
41        }
42    }
43}
44impl Encode for WriteShareGroupStateRequest<'_> {
45    fn encode<B: BufMut>(&self, buf: &mut B, version: i16) -> Result<(), ProtocolError> {
46        if !(MIN_VERSION..=MAX_VERSION).contains(&version) {
47            return Err(ProtocolError::UnsupportedVersion {
48                api_key: API_KEY,
49                version,
50            });
51        }
52        let flex = is_flexible(version);
53        if version >= 0 {
54            if flex {
55                let () = put_compact_string(buf, self.group_id);
56            } else {
57                let () = put_string(buf, self.group_id);
58            }
59        }
60        if version >= 0 {
61            {
62                crate::primitives::array::put_array_len(buf, (self.topics).len(), flex);
63                for it in &self.topics {
64                    it.encode(buf, version)?;
65                }
66            };
67        }
68        if flex {
69            let tagged = WriteTaggedFields::new();
70            tagged.write(buf, &self.unknown_tagged_fields);
71        }
72        Ok(())
73    }
74    fn encoded_len(&self, version: i16) -> usize {
75        let flex = is_flexible(version);
76        let mut n: usize = 0;
77        if version >= 0 {
78            n += if flex {
79                compact_string_len(self.group_id)
80            } else {
81                string_len(self.group_id)
82            };
83        }
84        if version >= 0 {
85            n += {
86                let prefix =
87                    crate::primitives::array::array_len_prefix_len((self.topics).len(), flex);
88                let body: usize = (self.topics).iter().map(|it| it.encoded_len(version)).sum();
89                prefix + body
90            };
91        }
92        if flex {
93            let known_pairs: Vec<(u32, usize)> = Vec::new();
94            n += tagged_fields_len(&known_pairs, &self.unknown_tagged_fields);
95        }
96        n
97    }
98}
99impl<'de> DecodeBorrow<'de> for WriteShareGroupStateRequest<'de> {
100    fn decode_borrow(buf: &mut &'de [u8], version: i16) -> Result<Self, ProtocolError> {
101        if !(MIN_VERSION..=MAX_VERSION).contains(&version) {
102            return Err(ProtocolError::UnsupportedVersion {
103                api_key: API_KEY,
104                version,
105            });
106        }
107        let flex = is_flexible(version);
108        let mut out = Self::default();
109        if version >= 0 {
110            out.group_id = if flex {
111                get_compact_string_borrowed(buf)?
112            } else {
113                get_string_borrowed(buf)?
114            };
115        }
116        if version >= 0 {
117            out.topics = {
118                let n = crate::primitives::array::get_array_len(buf, flex)?;
119                let mut v = Vec::with_capacity(n);
120                for _ in 0..n {
121                    v.push(WriteStateData::decode_borrow(buf, version)?);
122                }
123                v
124            };
125        }
126        if flex {
127            out.unknown_tagged_fields = read_tagged_fields(buf, |_tag, _payload| Ok(false))?;
128        }
129        Ok(out)
130    }
131}
132#[cfg(test)]
133impl WriteShareGroupStateRequest<'_> {
134    #[must_use]
135    pub fn populated(version: i16) -> Self {
136        let mut m = Self::default();
137        if version >= 0 {
138            m.group_id = "x";
139        }
140        if version >= 0 {
141            m.topics = vec![WriteStateData::populated(version)];
142        }
143        m
144    }
145}
146#[derive(Debug, Clone, PartialEq, Eq, Default)]
147pub struct WriteStateData {
148    pub topic_id: crate::primitives::uuid::Uuid,
149    pub partitions: Vec<PartitionData>,
150    pub unknown_tagged_fields: UnknownTaggedFields,
151}
152impl WriteStateData {
153    /// # Panics
154    ///
155    /// Panics if a records field contains an invalid encoded record batch.
156    #[must_use]
157    pub fn to_owned(&self) -> crate::owned::write_share_group_state_request::WriteStateData {
158        crate::owned::write_share_group_state_request::WriteStateData {
159            topic_id: (self.topic_id),
160            partitions: (self.partitions)
161                .iter()
162                .map(PartitionData::to_owned)
163                .collect(),
164            unknown_tagged_fields: self.unknown_tagged_fields.clone(),
165        }
166    }
167}
168impl Encode for WriteStateData {
169    fn encode<B: BufMut>(&self, buf: &mut B, version: i16) -> Result<(), ProtocolError> {
170        let flex = version >= 0;
171        if version >= 0 {
172            crate::primitives::uuid::put_uuid(buf, self.topic_id);
173        }
174        if version >= 0 {
175            {
176                crate::primitives::array::put_array_len(buf, (self.partitions).len(), flex);
177                for it in &self.partitions {
178                    it.encode(buf, version)?;
179                }
180            };
181        }
182        if flex {
183            let tagged = WriteTaggedFields::new();
184            tagged.write(buf, &self.unknown_tagged_fields);
185        }
186        Ok(())
187    }
188    fn encoded_len(&self, version: i16) -> usize {
189        let flex = version >= 0;
190        let mut n: usize = 0;
191        if version >= 0 {
192            n += 16;
193        }
194        if version >= 0 {
195            n += {
196                let prefix =
197                    crate::primitives::array::array_len_prefix_len((self.partitions).len(), flex);
198                let body: usize = (self.partitions)
199                    .iter()
200                    .map(|it| it.encoded_len(version))
201                    .sum();
202                prefix + body
203            };
204        }
205        if flex {
206            let known_pairs: Vec<(u32, usize)> = Vec::new();
207            n += tagged_fields_len(&known_pairs, &self.unknown_tagged_fields);
208        }
209        n
210    }
211}
212impl<'de> DecodeBorrow<'de> for WriteStateData {
213    fn decode_borrow(buf: &mut &'de [u8], version: i16) -> Result<Self, ProtocolError> {
214        let flex = version >= 0;
215        let mut out = Self::default();
216        if version >= 0 {
217            out.topic_id = crate::primitives::uuid::get_uuid(buf)?;
218        }
219        if version >= 0 {
220            out.partitions = {
221                let n = crate::primitives::array::get_array_len(buf, flex)?;
222                let mut v = Vec::with_capacity(n);
223                for _ in 0..n {
224                    v.push(PartitionData::decode_borrow(buf, version)?);
225                }
226                v
227            };
228        }
229        if flex {
230            out.unknown_tagged_fields = read_tagged_fields(buf, |_tag, _payload| Ok(false))?;
231        }
232        Ok(out)
233    }
234}
235#[cfg(test)]
236impl WriteStateData {
237    #[must_use]
238    pub fn populated(version: i16) -> Self {
239        let mut m = Self::default();
240        if version >= 0 {
241            m.topic_id = crate::primitives::uuid::Uuid([1u8; 16]);
242        }
243        if version >= 0 {
244            m.partitions = vec![PartitionData::populated(version)];
245        }
246        m
247    }
248}
249#[derive(Debug, Clone, PartialEq, Eq)]
250pub struct PartitionData {
251    pub partition: i32,
252    pub state_epoch: i32,
253    pub leader_epoch: i32,
254    pub start_offset: i64,
255    pub delivery_complete_count: i32,
256    pub state_batches: Vec<StateBatch>,
257    pub unknown_tagged_fields: UnknownTaggedFields,
258}
259impl Default for PartitionData {
260    fn default() -> Self {
261        Self {
262            partition: 0i32,
263            state_epoch: 0i32,
264            leader_epoch: 0i32,
265            start_offset: 0i64,
266            delivery_complete_count: -1i32,
267            state_batches: Vec::new(),
268            unknown_tagged_fields: UnknownTaggedFields::default(),
269        }
270    }
271}
272impl PartitionData {
273    /// # Panics
274    ///
275    /// Panics if a records field contains an invalid encoded record batch.
276    #[must_use]
277    pub fn to_owned(&self) -> crate::owned::write_share_group_state_request::PartitionData {
278        crate::owned::write_share_group_state_request::PartitionData {
279            partition: (self.partition),
280            state_epoch: (self.state_epoch),
281            leader_epoch: (self.leader_epoch),
282            start_offset: (self.start_offset),
283            delivery_complete_count: (self.delivery_complete_count),
284            state_batches: (self.state_batches)
285                .iter()
286                .map(StateBatch::to_owned)
287                .collect(),
288            unknown_tagged_fields: self.unknown_tagged_fields.clone(),
289        }
290    }
291}
292impl Encode for PartitionData {
293    fn encode<B: BufMut>(&self, buf: &mut B, version: i16) -> Result<(), ProtocolError> {
294        let flex = version >= 0;
295        if version >= 0 {
296            put_i32(buf, self.partition);
297        }
298        if version >= 0 {
299            put_i32(buf, self.state_epoch);
300        }
301        if version >= 0 {
302            put_i32(buf, self.leader_epoch);
303        }
304        if version >= 0 {
305            put_i64(buf, self.start_offset);
306        }
307        if version >= 1 {
308            put_i32(buf, self.delivery_complete_count);
309        }
310        if version >= 0 {
311            {
312                crate::primitives::array::put_array_len(buf, (self.state_batches).len(), flex);
313                for it in &self.state_batches {
314                    it.encode(buf, version)?;
315                }
316            };
317        }
318        if flex {
319            let tagged = WriteTaggedFields::new();
320            tagged.write(buf, &self.unknown_tagged_fields);
321        }
322        Ok(())
323    }
324    fn encoded_len(&self, version: i16) -> usize {
325        let flex = version >= 0;
326        let mut n: usize = 0;
327        if version >= 0 {
328            n += 4;
329        }
330        if version >= 0 {
331            n += 4;
332        }
333        if version >= 0 {
334            n += 4;
335        }
336        if version >= 0 {
337            n += 8;
338        }
339        if version >= 1 {
340            n += 4;
341        }
342        if version >= 0 {
343            n += {
344                let prefix = crate::primitives::array::array_len_prefix_len(
345                    (self.state_batches).len(),
346                    flex,
347                );
348                let body: usize = (self.state_batches)
349                    .iter()
350                    .map(|it| it.encoded_len(version))
351                    .sum();
352                prefix + body
353            };
354        }
355        if flex {
356            let known_pairs: Vec<(u32, usize)> = Vec::new();
357            n += tagged_fields_len(&known_pairs, &self.unknown_tagged_fields);
358        }
359        n
360    }
361}
362impl<'de> DecodeBorrow<'de> for PartitionData {
363    fn decode_borrow(buf: &mut &'de [u8], version: i16) -> Result<Self, ProtocolError> {
364        let flex = version >= 0;
365        let mut out = Self::default();
366        if version >= 0 {
367            out.partition = get_i32(buf)?;
368        }
369        if version >= 0 {
370            out.state_epoch = get_i32(buf)?;
371        }
372        if version >= 0 {
373            out.leader_epoch = get_i32(buf)?;
374        }
375        if version >= 0 {
376            out.start_offset = get_i64(buf)?;
377        }
378        if version >= 1 {
379            out.delivery_complete_count = get_i32(buf)?;
380        }
381        if version >= 0 {
382            out.state_batches = {
383                let n = crate::primitives::array::get_array_len(buf, flex)?;
384                let mut v = Vec::with_capacity(n);
385                for _ in 0..n {
386                    v.push(StateBatch::decode_borrow(buf, version)?);
387                }
388                v
389            };
390        }
391        if flex {
392            out.unknown_tagged_fields = read_tagged_fields(buf, |_tag, _payload| Ok(false))?;
393        }
394        Ok(out)
395    }
396}
397#[cfg(test)]
398impl PartitionData {
399    #[must_use]
400    pub fn populated(version: i16) -> Self {
401        let mut m = Self::default();
402        if version >= 0 {
403            m.partition = 1i32;
404        }
405        if version >= 0 {
406            m.state_epoch = 1i32;
407        }
408        if version >= 0 {
409            m.leader_epoch = 1i32;
410        }
411        if version >= 0 {
412            m.start_offset = 1i64;
413        }
414        if version >= 1 {
415            m.delivery_complete_count = 1i32;
416        }
417        if version >= 0 {
418            m.state_batches = vec![StateBatch::populated(version)];
419        }
420        m
421    }
422}
423#[derive(Debug, Clone, PartialEq, Eq, Default)]
424pub struct StateBatch {
425    pub first_offset: i64,
426    pub last_offset: i64,
427    pub delivery_state: i8,
428    pub delivery_count: i16,
429    pub unknown_tagged_fields: UnknownTaggedFields,
430}
431impl StateBatch {
432    /// # Panics
433    ///
434    /// Panics if a records field contains an invalid encoded record batch.
435    #[must_use]
436    pub fn to_owned(&self) -> crate::owned::write_share_group_state_request::StateBatch {
437        crate::owned::write_share_group_state_request::StateBatch {
438            first_offset: (self.first_offset),
439            last_offset: (self.last_offset),
440            delivery_state: (self.delivery_state),
441            delivery_count: (self.delivery_count),
442            unknown_tagged_fields: self.unknown_tagged_fields.clone(),
443        }
444    }
445}
446impl Encode for StateBatch {
447    fn encode<B: BufMut>(&self, buf: &mut B, version: i16) -> Result<(), ProtocolError> {
448        let flex = version >= 0;
449        if version >= 0 {
450            put_i64(buf, self.first_offset);
451        }
452        if version >= 0 {
453            put_i64(buf, self.last_offset);
454        }
455        if version >= 0 {
456            put_i8(buf, self.delivery_state);
457        }
458        if version >= 0 {
459            put_i16(buf, self.delivery_count);
460        }
461        if flex {
462            let tagged = WriteTaggedFields::new();
463            tagged.write(buf, &self.unknown_tagged_fields);
464        }
465        Ok(())
466    }
467    fn encoded_len(&self, version: i16) -> usize {
468        let flex = version >= 0;
469        let mut n: usize = 0;
470        if version >= 0 {
471            n += 8;
472        }
473        if version >= 0 {
474            n += 8;
475        }
476        if version >= 0 {
477            n += 1;
478        }
479        if version >= 0 {
480            n += 2;
481        }
482        if flex {
483            let known_pairs: Vec<(u32, usize)> = Vec::new();
484            n += tagged_fields_len(&known_pairs, &self.unknown_tagged_fields);
485        }
486        n
487    }
488}
489impl<'de> DecodeBorrow<'de> for StateBatch {
490    fn decode_borrow(buf: &mut &'de [u8], version: i16) -> Result<Self, ProtocolError> {
491        let flex = version >= 0;
492        let mut out = Self::default();
493        if version >= 0 {
494            out.first_offset = get_i64(buf)?;
495        }
496        if version >= 0 {
497            out.last_offset = get_i64(buf)?;
498        }
499        if version >= 0 {
500            out.delivery_state = get_i8(buf)?;
501        }
502        if version >= 0 {
503            out.delivery_count = get_i16(buf)?;
504        }
505        if flex {
506            out.unknown_tagged_fields = read_tagged_fields(buf, |_tag, _payload| Ok(false))?;
507        }
508        Ok(out)
509    }
510}
511#[cfg(test)]
512impl StateBatch {
513    #[must_use]
514    pub fn populated(version: i16) -> Self {
515        let mut m = Self::default();
516        if version >= 0 {
517            m.first_offset = 1i64;
518        }
519        if version >= 0 {
520            m.last_offset = 1i64;
521        }
522        if version >= 0 {
523            m.delivery_state = 1i8;
524        }
525        if version >= 0 {
526            m.delivery_count = 1i16;
527        }
528        m
529    }
530}
krabka_protocol/opt/rustwide/workdir/generated/WriteShareGroupStateRequest.borrowed.rs

krabka_protocol/opt/rustwide/workdir/generated/
WriteShareGroupStateRequest.borrowed.rs
