#![allow(unused)]
use std::borrow::Borrow;
use std::collections::BTreeMap;
use anyhow::bail;
use bytes::Bytes;
use uuid::Uuid;
use crate::protocol::{
buf::{ByteBuf, ByteBufMut},
compute_unknown_tagged_fields_size, types, write_unknown_tagged_fields, Builder, Decodable,
DecodeError, Decoder, Encodable, EncodeError, Encoder, HeaderVersion, MapDecodable,
MapEncodable, Message, StrBytes, VersionRange,
};
#[non_exhaustive]
#[derive(Debug, Clone, PartialEq, derive_builder::Builder)]
#[builder(default)]
pub struct LeaderAndIsrLiveLeader {
pub broker_id: super::BrokerId,
pub host_name: StrBytes,
pub port: i32,
pub unknown_tagged_fields: BTreeMap<i32, Bytes>,
}
impl Builder for LeaderAndIsrLiveLeader {
type Builder = LeaderAndIsrLiveLeaderBuilder;
fn builder() -> Self::Builder {
LeaderAndIsrLiveLeaderBuilder::default()
}
}
impl Encodable for LeaderAndIsrLiveLeader {
fn encode<B: ByteBufMut>(&self, buf: &mut B, version: i16) -> Result<(), EncodeError> {
types::Int32.encode(buf, &self.broker_id)?;
if version >= 4 {
types::CompactString.encode(buf, &self.host_name)?;
} else {
types::String.encode(buf, &self.host_name)?;
}
types::Int32.encode(buf, &self.port)?;
if version >= 4 {
let num_tagged_fields = self.unknown_tagged_fields.len();
if num_tagged_fields > std::u32::MAX as usize {
bail!(
"Too many tagged fields to encode ({} fields)",
num_tagged_fields
);
}
types::UnsignedVarInt.encode(buf, num_tagged_fields as u32)?;
write_unknown_tagged_fields(buf, 0.., &self.unknown_tagged_fields)?;
}
Ok(())
}
fn compute_size(&self, version: i16) -> Result<usize, EncodeError> {
let mut total_size = 0;
total_size += types::Int32.compute_size(&self.broker_id)?;
if version >= 4 {
total_size += types::CompactString.compute_size(&self.host_name)?;
} else {
total_size += types::String.compute_size(&self.host_name)?;
}
total_size += types::Int32.compute_size(&self.port)?;
if version >= 4 {
let num_tagged_fields = self.unknown_tagged_fields.len();
if num_tagged_fields > std::u32::MAX as usize {
bail!(
"Too many tagged fields to encode ({} fields)",
num_tagged_fields
);
}
total_size += types::UnsignedVarInt.compute_size(num_tagged_fields as u32)?;
total_size += compute_unknown_tagged_fields_size(&self.unknown_tagged_fields)?;
}
Ok(total_size)
}
}
impl Decodable for LeaderAndIsrLiveLeader {
fn decode<B: ByteBuf>(buf: &mut B, version: i16) -> Result<Self, DecodeError> {
let broker_id = types::Int32.decode(buf)?;
let host_name = if version >= 4 {
types::CompactString.decode(buf)?
} else {
types::String.decode(buf)?
};
let port = types::Int32.decode(buf)?;
let mut unknown_tagged_fields = BTreeMap::new();
if version >= 4 {
let num_tagged_fields = types::UnsignedVarInt.decode(buf)?;
for _ in 0..num_tagged_fields {
let tag: u32 = types::UnsignedVarInt.decode(buf)?;
let size: u32 = types::UnsignedVarInt.decode(buf)?;
let unknown_value = buf.try_get_bytes(size as usize)?;
unknown_tagged_fields.insert(tag as i32, unknown_value);
}
}
Ok(Self {
broker_id,
host_name,
port,
unknown_tagged_fields,
})
}
}
impl Default for LeaderAndIsrLiveLeader {
fn default() -> Self {
Self {
broker_id: (0).into(),
host_name: Default::default(),
port: 0,
unknown_tagged_fields: BTreeMap::new(),
}
}
}
impl Message for LeaderAndIsrLiveLeader {
const VERSIONS: VersionRange = VersionRange { min: 0, max: 7 };
const DEPRECATED_VERSIONS: Option<VersionRange> = None;
}
#[non_exhaustive]
#[derive(Debug, Clone, PartialEq, derive_builder::Builder)]
#[builder(default)]
pub struct LeaderAndIsrPartitionState {
pub topic_name: super::TopicName,
pub partition_index: i32,
pub controller_epoch: i32,
pub leader: super::BrokerId,
pub leader_epoch: i32,
pub isr: Vec<super::BrokerId>,
pub partition_epoch: i32,
pub replicas: Vec<super::BrokerId>,
pub adding_replicas: Vec<super::BrokerId>,
pub removing_replicas: Vec<super::BrokerId>,
pub is_new: bool,
pub leader_recovery_state: i8,
pub unknown_tagged_fields: BTreeMap<i32, Bytes>,
}
impl Builder for LeaderAndIsrPartitionState {
type Builder = LeaderAndIsrPartitionStateBuilder;
fn builder() -> Self::Builder {
LeaderAndIsrPartitionStateBuilder::default()
}
}
impl Encodable for LeaderAndIsrPartitionState {
fn encode<B: ByteBufMut>(&self, buf: &mut B, version: i16) -> Result<(), EncodeError> {
if version <= 1 {
types::String.encode(buf, &self.topic_name)?;
}
types::Int32.encode(buf, &self.partition_index)?;
types::Int32.encode(buf, &self.controller_epoch)?;
types::Int32.encode(buf, &self.leader)?;
types::Int32.encode(buf, &self.leader_epoch)?;
if version >= 4 {
types::CompactArray(types::Int32).encode(buf, &self.isr)?;
} else {
types::Array(types::Int32).encode(buf, &self.isr)?;
}
types::Int32.encode(buf, &self.partition_epoch)?;
if version >= 4 {
types::CompactArray(types::Int32).encode(buf, &self.replicas)?;
} else {
types::Array(types::Int32).encode(buf, &self.replicas)?;
}
if version >= 3 {
if version >= 4 {
types::CompactArray(types::Int32).encode(buf, &self.adding_replicas)?;
} else {
types::Array(types::Int32).encode(buf, &self.adding_replicas)?;
}
}
if version >= 3 {
if version >= 4 {
types::CompactArray(types::Int32).encode(buf, &self.removing_replicas)?;
} else {
types::Array(types::Int32).encode(buf, &self.removing_replicas)?;
}
}
if version >= 1 {
types::Boolean.encode(buf, &self.is_new)?;
}
if version >= 6 {
types::Int8.encode(buf, &self.leader_recovery_state)?;
} else {
if self.leader_recovery_state != 0 {
bail!("failed to encode");
}
}
if version >= 4 {
let num_tagged_fields = self.unknown_tagged_fields.len();
if num_tagged_fields > std::u32::MAX as usize {
bail!(
"Too many tagged fields to encode ({} fields)",
num_tagged_fields
);
}
types::UnsignedVarInt.encode(buf, num_tagged_fields as u32)?;
write_unknown_tagged_fields(buf, 0.., &self.unknown_tagged_fields)?;
}
Ok(())
}
fn compute_size(&self, version: i16) -> Result<usize, EncodeError> {
let mut total_size = 0;
if version <= 1 {
total_size += types::String.compute_size(&self.topic_name)?;
}
total_size += types::Int32.compute_size(&self.partition_index)?;
total_size += types::Int32.compute_size(&self.controller_epoch)?;
total_size += types::Int32.compute_size(&self.leader)?;
total_size += types::Int32.compute_size(&self.leader_epoch)?;
if version >= 4 {
total_size += types::CompactArray(types::Int32).compute_size(&self.isr)?;
} else {
total_size += types::Array(types::Int32).compute_size(&self.isr)?;
}
total_size += types::Int32.compute_size(&self.partition_epoch)?;
if version >= 4 {
total_size += types::CompactArray(types::Int32).compute_size(&self.replicas)?;
} else {
total_size += types::Array(types::Int32).compute_size(&self.replicas)?;
}
if version >= 3 {
if version >= 4 {
total_size +=
types::CompactArray(types::Int32).compute_size(&self.adding_replicas)?;
} else {
total_size += types::Array(types::Int32).compute_size(&self.adding_replicas)?;
}
}
if version >= 3 {
if version >= 4 {
total_size +=
types::CompactArray(types::Int32).compute_size(&self.removing_replicas)?;
} else {
total_size += types::Array(types::Int32).compute_size(&self.removing_replicas)?;
}
}
if version >= 1 {
total_size += types::Boolean.compute_size(&self.is_new)?;
}
if version >= 6 {
total_size += types::Int8.compute_size(&self.leader_recovery_state)?;
} else {
if self.leader_recovery_state != 0 {
bail!("failed to encode");
}
}
if version >= 4 {
let num_tagged_fields = self.unknown_tagged_fields.len();
if num_tagged_fields > std::u32::MAX as usize {
bail!(
"Too many tagged fields to encode ({} fields)",
num_tagged_fields
);
}
total_size += types::UnsignedVarInt.compute_size(num_tagged_fields as u32)?;
total_size += compute_unknown_tagged_fields_size(&self.unknown_tagged_fields)?;
}
Ok(total_size)
}
}
impl Decodable for LeaderAndIsrPartitionState {
fn decode<B: ByteBuf>(buf: &mut B, version: i16) -> Result<Self, DecodeError> {
let topic_name = if version <= 1 {
types::String.decode(buf)?
} else {
Default::default()
};
let partition_index = types::Int32.decode(buf)?;
let controller_epoch = types::Int32.decode(buf)?;
let leader = types::Int32.decode(buf)?;
let leader_epoch = types::Int32.decode(buf)?;
let isr = if version >= 4 {
types::CompactArray(types::Int32).decode(buf)?
} else {
types::Array(types::Int32).decode(buf)?
};
let partition_epoch = types::Int32.decode(buf)?;
let replicas = if version >= 4 {
types::CompactArray(types::Int32).decode(buf)?
} else {
types::Array(types::Int32).decode(buf)?
};
let adding_replicas = if version >= 3 {
if version >= 4 {
types::CompactArray(types::Int32).decode(buf)?
} else {
types::Array(types::Int32).decode(buf)?
}
} else {
Default::default()
};
let removing_replicas = if version >= 3 {
if version >= 4 {
types::CompactArray(types::Int32).decode(buf)?
} else {
types::Array(types::Int32).decode(buf)?
}
} else {
Default::default()
};
let is_new = if version >= 1 {
types::Boolean.decode(buf)?
} else {
false
};
let leader_recovery_state = if version >= 6 {
types::Int8.decode(buf)?
} else {
0
};
let mut unknown_tagged_fields = BTreeMap::new();
if version >= 4 {
let num_tagged_fields = types::UnsignedVarInt.decode(buf)?;
for _ in 0..num_tagged_fields {
let tag: u32 = types::UnsignedVarInt.decode(buf)?;
let size: u32 = types::UnsignedVarInt.decode(buf)?;
let unknown_value = buf.try_get_bytes(size as usize)?;
unknown_tagged_fields.insert(tag as i32, unknown_value);
}
}
Ok(Self {
topic_name,
partition_index,
controller_epoch,
leader,
leader_epoch,
isr,
partition_epoch,
replicas,
adding_replicas,
removing_replicas,
is_new,
leader_recovery_state,
unknown_tagged_fields,
})
}
}
impl Default for LeaderAndIsrPartitionState {
fn default() -> Self {
Self {
topic_name: Default::default(),
partition_index: 0,
controller_epoch: 0,
leader: (0).into(),
leader_epoch: 0,
isr: Default::default(),
partition_epoch: 0,
replicas: Default::default(),
adding_replicas: Default::default(),
removing_replicas: Default::default(),
is_new: false,
leader_recovery_state: 0,
unknown_tagged_fields: BTreeMap::new(),
}
}
}
impl Message for LeaderAndIsrPartitionState {
const VERSIONS: VersionRange = VersionRange { min: 0, max: 7 };
const DEPRECATED_VERSIONS: Option<VersionRange> = None;
}
#[non_exhaustive]
#[derive(Debug, Clone, PartialEq, derive_builder::Builder)]
#[builder(default)]
pub struct LeaderAndIsrRequest {
pub controller_id: super::BrokerId,
pub is_k_raft_controller: bool,
pub controller_epoch: i32,
pub broker_epoch: i64,
pub _type: i8,
pub ungrouped_partition_states: Vec<LeaderAndIsrPartitionState>,
pub topic_states: Vec<LeaderAndIsrTopicState>,
pub live_leaders: Vec<LeaderAndIsrLiveLeader>,
pub unknown_tagged_fields: BTreeMap<i32, Bytes>,
}
impl Builder for LeaderAndIsrRequest {
type Builder = LeaderAndIsrRequestBuilder;
fn builder() -> Self::Builder {
LeaderAndIsrRequestBuilder::default()
}
}
impl Encodable for LeaderAndIsrRequest {
fn encode<B: ByteBufMut>(&self, buf: &mut B, version: i16) -> Result<(), EncodeError> {
types::Int32.encode(buf, &self.controller_id)?;
if version >= 7 {
types::Boolean.encode(buf, &self.is_k_raft_controller)?;
} else {
if self.is_k_raft_controller {
bail!("failed to encode");
}
}
types::Int32.encode(buf, &self.controller_epoch)?;
if version >= 2 {
types::Int64.encode(buf, &self.broker_epoch)?;
}
if version >= 5 {
types::Int8.encode(buf, &self._type)?;
} else {
if self._type != 0 {
bail!("failed to encode");
}
}
if version <= 1 {
types::Array(types::Struct { version })
.encode(buf, &self.ungrouped_partition_states)?;
} else {
if !self.ungrouped_partition_states.is_empty() {
bail!("failed to encode");
}
}
if version >= 2 {
if version >= 4 {
types::CompactArray(types::Struct { version }).encode(buf, &self.topic_states)?;
} else {
types::Array(types::Struct { version }).encode(buf, &self.topic_states)?;
}
} else {
if !self.topic_states.is_empty() {
bail!("failed to encode");
}
}
if version >= 4 {
types::CompactArray(types::Struct { version }).encode(buf, &self.live_leaders)?;
} else {
types::Array(types::Struct { version }).encode(buf, &self.live_leaders)?;
}
if version >= 4 {
let num_tagged_fields = self.unknown_tagged_fields.len();
if num_tagged_fields > std::u32::MAX as usize {
bail!(
"Too many tagged fields to encode ({} fields)",
num_tagged_fields
);
}
types::UnsignedVarInt.encode(buf, num_tagged_fields as u32)?;
write_unknown_tagged_fields(buf, 0.., &self.unknown_tagged_fields)?;
}
Ok(())
}
fn compute_size(&self, version: i16) -> Result<usize, EncodeError> {
let mut total_size = 0;
total_size += types::Int32.compute_size(&self.controller_id)?;
if version >= 7 {
total_size += types::Boolean.compute_size(&self.is_k_raft_controller)?;
} else {
if self.is_k_raft_controller {
bail!("failed to encode");
}
}
total_size += types::Int32.compute_size(&self.controller_epoch)?;
if version >= 2 {
total_size += types::Int64.compute_size(&self.broker_epoch)?;
}
if version >= 5 {
total_size += types::Int8.compute_size(&self._type)?;
} else {
if self._type != 0 {
bail!("failed to encode");
}
}
if version <= 1 {
total_size += types::Array(types::Struct { version })
.compute_size(&self.ungrouped_partition_states)?;
} else {
if !self.ungrouped_partition_states.is_empty() {
bail!("failed to encode");
}
}
if version >= 2 {
if version >= 4 {
total_size += types::CompactArray(types::Struct { version })
.compute_size(&self.topic_states)?;
} else {
total_size +=
types::Array(types::Struct { version }).compute_size(&self.topic_states)?;
}
} else {
if !self.topic_states.is_empty() {
bail!("failed to encode");
}
}
if version >= 4 {
total_size +=
types::CompactArray(types::Struct { version }).compute_size(&self.live_leaders)?;
} else {
total_size +=
types::Array(types::Struct { version }).compute_size(&self.live_leaders)?;
}
if version >= 4 {
let num_tagged_fields = self.unknown_tagged_fields.len();
if num_tagged_fields > std::u32::MAX as usize {
bail!(
"Too many tagged fields to encode ({} fields)",
num_tagged_fields
);
}
total_size += types::UnsignedVarInt.compute_size(num_tagged_fields as u32)?;
total_size += compute_unknown_tagged_fields_size(&self.unknown_tagged_fields)?;
}
Ok(total_size)
}
}
impl Decodable for LeaderAndIsrRequest {
fn decode<B: ByteBuf>(buf: &mut B, version: i16) -> Result<Self, DecodeError> {
let controller_id = types::Int32.decode(buf)?;
let is_k_raft_controller = if version >= 7 {
types::Boolean.decode(buf)?
} else {
false
};
let controller_epoch = types::Int32.decode(buf)?;
let broker_epoch = if version >= 2 {
types::Int64.decode(buf)?
} else {
-1
};
let _type = if version >= 5 {
types::Int8.decode(buf)?
} else {
0
};
let ungrouped_partition_states = if version <= 1 {
types::Array(types::Struct { version }).decode(buf)?
} else {
Default::default()
};
let topic_states = if version >= 2 {
if version >= 4 {
types::CompactArray(types::Struct { version }).decode(buf)?
} else {
types::Array(types::Struct { version }).decode(buf)?
}
} else {
Default::default()
};
let live_leaders = if version >= 4 {
types::CompactArray(types::Struct { version }).decode(buf)?
} else {
types::Array(types::Struct { version }).decode(buf)?
};
let mut unknown_tagged_fields = BTreeMap::new();
if version >= 4 {
let num_tagged_fields = types::UnsignedVarInt.decode(buf)?;
for _ in 0..num_tagged_fields {
let tag: u32 = types::UnsignedVarInt.decode(buf)?;
let size: u32 = types::UnsignedVarInt.decode(buf)?;
let unknown_value = buf.try_get_bytes(size as usize)?;
unknown_tagged_fields.insert(tag as i32, unknown_value);
}
}
Ok(Self {
controller_id,
is_k_raft_controller,
controller_epoch,
broker_epoch,
_type,
ungrouped_partition_states,
topic_states,
live_leaders,
unknown_tagged_fields,
})
}
}
impl Default for LeaderAndIsrRequest {
fn default() -> Self {
Self {
controller_id: (0).into(),
is_k_raft_controller: false,
controller_epoch: 0,
broker_epoch: -1,
_type: 0,
ungrouped_partition_states: Default::default(),
topic_states: Default::default(),
live_leaders: Default::default(),
unknown_tagged_fields: BTreeMap::new(),
}
}
}
impl Message for LeaderAndIsrRequest {
const VERSIONS: VersionRange = VersionRange { min: 0, max: 7 };
const DEPRECATED_VERSIONS: Option<VersionRange> = None;
}
#[non_exhaustive]
#[derive(Debug, Clone, PartialEq, derive_builder::Builder)]
#[builder(default)]
pub struct LeaderAndIsrTopicState {
pub topic_name: super::TopicName,
pub topic_id: Uuid,
pub partition_states: Vec<LeaderAndIsrPartitionState>,
pub unknown_tagged_fields: BTreeMap<i32, Bytes>,
}
impl Builder for LeaderAndIsrTopicState {
type Builder = LeaderAndIsrTopicStateBuilder;
fn builder() -> Self::Builder {
LeaderAndIsrTopicStateBuilder::default()
}
}
impl Encodable for LeaderAndIsrTopicState {
fn encode<B: ByteBufMut>(&self, buf: &mut B, version: i16) -> Result<(), EncodeError> {
if version >= 2 {
if version >= 4 {
types::CompactString.encode(buf, &self.topic_name)?;
} else {
types::String.encode(buf, &self.topic_name)?;
}
} else {
if !self.topic_name.is_empty() {
bail!("failed to encode");
}
}
if version >= 5 {
types::Uuid.encode(buf, &self.topic_id)?;
}
if version >= 2 {
if version >= 4 {
types::CompactArray(types::Struct { version })
.encode(buf, &self.partition_states)?;
} else {
types::Array(types::Struct { version }).encode(buf, &self.partition_states)?;
}
} else {
if !self.partition_states.is_empty() {
bail!("failed to encode");
}
}
if version >= 4 {
let num_tagged_fields = self.unknown_tagged_fields.len();
if num_tagged_fields > std::u32::MAX as usize {
bail!(
"Too many tagged fields to encode ({} fields)",
num_tagged_fields
);
}
types::UnsignedVarInt.encode(buf, num_tagged_fields as u32)?;
write_unknown_tagged_fields(buf, 0.., &self.unknown_tagged_fields)?;
}
Ok(())
}
fn compute_size(&self, version: i16) -> Result<usize, EncodeError> {
let mut total_size = 0;
if version >= 2 {
if version >= 4 {
total_size += types::CompactString.compute_size(&self.topic_name)?;
} else {
total_size += types::String.compute_size(&self.topic_name)?;
}
} else {
if !self.topic_name.is_empty() {
bail!("failed to encode");
}
}
if version >= 5 {
total_size += types::Uuid.compute_size(&self.topic_id)?;
}
if version >= 2 {
if version >= 4 {
total_size += types::CompactArray(types::Struct { version })
.compute_size(&self.partition_states)?;
} else {
total_size +=
types::Array(types::Struct { version }).compute_size(&self.partition_states)?;
}
} else {
if !self.partition_states.is_empty() {
bail!("failed to encode");
}
}
if version >= 4 {
let num_tagged_fields = self.unknown_tagged_fields.len();
if num_tagged_fields > std::u32::MAX as usize {
bail!(
"Too many tagged fields to encode ({} fields)",
num_tagged_fields
);
}
total_size += types::UnsignedVarInt.compute_size(num_tagged_fields as u32)?;
total_size += compute_unknown_tagged_fields_size(&self.unknown_tagged_fields)?;
}
Ok(total_size)
}
}
impl Decodable for LeaderAndIsrTopicState {
fn decode<B: ByteBuf>(buf: &mut B, version: i16) -> Result<Self, DecodeError> {
let topic_name = if version >= 2 {
if version >= 4 {
types::CompactString.decode(buf)?
} else {
types::String.decode(buf)?
}
} else {
Default::default()
};
let topic_id = if version >= 5 {
types::Uuid.decode(buf)?
} else {
Uuid::nil()
};
let partition_states = if version >= 2 {
if version >= 4 {
types::CompactArray(types::Struct { version }).decode(buf)?
} else {
types::Array(types::Struct { version }).decode(buf)?
}
} else {
Default::default()
};
let mut unknown_tagged_fields = BTreeMap::new();
if version >= 4 {
let num_tagged_fields = types::UnsignedVarInt.decode(buf)?;
for _ in 0..num_tagged_fields {
let tag: u32 = types::UnsignedVarInt.decode(buf)?;
let size: u32 = types::UnsignedVarInt.decode(buf)?;
let unknown_value = buf.try_get_bytes(size as usize)?;
unknown_tagged_fields.insert(tag as i32, unknown_value);
}
}
Ok(Self {
topic_name,
topic_id,
partition_states,
unknown_tagged_fields,
})
}
}
impl Default for LeaderAndIsrTopicState {
fn default() -> Self {
Self {
topic_name: Default::default(),
topic_id: Uuid::nil(),
partition_states: Default::default(),
unknown_tagged_fields: BTreeMap::new(),
}
}
}
impl Message for LeaderAndIsrTopicState {
const VERSIONS: VersionRange = VersionRange { min: 0, max: 7 };
const DEPRECATED_VERSIONS: Option<VersionRange> = None;
}
impl HeaderVersion for LeaderAndIsrRequest {
fn header_version(version: i16) -> i16 {
if version >= 4 {
2
} else {
1
}
}
}