impl ::std::default::Default for PartitionValuesRequest {
fn default() -> Self {
PartitionValuesRequest {
db_name: ::std::default::Default::default(),
tbl_name: ::std::default::Default::default(),
partition_keys: ::std::default::Default::default(),
apply_distinct: Some(true),
filter: ::std::default::Default::default(),
partition_order: ::std::default::Default::default(),
ascending: Some(true),
max_parts: Some(-1i64),
cat_name: ::std::default::Default::default(),
}
}
}
#[derive(PartialOrd, Hash, Eq, Ord, Debug, Clone, PartialEq)]
pub struct PartitionValuesRequest {
pub db_name: ::pilota::FastStr,
pub tbl_name: ::pilota::FastStr,
pub partition_keys: ::std::vec::Vec<FieldSchema>,
pub apply_distinct: ::std::option::Option<bool>,
pub filter: ::std::option::Option<::pilota::FastStr>,
pub partition_order: ::std::option::Option<::std::vec::Vec<FieldSchema>>,
pub ascending: ::std::option::Option<bool>,
pub max_parts: ::std::option::Option<i64>,
pub cat_name: ::std::option::Option<::pilota::FastStr>,
}
impl ::pilota::thrift::Message for PartitionValuesRequest {
fn encode<T: ::pilota::thrift::TOutputProtocol>(
&self,
__protocol: &mut T,
) -> ::std::result::Result<(), ::pilota::thrift::ThriftException> {
#[allow(unused_imports)]
use ::pilota::thrift::TOutputProtocolExt;
let struct_ident = ::pilota::thrift::TStructIdentifier {
name: "PartitionValuesRequest",
};
__protocol.write_struct_begin(&struct_ident)?;
__protocol.write_faststr_field(1, (&self.db_name).clone())?;
__protocol.write_faststr_field(2, (&self.tbl_name).clone())?;
__protocol.write_list_field(
3,
::pilota::thrift::TType::Struct,
&&self.partition_keys,
|__protocol, val| {
__protocol.write_struct(val)?;
::std::result::Result::Ok(())
},
)?;
if let Some(value) = self.apply_distinct.as_ref() {
__protocol.write_bool_field(4, *value)?;
}
if let Some(value) = self.filter.as_ref() {
__protocol.write_faststr_field(5, (value).clone())?;
}
if let Some(value) = self.partition_order.as_ref() {
__protocol.write_list_field(
6,
::pilota::thrift::TType::Struct,
&value,
|__protocol, val| {
__protocol.write_struct(val)?;
::std::result::Result::Ok(())
},
)?;
}
if let Some(value) = self.ascending.as_ref() {
__protocol.write_bool_field(7, *value)?;
}
if let Some(value) = self.max_parts.as_ref() {
__protocol.write_i64_field(8, *value)?;
}
if let Some(value) = self.cat_name.as_ref() {
__protocol.write_faststr_field(9, (value).clone())?;
}
__protocol.write_field_stop()?;
__protocol.write_struct_end()?;
::std::result::Result::Ok(())
}
fn decode<T: ::pilota::thrift::TInputProtocol>(
__protocol: &mut T,
) -> ::std::result::Result<Self, ::pilota::thrift::ThriftException> {
#[allow(unused_imports)]
use ::pilota::{thrift::TLengthProtocolExt, Buf};
let mut var_1 = None;
let mut var_2 = None;
let mut var_3 = None;
let mut var_4 = Some(true);
let mut var_5 = None;
let mut var_6 = None;
let mut var_7 = Some(true);
let mut var_8 = Some(-1i64);
let mut var_9 = None;
let mut __pilota_decoding_field_id = None;
__protocol.read_struct_begin()?;
if let ::std::result::Result::Err(mut err) = (|| {
loop {
let field_ident = __protocol.read_field_begin()?;
if field_ident.field_type == ::pilota::thrift::TType::Stop {
__protocol.field_stop_len();
break;
} else {
__protocol.field_begin_len(field_ident.field_type, field_ident.id);
}
__pilota_decoding_field_id = field_ident.id;
match field_ident.id {
Some(1) if field_ident.field_type == ::pilota::thrift::TType::Binary => {
var_1 = Some(__protocol.read_faststr()?);
}
Some(2) if field_ident.field_type == ::pilota::thrift::TType::Binary => {
var_2 = Some(__protocol.read_faststr()?);
}
Some(3) if field_ident.field_type == ::pilota::thrift::TType::List => {
var_3 = Some(unsafe {
let list_ident = __protocol.read_list_begin()?;
let mut val: ::std::vec::Vec<FieldSchema> =
::std::vec::Vec::with_capacity(list_ident.size);
for i in 0..list_ident.size {
val.as_mut_ptr()
.offset(i as isize)
.write(::pilota::thrift::Message::decode(__protocol)?);
}
val.set_len(list_ident.size);
__protocol.read_list_end()?;
val
});
}
Some(4) if field_ident.field_type == ::pilota::thrift::TType::Bool => {
var_4 = Some(__protocol.read_bool()?);
}
Some(5) if field_ident.field_type == ::pilota::thrift::TType::Binary => {
var_5 = Some(__protocol.read_faststr()?);
}
Some(6) if field_ident.field_type == ::pilota::thrift::TType::List => {
var_6 = Some(unsafe {
let list_ident = __protocol.read_list_begin()?;
let mut val: ::std::vec::Vec<FieldSchema> =
::std::vec::Vec::with_capacity(list_ident.size);
for i in 0..list_ident.size {
val.as_mut_ptr()
.offset(i as isize)
.write(::pilota::thrift::Message::decode(__protocol)?);
}
val.set_len(list_ident.size);
__protocol.read_list_end()?;
val
});
}
Some(7) if field_ident.field_type == ::pilota::thrift::TType::Bool => {
var_7 = Some(__protocol.read_bool()?);
}
Some(8) if field_ident.field_type == ::pilota::thrift::TType::I64 => {
var_8 = Some(__protocol.read_i64()?);
}
Some(9) if field_ident.field_type == ::pilota::thrift::TType::Binary => {
var_9 = Some(__protocol.read_faststr()?);
}
_ => {
__protocol.skip(field_ident.field_type)?;
}
}
__protocol.read_field_end()?;
__protocol.field_end_len();
}
::std::result::Result::Ok::<_, ::pilota::thrift::ThriftException>(())
})() {
if let Some(field_id) = __pilota_decoding_field_id {
err.prepend_msg(&format!(
"decode struct `PartitionValuesRequest` field(#{}) failed, caused by: ",
field_id
));
}
return ::std::result::Result::Err(err);
};
__protocol.read_struct_end()?;
let Some(var_1) = var_1 else {
return ::std::result::Result::Err(::pilota::thrift::new_protocol_exception(
::pilota::thrift::ProtocolExceptionKind::InvalidData,
"field db_name is required".to_string(),
));
};
let Some(var_2) = var_2 else {
return ::std::result::Result::Err(::pilota::thrift::new_protocol_exception(
::pilota::thrift::ProtocolExceptionKind::InvalidData,
"field tbl_name is required".to_string(),
));
};
let Some(var_3) = var_3 else {
return ::std::result::Result::Err(::pilota::thrift::new_protocol_exception(
::pilota::thrift::ProtocolExceptionKind::InvalidData,
"field partition_keys is required".to_string(),
));
};
let data = Self {
db_name: var_1,
tbl_name: var_2,
partition_keys: var_3,
apply_distinct: var_4,
filter: var_5,
partition_order: var_6,
ascending: var_7,
max_parts: var_8,
cat_name: var_9,
};
::std::result::Result::Ok(data)
}
fn decode_async<'a, T: ::pilota::thrift::TAsyncInputProtocol>(
__protocol: &'a mut T,
) -> ::std::pin::Pin<
::std::boxed::Box<
dyn ::std::future::Future<
Output = ::std::result::Result<Self, ::pilota::thrift::ThriftException>,
> + Send
+ 'a,
>,
> {
::std::boxed::Box::pin(async move {
let mut var_1 = None;
let mut var_2 = None;
let mut var_3 = None;
let mut var_4 = Some(true);
let mut var_5 = None;
let mut var_6 = None;
let mut var_7 = Some(true);
let mut var_8 = Some(-1i64);
let mut var_9 = None;
let mut __pilota_decoding_field_id = None;
__protocol.read_struct_begin().await?;
if let ::std::result::Result::Err(mut err) = async {
loop {
let field_ident = __protocol.read_field_begin().await?;
if field_ident.field_type == ::pilota::thrift::TType::Stop {
break;
} else {
}
__pilota_decoding_field_id = field_ident.id;
match field_ident.id {
Some(1) if field_ident.field_type == ::pilota::thrift::TType::Binary => {
var_1 = Some(__protocol.read_faststr().await?);
}
Some(2) if field_ident.field_type == ::pilota::thrift::TType::Binary => {
var_2 = Some(__protocol.read_faststr().await?);
}
Some(3) if field_ident.field_type == ::pilota::thrift::TType::List => {
var_3 = Some({
let list_ident = __protocol.read_list_begin().await?;
let mut val = ::std::vec::Vec::with_capacity(list_ident.size);
for _ in 0..list_ident.size {
val.push(
<FieldSchema as ::pilota::thrift::Message>::decode_async(
__protocol,
)
.await?,
);
}
__protocol.read_list_end().await?;
val
});
}
Some(4) if field_ident.field_type == ::pilota::thrift::TType::Bool => {
var_4 = Some(__protocol.read_bool().await?);
}
Some(5) if field_ident.field_type == ::pilota::thrift::TType::Binary => {
var_5 = Some(__protocol.read_faststr().await?);
}
Some(6) if field_ident.field_type == ::pilota::thrift::TType::List => {
var_6 = Some({
let list_ident = __protocol.read_list_begin().await?;
let mut val = ::std::vec::Vec::with_capacity(list_ident.size);
for _ in 0..list_ident.size {
val.push(
<FieldSchema as ::pilota::thrift::Message>::decode_async(
__protocol,
)
.await?,
);
}
__protocol.read_list_end().await?;
val
});
}
Some(7) if field_ident.field_type == ::pilota::thrift::TType::Bool => {
var_7 = Some(__protocol.read_bool().await?);
}
Some(8) if field_ident.field_type == ::pilota::thrift::TType::I64 => {
var_8 = Some(__protocol.read_i64().await?);
}
Some(9) if field_ident.field_type == ::pilota::thrift::TType::Binary => {
var_9 = Some(__protocol.read_faststr().await?);
}
_ => {
__protocol.skip(field_ident.field_type).await?;
}
}
__protocol.read_field_end().await?;
}
::std::result::Result::Ok::<_, ::pilota::thrift::ThriftException>(())
}
.await
{
if let Some(field_id) = __pilota_decoding_field_id {
err.prepend_msg(&format!(
"decode struct `PartitionValuesRequest` field(#{}) failed, caused by: ",
field_id
));
}
return ::std::result::Result::Err(err);
};
__protocol.read_struct_end().await?;
let Some(var_1) = var_1 else {
return ::std::result::Result::Err(::pilota::thrift::new_protocol_exception(
::pilota::thrift::ProtocolExceptionKind::InvalidData,
"field db_name is required".to_string(),
));
};
let Some(var_2) = var_2 else {
return ::std::result::Result::Err(::pilota::thrift::new_protocol_exception(
::pilota::thrift::ProtocolExceptionKind::InvalidData,
"field tbl_name is required".to_string(),
));
};
let Some(var_3) = var_3 else {
return ::std::result::Result::Err(::pilota::thrift::new_protocol_exception(
::pilota::thrift::ProtocolExceptionKind::InvalidData,
"field partition_keys is required".to_string(),
));
};
let data = Self {
db_name: var_1,
tbl_name: var_2,
partition_keys: var_3,
apply_distinct: var_4,
filter: var_5,
partition_order: var_6,
ascending: var_7,
max_parts: var_8,
cat_name: var_9,
};
::std::result::Result::Ok(data)
})
}
fn size<T: ::pilota::thrift::TLengthProtocol>(&self, __protocol: &mut T) -> usize {
#[allow(unused_imports)]
use ::pilota::thrift::TLengthProtocolExt;
__protocol.struct_begin_len(&::pilota::thrift::TStructIdentifier {
name: "PartitionValuesRequest",
}) + __protocol.faststr_field_len(Some(1), &self.db_name)
+ __protocol.faststr_field_len(Some(2), &self.tbl_name)
+ __protocol.list_field_len(
Some(3),
::pilota::thrift::TType::Struct,
&self.partition_keys,
|__protocol, el| __protocol.struct_len(el),
)
+ self
.apply_distinct
.as_ref()
.map_or(0, |value| __protocol.bool_field_len(Some(4), *value))
+ self
.filter
.as_ref()
.map_or(0, |value| __protocol.faststr_field_len(Some(5), value))
+ self.partition_order.as_ref().map_or(0, |value| {
__protocol.list_field_len(
Some(6),
::pilota::thrift::TType::Struct,
value,
|__protocol, el| __protocol.struct_len(el),
)
})
+ self
.ascending
.as_ref()
.map_or(0, |value| __protocol.bool_field_len(Some(7), *value))
+ self
.max_parts
.as_ref()
.map_or(0, |value| __protocol.i64_field_len(Some(8), *value))
+ self
.cat_name
.as_ref()
.map_or(0, |value| __protocol.faststr_field_len(Some(9), value))
+ __protocol.field_stop_len()
+ __protocol.struct_end_len()
}
}