use std::borrow::Cow;
use derive_more::{Deref, Display, From};
use schemars::JsonSchema;
use serde::de::Deserializer;
use serde::{Deserialize, Serialize};
use crate::Validate;
#[derive(Debug, Clone, PartialEq, Eq, Hash, Deref, From, Display)]
pub struct MetadataKey(tonic::metadata::AsciiMetadataKey);
impl<'de> Deserialize<'de> for MetadataKey {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
let s = String::deserialize(deserializer)?;
s.parse::<tonic::metadata::AsciiMetadataKey>()
.map(Self)
.map_err(serde::de::Error::custom)
}
}
impl Serialize for MetadataKey {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
self.0.as_str().serialize(serializer)
}
}
impl JsonSchema for MetadataKey {
fn schema_name() -> Cow<'static, str> {
Cow::Borrowed("MetadataKey")
}
fn json_schema(generator: &mut schemars::SchemaGenerator) -> schemars::Schema {
String::json_schema(generator)
}
}
impl Validate for MetadataKey {}
#[derive(Debug, Clone, PartialEq, Eq, Hash, Deref, From)]
pub struct MetadataValue(tonic::metadata::AsciiMetadataValue);
impl<'de> Deserialize<'de> for MetadataValue {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
let s = String::deserialize(deserializer)?;
s.parse::<tonic::metadata::AsciiMetadataValue>()
.map(Self)
.map_err(serde::de::Error::custom)
}
}
impl Serialize for MetadataValue {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
self.0
.to_str()
.map_err(serde::ser::Error::custom)?
.serialize(serializer)
}
}
impl JsonSchema for MetadataValue {
fn schema_name() -> Cow<'static, str> {
Cow::Borrowed("MetadataValue")
}
fn json_schema(generator: &mut schemars::SchemaGenerator) -> schemars::Schema {
String::json_schema(generator)
}
}
impl Validate for MetadataValue {}