1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136
use std::iter::FromIterator;
use self::sealed::AsConnectionOption;
use util::FlatCsv;
use {HeaderName, HeaderValue};
/// `Connection` header, defined in
/// [RFC7230](http://tools.ietf.org/html/rfc7230#section-6.1)
///
/// The `Connection` header field allows the sender to indicate desired
/// control options for the current connection. In order to avoid
/// confusing downstream recipients, a proxy or gateway MUST remove or
/// replace any received connection options before forwarding the
/// message.
///
/// # ABNF
///
/// ```text
/// Connection = 1#connection-option
/// connection-option = token
///
/// # Example values
/// * `close`
/// * `keep-alive`
/// * `upgrade`
/// ```
///
/// # Examples
///
/// ```
/// # extern crate headers;
/// use headers::Connection;
///
/// let keep_alive = Connection::keep_alive();
/// ```
// This is frequently just 1 or 2 values, so optimize for that case.
#[derive(Clone, Debug)]
pub struct Connection(FlatCsv);
derive_header! {
Connection(_),
name: CONNECTION
}
impl Connection {
/// A constructor to easily create a `Connection: close` header.
#[inline]
pub fn close() -> Connection {
Connection(HeaderValue::from_static("close").into())
}
/// A constructor to easily create a `Connection: keep-alive` header.
#[inline]
pub fn keep_alive() -> Connection {
Connection(HeaderValue::from_static("keep-alive").into())
}
/// A constructor to easily create a `Connection: Upgrade` header.
#[inline]
pub fn upgrade() -> Connection {
Connection(HeaderValue::from_static("upgrade").into())
}
/// Check if this header contains a given "connection option".
///
/// This can be used with various argument types:
///
/// - `&str`
/// - `&HeaderName`
/// - `HeaderName`
///
/// # Example
///
/// ```
/// # extern crate headers;
/// extern crate http;
///
/// use http::header::UPGRADE;
/// use headers::Connection;
///
/// let conn = Connection::keep_alive();
///
/// assert!(!conn.contains("close"));
/// assert!(!conn.contains(UPGRADE));
/// assert!(conn.contains("keep-alive"));
/// assert!(conn.contains("Keep-Alive"));
/// ```
pub fn contains(&self, name: impl AsConnectionOption) -> bool {
let s = name.as_connection_option();
self.0
.iter()
.find(|&opt| opt.eq_ignore_ascii_case(s))
.is_some()
}
}
impl FromIterator<HeaderName> for Connection {
fn from_iter<I>(iter: I) -> Self
where
I: IntoIterator<Item = HeaderName>,
{
let flat = iter.into_iter().map(HeaderValue::from).collect();
Connection(flat)
}
}
mod sealed {
pub trait AsConnectionOption: Sealed {
fn as_connection_option(&self) -> &str;
}
pub trait Sealed {}
impl<'a> AsConnectionOption for &'a str {
fn as_connection_option(&self) -> &str {
*self
}
}
impl<'a> Sealed for &'a str {}
impl<'a> AsConnectionOption for &'a ::HeaderName {
fn as_connection_option(&self) -> &str {
self.as_ref()
}
}
impl<'a> Sealed for &'a ::HeaderName {}
impl AsConnectionOption for ::HeaderName {
fn as_connection_option(&self) -> &str {
self.as_ref()
}
}
impl Sealed for ::HeaderName {}
}