#[non_exhaustive]
#[derive(Debug)]
pub struct ResponseStreamUnmarshaller;
impl ResponseStreamUnmarshaller {
pub fn new() -> Self {
ResponseStreamUnmarshaller
}
}
impl ::aws_smithy_eventstream::frame::UnmarshallMessage for ResponseStreamUnmarshaller {
type Output = crate::types::ResponseStream;
type Error = crate::types::error::ResponseStreamError;
fn unmarshall(
&self,
message: &::aws_smithy_types::event_stream::Message,
) -> std::result::Result<::aws_smithy_eventstream::frame::UnmarshalledMessage<Self::Output, Self::Error>, ::aws_smithy_eventstream::error::Error>
{
let response_headers = ::aws_smithy_eventstream::smithy::parse_response_headers(message)?;
match response_headers.message_type.as_str() {
"event" => match response_headers.smithy_type.as_str() {
"PayloadPart" => {
let mut builder = crate::types::builders::PayloadPartBuilder::default();
let content_type = response_headers.content_type().unwrap_or_default();
if content_type != "application/octet-stream" {
return Err(::aws_smithy_eventstream::error::Error::unmarshalling(format!(
"expected :content-type to be 'application/octet-stream', but was '{content_type}'"
)));
}
builder = builder.set_bytes(Some(::aws_smithy_types::Blob::new(message.payload().as_ref())));
Ok(::aws_smithy_eventstream::frame::UnmarshalledMessage::Event(
crate::types::ResponseStream::PayloadPart(builder.build()),
))
}
_unknown_variant => Ok(::aws_smithy_eventstream::frame::UnmarshalledMessage::Event(
crate::types::ResponseStream::Unknown,
)),
},
"exception" => {
let generic = match crate::protocol_serde::parse_event_stream_error_metadata(message.payload()) {
Ok(builder) => builder.build(),
Err(err) => {
return Ok(::aws_smithy_eventstream::frame::UnmarshalledMessage::Error(
crate::types::error::ResponseStreamError::unhandled(err),
))
}
};
match response_headers.smithy_type.as_str() {
"ModelStreamError" => {
let mut builder = crate::types::error::builders::ModelStreamErrorBuilder::default();
builder = crate::protocol_serde::shape_model_stream_error::de_model_stream_error_json_err(&message.payload()[..], builder)
.map_err(|err| {
::aws_smithy_eventstream::error::Error::unmarshalling(format!("failed to unmarshall ModelStreamError: {err}"))
})?;
builder.set_meta(Some(generic));
return Ok(::aws_smithy_eventstream::frame::UnmarshalledMessage::Error(
crate::types::error::ResponseStreamError::ModelStreamError(builder.build()),
));
}
"InternalStreamFailure" => {
let mut builder = crate::types::error::builders::InternalStreamFailureBuilder::default();
builder = crate::protocol_serde::shape_internal_stream_failure::de_internal_stream_failure_json_err(
&message.payload()[..],
builder,
)
.map_err(|err| {
::aws_smithy_eventstream::error::Error::unmarshalling(format!("failed to unmarshall InternalStreamFailure: {err}"))
})?;
builder.set_meta(Some(generic));
return Ok(::aws_smithy_eventstream::frame::UnmarshalledMessage::Error(
crate::types::error::ResponseStreamError::InternalStreamFailure(builder.build()),
));
}
_ => {}
}
Ok(::aws_smithy_eventstream::frame::UnmarshalledMessage::Error(
crate::types::error::ResponseStreamError::generic(generic),
))
}
value => {
return Err(::aws_smithy_eventstream::error::Error::unmarshalling(format!(
"unrecognized :message-type: {value}"
)));
}
}
}
}