mirror of https://github.com/fafhrd91/actix-web
implement IntoHeaderPair for all T: Header
This commit is contained in:
parent
c34e6dccc7
commit
f5ff40e36b
|
@ -75,6 +75,7 @@ regex = "1.3"
|
||||||
serde = "1.0"
|
serde = "1.0"
|
||||||
serde_json = "1.0"
|
serde_json = "1.0"
|
||||||
sha-1 = "0.9"
|
sha-1 = "0.9"
|
||||||
|
smallvec = "1.6"
|
||||||
slab = "0.4"
|
slab = "0.4"
|
||||||
serde_urlencoded = "0.7"
|
serde_urlencoded = "0.7"
|
||||||
time = { version = "0.2.7", default-features = false, features = ["std"] }
|
time = { version = "0.2.7", default-features = false, features = ["std"] }
|
||||||
|
|
|
@ -45,7 +45,7 @@ where
|
||||||
Some(port) => write!(wrt, "{}:{}", host, port),
|
Some(port) => write!(wrt, "{}:{}", host, port),
|
||||||
};
|
};
|
||||||
|
|
||||||
match wrt.get_mut().split().freeze().try_into() {
|
match wrt.get_mut().split().freeze().try_into_value() {
|
||||||
Ok(value) => match head {
|
Ok(value) => match head {
|
||||||
RequestHeadType::Owned(ref mut head) => {
|
RequestHeadType::Owned(ref mut head) => {
|
||||||
head.headers.insert(HOST, value)
|
head.headers.insert(HOST, value)
|
||||||
|
|
|
@ -82,7 +82,7 @@ impl fmt::Display for CacheControl {
|
||||||
impl IntoHeaderValue for CacheControl {
|
impl IntoHeaderValue for CacheControl {
|
||||||
type Error = header::InvalidHeaderValue;
|
type Error = header::InvalidHeaderValue;
|
||||||
|
|
||||||
fn try_into(self) -> Result<header::HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<header::HeaderValue, Self::Error> {
|
||||||
let mut writer = Writer::new();
|
let mut writer = Writer::new();
|
||||||
let _ = write!(&mut writer, "{}", self);
|
let _ = write!(&mut writer, "{}", self);
|
||||||
header::HeaderValue::from_maybe_shared(writer.take())
|
header::HeaderValue::from_maybe_shared(writer.take())
|
||||||
|
|
|
@ -454,7 +454,7 @@ impl ContentDisposition {
|
||||||
impl IntoHeaderValue for ContentDisposition {
|
impl IntoHeaderValue for ContentDisposition {
|
||||||
type Error = header::InvalidHeaderValue;
|
type Error = header::InvalidHeaderValue;
|
||||||
|
|
||||||
fn try_into(self) -> Result<header::HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<header::HeaderValue, Self::Error> {
|
||||||
let mut writer = Writer::new();
|
let mut writer = Writer::new();
|
||||||
let _ = write!(&mut writer, "{}", self);
|
let _ = write!(&mut writer, "{}", self);
|
||||||
header::HeaderValue::from_maybe_shared(writer.take())
|
header::HeaderValue::from_maybe_shared(writer.take())
|
||||||
|
|
|
@ -200,7 +200,7 @@ impl Display for ContentRangeSpec {
|
||||||
impl IntoHeaderValue for ContentRangeSpec {
|
impl IntoHeaderValue for ContentRangeSpec {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
let mut writer = Writer::new();
|
let mut writer = Writer::new();
|
||||||
let _ = write!(&mut writer, "{}", self);
|
let _ = write!(&mut writer, "{}", self);
|
||||||
HeaderValue::from_maybe_shared(writer.take())
|
HeaderValue::from_maybe_shared(writer.take())
|
||||||
|
|
|
@ -99,6 +99,7 @@ impl ContentType {
|
||||||
pub fn form_url_encoded() -> ContentType {
|
pub fn form_url_encoded() -> ContentType {
|
||||||
ContentType(mime::APPLICATION_WWW_FORM_URLENCODED)
|
ContentType(mime::APPLICATION_WWW_FORM_URLENCODED)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// A constructor to easily create a `Content-Type: image/jpeg` header.
|
/// A constructor to easily create a `Content-Type: image/jpeg` header.
|
||||||
#[inline]
|
#[inline]
|
||||||
pub fn jpeg() -> ContentType {
|
pub fn jpeg() -> ContentType {
|
||||||
|
|
|
@ -98,7 +98,7 @@ impl Display for IfRange {
|
||||||
impl IntoHeaderValue for IfRange {
|
impl IntoHeaderValue for IfRange {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
let mut writer = Writer::new();
|
let mut writer = Writer::new();
|
||||||
let _ = write!(&mut writer, "{}", self);
|
let _ = write!(&mut writer, "{}", self);
|
||||||
HeaderValue::from_maybe_shared(writer.take())
|
HeaderValue::from_maybe_shared(writer.take())
|
||||||
|
|
|
@ -169,7 +169,7 @@ macro_rules! header {
|
||||||
impl $crate::http::header::IntoHeaderValue for $id {
|
impl $crate::http::header::IntoHeaderValue for $id {
|
||||||
type Error = $crate::http::header::InvalidHeaderValue;
|
type Error = $crate::http::header::InvalidHeaderValue;
|
||||||
|
|
||||||
fn try_into(self) -> Result<$crate::http::header::HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<$crate::http::header::HeaderValue, Self::Error> {
|
||||||
use std::fmt::Write;
|
use std::fmt::Write;
|
||||||
let mut writer = $crate::http::header::Writer::new();
|
let mut writer = $crate::http::header::Writer::new();
|
||||||
let _ = write!(&mut writer, "{}", self);
|
let _ = write!(&mut writer, "{}", self);
|
||||||
|
@ -205,7 +205,7 @@ macro_rules! header {
|
||||||
impl $crate::http::header::IntoHeaderValue for $id {
|
impl $crate::http::header::IntoHeaderValue for $id {
|
||||||
type Error = $crate::http::header::InvalidHeaderValue;
|
type Error = $crate::http::header::InvalidHeaderValue;
|
||||||
|
|
||||||
fn try_into(self) -> Result<$crate::http::header::HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<$crate::http::header::HeaderValue, Self::Error> {
|
||||||
use std::fmt::Write;
|
use std::fmt::Write;
|
||||||
let mut writer = $crate::http::header::Writer::new();
|
let mut writer = $crate::http::header::Writer::new();
|
||||||
let _ = write!(&mut writer, "{}", self);
|
let _ = write!(&mut writer, "{}", self);
|
||||||
|
@ -241,8 +241,8 @@ macro_rules! header {
|
||||||
impl $crate::http::header::IntoHeaderValue for $id {
|
impl $crate::http::header::IntoHeaderValue for $id {
|
||||||
type Error = $crate::http::header::InvalidHeaderValue;
|
type Error = $crate::http::header::InvalidHeaderValue;
|
||||||
|
|
||||||
fn try_into(self) -> Result<$crate::http::header::HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<$crate::http::header::HeaderValue, Self::Error> {
|
||||||
self.0.try_into()
|
self.0.try_into_value()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
@ -290,7 +290,7 @@ macro_rules! header {
|
||||||
impl $crate::http::header::IntoHeaderValue for $id {
|
impl $crate::http::header::IntoHeaderValue for $id {
|
||||||
type Error = $crate::http::header::InvalidHeaderValue;
|
type Error = $crate::http::header::InvalidHeaderValue;
|
||||||
|
|
||||||
fn try_into(self) -> Result<$crate::http::header::HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<$crate::http::header::HeaderValue, Self::Error> {
|
||||||
use std::fmt::Write;
|
use std::fmt::Write;
|
||||||
let mut writer = $crate::http::header::Writer::new();
|
let mut writer = $crate::http::header::Writer::new();
|
||||||
let _ = write!(&mut writer, "{}", self);
|
let _ = write!(&mut writer, "{}", self);
|
||||||
|
|
|
@ -3,9 +3,11 @@ use std::convert::{Infallible, TryFrom};
|
||||||
use either::Either;
|
use either::Either;
|
||||||
use http::{
|
use http::{
|
||||||
header::{HeaderName, InvalidHeaderName, InvalidHeaderValue},
|
header::{HeaderName, InvalidHeaderName, InvalidHeaderValue},
|
||||||
HeaderValue,
|
Error as HttpError, HeaderValue,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
use super::{Header, IntoHeaderValue};
|
||||||
|
|
||||||
/// Transforms structures into header K/V pairs for inserting into `HeaderMap`s.
|
/// Transforms structures into header K/V pairs for inserting into `HeaderMap`s.
|
||||||
pub trait IntoHeaderPair: Sized {
|
pub trait IntoHeaderPair: Sized {
|
||||||
type Error;
|
type Error;
|
||||||
|
@ -41,13 +43,28 @@ impl IntoHeaderPair for (&str, HeaderValue) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum InvalidHeaderPart {
|
||||||
|
Name(InvalidHeaderName),
|
||||||
|
Value(InvalidHeaderValue),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<InvalidHeaderPart> for HttpError {
|
||||||
|
fn from(part_err: InvalidHeaderPart) -> Self {
|
||||||
|
match part_err {
|
||||||
|
InvalidHeaderPart::Name(err) => err.into(),
|
||||||
|
InvalidHeaderPart::Value(err) => err.into(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl IntoHeaderPair for (&str, &str) {
|
impl IntoHeaderPair for (&str, &str) {
|
||||||
type Error = Either<InvalidHeaderName, InvalidHeaderValue>;
|
type Error = InvalidHeaderPart;
|
||||||
|
|
||||||
fn try_into_header_pair(self) -> Result<(HeaderName, HeaderValue), Self::Error> {
|
fn try_into_header_pair(self) -> Result<(HeaderName, HeaderValue), Self::Error> {
|
||||||
let (name, value) = self;
|
let (name, value) = self;
|
||||||
let name = HeaderName::try_from(name).map_err(Either::Left)?;
|
let name = HeaderName::try_from(name).map_err(InvalidHeaderPart::Name)?;
|
||||||
let value = HeaderValue::try_from(value).map_err(Either::Right)?;
|
let value = HeaderValue::try_from(value).map_err(InvalidHeaderPart::Value)?;
|
||||||
Ok((name, value))
|
Ok((name, value))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -82,3 +99,14 @@ impl IntoHeaderPair for (String, String) {
|
||||||
Ok((name, value))
|
Ok((name, value))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl<T> IntoHeaderPair for T
|
||||||
|
where
|
||||||
|
T: Header,
|
||||||
|
{
|
||||||
|
type Error = <T as IntoHeaderValue>::Error;
|
||||||
|
|
||||||
|
fn try_into_header_pair(self) -> Result<(HeaderName, HeaderValue), Self::Error> {
|
||||||
|
Ok((T::name(), self.try_into_value()?))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
@ -10,14 +10,14 @@ pub trait IntoHeaderValue: Sized {
|
||||||
type Error: Into<HttpError>;
|
type Error: Into<HttpError>;
|
||||||
|
|
||||||
/// Try to convert value to a Header pair value.
|
/// Try to convert value to a Header pair value.
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error>;
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error>;
|
||||||
}
|
}
|
||||||
|
|
||||||
impl IntoHeaderValue for HeaderValue {
|
impl IntoHeaderValue for HeaderValue {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
Ok(self)
|
Ok(self)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -26,7 +26,7 @@ impl<'a> IntoHeaderValue for &'a str {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
self.parse()
|
self.parse()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -35,7 +35,7 @@ impl<'a> IntoHeaderValue for &'a [u8] {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
HeaderValue::from_bytes(self)
|
HeaderValue::from_bytes(self)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -44,7 +44,7 @@ impl IntoHeaderValue for Bytes {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
HeaderValue::from_maybe_shared(self)
|
HeaderValue::from_maybe_shared(self)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -53,7 +53,7 @@ impl IntoHeaderValue for Vec<u8> {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
HeaderValue::try_from(self)
|
HeaderValue::try_from(self)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -62,7 +62,7 @@ impl IntoHeaderValue for String {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
HeaderValue::try_from(self)
|
HeaderValue::try_from(self)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -71,7 +71,7 @@ impl IntoHeaderValue for usize {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
let s = format!("{}", self);
|
let s = format!("{}", self);
|
||||||
HeaderValue::try_from(s)
|
HeaderValue::try_from(s)
|
||||||
}
|
}
|
||||||
|
@ -81,7 +81,7 @@ impl IntoHeaderValue for u64 {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
let s = format!("{}", self);
|
let s = format!("{}", self);
|
||||||
HeaderValue::try_from(s)
|
HeaderValue::try_from(s)
|
||||||
}
|
}
|
||||||
|
@ -91,7 +91,7 @@ impl IntoHeaderValue for Mime {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
HeaderValue::try_from(format!("{}", self))
|
HeaderValue::try_from(format!("{}", self))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,12 +1,16 @@
|
||||||
use std::{collections::hash_map::{self, Entry}, convert::TryFrom};
|
use std::{
|
||||||
|
collections::hash_map::{self, Entry},
|
||||||
|
convert::TryFrom,
|
||||||
|
};
|
||||||
|
|
||||||
use ahash::AHashMap;
|
use ahash::AHashMap;
|
||||||
use either::Either;
|
use either::Either;
|
||||||
use http::header::{HeaderName, HeaderValue};
|
use http::header::{HeaderName, HeaderValue};
|
||||||
|
use smallvec::{smallvec, SmallVec};
|
||||||
|
|
||||||
/// A set of HTTP headers
|
/// A multi-map of HTTP headers.
|
||||||
///
|
///
|
||||||
/// `HeaderMap` is an multi-map of [`HeaderName`] to values.
|
/// `HeaderMap` is a "multi-map" of [`HeaderName`] to one or more values.
|
||||||
#[derive(Debug, Clone, Default)]
|
#[derive(Debug, Clone, Default)]
|
||||||
pub struct HeaderMap {
|
pub struct HeaderMap {
|
||||||
pub(crate) inner: AHashMap<HeaderName, Value>,
|
pub(crate) inner: AHashMap<HeaderName, Value>,
|
||||||
|
@ -15,18 +19,18 @@ pub struct HeaderMap {
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub(crate) enum Value {
|
pub(crate) enum Value {
|
||||||
One(HeaderValue),
|
One(HeaderValue),
|
||||||
Multi(Vec<HeaderValue>),
|
Multi(SmallVec<[HeaderValue; 4]>),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Value {
|
impl Value {
|
||||||
fn get(&self) -> &HeaderValue {
|
fn first(&self) -> &HeaderValue {
|
||||||
match self {
|
match self {
|
||||||
Value::One(ref val) => val,
|
Value::One(ref val) => val,
|
||||||
Value::Multi(ref val) => &val[0],
|
Value::Multi(ref val) => &val[0],
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_mut(&mut self) -> &mut HeaderValue {
|
fn first_mut(&mut self) -> &mut HeaderValue {
|
||||||
match self {
|
match self {
|
||||||
Value::One(ref mut val) => val,
|
Value::One(ref mut val) => val,
|
||||||
Value::Multi(ref mut val) => &mut val[0],
|
Value::Multi(ref mut val) => &mut val[0],
|
||||||
|
@ -36,7 +40,7 @@ impl Value {
|
||||||
fn append(&mut self, val: HeaderValue) {
|
fn append(&mut self, val: HeaderValue) {
|
||||||
match self {
|
match self {
|
||||||
Value::One(_) => {
|
Value::One(_) => {
|
||||||
let data = std::mem::replace(self, Value::Multi(vec![val]));
|
let data = std::mem::replace(self, Value::Multi(smallvec![val]));
|
||||||
match data {
|
match data {
|
||||||
Value::One(val) => self.append(val),
|
Value::One(val) => self.append(val),
|
||||||
Value::Multi(_) => unreachable!(),
|
Value::Multi(_) => unreachable!(),
|
||||||
|
@ -117,7 +121,7 @@ impl HeaderMap {
|
||||||
/// is returned. Use `get_all` to get all values associated with a given
|
/// is returned. Use `get_all` to get all values associated with a given
|
||||||
/// key. Returns `None` if there are no values associated with the key.
|
/// key. Returns `None` if there are no values associated with the key.
|
||||||
pub fn get<N: AsName>(&self, name: N) -> Option<&HeaderValue> {
|
pub fn get<N: AsName>(&self, name: N) -> Option<&HeaderValue> {
|
||||||
self.get2(name).map(|v| v.get())
|
self.get2(name).map(|v| v.first())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get2<N: AsName>(&self, name: N) -> Option<&Value> {
|
fn get2<N: AsName>(&self, name: N) -> Option<&Value> {
|
||||||
|
@ -133,11 +137,11 @@ impl HeaderMap {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns a view of all values associated with a key.
|
/// Returns an iterator of all values associated with a key.
|
||||||
///
|
///
|
||||||
/// The returned view does not incur any allocations and allows iterating
|
/// The returned view does not incur any allocations and allows iterating the values associated
|
||||||
/// the values associated with the key. See [`GetAll`] for more details.
|
/// with the key. Returns `None` if there are no values associated with the key. Iteration order
|
||||||
/// Returns `None` if there are no values associated with the key.
|
/// is not guaranteed to be the same as insertion order.
|
||||||
pub fn get_all<N: AsName>(&self, name: N) -> GetAll<'_> {
|
pub fn get_all<N: AsName>(&self, name: N) -> GetAll<'_> {
|
||||||
GetAll {
|
GetAll {
|
||||||
idx: 0,
|
idx: 0,
|
||||||
|
@ -152,10 +156,10 @@ impl HeaderMap {
|
||||||
/// key. Returns `None` if there are no values associated with the key.
|
/// key. Returns `None` if there are no values associated with the key.
|
||||||
pub fn get_mut<N: AsName>(&mut self, name: N) -> Option<&mut HeaderValue> {
|
pub fn get_mut<N: AsName>(&mut self, name: N) -> Option<&mut HeaderValue> {
|
||||||
match name.as_name() {
|
match name.as_name() {
|
||||||
Either::Left(name) => self.inner.get_mut(name).map(|v| v.get_mut()),
|
Either::Left(name) => self.inner.get_mut(name).map(|v| v.first_mut()),
|
||||||
Either::Right(s) => {
|
Either::Right(s) => {
|
||||||
if let Ok(name) = HeaderName::try_from(s) {
|
if let Ok(name) = HeaderName::try_from(s) {
|
||||||
self.inner.get_mut(&name).map(|v| v.get_mut())
|
self.inner.get_mut(&name).map(|v| v.first_mut())
|
||||||
} else {
|
} else {
|
||||||
None
|
None
|
||||||
}
|
}
|
||||||
|
@ -336,7 +340,7 @@ impl<'a> IntoIterator for &'a HeaderMap {
|
||||||
|
|
||||||
pub struct Iter<'a> {
|
pub struct Iter<'a> {
|
||||||
idx: usize,
|
idx: usize,
|
||||||
current: Option<(&'a HeaderName, &'a Vec<HeaderValue>)>,
|
current: Option<(&'a HeaderName, &'a SmallVec<[HeaderValue; 4]>)>,
|
||||||
iter: hash_map::Iter<'a, HeaderName, Value>,
|
iter: hash_map::Iter<'a, HeaderName, Value>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -161,7 +161,7 @@ impl FromStr for EntityTag {
|
||||||
impl IntoHeaderValue for EntityTag {
|
impl IntoHeaderValue for EntityTag {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
let mut wrt = Writer::new();
|
let mut wrt = Writer::new();
|
||||||
write!(wrt, "{}", self).unwrap();
|
write!(wrt, "{}", self).unwrap();
|
||||||
HeaderValue::from_maybe_shared(wrt.take())
|
HeaderValue::from_maybe_shared(wrt.take())
|
||||||
|
|
|
@ -48,7 +48,7 @@ impl From<SystemTime> for HttpDate {
|
||||||
impl IntoHeaderValue for HttpDate {
|
impl IntoHeaderValue for HttpDate {
|
||||||
type Error = InvalidHeaderValue;
|
type Error = InvalidHeaderValue;
|
||||||
|
|
||||||
fn try_into(self) -> Result<HeaderValue, Self::Error> {
|
fn try_into_value(self) -> Result<HeaderValue, Self::Error> {
|
||||||
let mut wrt = BytesMut::with_capacity(29).writer();
|
let mut wrt = BytesMut::with_capacity(29).writer();
|
||||||
write!(
|
write!(
|
||||||
wrt,
|
wrt,
|
||||||
|
|
|
@ -53,7 +53,7 @@ pub use self::response::{Response, ResponseBuilder};
|
||||||
pub use self::service::HttpService;
|
pub use self::service::HttpService;
|
||||||
|
|
||||||
pub mod http {
|
pub mod http {
|
||||||
//! Various HTTP related types
|
//! Various HTTP related types.
|
||||||
|
|
||||||
// re-exports
|
// re-exports
|
||||||
pub use http::header::{HeaderName, HeaderValue};
|
pub use http::header::{HeaderName, HeaderValue};
|
||||||
|
@ -64,7 +64,7 @@ pub mod http {
|
||||||
pub use crate::cookie::{Cookie, CookieBuilder};
|
pub use crate::cookie::{Cookie, CookieBuilder};
|
||||||
pub use crate::header::HeaderMap;
|
pub use crate::header::HeaderMap;
|
||||||
|
|
||||||
/// Various http headers
|
/// A collection of HTTP headers and helpers.
|
||||||
pub mod header {
|
pub mod header {
|
||||||
pub use crate::header::*;
|
pub use crate::header::*;
|
||||||
}
|
}
|
||||||
|
|
|
@ -361,7 +361,7 @@ impl ResponseBuilder {
|
||||||
#[doc(hidden)]
|
#[doc(hidden)]
|
||||||
pub fn set<H: Header>(&mut self, hdr: H) -> &mut Self {
|
pub fn set<H: Header>(&mut self, hdr: H) -> &mut Self {
|
||||||
if let Some(parts) = parts(&mut self.head, &self.err) {
|
if let Some(parts) = parts(&mut self.head, &self.err) {
|
||||||
match hdr.try_into() {
|
match hdr.try_into_value() {
|
||||||
Ok(value) => {
|
Ok(value) => {
|
||||||
parts.headers.append(H::name(), value);
|
parts.headers.append(H::name(), value);
|
||||||
}
|
}
|
||||||
|
@ -391,7 +391,7 @@ impl ResponseBuilder {
|
||||||
{
|
{
|
||||||
if let Some(parts) = parts(&mut self.head, &self.err) {
|
if let Some(parts) = parts(&mut self.head, &self.err) {
|
||||||
match HeaderName::try_from(key) {
|
match HeaderName::try_from(key) {
|
||||||
Ok(key) => match value.try_into() {
|
Ok(key) => match value.try_into_value() {
|
||||||
Ok(value) => {
|
Ok(value) => {
|
||||||
parts.headers.append(key, value);
|
parts.headers.append(key, value);
|
||||||
}
|
}
|
||||||
|
@ -485,8 +485,7 @@ impl ResponseBuilder {
|
||||||
parts.set_connection_type(ConnectionType::Upgrade);
|
parts.set_connection_type(ConnectionType::Upgrade);
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: fix signature
|
if let Ok(value) = value.try_into_value() {
|
||||||
if let Ok(value) = value.try_into() {
|
|
||||||
self.insert_header((header::UPGRADE, value));
|
self.insert_header((header::UPGRADE, value));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -880,6 +879,8 @@ impl From<BytesMut> for Response {
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
|
use serde_json::json;
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::body::Body;
|
use crate::body::Body;
|
||||||
use crate::http::header::{HeaderValue, CONTENT_TYPE, COOKIE, SET_COOKIE};
|
use crate::http::header::{HeaderValue, CONTENT_TYPE, COOKIE, SET_COOKIE};
|
||||||
|
@ -1113,4 +1114,54 @@ mod tests {
|
||||||
let cookie = resp.cookies().next().unwrap();
|
let cookie = resp.cookies().next().unwrap();
|
||||||
assert_eq!((cookie.name(), cookie.value()), ("cookie1", "val100"));
|
assert_eq!((cookie.name(), cookie.value()), ("cookie1", "val100"));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn response_builder_header_insert_kv() {
|
||||||
|
let mut res = Response::Ok();
|
||||||
|
res.insert_header(("Content-Type", "application/octet-stream"));
|
||||||
|
let res = res.finish();
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
res.headers().get("Content-Type"),
|
||||||
|
Some(&HeaderValue::from_static("application/octet-stream"))
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn response_builder_header_insert_typed() {
|
||||||
|
let mut res = Response::Ok();
|
||||||
|
res.insert_header(header::ContentType(mime::APPLICATION_OCTET_STREAM));
|
||||||
|
let res = res.finish();
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
res.headers().get("Content-Type"),
|
||||||
|
Some(&HeaderValue::from_static("application/octet-stream"))
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn response_builder_header_append_kv() {
|
||||||
|
let mut res = Response::Ok();
|
||||||
|
res.append_header(("Content-Type", "application/octet-stream"));
|
||||||
|
res.append_header(("Content-Type", "application/json"));
|
||||||
|
let res = res.finish();
|
||||||
|
|
||||||
|
let headers: Vec<_> = res.headers().get_all("Content-Type").cloned().collect();
|
||||||
|
assert_eq!(headers.len(), 2);
|
||||||
|
assert!(headers.contains(&HeaderValue::from_static("application/octet-stream")));
|
||||||
|
assert!(headers.contains(&HeaderValue::from_static("application/json")));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn response_builder_header_append_typed() {
|
||||||
|
let mut res = Response::Ok();
|
||||||
|
res.append_header(header::ContentType(mime::APPLICATION_OCTET_STREAM));
|
||||||
|
res.append_header(header::ContentType(mime::APPLICATION_JSON));
|
||||||
|
let res = res.finish();
|
||||||
|
|
||||||
|
let headers: Vec<_> = res.headers().get_all("Content-Type").cloned().collect();
|
||||||
|
assert_eq!(headers.len(), 2);
|
||||||
|
assert!(headers.contains(&HeaderValue::from_static("application/octet-stream")));
|
||||||
|
assert!(headers.contains(&HeaderValue::from_static("application/json")));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -109,7 +109,7 @@ impl TestRequest {
|
||||||
|
|
||||||
/// Set a header
|
/// Set a header
|
||||||
pub fn set<H: Header>(&mut self, hdr: H) -> &mut Self {
|
pub fn set<H: Header>(&mut self, hdr: H) -> &mut Self {
|
||||||
if let Ok(value) = hdr.try_into() {
|
if let Ok(value) = hdr.try_into_value() {
|
||||||
parts(&mut self.0).headers.append(H::name(), value);
|
parts(&mut self.0).headers.append(H::name(), value);
|
||||||
return self;
|
return self;
|
||||||
}
|
}
|
||||||
|
@ -124,7 +124,7 @@ impl TestRequest {
|
||||||
V: IntoHeaderValue,
|
V: IntoHeaderValue,
|
||||||
{
|
{
|
||||||
if let Ok(key) = HeaderName::try_from(key) {
|
if let Ok(key) = HeaderName::try_from(key) {
|
||||||
if let Ok(value) = value.try_into() {
|
if let Ok(value) = value.try_into_value() {
|
||||||
parts(&mut self.0).headers.append(key, value);
|
parts(&mut self.0).headers.append(key, value);
|
||||||
return self;
|
return self;
|
||||||
}
|
}
|
||||||
|
|
|
@ -133,7 +133,7 @@ impl ClientBuilder {
|
||||||
V::Error: fmt::Debug,
|
V::Error: fmt::Debug,
|
||||||
{
|
{
|
||||||
match HeaderName::try_from(key) {
|
match HeaderName::try_from(key) {
|
||||||
Ok(key) => match value.try_into() {
|
Ok(key) => match value.try_into_value() {
|
||||||
Ok(value) => {
|
Ok(value) => {
|
||||||
self.headers.append(key, value);
|
self.headers.append(key, value);
|
||||||
}
|
}
|
||||||
|
|
|
@ -144,7 +144,7 @@ impl FrozenSendBuilder {
|
||||||
V: IntoHeaderValue,
|
V: IntoHeaderValue,
|
||||||
{
|
{
|
||||||
match HeaderName::try_from(key) {
|
match HeaderName::try_from(key) {
|
||||||
Ok(key) => match value.try_into() {
|
Ok(key) => match value.try_into_value() {
|
||||||
Ok(value) => self.extra_headers.insert(key, value),
|
Ok(value) => self.extra_headers.insert(key, value),
|
||||||
Err(e) => self.err = Some(e.into()),
|
Err(e) => self.err = Some(e.into()),
|
||||||
},
|
},
|
||||||
|
|
|
@ -169,7 +169,7 @@ impl ClientRequest {
|
||||||
/// }
|
/// }
|
||||||
/// ```
|
/// ```
|
||||||
pub fn set<H: Header>(mut self, hdr: H) -> Self {
|
pub fn set<H: Header>(mut self, hdr: H) -> Self {
|
||||||
match hdr.try_into() {
|
match hdr.try_into_value() {
|
||||||
Ok(value) => {
|
Ok(value) => {
|
||||||
self.head.headers.insert(H::name(), value);
|
self.head.headers.insert(H::name(), value);
|
||||||
}
|
}
|
||||||
|
@ -203,7 +203,7 @@ impl ClientRequest {
|
||||||
V: IntoHeaderValue,
|
V: IntoHeaderValue,
|
||||||
{
|
{
|
||||||
match HeaderName::try_from(key) {
|
match HeaderName::try_from(key) {
|
||||||
Ok(key) => match value.try_into() {
|
Ok(key) => match value.try_into_value() {
|
||||||
Ok(value) => self.head.headers.append(key, value),
|
Ok(value) => self.head.headers.append(key, value),
|
||||||
Err(e) => self.err = Some(e.into()),
|
Err(e) => self.err = Some(e.into()),
|
||||||
},
|
},
|
||||||
|
@ -220,7 +220,7 @@ impl ClientRequest {
|
||||||
V: IntoHeaderValue,
|
V: IntoHeaderValue,
|
||||||
{
|
{
|
||||||
match HeaderName::try_from(key) {
|
match HeaderName::try_from(key) {
|
||||||
Ok(key) => match value.try_into() {
|
Ok(key) => match value.try_into_value() {
|
||||||
Ok(value) => self.head.headers.insert(key, value),
|
Ok(value) => self.head.headers.insert(key, value),
|
||||||
Err(e) => self.err = Some(e.into()),
|
Err(e) => self.err = Some(e.into()),
|
||||||
},
|
},
|
||||||
|
@ -239,7 +239,7 @@ impl ClientRequest {
|
||||||
match HeaderName::try_from(key) {
|
match HeaderName::try_from(key) {
|
||||||
Ok(key) => {
|
Ok(key) => {
|
||||||
if !self.head.headers.contains_key(&key) {
|
if !self.head.headers.contains_key(&key) {
|
||||||
match value.try_into() {
|
match value.try_into_value() {
|
||||||
Ok(value) => self.head.headers.insert(key, value),
|
Ok(value) => self.head.headers.insert(key, value),
|
||||||
Err(e) => self.err = Some(e.into()),
|
Err(e) => self.err = Some(e.into()),
|
||||||
}
|
}
|
||||||
|
|
|
@ -296,7 +296,7 @@ impl RequestSender {
|
||||||
match self {
|
match self {
|
||||||
RequestSender::Owned(head) => {
|
RequestSender::Owned(head) => {
|
||||||
if !head.headers.contains_key(&key) {
|
if !head.headers.contains_key(&key) {
|
||||||
match value.try_into() {
|
match value.try_into_value() {
|
||||||
Ok(value) => head.headers.insert(key, value),
|
Ok(value) => head.headers.insert(key, value),
|
||||||
Err(e) => return Err(e.into()),
|
Err(e) => return Err(e.into()),
|
||||||
}
|
}
|
||||||
|
@ -306,7 +306,7 @@ impl RequestSender {
|
||||||
if !head.headers.contains_key(&key)
|
if !head.headers.contains_key(&key)
|
||||||
&& !extra_headers.iter().any(|h| h.contains_key(&key))
|
&& !extra_headers.iter().any(|h| h.contains_key(&key))
|
||||||
{
|
{
|
||||||
match value.try_into() {
|
match value.try_into_value() {
|
||||||
Ok(v) => {
|
Ok(v) => {
|
||||||
let h = extra_headers.get_or_insert(HeaderMap::new());
|
let h = extra_headers.get_or_insert(HeaderMap::new());
|
||||||
h.insert(key, v)
|
h.insert(key, v)
|
||||||
|
|
|
@ -45,7 +45,7 @@ impl TestResponse {
|
||||||
|
|
||||||
/// Set a header
|
/// Set a header
|
||||||
pub fn set<H: Header>(mut self, hdr: H) -> Self {
|
pub fn set<H: Header>(mut self, hdr: H) -> Self {
|
||||||
if let Ok(value) = hdr.try_into() {
|
if let Ok(value) = hdr.try_into_value() {
|
||||||
self.head.headers.append(H::name(), value);
|
self.head.headers.append(H::name(), value);
|
||||||
return self;
|
return self;
|
||||||
}
|
}
|
||||||
|
@ -60,7 +60,7 @@ impl TestResponse {
|
||||||
V: IntoHeaderValue,
|
V: IntoHeaderValue,
|
||||||
{
|
{
|
||||||
if let Ok(key) = HeaderName::try_from(key) {
|
if let Ok(key) = HeaderName::try_from(key) {
|
||||||
if let Ok(value) = value.try_into() {
|
if let Ok(value) = value.try_into_value() {
|
||||||
self.head.headers.append(key, value);
|
self.head.headers.append(key, value);
|
||||||
return self;
|
return self;
|
||||||
}
|
}
|
||||||
|
|
|
@ -170,7 +170,7 @@ impl WebsocketsRequest {
|
||||||
V: IntoHeaderValue,
|
V: IntoHeaderValue,
|
||||||
{
|
{
|
||||||
match HeaderName::try_from(key) {
|
match HeaderName::try_from(key) {
|
||||||
Ok(key) => match value.try_into() {
|
Ok(key) => match value.try_into_value() {
|
||||||
Ok(value) => {
|
Ok(value) => {
|
||||||
self.head.headers.append(key, value);
|
self.head.headers.append(key, value);
|
||||||
}
|
}
|
||||||
|
@ -189,7 +189,7 @@ impl WebsocketsRequest {
|
||||||
V: IntoHeaderValue,
|
V: IntoHeaderValue,
|
||||||
{
|
{
|
||||||
match HeaderName::try_from(key) {
|
match HeaderName::try_from(key) {
|
||||||
Ok(key) => match value.try_into() {
|
Ok(key) => match value.try_into_value() {
|
||||||
Ok(value) => {
|
Ok(value) => {
|
||||||
self.head.headers.insert(key, value);
|
self.head.headers.insert(key, value);
|
||||||
}
|
}
|
||||||
|
@ -210,7 +210,7 @@ impl WebsocketsRequest {
|
||||||
match HeaderName::try_from(key) {
|
match HeaderName::try_from(key) {
|
||||||
Ok(key) => {
|
Ok(key) => {
|
||||||
if !self.head.headers.contains_key(&key) {
|
if !self.head.headers.contains_key(&key) {
|
||||||
match value.try_into() {
|
match value.try_into_value() {
|
||||||
Ok(value) => {
|
Ok(value) => {
|
||||||
self.head.headers.insert(key, value);
|
self.head.headers.insert(key, value);
|
||||||
}
|
}
|
||||||
|
|
|
@ -2,7 +2,7 @@ use std::convert::TryFrom;
|
||||||
|
|
||||||
use actix_http::error::InternalError;
|
use actix_http::error::InternalError;
|
||||||
use actix_http::http::{
|
use actix_http::http::{
|
||||||
header::IntoHeaderValue, Error as HttpError, HeaderMap, HeaderName, StatusCode,
|
header::IntoHeaderPair, Error as HttpError, HeaderMap, HeaderName, StatusCode,
|
||||||
};
|
};
|
||||||
use actix_http::ResponseBuilder;
|
use actix_http::ResponseBuilder;
|
||||||
use bytes::{Bytes, BytesMut};
|
use bytes::{Bytes, BytesMut};
|
||||||
|
@ -203,29 +203,24 @@ impl<T: Responder> CustomResponder<T> {
|
||||||
/// web::Json(
|
/// web::Json(
|
||||||
/// MyObj{name: "Name".to_string()}
|
/// MyObj{name: "Name".to_string()}
|
||||||
/// )
|
/// )
|
||||||
/// .with_header("x-version", "1.2.3")
|
/// .with_header(("x-version", "1.2.3"))
|
||||||
/// }
|
/// }
|
||||||
/// # fn main() {}
|
/// # fn main() {}
|
||||||
/// ```
|
/// ```
|
||||||
pub fn with_header<K, V>(mut self, key: K, value: V) -> Self
|
pub fn with_header<H>(mut self, header: H) -> Self
|
||||||
where
|
where
|
||||||
HeaderName: TryFrom<K>,
|
H: IntoHeaderPair,
|
||||||
<HeaderName as TryFrom<K>>::Error: Into<HttpError>,
|
H::Error: Into<HttpError>,
|
||||||
V: IntoHeaderValue,
|
|
||||||
{
|
{
|
||||||
if self.headers.is_none() {
|
if self.headers.is_none() {
|
||||||
self.headers = Some(HeaderMap::new());
|
self.headers = Some(HeaderMap::new());
|
||||||
}
|
}
|
||||||
|
|
||||||
match HeaderName::try_from(key) {
|
match header.try_into_header_pair() {
|
||||||
Ok(key) => match value.try_into() {
|
Ok((key, value)) => self.headers.as_mut().unwrap().append(key, value),
|
||||||
Ok(value) => {
|
|
||||||
self.headers.as_mut().unwrap().append(key, value);
|
|
||||||
}
|
|
||||||
Err(e) => self.error = Some(e.into()),
|
|
||||||
},
|
|
||||||
Err(e) => self.error = Some(e.into()),
|
Err(e) => self.error = Some(e.into()),
|
||||||
};
|
};
|
||||||
|
|
||||||
self
|
self
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue