crabka_protocol/opt/rustwide/workdir/generated/
EnvelopeRequest.borrowed.rs1use crate::primitives::string_bytes::{
4 put_bytes, put_compact_bytes, put_compact_nullable_bytes, put_nullable_bytes,
5};
6use crate::primitives::string_bytes_borrowed::{
7 get_bytes_borrowed, get_compact_bytes_borrowed, get_compact_nullable_bytes_borrowed,
8 get_nullable_bytes_borrowed,
9};
10use crate::tagged_fields::{WriteTaggedFields, read_tagged_fields, tagged_fields_len};
11use crate::{DecodeBorrow, Encode, ProtocolError, UnknownTaggedFields};
12use bytes::{BufMut, Bytes};
13
14pub const API_KEY: i16 = 58;
15pub const MIN_VERSION: i16 = 0;
16pub const MAX_VERSION: i16 = 0;
17pub const FLEXIBLE_MIN: i16 = 0;
18
19#[inline]
20fn is_flexible(version: i16) -> bool {
21 version >= FLEXIBLE_MIN
22}
23
24#[derive(Debug, Clone, PartialEq, Eq, Default)]
25pub struct EnvelopeRequest<'a> {
26 pub request_data: &'a [u8],
27 pub request_principal: Option<&'a [u8]>,
28 pub client_host_address: &'a [u8],
29 pub unknown_tagged_fields: UnknownTaggedFields,
30}
31impl EnvelopeRequest<'_> {
32 pub fn to_owned(&self) -> crate::owned::envelope_request::EnvelopeRequest {
33 crate::owned::envelope_request::EnvelopeRequest {
34 request_data: Bytes::copy_from_slice(self.request_data),
35 request_principal: (self.request_principal).map(Bytes::copy_from_slice),
36 client_host_address: Bytes::copy_from_slice(self.client_host_address),
37 unknown_tagged_fields: self.unknown_tagged_fields.clone(),
38 }
39 }
40}
41impl Encode for EnvelopeRequest<'_> {
42 fn encode<B: BufMut>(&self, buf: &mut B, version: i16) -> Result<(), ProtocolError> {
43 if !(MIN_VERSION..=MAX_VERSION).contains(&version) {
44 return Err(ProtocolError::UnsupportedVersion {
45 api_key: API_KEY,
46 version,
47 });
48 }
49 let flex = is_flexible(version);
50 if version >= 0 {
51 if flex {
52 put_compact_bytes(buf, self.request_data);
53 } else {
54 put_bytes(buf, self.request_data);
55 }
56 }
57 if version >= 0 {
58 if flex {
59 put_compact_nullable_bytes(buf, self.request_principal);
60 } else {
61 put_nullable_bytes(buf, self.request_principal);
62 }
63 }
64 if version >= 0 {
65 if flex {
66 put_compact_bytes(buf, self.client_host_address);
67 } else {
68 put_bytes(buf, self.client_host_address);
69 }
70 }
71 if flex {
72 let tagged = WriteTaggedFields::new();
73 tagged.write(buf, &self.unknown_tagged_fields);
74 }
75 Ok(())
76 }
77 fn encoded_len(&self, version: i16) -> usize {
78 let flex = is_flexible(version);
79 let mut n: usize = 0;
80 if version >= 0 {
81 n += if flex {
82 crate::primitives::varint::uvarint_len(
83 u32::try_from((self.request_data).len() + 1).unwrap(),
84 ) + (self.request_data).len()
85 } else {
86 4 + (self.request_data).len()
87 };
88 }
89 if version >= 0 {
90 n += match self.request_principal {
91 None => {
92 if flex {
93 1
94 } else {
95 4
96 }
97 }
98 Some(b) => {
99 if flex {
100 crate::primitives::varint::uvarint_len(u32::try_from(b.len() + 1).unwrap())
101 + b.len()
102 } else {
103 4 + b.len()
104 }
105 }
106 };
107 }
108 if version >= 0 {
109 n += if flex {
110 crate::primitives::varint::uvarint_len(
111 u32::try_from((self.client_host_address).len() + 1).unwrap(),
112 ) + (self.client_host_address).len()
113 } else {
114 4 + (self.client_host_address).len()
115 };
116 }
117 if flex {
118 let known_pairs: Vec<(u32, usize)> = Vec::new();
119 n += tagged_fields_len(&known_pairs, &self.unknown_tagged_fields);
120 }
121 n
122 }
123}
124impl<'de> DecodeBorrow<'de> for EnvelopeRequest<'de> {
125 fn decode_borrow(buf: &mut &'de [u8], version: i16) -> Result<Self, ProtocolError> {
126 if !(MIN_VERSION..=MAX_VERSION).contains(&version) {
127 return Err(ProtocolError::UnsupportedVersion {
128 api_key: API_KEY,
129 version,
130 });
131 }
132 let flex = is_flexible(version);
133 let mut out = Self::default();
134 if version >= 0 {
135 out.request_data = if flex {
136 get_compact_bytes_borrowed(buf)?
137 } else {
138 get_bytes_borrowed(buf)?
139 };
140 }
141 if version >= 0 {
142 out.request_principal = if flex {
143 get_compact_nullable_bytes_borrowed(buf)?
144 } else {
145 get_nullable_bytes_borrowed(buf)?
146 };
147 }
148 if version >= 0 {
149 out.client_host_address = if flex {
150 get_compact_bytes_borrowed(buf)?
151 } else {
152 get_bytes_borrowed(buf)?
153 };
154 }
155 if flex {
156 out.unknown_tagged_fields = read_tagged_fields(buf, |_tag, _payload| Ok(false))?;
157 }
158 Ok(out)
159 }
160}
161#[cfg(test)]
162impl EnvelopeRequest<'_> {
163 #[must_use]
164 pub fn populated(version: i16) -> Self {
165 let mut m = Self::default();
166 if version >= 0 {
167 m.request_data = &b"x"[..];
168 }
169 if version >= 0 {
170 m.request_principal = Some(&b"x"[..]);
171 }
172 if version >= 0 {
173 m.client_host_address = &b"x"[..];
174 }
175 m
176 }
177}