use conjure_object::serde::{ser, de};
use conjure_object::serde::ser::SerializeMap as SerializeMap_;
use conjure_object::private::{UnionField_, UnionTypeField_};
use std::fmt;
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub enum HealthMessage {
Ping(super::Ping),
Pong(super::Pong),
ShutdownNotice(super::ShutdownNotice),
ClientMessageError(super::ClientMessageError),
Unknown(Unknown),
}
impl ser::Serialize for HealthMessage {
fn serialize<S>(&self, s: S) -> Result<S::Ok, S::Error>
where
S: ser::Serializer,
{
let mut map = s.serialize_map(Some(2))?;
match self {
HealthMessage::Ping(value) => {
map.serialize_entry(&"type", &"ping")?;
map.serialize_entry(&"ping", value)?;
}
HealthMessage::Pong(value) => {
map.serialize_entry(&"type", &"pong")?;
map.serialize_entry(&"pong", value)?;
}
HealthMessage::ShutdownNotice(value) => {
map.serialize_entry(&"type", &"shutdownNotice")?;
map.serialize_entry(&"shutdownNotice", value)?;
}
HealthMessage::ClientMessageError(value) => {
map.serialize_entry(&"type", &"clientMessageError")?;
map.serialize_entry(&"clientMessageError", value)?;
}
HealthMessage::Unknown(value) => {
map.serialize_entry(&"type", &value.type_)?;
map.serialize_entry(&value.type_, &value.value)?;
}
}
map.end()
}
}
impl<'de> de::Deserialize<'de> for HealthMessage {
fn deserialize<D>(d: D) -> Result<HealthMessage, D::Error>
where
D: de::Deserializer<'de>,
{
d.deserialize_map(Visitor_)
}
}
struct Visitor_;
impl<'de> de::Visitor<'de> for Visitor_ {
type Value = HealthMessage;
fn expecting(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
fmt.write_str("union HealthMessage")
}
fn visit_map<A>(self, mut map: A) -> Result<HealthMessage, A::Error>
where
A: de::MapAccess<'de>,
{
let v = match map.next_key::<UnionField_<Variant_>>()? {
Some(UnionField_::Type) => {
let variant = map.next_value()?;
let key = map.next_key()?;
match (variant, key) {
(Variant_::Ping, Some(Variant_::Ping)) => {
let value = map.next_value()?;
HealthMessage::Ping(value)
}
(Variant_::Pong, Some(Variant_::Pong)) => {
let value = map.next_value()?;
HealthMessage::Pong(value)
}
(Variant_::ShutdownNotice, Some(Variant_::ShutdownNotice)) => {
let value = map.next_value()?;
HealthMessage::ShutdownNotice(value)
}
(
Variant_::ClientMessageError,
Some(Variant_::ClientMessageError),
) => {
let value = map.next_value()?;
HealthMessage::ClientMessageError(value)
}
(Variant_::Unknown(type_), Some(Variant_::Unknown(b))) => {
if type_ == b {
let value = map.next_value()?;
HealthMessage::Unknown(Unknown { type_, value })
} else {
return Err(
de::Error::invalid_value(de::Unexpected::Str(&type_), &&*b),
)
}
}
(variant, Some(key)) => {
return Err(
de::Error::invalid_value(
de::Unexpected::Str(key.as_str()),
&variant.as_str(),
),
);
}
(variant, None) => {
return Err(de::Error::missing_field(variant.as_str()));
}
}
}
Some(UnionField_::Value(variant)) => {
let value = match &variant {
Variant_::Ping => {
let value = map.next_value()?;
HealthMessage::Ping(value)
}
Variant_::Pong => {
let value = map.next_value()?;
HealthMessage::Pong(value)
}
Variant_::ShutdownNotice => {
let value = map.next_value()?;
HealthMessage::ShutdownNotice(value)
}
Variant_::ClientMessageError => {
let value = map.next_value()?;
HealthMessage::ClientMessageError(value)
}
Variant_::Unknown(type_) => {
let value = map.next_value()?;
HealthMessage::Unknown(Unknown {
type_: type_.clone(),
value,
})
}
};
if map.next_key::<UnionTypeField_>()?.is_none() {
return Err(de::Error::missing_field("type"));
}
let type_variant = map.next_value::<Variant_>()?;
if variant != type_variant {
return Err(
de::Error::invalid_value(
de::Unexpected::Str(type_variant.as_str()),
&variant.as_str(),
),
);
}
value
}
None => return Err(de::Error::missing_field("type")),
};
if map.next_key::<UnionField_<Variant_>>()?.is_some() {
return Err(de::Error::invalid_length(3, &"type and value fields"));
}
Ok(v)
}
}
#[derive(PartialEq)]
enum Variant_ {
Ping,
Pong,
ShutdownNotice,
ClientMessageError,
Unknown(Box<str>),
}
impl Variant_ {
fn as_str(&self) -> &'static str {
match *self {
Variant_::Ping => "ping",
Variant_::Pong => "pong",
Variant_::ShutdownNotice => "shutdownNotice",
Variant_::ClientMessageError => "clientMessageError",
Variant_::Unknown(_) => "unknown variant",
}
}
}
impl<'de> de::Deserialize<'de> for Variant_ {
fn deserialize<D>(d: D) -> Result<Variant_, D::Error>
where
D: de::Deserializer<'de>,
{
d.deserialize_str(VariantVisitor_)
}
}
struct VariantVisitor_;
impl<'de> de::Visitor<'de> for VariantVisitor_ {
type Value = Variant_;
fn expecting(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
fmt.write_str("string")
}
fn visit_str<E>(self, value: &str) -> Result<Variant_, E>
where
E: de::Error,
{
let v = match value {
"ping" => Variant_::Ping,
"pong" => Variant_::Pong,
"shutdownNotice" => Variant_::ShutdownNotice,
"clientMessageError" => Variant_::ClientMessageError,
value => Variant_::Unknown(value.to_string().into_boxed_str()),
};
Ok(v)
}
}
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct Unknown {
type_: Box<str>,
value: conjure_object::Any,
}
impl Unknown {
#[inline]
pub fn type_(&self) -> &str {
&self.type_
}
}