kafka_protocol/messages/
describe_producers_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 DescribeProducersRequest {
24 pub topics: Vec<TopicRequest>,
28
29 pub unknown_tagged_fields: BTreeMap<i32, Bytes>,
31}
32
33impl DescribeProducersRequest {
34 pub fn with_topics(mut self, value: Vec<TopicRequest>) -> Self {
40 self.topics = 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 DescribeProducersRequest {
57 fn encode<B: ByteBufMut>(&self, buf: &mut B, version: i16) -> Result<()> {
58 types::CompactArray(types::Struct { version }).encode(buf, &self.topics)?;
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::CompactArray(types::Struct { version }).compute_size(&self.topics)?;
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 DescribeProducersRequest {
90 fn decode<B: ByteBuf>(buf: &mut B, version: i16) -> Result<Self> {
91 let topics = types::CompactArray(types::Struct { version }).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 topics,
102 unknown_tagged_fields,
103 })
104 }
105}
106
107impl Default for DescribeProducersRequest {
108 fn default() -> Self {
109 Self {
110 topics: Default::default(),
111 unknown_tagged_fields: BTreeMap::new(),
112 }
113 }
114}
115
116impl Message for DescribeProducersRequest {
117 const VERSIONS: VersionRange = VersionRange { min: 0, max: 0 };
118 const DEPRECATED_VERSIONS: Option<VersionRange> = None;
119}
120
121#[non_exhaustive]
123#[derive(Debug, Clone, PartialEq)]
124pub struct TopicRequest {
125 pub name: super::TopicName,
129
130 pub partition_indexes: Vec<i32>,
134
135 pub unknown_tagged_fields: BTreeMap<i32, Bytes>,
137}
138
139impl TopicRequest {
140 pub fn with_name(mut self, value: super::TopicName) -> Self {
146 self.name = value;
147 self
148 }
149 pub fn with_partition_indexes(mut self, value: Vec<i32>) -> Self {
155 self.partition_indexes = value;
156 self
157 }
158 pub fn with_unknown_tagged_fields(mut self, value: BTreeMap<i32, Bytes>) -> Self {
160 self.unknown_tagged_fields = value;
161 self
162 }
163 pub fn with_unknown_tagged_field(mut self, key: i32, value: Bytes) -> Self {
165 self.unknown_tagged_fields.insert(key, value);
166 self
167 }
168}
169
170#[cfg(feature = "client")]
171impl Encodable for TopicRequest {
172 fn encode<B: ByteBufMut>(&self, buf: &mut B, version: i16) -> Result<()> {
173 types::CompactString.encode(buf, &self.name)?;
174 types::CompactArray(types::Int32).encode(buf, &self.partition_indexes)?;
175 let num_tagged_fields = self.unknown_tagged_fields.len();
176 if num_tagged_fields > std::u32::MAX as usize {
177 bail!(
178 "Too many tagged fields to encode ({} fields)",
179 num_tagged_fields
180 );
181 }
182 types::UnsignedVarInt.encode(buf, num_tagged_fields as u32)?;
183
184 write_unknown_tagged_fields(buf, 0.., &self.unknown_tagged_fields)?;
185 Ok(())
186 }
187 fn compute_size(&self, version: i16) -> Result<usize> {
188 let mut total_size = 0;
189 total_size += types::CompactString.compute_size(&self.name)?;
190 total_size += types::CompactArray(types::Int32).compute_size(&self.partition_indexes)?;
191 let num_tagged_fields = self.unknown_tagged_fields.len();
192 if num_tagged_fields > std::u32::MAX as usize {
193 bail!(
194 "Too many tagged fields to encode ({} fields)",
195 num_tagged_fields
196 );
197 }
198 total_size += types::UnsignedVarInt.compute_size(num_tagged_fields as u32)?;
199
200 total_size += compute_unknown_tagged_fields_size(&self.unknown_tagged_fields)?;
201 Ok(total_size)
202 }
203}
204
205#[cfg(feature = "broker")]
206impl Decodable for TopicRequest {
207 fn decode<B: ByteBuf>(buf: &mut B, version: i16) -> Result<Self> {
208 let name = types::CompactString.decode(buf)?;
209 let partition_indexes = types::CompactArray(types::Int32).decode(buf)?;
210 let mut unknown_tagged_fields = BTreeMap::new();
211 let num_tagged_fields = types::UnsignedVarInt.decode(buf)?;
212 for _ in 0..num_tagged_fields {
213 let tag: u32 = types::UnsignedVarInt.decode(buf)?;
214 let size: u32 = types::UnsignedVarInt.decode(buf)?;
215 let unknown_value = buf.try_get_bytes(size as usize)?;
216 unknown_tagged_fields.insert(tag as i32, unknown_value);
217 }
218 Ok(Self {
219 name,
220 partition_indexes,
221 unknown_tagged_fields,
222 })
223 }
224}
225
226impl Default for TopicRequest {
227 fn default() -> Self {
228 Self {
229 name: Default::default(),
230 partition_indexes: Default::default(),
231 unknown_tagged_fields: BTreeMap::new(),
232 }
233 }
234}
235
236impl Message for TopicRequest {
237 const VERSIONS: VersionRange = VersionRange { min: 0, max: 0 };
238 const DEPRECATED_VERSIONS: Option<VersionRange> = None;
239}
240
241impl HeaderVersion for DescribeProducersRequest {
242 fn header_version(version: i16) -> i16 {
243 2
244 }
245}