#![allow(
missing_docs,
clippy::all,
clippy::pedantic,
clippy::nursery,
clippy::arithmetic_side_effects,
reason = "Generated protocol modules mirror Kafka's schema shape and intentionally trade \
hand-written lint style for reproducible wire-code output."
)]
use bytes::{Bytes, BytesMut};
use crate::*;
#[derive(Debug, Clone, PartialEq)]
pub struct BrokerRegistrationRequestData {
pub broker_id: i32,
pub cluster_id: KafkaString,
pub incarnation_id: KafkaUuid,
pub listeners: Vec<Listener>,
pub features: Vec<Feature>,
pub rack: Option<KafkaString>,
pub is_migrating_zk_broker: bool,
pub log_dirs: Vec<KafkaUuid>,
pub previous_broker_epoch: i64,
pub _unknown_tagged_fields: Vec<RawTaggedField>,
}
impl Default for BrokerRegistrationRequestData {
fn default() -> Self {
Self {
broker_id: 0_i32,
cluster_id: KafkaString::default(),
incarnation_id: KafkaUuid::ZERO,
listeners: Vec::new(),
features: Vec::new(),
rack: None,
is_migrating_zk_broker: false,
log_dirs: Vec::new(),
previous_broker_epoch: -1i64,
_unknown_tagged_fields: Vec::new(),
}
}
}
impl BrokerRegistrationRequestData {
pub fn with_broker_id(mut self, value: i32) -> Self {
self.broker_id = value;
self
}
pub fn with_cluster_id(mut self, value: KafkaString) -> Self {
self.cluster_id = value;
self
}
pub fn with_incarnation_id(mut self, value: KafkaUuid) -> Self {
self.incarnation_id = value;
self
}
pub fn with_listeners(mut self, value: Vec<Listener>) -> Self {
self.listeners = value;
self
}
pub fn with_features(mut self, value: Vec<Feature>) -> Self {
self.features = value;
self
}
pub fn with_rack(mut self, value: Option<KafkaString>) -> Self {
self.rack = value;
self
}
pub fn with_is_migrating_zk_broker(mut self, value: bool) -> Self {
self.is_migrating_zk_broker = value;
self
}
pub fn with_log_dirs(mut self, value: Vec<KafkaUuid>) -> Self {
self.log_dirs = value;
self
}
pub fn with_previous_broker_epoch(mut self, value: i64) -> Self {
self.previous_broker_epoch = value;
self
}
pub fn read(buf: &mut Bytes, version: i16) -> Result<Self> {
if version < 0 || version > 4 {
return Err(UnsupportedVersion::new(62, version).into());
}
let broker_id;
let cluster_id;
let incarnation_id;
let listeners;
let features;
let rack;
let mut is_migrating_zk_broker = false;
let mut log_dirs = Vec::new();
let mut previous_broker_epoch = -1i64;
let mut _unknown_tagged_fields: Vec<RawTaggedField> = Vec::new();
broker_id = read_i32(buf)?;
cluster_id = read_compact_string(buf)?;
incarnation_id = read_uuid(buf)?;
listeners = {
let len = read_compact_array_length(buf)?;
let mut arr = Vec::with_capacity(array_read_capacity(len, (buf).len()));
for _ in 0..len {
arr.push(Listener::read(buf, version)?);
}
arr
};
features = {
let len = read_compact_array_length(buf)?;
let mut arr = Vec::with_capacity(array_read_capacity(len, (buf).len()));
for _ in 0..len {
arr.push(Feature::read(buf, version)?);
}
arr
};
rack = read_compact_nullable_string(buf)?;
if version >= 1 {
is_migrating_zk_broker = read_bool(buf)?;
}
if version >= 2 {
log_dirs = {
let len = read_compact_array_length(buf)?;
let mut arr = Vec::with_capacity(array_read_capacity(len, (buf).len()));
for _ in 0..len {
arr.push(read_uuid(buf)?);
}
arr
};
}
if version >= 3 {
previous_broker_epoch = read_i64(buf)?;
}
let tagged_fields = read_tagged_fields(buf)?;
for field in &tagged_fields {
match field.tag {
_ => {
_unknown_tagged_fields.push(field.clone());
},
}
}
Ok(Self {
broker_id,
cluster_id,
incarnation_id,
listeners,
features,
rack,
is_migrating_zk_broker,
log_dirs,
previous_broker_epoch,
_unknown_tagged_fields,
})
}
pub fn write(&self, buf: &mut BytesMut, version: i16) -> Result<()> {
if version < 0 || version > 4 {
return Err(UnsupportedVersion::new(62, version).into());
}
write_i32(buf, self.broker_id);
write_compact_string(buf, &self.cluster_id)?;
write_uuid(buf, &self.incarnation_id);
write_compact_array_length(buf, self.listeners.len() as i32);
for el in &self.listeners {
el.write(buf, version)?;
}
write_compact_array_length(buf, self.features.len() as i32);
for el in &self.features {
el.write(buf, version)?;
}
write_compact_nullable_string(buf, self.rack.as_ref())?;
if version >= 1 {
write_bool(buf, self.is_migrating_zk_broker);
} else if self.is_migrating_zk_broker != false {
return Err(UnsupportedFieldVersion::new(62, "is_migrating_zk_broker", version).into());
}
if version >= 2 {
write_compact_array_length(buf, self.log_dirs.len() as i32);
for el in &self.log_dirs {
write_uuid(buf, el);
}
} else if self.log_dirs != Vec::new() {
return Err(UnsupportedFieldVersion::new(62, "log_dirs", version).into());
}
if version >= 3 {
write_i64(buf, self.previous_broker_epoch);
} else if self.previous_broker_epoch != -1i64 {
return Err(UnsupportedFieldVersion::new(62, "previous_broker_epoch", version).into());
}
let mut all_tags: Vec<RawTaggedField> = self._unknown_tagged_fields.clone();
all_tags.sort_by_key(|f| f.tag);
write_tagged_fields(buf, &all_tags)?;
Ok(())
}
pub fn encoded_len(&self, version: i16) -> Result<usize> {
if version < 0 || version > 4 {
return Err(UnsupportedVersion::new(62, version).into());
}
let mut len: usize = 0;
len += 4;
len += compact_string_len(&self.cluster_id)?;
len += 16;
len += compact_array_length_len(self.listeners.len() as i32);
for el in &self.listeners {
len += el.encoded_len(version)?;
}
len += compact_array_length_len(self.features.len() as i32);
for el in &self.features {
len += el.encoded_len(version)?;
}
len += compact_nullable_string_len(self.rack.as_ref())?;
if version >= 1 {
len += 1;
} else if self.is_migrating_zk_broker != false {
return Err(UnsupportedFieldVersion::new(62, "is_migrating_zk_broker", version).into());
}
if version >= 2 {
len += compact_array_length_len(self.log_dirs.len() as i32);
len += self.log_dirs.len() * 16usize;
} else if self.log_dirs != Vec::new() {
return Err(UnsupportedFieldVersion::new(62, "log_dirs", version).into());
}
if version >= 3 {
len += 8;
} else if self.previous_broker_epoch != -1i64 {
return Err(UnsupportedFieldVersion::new(62, "previous_broker_epoch", version).into());
}
let mut all_tags: Vec<RawTaggedField> = self._unknown_tagged_fields.clone();
all_tags.sort_by_key(|f| f.tag);
len += tagged_fields_len(&all_tags)?;
Ok(len)
}
}
#[derive(Debug, Clone, PartialEq)]
pub struct Listener {
pub name: KafkaString,
pub host: KafkaString,
pub port: u16,
pub security_protocol: i16,
pub _unknown_tagged_fields: Vec<RawTaggedField>,
}
impl Default for Listener {
fn default() -> Self {
Self {
name: KafkaString::default(),
host: KafkaString::default(),
port: 0_u16,
security_protocol: 0_i16,
_unknown_tagged_fields: Vec::new(),
}
}
}
impl Listener {
pub fn with_name(mut self, value: KafkaString) -> Self {
self.name = value;
self
}
pub fn with_host(mut self, value: KafkaString) -> Self {
self.host = value;
self
}
pub fn with_port(mut self, value: u16) -> Self {
self.port = value;
self
}
pub fn with_security_protocol(mut self, value: i16) -> Self {
self.security_protocol = value;
self
}
pub fn read(buf: &mut Bytes, _version: i16) -> Result<Self> {
let name;
let host;
let port;
let security_protocol;
let mut _unknown_tagged_fields: Vec<RawTaggedField> = Vec::new();
name = read_compact_string(buf)?;
host = read_compact_string(buf)?;
port = read_u16(buf)?;
security_protocol = read_i16(buf)?;
let tagged_fields = read_tagged_fields(buf)?;
for field in &tagged_fields {
match field.tag {
_ => {
_unknown_tagged_fields.push(field.clone());
},
}
}
Ok(Self {
name,
host,
port,
security_protocol,
_unknown_tagged_fields,
})
}
pub fn write(&self, buf: &mut BytesMut, _version: i16) -> Result<()> {
write_compact_string(buf, &self.name)?;
write_compact_string(buf, &self.host)?;
write_u16(buf, self.port);
write_i16(buf, self.security_protocol);
let mut all_tags: Vec<RawTaggedField> = self._unknown_tagged_fields.clone();
all_tags.sort_by_key(|f| f.tag);
write_tagged_fields(buf, &all_tags)?;
Ok(())
}
pub fn encoded_len(&self, _version: i16) -> Result<usize> {
let mut len: usize = 0;
len += compact_string_len(&self.name)?;
len += compact_string_len(&self.host)?;
len += 2;
len += 2;
let mut all_tags: Vec<RawTaggedField> = self._unknown_tagged_fields.clone();
all_tags.sort_by_key(|f| f.tag);
len += tagged_fields_len(&all_tags)?;
Ok(len)
}
}
#[derive(Debug, Clone, PartialEq)]
pub struct Feature {
pub name: KafkaString,
pub min_supported_version: i16,
pub max_supported_version: i16,
pub _unknown_tagged_fields: Vec<RawTaggedField>,
}
impl Default for Feature {
fn default() -> Self {
Self {
name: KafkaString::default(),
min_supported_version: 0_i16,
max_supported_version: 0_i16,
_unknown_tagged_fields: Vec::new(),
}
}
}
impl Feature {
pub fn with_name(mut self, value: KafkaString) -> Self {
self.name = value;
self
}
pub fn with_min_supported_version(mut self, value: i16) -> Self {
self.min_supported_version = value;
self
}
pub fn with_max_supported_version(mut self, value: i16) -> Self {
self.max_supported_version = value;
self
}
pub fn read(buf: &mut Bytes, _version: i16) -> Result<Self> {
let name;
let min_supported_version;
let max_supported_version;
let mut _unknown_tagged_fields: Vec<RawTaggedField> = Vec::new();
name = read_compact_string(buf)?;
min_supported_version = read_i16(buf)?;
max_supported_version = read_i16(buf)?;
let tagged_fields = read_tagged_fields(buf)?;
for field in &tagged_fields {
match field.tag {
_ => {
_unknown_tagged_fields.push(field.clone());
},
}
}
Ok(Self {
name,
min_supported_version,
max_supported_version,
_unknown_tagged_fields,
})
}
pub fn write(&self, buf: &mut BytesMut, _version: i16) -> Result<()> {
write_compact_string(buf, &self.name)?;
write_i16(buf, self.min_supported_version);
write_i16(buf, self.max_supported_version);
let mut all_tags: Vec<RawTaggedField> = self._unknown_tagged_fields.clone();
all_tags.sort_by_key(|f| f.tag);
write_tagged_fields(buf, &all_tags)?;
Ok(())
}
pub fn encoded_len(&self, _version: i16) -> Result<usize> {
let mut len: usize = 0;
len += compact_string_len(&self.name)?;
len += 2;
len += 2;
let mut all_tags: Vec<RawTaggedField> = self._unknown_tagged_fields.clone();
all_tags.sort_by_key(|f| f.tag);
len += tagged_fields_len(&all_tags)?;
Ok(len)
}
}