kafka_protocol/messages/
unregister_broker_request.rs1#![allow(unused)]
6
7use std::borrow::Borrow;
8use std::collections::BTreeMap;
9
10use anyhow::{bail, Result};
11use bytes::Bytes;
12use uuid::Uuid;
13
14use crate::protocol::{
15 buf::{ByteBuf, ByteBufMut},
16 compute_unknown_tagged_fields_size, types, write_unknown_tagged_fields, Decodable, Decoder,
17 Encodable, Encoder, HeaderVersion, Message, StrBytes, VersionRange,
18};
19
20#[non_exhaustive]
22#[derive(Debug, Clone, PartialEq)]
23pub struct UnregisterBrokerRequest {
24 pub broker_id: super::BrokerId,
28
29 pub unknown_tagged_fields: BTreeMap<i32, Bytes>,
31}
32
33impl UnregisterBrokerRequest {
34 pub fn with_broker_id(mut self, value: super::BrokerId) -> Self {
40 self.broker_id = value;
41 self
42 }
43 pub fn with_unknown_tagged_fields(mut self, value: BTreeMap<i32, Bytes>) -> Self {
45 self.unknown_tagged_fields = value;
46 self
47 }
48 pub fn with_unknown_tagged_field(mut self, key: i32, value: Bytes) -> Self {
50 self.unknown_tagged_fields.insert(key, value);
51 self
52 }
53}
54
55#[cfg(feature = "client")]
56impl Encodable for UnregisterBrokerRequest {
57 fn encode<B: ByteBufMut>(&self, buf: &mut B, version: i16) -> Result<()> {
58 types::Int32.encode(buf, &self.broker_id)?;
59 let num_tagged_fields = self.unknown_tagged_fields.len();
60 if num_tagged_fields > std::u32::MAX as usize {
61 bail!(
62 "Too many tagged fields to encode ({} fields)",
63 num_tagged_fields
64 );
65 }
66 types::UnsignedVarInt.encode(buf, num_tagged_fields as u32)?;
67
68 write_unknown_tagged_fields(buf, 0.., &self.unknown_tagged_fields)?;
69 Ok(())
70 }
71 fn compute_size(&self, version: i16) -> Result<usize> {
72 let mut total_size = 0;
73 total_size += types::Int32.compute_size(&self.broker_id)?;
74 let num_tagged_fields = self.unknown_tagged_fields.len();
75 if num_tagged_fields > std::u32::MAX as usize {
76 bail!(
77 "Too many tagged fields to encode ({} fields)",
78 num_tagged_fields
79 );
80 }
81 total_size += types::UnsignedVarInt.compute_size(num_tagged_fields as u32)?;
82
83 total_size += compute_unknown_tagged_fields_size(&self.unknown_tagged_fields)?;
84 Ok(total_size)
85 }
86}
87
88#[cfg(feature = "broker")]
89impl Decodable for UnregisterBrokerRequest {
90 fn decode<B: ByteBuf>(buf: &mut B, version: i16) -> Result<Self> {
91 let broker_id = types::Int32.decode(buf)?;
92 let mut unknown_tagged_fields = BTreeMap::new();
93 let num_tagged_fields = types::UnsignedVarInt.decode(buf)?;
94 for _ in 0..num_tagged_fields {
95 let tag: u32 = types::UnsignedVarInt.decode(buf)?;
96 let size: u32 = types::UnsignedVarInt.decode(buf)?;
97 let unknown_value = buf.try_get_bytes(size as usize)?;
98 unknown_tagged_fields.insert(tag as i32, unknown_value);
99 }
100 Ok(Self {
101 broker_id,
102 unknown_tagged_fields,
103 })
104 }
105}
106
107impl Default for UnregisterBrokerRequest {
108 fn default() -> Self {
109 Self {
110 broker_id: (0).into(),
111 unknown_tagged_fields: BTreeMap::new(),
112 }
113 }
114}
115
116impl Message for UnregisterBrokerRequest {
117 const VERSIONS: VersionRange = VersionRange { min: 0, max: 0 };
118 const DEPRECATED_VERSIONS: Option<VersionRange> = None;
119}
120
121impl HeaderVersion for UnregisterBrokerRequest {
122 fn header_version(version: i16) -> i16 {
123 2
124 }
125}