use super::*;
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
pub struct ContentLength(u64);
impl std::ops::Deref for ContentLength {
type Target = u64;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl std::ops::DerefMut for ContentLength {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.0
}
}
impl AsRef<u64> for ContentLength {
fn as_ref(&self) -> &u64 {
&self.0
}
}
impl AsMut<u64> for ContentLength {
fn as_mut(&mut self) -> &mut u64 {
&mut self.0
}
}
impl From<u64> for ContentLength {
fn from(v: u64) -> ContentLength {
ContentLength(v)
}
}
impl From<ContentLength> for u64 {
fn from(v: ContentLength) -> u64 {
v.0
}
}
impl super::TypedHeader for ContentLength {
fn from_headers(headers: impl AsRef<Headers>) -> Result<Option<Self>, HeaderParseError> {
let headers = headers.as_ref();
let header = match headers.get(&CONTENT_LENGTH) {
None => return Ok(None),
Some(header) => header,
};
let length = header
.as_str()
.parse::<u64>()
.map(ContentLength)
.map_err(|_| HeaderParseError)?;
Ok(Some(length))
}
fn insert_into(&self, mut headers: impl AsMut<Headers>) {
let headers = headers.as_mut();
headers.insert(CONTENT_LENGTH, self.0.to_string());
}
}