#![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 InitProducerIdRequestData {
pub transactional_id: Option<KafkaString>,
pub transaction_timeout_ms: i32,
pub producer_id: i64,
pub producer_epoch: i16,
pub enable2_pc: bool,
pub keep_prepared_txn: bool,
pub _unknown_tagged_fields: Vec<RawTaggedField>,
}
impl Default for InitProducerIdRequestData {
fn default() -> Self {
Self {
transactional_id: None,
transaction_timeout_ms: 0_i32,
producer_id: -1i64,
producer_epoch: -1i16,
enable2_pc: false,
keep_prepared_txn: false,
_unknown_tagged_fields: Vec::new(),
}
}
}
impl InitProducerIdRequestData {
pub fn with_transactional_id(mut self, value: Option<KafkaString>) -> Self {
self.transactional_id = value;
self
}
pub fn with_transaction_timeout_ms(mut self, value: i32) -> Self {
self.transaction_timeout_ms = value;
self
}
pub fn with_producer_id(mut self, value: i64) -> Self {
self.producer_id = value;
self
}
pub fn with_producer_epoch(mut self, value: i16) -> Self {
self.producer_epoch = value;
self
}
pub fn with_enable2_pc(mut self, value: bool) -> Self {
self.enable2_pc = value;
self
}
pub fn with_keep_prepared_txn(mut self, value: bool) -> Self {
self.keep_prepared_txn = value;
self
}
pub fn read(buf: &mut Bytes, version: i16) -> Result<Self> {
if version < 0 || version > 6 {
return Err(UnsupportedVersion::new(22, version).into());
}
let transactional_id;
let transaction_timeout_ms;
let mut producer_id = -1i64;
let mut producer_epoch = -1i16;
let mut enable2_pc = false;
let mut keep_prepared_txn = false;
let mut _unknown_tagged_fields: Vec<RawTaggedField> = Vec::new();
if version >= 2 {
transactional_id = read_compact_nullable_string(buf)?;
} else {
transactional_id = read_nullable_string(buf)?;
}
transaction_timeout_ms = read_i32(buf)?;
if version >= 3 {
producer_id = read_i64(buf)?;
}
if version >= 3 {
producer_epoch = read_i16(buf)?;
}
if version >= 6 {
enable2_pc = read_bool(buf)?;
}
if version >= 6 {
keep_prepared_txn = read_bool(buf)?;
}
if version >= 2 {
let tagged_fields = read_tagged_fields(buf)?;
for field in &tagged_fields {
match field.tag {
_ => {
_unknown_tagged_fields.push(field.clone());
},
}
}
}
Ok(Self {
transactional_id,
transaction_timeout_ms,
producer_id,
producer_epoch,
enable2_pc,
keep_prepared_txn,
_unknown_tagged_fields,
})
}
pub fn write(&self, buf: &mut BytesMut, version: i16) -> Result<()> {
if version < 0 || version > 6 {
return Err(UnsupportedVersion::new(22, version).into());
}
if version >= 2 {
write_compact_nullable_string(buf, self.transactional_id.as_ref())?;
} else {
write_nullable_string(buf, self.transactional_id.as_ref())?;
}
write_i32(buf, self.transaction_timeout_ms);
if version >= 3 {
write_i64(buf, self.producer_id);
} else if self.producer_id != -1i64 {
return Err(UnsupportedFieldVersion::new(22, "producer_id", version).into());
}
if version >= 3 {
write_i16(buf, self.producer_epoch);
} else if self.producer_epoch != -1i16 {
return Err(UnsupportedFieldVersion::new(22, "producer_epoch", version).into());
}
if version >= 6 {
write_bool(buf, self.enable2_pc);
} else if self.enable2_pc != false {
return Err(UnsupportedFieldVersion::new(22, "enable2_pc", version).into());
}
if version >= 6 {
write_bool(buf, self.keep_prepared_txn);
} else if self.keep_prepared_txn != false {
return Err(UnsupportedFieldVersion::new(22, "keep_prepared_txn", version).into());
}
if version >= 2 {
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 > 6 {
return Err(UnsupportedVersion::new(22, version).into());
}
let mut len: usize = 0;
if version >= 2 {
len += compact_nullable_string_len(self.transactional_id.as_ref())?;
} else {
len += nullable_string_len(self.transactional_id.as_ref())?;
}
len += 4;
if version >= 3 {
len += 8;
} else if self.producer_id != -1i64 {
return Err(UnsupportedFieldVersion::new(22, "producer_id", version).into());
}
if version >= 3 {
len += 2;
} else if self.producer_epoch != -1i16 {
return Err(UnsupportedFieldVersion::new(22, "producer_epoch", version).into());
}
if version >= 6 {
len += 1;
} else if self.enable2_pc != false {
return Err(UnsupportedFieldVersion::new(22, "enable2_pc", version).into());
}
if version >= 6 {
len += 1;
} else if self.keep_prepared_txn != false {
return Err(UnsupportedFieldVersion::new(22, "keep_prepared_txn", version).into());
}
if version >= 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)
}
}