kafka_protocol/messages/
envelope_request.rs

1//! EnvelopeRequest
2//!
3//! See the schema for this message [here](https://github.com/apache/kafka/blob/trunk/clients/src/main/resources/common/message/EnvelopeRequest.json).
4// WARNING: the items of this module are generated and should not be edited directly
5#![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/// Valid versions: 0
21#[non_exhaustive]
22#[derive(Debug, Clone, PartialEq)]
23pub struct EnvelopeRequest {
24    /// The embedded request header and data.
25    ///
26    /// Supported API versions: 0
27    pub request_data: Bytes,
28
29    /// Value of the initial client principal when the request is redirected by a broker.
30    ///
31    /// Supported API versions: 0
32    pub request_principal: Option<Bytes>,
33
34    /// The original client's address in bytes.
35    ///
36    /// Supported API versions: 0
37    pub client_host_address: Bytes,
38
39    /// Other tagged fields
40    pub unknown_tagged_fields: BTreeMap<i32, Bytes>,
41}
42
43impl EnvelopeRequest {
44    /// Sets `request_data` to the passed value.
45    ///
46    /// The embedded request header and data.
47    ///
48    /// Supported API versions: 0
49    pub fn with_request_data(mut self, value: Bytes) -> Self {
50        self.request_data = value;
51        self
52    }
53    /// Sets `request_principal` to the passed value.
54    ///
55    /// Value of the initial client principal when the request is redirected by a broker.
56    ///
57    /// Supported API versions: 0
58    pub fn with_request_principal(mut self, value: Option<Bytes>) -> Self {
59        self.request_principal = value;
60        self
61    }
62    /// Sets `client_host_address` to the passed value.
63    ///
64    /// The original client's address in bytes.
65    ///
66    /// Supported API versions: 0
67    pub fn with_client_host_address(mut self, value: Bytes) -> Self {
68        self.client_host_address = value;
69        self
70    }
71    /// Sets unknown_tagged_fields to the passed value.
72    pub fn with_unknown_tagged_fields(mut self, value: BTreeMap<i32, Bytes>) -> Self {
73        self.unknown_tagged_fields = value;
74        self
75    }
76    /// Inserts an entry into unknown_tagged_fields.
77    pub fn with_unknown_tagged_field(mut self, key: i32, value: Bytes) -> Self {
78        self.unknown_tagged_fields.insert(key, value);
79        self
80    }
81}
82
83#[cfg(feature = "client")]
84impl Encodable for EnvelopeRequest {
85    fn encode<B: ByteBufMut>(&self, buf: &mut B, version: i16) -> Result<()> {
86        if version != 0 {
87            bail!("specified version not supported by this message type");
88        }
89        types::CompactBytes.encode(buf, &self.request_data)?;
90        types::CompactBytes.encode(buf, &self.request_principal)?;
91        types::CompactBytes.encode(buf, &self.client_host_address)?;
92        let num_tagged_fields = self.unknown_tagged_fields.len();
93        if num_tagged_fields > std::u32::MAX as usize {
94            bail!(
95                "Too many tagged fields to encode ({} fields)",
96                num_tagged_fields
97            );
98        }
99        types::UnsignedVarInt.encode(buf, num_tagged_fields as u32)?;
100
101        write_unknown_tagged_fields(buf, 0.., &self.unknown_tagged_fields)?;
102        Ok(())
103    }
104    fn compute_size(&self, version: i16) -> Result<usize> {
105        let mut total_size = 0;
106        total_size += types::CompactBytes.compute_size(&self.request_data)?;
107        total_size += types::CompactBytes.compute_size(&self.request_principal)?;
108        total_size += types::CompactBytes.compute_size(&self.client_host_address)?;
109        let num_tagged_fields = self.unknown_tagged_fields.len();
110        if num_tagged_fields > std::u32::MAX as usize {
111            bail!(
112                "Too many tagged fields to encode ({} fields)",
113                num_tagged_fields
114            );
115        }
116        total_size += types::UnsignedVarInt.compute_size(num_tagged_fields as u32)?;
117
118        total_size += compute_unknown_tagged_fields_size(&self.unknown_tagged_fields)?;
119        Ok(total_size)
120    }
121}
122
123#[cfg(feature = "broker")]
124impl Decodable for EnvelopeRequest {
125    fn decode<B: ByteBuf>(buf: &mut B, version: i16) -> Result<Self> {
126        if version != 0 {
127            bail!("specified version not supported by this message type");
128        }
129        let request_data = types::CompactBytes.decode(buf)?;
130        let request_principal = types::CompactBytes.decode(buf)?;
131        let client_host_address = types::CompactBytes.decode(buf)?;
132        let mut unknown_tagged_fields = BTreeMap::new();
133        let num_tagged_fields = types::UnsignedVarInt.decode(buf)?;
134        for _ in 0..num_tagged_fields {
135            let tag: u32 = types::UnsignedVarInt.decode(buf)?;
136            let size: u32 = types::UnsignedVarInt.decode(buf)?;
137            let unknown_value = buf.try_get_bytes(size as usize)?;
138            unknown_tagged_fields.insert(tag as i32, unknown_value);
139        }
140        Ok(Self {
141            request_data,
142            request_principal,
143            client_host_address,
144            unknown_tagged_fields,
145        })
146    }
147}
148
149impl Default for EnvelopeRequest {
150    fn default() -> Self {
151        Self {
152            request_data: Default::default(),
153            request_principal: Some(Default::default()),
154            client_host_address: Default::default(),
155            unknown_tagged_fields: BTreeMap::new(),
156        }
157    }
158}
159
160impl Message for EnvelopeRequest {
161    const VERSIONS: VersionRange = VersionRange { min: 0, max: 0 };
162    const DEPRECATED_VERSIONS: Option<VersionRange> = None;
163}
164
165impl HeaderVersion for EnvelopeRequest {
166    fn header_version(version: i16) -> i16 {
167        2
168    }
169}
kafka_protocol/messages/envelope_request.rs

kafka_protocol/messages/
envelope_request.rs
