1
0
mirror of https://github.com/fafhrd91/actix-web synced 2024-11-24 16:32:59 +01:00
actix-web/src/httpresponse.rs

1152 lines
35 KiB
Rust
Raw Normal View History

2018-01-28 07:03:03 +01:00
//! Http response
2018-03-20 23:51:19 +01:00
use std::cell::UnsafeCell;
2017-12-16 01:24:15 +01:00
use std::collections::VecDeque;
2018-04-14 01:02:01 +02:00
use std::io::Write;
use std::rc::Rc;
use std::{fmt, mem, str};
2017-10-07 06:48:14 +02:00
2018-04-14 01:02:01 +02:00
use bytes::{BufMut, Bytes, BytesMut};
2018-01-28 07:03:03 +01:00
use cookie::{Cookie, CookieJar};
2018-03-08 02:40:13 +01:00
use futures::Stream;
2017-10-10 08:07:32 +02:00
use http::header::{self, HeaderName, HeaderValue};
2018-04-14 01:02:01 +02:00
use http::{Error as HttpError, HeaderMap, HttpTryFrom, StatusCode, Version};
2017-11-27 19:39:47 +01:00
use serde::Serialize;
2018-04-14 01:02:01 +02:00
use serde_json;
2017-12-08 01:40:29 +01:00
2017-10-24 08:25:32 +02:00
use body::Body;
2018-04-14 01:02:01 +02:00
use client::ClientResponse;
2017-11-25 18:03:44 +01:00
use error::Error;
2017-12-14 18:43:42 +01:00
use handler::Responder;
2018-04-14 01:02:01 +02:00
use header::{ContentEncoding, Header, IntoHeaderValue};
use httpmessage::HttpMessage;
2018-04-14 01:02:01 +02:00
use httprequest::HttpRequest;
2018-03-09 19:00:15 +01:00
/// max write buffer size 64k
pub(crate) const MAX_WRITE_BUFFER_SIZE: usize = 65_536;
/// Represents various types of connection
2017-10-08 06:48:00 +02:00
#[derive(Copy, Clone, PartialEq, Debug)]
pub enum ConnectionType {
/// Close connection after response
2017-10-08 06:48:00 +02:00
Close,
/// Keep connection alive after response
2017-10-08 06:48:00 +02:00
KeepAlive,
/// Connection is upgraded to different type
2017-10-08 06:48:00 +02:00
Upgrade,
}
2017-12-16 01:24:15 +01:00
/// An HTTP Response
2018-04-14 01:02:01 +02:00
pub struct HttpResponse(
Option<Box<InnerHttpResponse>>,
Rc<UnsafeCell<HttpResponsePool>>,
);
2017-12-16 01:24:15 +01:00
impl Drop for HttpResponse {
fn drop(&mut self) {
if let Some(inner) = self.0.take() {
HttpResponsePool::release(&self.1, inner)
2017-12-16 01:24:15 +01:00
}
}
}
2017-10-07 08:14:13 +02:00
impl HttpResponse {
2017-12-16 01:24:15 +01:00
#[inline(always)]
2017-12-16 05:00:12 +01:00
#[cfg_attr(feature = "cargo-clippy", allow(inline_always))]
2017-12-16 01:24:15 +01:00
fn get_ref(&self) -> &InnerHttpResponse {
self.0.as_ref().unwrap()
}
#[inline(always)]
2017-12-16 05:00:12 +01:00
#[cfg_attr(feature = "cargo-clippy", allow(inline_always))]
2017-12-16 01:24:15 +01:00
fn get_mut(&mut self) -> &mut InnerHttpResponse {
self.0.as_mut().unwrap()
}
2017-11-27 07:31:29 +01:00
/// Create http response builder with specific status.
2017-10-11 01:03:32 +02:00
#[inline]
2017-11-27 07:31:29 +01:00
pub fn build(status: StatusCode) -> HttpResponseBuilder {
HttpResponsePool::get(status)
2017-10-11 01:03:32 +02:00
}
/// Create http response builder
#[inline]
pub fn build_from<T: Into<HttpResponseBuilder>>(source: T) -> HttpResponseBuilder {
source.into()
}
2017-10-07 08:14:13 +02:00
/// Constructs a response
2017-10-07 06:48:14 +02:00
#[inline]
pub fn new(status: StatusCode) -> HttpResponse {
HttpResponsePool::with_body(status, Body::Empty)
}
/// Constructs a response with body
#[inline]
pub fn with_body<B: Into<Body>>(status: StatusCode, body: B) -> HttpResponse {
HttpResponsePool::with_body(status, body.into())
2017-10-07 06:48:14 +02:00
}
2017-11-25 18:03:44 +01:00
/// Constructs a error response
#[inline]
pub fn from_error(error: Error) -> HttpResponse {
let mut resp = error.cause().error_response();
2017-12-16 01:24:15 +01:00
resp.get_mut().error = Some(error);
2017-11-25 18:03:44 +01:00
resp
}
/// Convert `HttpResponse` to a `HttpResponseBuilder`
#[inline]
pub fn into_builder(mut self) -> HttpResponseBuilder {
let response = self.0.take();
let pool = Some(Rc::clone(&self.1));
HttpResponseBuilder {
response,
pool,
err: None,
2018-04-14 01:02:01 +02:00
cookies: None, // TODO: convert set-cookie headers
}
}
2017-11-25 18:03:44 +01:00
/// The source `error` for this response
#[inline]
pub fn error(&self) -> Option<&Error> {
2017-12-16 01:24:15 +01:00
self.get_ref().error.as_ref()
2017-11-25 18:03:44 +01:00
}
/// Get the HTTP version of this response
2017-10-07 06:48:14 +02:00
#[inline]
2017-10-11 02:14:30 +02:00
pub fn version(&self) -> Option<Version> {
2017-12-16 01:24:15 +01:00
self.get_ref().version
2017-10-07 06:48:14 +02:00
}
/// Get the headers from the response
2017-10-07 06:48:14 +02:00
#[inline]
2017-10-10 08:07:32 +02:00
pub fn headers(&self) -> &HeaderMap {
2017-12-16 01:24:15 +01:00
&self.get_ref().headers
2017-10-07 06:48:14 +02:00
}
/// Get a mutable reference to the headers
2017-10-07 06:48:14 +02:00
#[inline]
2017-10-10 08:07:32 +02:00
pub fn headers_mut(&mut self) -> &mut HeaderMap {
2017-12-16 01:24:15 +01:00
&mut self.get_mut().headers
2017-10-07 06:48:14 +02:00
}
/// Get the response status code
2017-10-07 06:48:14 +02:00
#[inline]
pub fn status(&self) -> StatusCode {
2017-12-16 01:24:15 +01:00
self.get_ref().status
2017-10-07 06:48:14 +02:00
}
/// Set the `StatusCode` for this response
2017-10-07 06:48:14 +02:00
#[inline]
2017-10-11 01:03:32 +02:00
pub fn status_mut(&mut self) -> &mut StatusCode {
2017-12-16 01:24:15 +01:00
&mut self.get_mut().status
2017-10-07 06:48:14 +02:00
}
/// Get custom reason for the response
2017-10-14 19:01:53 +02:00
#[inline]
pub fn reason(&self) -> &str {
2017-12-16 01:24:15 +01:00
if let Some(reason) = self.get_ref().reason {
2017-10-14 19:01:53 +02:00
reason
} else {
2018-04-29 07:55:47 +02:00
self.get_ref().status.canonical_reason().unwrap_or("<unknown status code>")
2017-10-14 19:01:53 +02:00
}
}
/// Set the custom reason for the response
2017-10-08 06:48:00 +02:00
#[inline]
2017-10-11 01:03:32 +02:00
pub fn set_reason(&mut self, reason: &'static str) -> &mut Self {
2017-12-16 01:24:15 +01:00
self.get_mut().reason = Some(reason);
2017-10-08 06:48:00 +02:00
self
}
/// Set connection type
2017-10-14 19:01:53 +02:00
pub fn set_connection_type(&mut self, conn: ConnectionType) -> &mut Self {
2017-12-16 01:24:15 +01:00
self.get_mut().connection_type = Some(conn);
2017-10-08 06:48:00 +02:00
self
}
/// Connection upgrade status
2018-01-01 02:26:32 +01:00
#[inline]
2017-10-08 06:48:00 +02:00
pub fn upgrade(&self) -> bool {
2017-12-16 01:24:15 +01:00
self.get_ref().connection_type == Some(ConnectionType::Upgrade)
2017-10-08 06:48:00 +02:00
}
2017-10-07 06:48:14 +02:00
/// Keep-alive status for this connection
2017-10-11 01:03:32 +02:00
pub fn keep_alive(&self) -> Option<bool> {
2017-12-16 01:24:15 +01:00
if let Some(ct) = self.get_ref().connection_type {
2017-10-23 07:54:11 +02:00
match ct {
ConnectionType::KeepAlive => Some(true),
ConnectionType::Close | ConnectionType::Upgrade => Some(false),
}
2017-10-07 06:48:14 +02:00
} else {
2017-10-11 01:03:32 +02:00
None
2017-10-07 06:48:14 +02:00
}
}
2017-10-08 06:48:00 +02:00
2017-10-07 06:48:14 +02:00
/// is chunked encoding enabled
2018-01-01 02:26:32 +01:00
#[inline]
pub fn chunked(&self) -> Option<bool> {
2017-12-16 01:24:15 +01:00
self.get_ref().chunked
2017-10-07 06:48:14 +02:00
}
/// Content encoding
2018-01-01 02:26:32 +01:00
#[inline]
pub fn content_encoding(&self) -> Option<ContentEncoding> {
self.get_ref().encoding
}
/// Set content encoding
pub fn set_content_encoding(&mut self, enc: ContentEncoding) -> &mut Self {
self.get_mut().encoding = Some(enc);
self
}
2017-10-07 08:14:13 +02:00
/// Get body os this response
2018-01-01 02:26:32 +01:00
#[inline]
2017-10-07 06:48:14 +02:00
pub fn body(&self) -> &Body {
2017-12-16 01:24:15 +01:00
&self.get_ref().body
2017-10-07 06:48:14 +02:00
}
2017-10-08 06:48:00 +02:00
/// Set a body
2017-10-11 01:03:32 +02:00
pub fn set_body<B: Into<Body>>(&mut self, body: B) {
2017-12-16 01:24:15 +01:00
self.get_mut().body = body.into();
2017-10-08 06:48:00 +02:00
}
2017-10-07 08:14:13 +02:00
/// Set a body and return previous body value
2017-10-08 06:48:00 +02:00
pub fn replace_body<B: Into<Body>>(&mut self, body: B) -> Body {
2017-12-16 01:24:15 +01:00
mem::replace(&mut self.get_mut().body, body.into())
2017-10-07 06:48:14 +02:00
}
2017-11-10 21:29:54 +01:00
/// Size of response in bytes, excluding HTTP headers
pub fn response_size(&self) -> u64 {
2017-12-16 01:24:15 +01:00
self.get_ref().response_size
2017-11-10 21:29:54 +01:00
}
/// Set content encoding
pub(crate) fn set_response_size(&mut self, size: u64) {
2017-12-16 01:24:15 +01:00
self.get_mut().response_size = size;
2017-11-10 21:29:54 +01:00
}
2018-03-09 19:00:15 +01:00
/// Set write buffer capacity
pub fn write_buffer_capacity(&self) -> usize {
self.get_ref().write_capacity
}
/// Set write buffer capacity
pub fn set_write_buffer_capacity(&mut self, cap: usize) {
self.get_mut().write_capacity = cap;
}
2017-10-07 06:48:14 +02:00
}
2017-10-11 01:03:32 +02:00
2017-11-09 04:31:25 +01:00
impl fmt::Debug for HttpResponse {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
2018-04-14 01:02:01 +02:00
let res = writeln!(
f,
"\nHttpResponse {:?} {}{}",
self.get_ref().version,
self.get_ref().status,
self.get_ref().reason.unwrap_or("")
);
2018-04-09 23:25:30 +02:00
let _ = writeln!(f, " encoding: {:?}", self.get_ref().encoding);
let _ = writeln!(f, " headers:");
2018-03-08 00:41:46 +01:00
for (key, val) in self.get_ref().headers.iter() {
2018-04-09 23:25:30 +02:00
let _ = writeln!(f, " {:?}: {:?}", key, val);
2017-11-09 04:31:25 +01:00
}
res
}
}
2017-10-11 01:03:32 +02:00
/// An HTTP response builder
///
/// This type can be used to construct an instance of `HttpResponse` through a
/// builder-like pattern.
2017-10-15 23:17:41 +02:00
pub struct HttpResponseBuilder {
2017-12-16 01:24:15 +01:00
response: Option<Box<InnerHttpResponse>>,
pool: Option<Rc<UnsafeCell<HttpResponsePool>>>,
2017-10-22 18:13:29 +02:00
err: Option<HttpError>,
2017-12-14 01:44:35 +01:00
cookies: Option<CookieJar>,
2017-10-11 01:03:32 +02:00
}
2017-10-15 23:17:41 +02:00
impl HttpResponseBuilder {
2018-03-06 09:43:25 +01:00
/// Set HTTP status code of this response.
#[inline]
pub fn status(&mut self, status: StatusCode) -> &mut Self {
if let Some(parts) = parts(&mut self.response, &self.err) {
parts.status = status;
}
self
}
2017-12-21 08:19:21 +01:00
/// Set HTTP version of this response.
///
/// By default response's http version depends on request's version.
2017-10-11 01:03:32 +02:00
#[inline]
pub fn version(&mut self, version: Version) -> &mut Self {
2017-12-14 01:44:35 +01:00
if let Some(parts) = parts(&mut self.response, &self.err) {
2017-10-11 02:14:30 +02:00
parts.version = Some(version);
2017-10-11 01:03:32 +02:00
}
self
}
2018-03-06 04:28:42 +01:00
/// Set a header.
///
/// ```rust
/// # extern crate actix_web;
/// use actix_web::{HttpRequest, HttpResponse, Result, http};
2018-03-06 04:28:42 +01:00
///
/// fn index(req: HttpRequest) -> Result<HttpResponse> {
/// Ok(HttpResponse::Ok()
2018-03-31 02:31:18 +02:00
/// .set(http::header::IfModifiedSince("Sun, 07 Nov 1994 08:48:37 GMT".parse()?))
/// .finish())
2018-03-06 04:28:42 +01:00
/// }
/// fn main() {}
/// ```
#[doc(hidden)]
2018-04-14 01:02:01 +02:00
pub fn set<H: Header>(&mut self, hdr: H) -> &mut Self {
2018-03-06 04:28:42 +01:00
if let Some(parts) = parts(&mut self.response, &self.err) {
match hdr.try_into() {
2018-04-14 01:02:01 +02:00
Ok(value) => {
parts.headers.append(H::name(), value);
}
2018-03-06 04:28:42 +01:00
Err(e) => self.err = Some(e.into()),
}
}
self
}
2017-10-11 01:03:32 +02:00
/// Set a header.
2017-12-21 08:19:21 +01:00
///
/// ```rust
/// # extern crate actix_web;
2018-03-31 09:16:55 +02:00
/// use actix_web::{http, HttpRequest, HttpResponse};
2017-12-21 08:19:21 +01:00
///
/// fn index(req: HttpRequest) -> HttpResponse {
/// HttpResponse::Ok()
2017-12-21 08:19:21 +01:00
/// .header("X-TEST", "value")
/// .header(http::header::CONTENT_TYPE, "application/json")
/// .finish()
2017-12-21 08:19:21 +01:00
/// }
/// fn main() {}
/// ```
2017-10-11 01:03:32 +02:00
pub fn header<K, V>(&mut self, key: K, value: V) -> &mut Self
2018-04-14 01:02:01 +02:00
where
HeaderName: HttpTryFrom<K>,
V: IntoHeaderValue,
2017-10-11 01:03:32 +02:00
{
2017-12-14 01:44:35 +01:00
if let Some(parts) = parts(&mut self.response, &self.err) {
2017-10-11 01:03:32 +02:00
match HeaderName::try_from(key) {
2018-04-14 01:02:01 +02:00
Ok(key) => match value.try_into() {
Ok(value) => {
parts.headers.append(key, value);
2017-10-11 01:03:32 +02:00
}
2018-04-14 01:02:01 +02:00
Err(e) => self.err = Some(e.into()),
2017-10-11 01:03:32 +02:00
},
Err(e) => self.err = Some(e.into()),
};
}
self
}
/// Set the custom reason for the response.
#[inline]
pub fn reason(&mut self, reason: &'static str) -> &mut Self {
2017-12-14 01:44:35 +01:00
if let Some(parts) = parts(&mut self.response, &self.err) {
2017-10-11 01:03:32 +02:00
parts.reason = Some(reason);
}
self
}
2017-11-06 10:24:49 +01:00
/// Set content encoding.
///
/// By default `ContentEncoding::Auto` is used, which automatically
2018-04-14 01:02:01 +02:00
/// negotiates content encoding based on request's `Accept-Encoding`
/// headers. To enforce specific encoding, use specific
/// ContentEncoding` value.
2017-12-04 05:47:15 +01:00
#[inline]
2017-11-06 10:24:49 +01:00
pub fn content_encoding(&mut self, enc: ContentEncoding) -> &mut Self {
2017-12-14 01:44:35 +01:00
if let Some(parts) = parts(&mut self.response, &self.err) {
parts.encoding = Some(enc);
2017-11-06 10:24:49 +01:00
}
self
}
2017-10-11 01:03:32 +02:00
/// Set connection type
2017-12-04 05:47:15 +01:00
#[inline]
2017-12-21 08:19:21 +01:00
#[doc(hidden)]
2017-10-14 19:01:53 +02:00
pub fn connection_type(&mut self, conn: ConnectionType) -> &mut Self {
2017-12-14 01:44:35 +01:00
if let Some(parts) = parts(&mut self.response, &self.err) {
2017-10-11 01:03:32 +02:00
parts.connection_type = Some(conn);
}
self
}
2017-10-14 19:01:53 +02:00
/// Set connection type to Upgrade
2017-12-04 05:47:15 +01:00
#[inline]
2017-12-21 08:19:21 +01:00
#[doc(hidden)]
2017-10-14 19:01:53 +02:00
pub fn upgrade(&mut self) -> &mut Self {
self.connection_type(ConnectionType::Upgrade)
}
/// Force close connection, even if it is marked as keep-alive
2017-12-04 05:47:15 +01:00
#[inline]
2017-10-14 19:01:53 +02:00
pub fn force_close(&mut self) -> &mut Self {
self.connection_type(ConnectionType::Close)
}
2017-10-11 01:03:32 +02:00
/// Enables automatic chunked transfer encoding
2017-12-04 05:47:15 +01:00
#[inline]
2017-12-14 07:36:28 +01:00
pub fn chunked(&mut self) -> &mut Self {
2017-12-14 01:44:35 +01:00
if let Some(parts) = parts(&mut self.response, &self.err) {
parts.chunked = Some(true);
}
self
}
/// Force disable chunked encoding
#[inline]
pub fn no_chunking(&mut self) -> &mut Self {
if let Some(parts) = parts(&mut self.response, &self.err) {
parts.chunked = Some(false);
2017-10-11 01:03:32 +02:00
}
self
}
2017-10-14 19:01:53 +02:00
/// Set response content type
2017-12-04 05:47:15 +01:00
#[inline]
2017-10-14 19:01:53 +02:00
pub fn content_type<V>(&mut self, value: V) -> &mut Self
2018-04-14 01:02:01 +02:00
where
HeaderValue: HttpTryFrom<V>,
2017-10-14 19:01:53 +02:00
{
2017-12-14 01:44:35 +01:00
if let Some(parts) = parts(&mut self.response, &self.err) {
2017-10-14 19:01:53 +02:00
match HeaderValue::try_from(value) {
2018-04-14 01:02:01 +02:00
Ok(value) => {
parts.headers.insert(header::CONTENT_TYPE, value);
}
2017-10-14 19:01:53 +02:00
Err(e) => self.err = Some(e.into()),
};
}
self
}
2018-01-12 01:22:27 +01:00
/// Set content length
#[inline]
pub fn content_length(&mut self, len: u64) -> &mut Self {
let mut wrt = BytesMut::new().writer();
let _ = write!(wrt, "{}", len);
self.header(header::CONTENT_LENGTH, wrt.get_mut().take().freeze())
}
2017-10-14 19:40:58 +02:00
/// Set a cookie
2017-12-21 08:19:21 +01:00
///
/// ```rust
/// # extern crate actix_web;
/// use actix_web::{http, HttpRequest, HttpResponse, Result};
2017-12-21 08:19:21 +01:00
///
/// fn index(req: HttpRequest) -> HttpResponse {
/// HttpResponse::Ok()
2017-12-21 08:19:21 +01:00
/// .cookie(
2018-03-31 02:31:18 +02:00
/// http::Cookie::build("name", "value")
2017-12-21 08:19:21 +01:00
/// .domain("www.rust-lang.org")
/// .path("/")
/// .secure(true)
/// .http_only(true)
/// .finish())
/// .finish()
2017-12-21 08:19:21 +01:00
/// }
/// fn main() {}
/// ```
2017-10-14 19:40:58 +02:00
pub fn cookie<'c>(&mut self, cookie: Cookie<'c>) -> &mut Self {
2017-12-14 01:44:35 +01:00
if self.cookies.is_none() {
let mut jar = CookieJar::new();
jar.add(cookie.into_owned());
self.cookies = Some(jar)
} else {
2018-04-29 07:55:47 +02:00
self.cookies.as_mut().unwrap().add(cookie.into_owned());
2017-10-14 19:40:58 +02:00
}
self
}
2018-04-14 01:02:01 +02:00
/// Remove cookie, cookie has to be cookie from `HttpRequest::cookies()`
/// method.
2017-10-14 19:40:58 +02:00
pub fn del_cookie<'a>(&mut self, cookie: &Cookie<'a>) -> &mut Self {
2017-12-14 01:44:35 +01:00
{
if self.cookies.is_none() {
self.cookies = Some(CookieJar::new())
2017-12-13 06:32:58 +01:00
}
2017-12-14 01:44:35 +01:00
let jar = self.cookies.as_mut().unwrap();
2017-10-14 19:40:58 +02:00
let cookie = cookie.clone().into_owned();
2017-12-13 06:32:58 +01:00
jar.add_original(cookie.clone());
jar.remove(cookie);
2017-10-14 19:40:58 +02:00
}
self
}
2018-04-14 01:02:01 +02:00
/// This method calls provided closure with builder reference if value is
/// true.
2017-10-29 22:51:02 +01:00
pub fn if_true<F>(&mut self, value: bool, f: F) -> &mut Self
2018-04-14 01:02:01 +02:00
where
F: FnOnce(&mut HttpResponseBuilder),
2017-10-29 22:51:02 +01:00
{
if value {
f(self);
}
self
}
2018-04-14 01:02:01 +02:00
/// This method calls provided closure with builder reference if value is
/// Some.
pub fn if_some<T, F>(&mut self, value: Option<T>, f: F) -> &mut Self
2018-04-14 01:02:01 +02:00
where
F: FnOnce(T, &mut HttpResponseBuilder),
2018-01-10 08:55:42 +01:00
{
if let Some(val) = value {
f(val, self);
}
self
}
2018-03-09 19:00:15 +01:00
/// Set write buffer capacity
///
/// This parameter makes sense only for streaming response
2018-04-14 01:02:01 +02:00
/// or actor. If write buffer reaches specified capacity, stream or actor
/// get paused.
2018-03-09 19:00:15 +01:00
///
/// Default write buffer capacity is 64kb
pub fn write_buffer_capacity(&mut self, cap: usize) -> &mut Self {
if let Some(parts) = parts(&mut self.response, &self.err) {
parts.write_capacity = cap;
}
self
}
2017-11-27 19:39:47 +01:00
/// Set a body and generate `HttpResponse`.
2017-12-21 08:19:21 +01:00
///
2017-11-27 19:39:47 +01:00
/// `HttpResponseBuilder` can not be used after this call.
pub fn body<B: Into<Body>>(&mut self, body: B) -> HttpResponse {
2017-10-11 01:03:32 +02:00
if let Some(e) = self.err.take() {
2018-04-14 01:02:01 +02:00
return Error::from(e).into();
2017-10-11 01:03:32 +02:00
}
2018-04-29 07:55:47 +02:00
let mut response = self.response.take().expect("cannot reuse response builder");
2017-12-14 01:44:35 +01:00
if let Some(ref jar) = self.cookies {
2017-12-13 06:32:58 +01:00
for cookie in jar.delta() {
match HeaderValue::from_str(&cookie.to_string()) {
Ok(val) => response.headers.append(header::SET_COOKIE, val),
Err(e) => return Error::from(e).into(),
};
2017-12-13 06:32:58 +01:00
}
2017-10-14 19:40:58 +02:00
}
2017-12-14 01:44:35 +01:00
response.body = body.into();
HttpResponse(Some(response), self.pool.take().unwrap())
2017-10-11 01:03:32 +02:00
}
2017-10-24 08:52:20 +02:00
#[inline]
2018-03-08 02:40:13 +01:00
/// Set a streaming body and generate `HttpResponse`.
///
/// `HttpResponseBuilder` can not be used after this call.
pub fn streaming<S, E>(&mut self, stream: S) -> HttpResponse
2018-04-14 01:02:01 +02:00
where
S: Stream<Item = Bytes, Error = E> + 'static,
E: Into<Error>,
2018-03-08 02:40:13 +01:00
{
2018-04-29 07:55:47 +02:00
self.body(Body::Streaming(Box::new(stream.map_err(|e| e.into()))))
2018-03-08 02:40:13 +01:00
}
2017-11-27 19:39:47 +01:00
/// Set a json body and generate `HttpResponse`
2017-12-21 08:19:21 +01:00
///
/// `HttpResponseBuilder` can not be used after this call.
pub fn json<T: Serialize>(&mut self, value: T) -> HttpResponse {
match serde_json::to_string(&value) {
Ok(body) => {
2018-04-14 01:02:01 +02:00
let contains = if let Some(parts) = parts(&mut self.response, &self.err)
{
parts.headers.contains_key(header::CONTENT_TYPE)
} else {
true
};
if !contains {
self.header(header::CONTENT_TYPE, "application/json");
}
self.body(body)
2018-04-14 01:02:01 +02:00
}
Err(e) => Error::from(e).into(),
2017-11-27 19:39:47 +01:00
}
}
#[inline]
2017-11-27 19:39:47 +01:00
/// Set an empty body and generate `HttpResponse`
2017-12-21 08:19:21 +01:00
///
/// `HttpResponseBuilder` can not be used after this call.
pub fn finish(&mut self) -> HttpResponse {
2017-10-24 08:52:20 +02:00
self.body(Body::Empty)
}
2018-01-01 02:26:32 +01:00
/// This method construct new `HttpResponseBuilder`
pub fn take(&mut self) -> HttpResponseBuilder {
HttpResponseBuilder {
response: self.response.take(),
2018-03-20 23:51:19 +01:00
pool: self.pool.take(),
2018-01-01 02:26:32 +01:00
err: self.err.take(),
cookies: self.cookies.take(),
}
}
2017-10-11 01:03:32 +02:00
}
2017-12-16 05:00:12 +01:00
#[inline]
2017-12-16 01:24:15 +01:00
#[cfg_attr(feature = "cargo-clippy", allow(borrowed_box))]
2018-04-14 01:02:01 +02:00
fn parts<'a>(
2018-04-29 07:55:47 +02:00
parts: &'a mut Option<Box<InnerHttpResponse>>, err: &Option<HttpError>,
2018-04-14 01:02:01 +02:00
) -> Option<&'a mut Box<InnerHttpResponse>> {
2017-10-11 01:03:32 +02:00
if err.is_some() {
2018-04-14 01:02:01 +02:00
return None;
2017-10-11 01:03:32 +02:00
}
parts.as_mut()
}
2017-10-23 07:54:11 +02:00
2017-11-28 22:52:53 +01:00
/// Helper converters
impl<I: Into<HttpResponse>, E: Into<Error>> From<Result<I, E>> for HttpResponse {
fn from(res: Result<I, E>) -> Self {
match res {
Ok(val) => val.into(),
Err(err) => err.into().into(),
}
}
}
2017-11-29 18:17:00 +01:00
impl From<HttpResponseBuilder> for HttpResponse {
fn from(mut builder: HttpResponseBuilder) -> Self {
builder.finish()
2017-11-29 18:17:00 +01:00
}
}
2017-12-14 18:43:42 +01:00
impl Responder for HttpResponseBuilder {
type Item = HttpResponse;
type Error = Error;
2017-12-16 05:00:12 +01:00
#[inline]
fn respond_to(mut self, _: HttpRequest) -> Result<HttpResponse, Error> {
Ok(self.finish())
2017-12-03 02:14:55 +01:00
}
}
2017-11-28 22:52:53 +01:00
impl From<&'static str> for HttpResponse {
2017-11-29 18:17:00 +01:00
fn from(val: &'static str) -> Self {
2018-04-29 07:55:47 +02:00
HttpResponse::Ok().content_type("text/plain; charset=utf-8").body(val)
2017-11-28 22:52:53 +01:00
}
}
2017-12-14 18:43:42 +01:00
impl Responder for &'static str {
type Item = HttpResponse;
type Error = Error;
2018-04-04 02:37:17 +02:00
fn respond_to(self, req: HttpRequest) -> Result<HttpResponse, Error> {
Ok(req.build_response(StatusCode::OK)
2018-04-14 01:02:01 +02:00
.content_type("text/plain; charset=utf-8")
.body(self))
2017-12-03 02:14:55 +01:00
}
}
2017-11-28 22:52:53 +01:00
impl From<&'static [u8]> for HttpResponse {
2017-11-29 18:17:00 +01:00
fn from(val: &'static [u8]) -> Self {
2018-04-29 07:55:47 +02:00
HttpResponse::Ok().content_type("application/octet-stream").body(val)
2017-11-28 22:52:53 +01:00
}
}
2017-12-14 18:43:42 +01:00
impl Responder for &'static [u8] {
type Item = HttpResponse;
type Error = Error;
2018-04-04 02:37:17 +02:00
fn respond_to(self, req: HttpRequest) -> Result<HttpResponse, Error> {
Ok(req.build_response(StatusCode::OK)
2018-04-14 01:02:01 +02:00
.content_type("application/octet-stream")
.body(self))
2017-12-03 02:14:55 +01:00
}
}
2017-11-28 22:52:53 +01:00
impl From<String> for HttpResponse {
2017-11-29 18:17:00 +01:00
fn from(val: String) -> Self {
2018-04-29 07:55:47 +02:00
HttpResponse::Ok().content_type("text/plain; charset=utf-8").body(val)
2017-11-28 22:52:53 +01:00
}
}
2017-12-14 18:43:42 +01:00
impl Responder for String {
type Item = HttpResponse;
type Error = Error;
2018-04-04 02:37:17 +02:00
fn respond_to(self, req: HttpRequest) -> Result<HttpResponse, Error> {
Ok(req.build_response(StatusCode::OK)
2018-04-14 01:02:01 +02:00
.content_type("text/plain; charset=utf-8")
.body(self))
2017-12-03 02:14:55 +01:00
}
}
2017-11-28 22:52:53 +01:00
impl<'a> From<&'a String> for HttpResponse {
2017-11-29 18:17:00 +01:00
fn from(val: &'a String) -> Self {
2017-11-28 22:52:53 +01:00
HttpResponse::build(StatusCode::OK)
.content_type("text/plain; charset=utf-8")
.body(val)
}
}
2017-12-14 18:43:42 +01:00
impl<'a> Responder for &'a String {
type Item = HttpResponse;
type Error = Error;
2018-04-04 02:37:17 +02:00
fn respond_to(self, req: HttpRequest) -> Result<HttpResponse, Error> {
Ok(req.build_response(StatusCode::OK)
2018-04-14 01:02:01 +02:00
.content_type("text/plain; charset=utf-8")
.body(self))
2017-12-03 02:14:55 +01:00
}
}
2017-11-28 22:52:53 +01:00
impl From<Bytes> for HttpResponse {
2017-11-29 18:17:00 +01:00
fn from(val: Bytes) -> Self {
2018-04-29 07:55:47 +02:00
HttpResponse::Ok().content_type("application/octet-stream").body(val)
2017-11-28 22:52:53 +01:00
}
}
2017-12-14 18:43:42 +01:00
impl Responder for Bytes {
type Item = HttpResponse;
type Error = Error;
2018-04-04 02:37:17 +02:00
fn respond_to(self, req: HttpRequest) -> Result<HttpResponse, Error> {
Ok(req.build_response(StatusCode::OK)
2018-04-14 01:02:01 +02:00
.content_type("application/octet-stream")
.body(self))
2017-12-03 02:14:55 +01:00
}
}
2017-11-28 22:52:53 +01:00
impl From<BytesMut> for HttpResponse {
2017-11-29 18:17:00 +01:00
fn from(val: BytesMut) -> Self {
2018-04-29 07:55:47 +02:00
HttpResponse::Ok().content_type("application/octet-stream").body(val)
2017-11-28 22:52:53 +01:00
}
}
2017-12-14 18:43:42 +01:00
impl Responder for BytesMut {
type Item = HttpResponse;
type Error = Error;
2018-04-04 02:37:17 +02:00
fn respond_to(self, req: HttpRequest) -> Result<HttpResponse, Error> {
Ok(req.build_response(StatusCode::OK)
2018-04-14 01:02:01 +02:00
.content_type("application/octet-stream")
.body(self))
2017-12-03 02:14:55 +01:00
}
}
/// Create `HttpResponseBuilder` from `ClientResponse`
///
/// It is useful for proxy response. This implementation
/// copies all responses's headers and status.
impl<'a> From<&'a ClientResponse> for HttpResponseBuilder {
fn from(resp: &'a ClientResponse) -> HttpResponseBuilder {
let mut builder = HttpResponse::build(resp.status());
for (key, value) in resp.headers() {
builder.header(key.clone(), value.clone());
}
builder
}
}
impl<'a, S> From<&'a HttpRequest<S>> for HttpResponseBuilder {
fn from(req: &'a HttpRequest<S>) -> HttpResponseBuilder {
if let Some(router) = req.router() {
2018-04-29 07:55:47 +02:00
router.server_settings().get_response_builder(StatusCode::OK)
} else {
HttpResponse::Ok()
}
}
}
2017-12-16 05:00:12 +01:00
#[derive(Debug)]
struct InnerHttpResponse {
version: Option<Version>,
headers: HeaderMap,
status: StatusCode,
reason: Option<&'static str>,
body: Body,
chunked: Option<bool>,
encoding: Option<ContentEncoding>,
2017-12-16 05:00:12 +01:00
connection_type: Option<ConnectionType>,
2018-03-09 19:00:15 +01:00
write_capacity: usize,
2017-12-16 05:00:12 +01:00
response_size: u64,
error: Option<Error>,
}
impl InnerHttpResponse {
#[inline]
fn new(status: StatusCode, body: Body) -> InnerHttpResponse {
InnerHttpResponse {
2018-02-26 23:33:56 +01:00
status,
body,
2017-12-16 05:00:12 +01:00
version: None,
2017-12-16 07:49:48 +01:00
headers: HeaderMap::with_capacity(16),
2017-12-16 05:00:12 +01:00
reason: None,
chunked: None,
encoding: None,
2017-12-16 05:00:12 +01:00
connection_type: None,
response_size: 0,
2018-03-09 19:00:15 +01:00
write_capacity: MAX_WRITE_BUFFER_SIZE,
2017-12-16 05:00:12 +01:00
error: None,
}
}
}
/// Internal use only! unsafe
pub(crate) struct HttpResponsePool(VecDeque<Box<InnerHttpResponse>>);
thread_local!(static POOL: Rc<UnsafeCell<HttpResponsePool>> = HttpResponsePool::pool());
2017-12-16 05:00:12 +01:00
impl HttpResponsePool {
pub fn pool() -> Rc<UnsafeCell<HttpResponsePool>> {
2018-04-29 07:55:47 +02:00
Rc::new(UnsafeCell::new(HttpResponsePool(VecDeque::with_capacity(128))))
}
2017-12-16 05:00:12 +01:00
2017-12-16 07:49:48 +01:00
#[inline]
2018-04-14 01:02:01 +02:00
pub fn get_builder(
2018-04-29 07:55:47 +02:00
pool: &Rc<UnsafeCell<HttpResponsePool>>, status: StatusCode,
2018-04-14 01:02:01 +02:00
) -> HttpResponseBuilder {
let p = unsafe { &mut *pool.as_ref().get() };
if let Some(mut msg) = p.0.pop_front() {
msg.status = status;
HttpResponseBuilder {
response: Some(msg),
pool: Some(Rc::clone(pool)),
err: None,
2018-04-14 01:02:01 +02:00
cookies: None,
}
} else {
let msg = Box::new(InnerHttpResponse::new(status, Body::Empty));
HttpResponseBuilder {
response: Some(msg),
pool: Some(Rc::clone(pool)),
err: None,
2018-04-14 01:02:01 +02:00
cookies: None,
}
}
2017-12-16 05:00:12 +01:00
}
2017-12-16 07:49:48 +01:00
#[inline]
2018-04-14 01:02:01 +02:00
pub fn get_response(
2018-04-29 07:55:47 +02:00
pool: &Rc<UnsafeCell<HttpResponsePool>>, status: StatusCode, body: Body,
2018-04-14 01:02:01 +02:00
) -> HttpResponse {
let p = unsafe { &mut *pool.as_ref().get() };
if let Some(mut msg) = p.0.pop_front() {
msg.status = status;
msg.body = body;
HttpResponse(Some(msg), Rc::clone(pool))
} else {
let msg = Box::new(InnerHttpResponse::new(status, body));
HttpResponse(Some(msg), Rc::clone(pool))
}
}
#[inline]
fn get(status: StatusCode) -> HttpResponseBuilder {
POOL.with(|pool| HttpResponsePool::get_builder(pool, status))
}
#[inline]
fn with_body(status: StatusCode, body: Body) -> HttpResponse {
POOL.with(|pool| HttpResponsePool::get_response(pool, status, body))
2017-12-16 05:00:12 +01:00
}
2017-12-16 07:49:48 +01:00
#[inline(always)]
#[cfg_attr(feature = "cargo-clippy", allow(boxed_local, inline_always))]
2018-04-14 01:02:01 +02:00
fn release(
2018-04-29 07:55:47 +02:00
pool: &Rc<UnsafeCell<HttpResponsePool>>, mut inner: Box<InnerHttpResponse>,
2018-04-14 01:02:01 +02:00
) {
let pool = unsafe { &mut *pool.as_ref().get() };
2018-03-20 23:51:19 +01:00
if pool.0.len() < 128 {
inner.headers.clear();
inner.version = None;
inner.chunked = None;
inner.reason = None;
inner.encoding = None;
inner.connection_type = None;
inner.response_size = 0;
inner.error = None;
inner.write_capacity = MAX_WRITE_BUFFER_SIZE;
pool.0.push_front(inner);
}
2017-12-16 05:00:12 +01:00
}
}
2017-10-23 07:54:11 +02:00
#[cfg(test)]
mod tests {
use super::*;
2017-11-28 22:52:53 +01:00
use body::Binary;
use http;
2018-04-14 01:02:01 +02:00
use http::header::{HeaderValue, CONTENT_TYPE, COOKIE};
use http::{Method, Uri};
use std::str::FromStr;
use time::Duration;
2017-12-13 20:16:26 +01:00
#[test]
fn test_debug() {
2018-03-08 07:40:46 +01:00
let resp = HttpResponse::Ok()
.header(COOKIE, HeaderValue::from_static("cookie1=value1; "))
.header(COOKIE, HeaderValue::from_static("cookie2=value2; "))
.finish();
2017-12-13 20:16:26 +01:00
let dbg = format!("{:?}", resp);
assert!(dbg.contains("HttpResponse"));
}
#[test]
fn test_response_cookies() {
let mut headers = HeaderMap::new();
2018-03-08 20:16:54 +01:00
headers.insert(COOKIE, HeaderValue::from_static("cookie1=value1"));
headers.insert(COOKIE, HeaderValue::from_static("cookie2=value2"));
2017-12-13 20:16:26 +01:00
let req = HttpRequest::new(
2018-04-14 01:02:01 +02:00
Method::GET,
Uri::from_str("/").unwrap(),
Version::HTTP_11,
headers,
None,
);
2017-12-13 20:16:26 +01:00
let cookies = req.cookies().unwrap();
let resp = HttpResponse::Ok()
2018-04-14 01:02:01 +02:00
.cookie(
http::Cookie::build("name", "value")
2017-12-13 20:16:26 +01:00
.domain("www.rust-lang.org")
.path("/test")
.http_only(true)
.max_age(Duration::days(1))
2018-04-14 01:02:01 +02:00
.finish(),
)
2017-12-13 20:16:26 +01:00
.del_cookie(&cookies[0])
.finish();
2017-12-13 20:16:26 +01:00
2018-04-14 01:02:01 +02:00
let mut val: Vec<_> = resp.headers()
.get_all("Set-Cookie")
.iter()
.map(|v| v.to_str().unwrap().to_owned())
.collect();
2017-12-13 20:16:26 +01:00
val.sort();
2018-03-08 20:16:54 +01:00
assert!(val[0].starts_with("cookie2=; Max-Age=0;"));
2017-12-13 20:16:26 +01:00
assert_eq!(
2018-04-14 01:02:01 +02:00
val[1],
"name=value; HttpOnly; Path=/test; Domain=www.rust-lang.org; Max-Age=86400"
);
2017-12-13 20:16:26 +01:00
}
2017-10-23 07:54:11 +02:00
2017-11-28 23:29:22 +01:00
#[test]
fn test_basic_builder() {
let resp = HttpResponse::Ok()
2017-12-04 05:09:46 +01:00
.header("X-TEST", "value")
2017-11-28 23:29:22 +01:00
.version(Version::HTTP_10)
.finish();
2017-11-28 23:29:22 +01:00
assert_eq!(resp.version(), Some(Version::HTTP_10));
2017-12-21 08:36:52 +01:00
assert_eq!(resp.status(), StatusCode::OK);
2017-11-28 23:29:22 +01:00
}
2017-10-23 07:54:11 +02:00
#[test]
fn test_upgrade() {
let resp = HttpResponse::build(StatusCode::OK).upgrade().finish();
2017-10-23 07:54:11 +02:00
assert!(resp.upgrade())
}
#[test]
fn test_force_close() {
2018-04-29 07:55:47 +02:00
let resp = HttpResponse::build(StatusCode::OK).force_close().finish();
2017-10-23 07:54:11 +02:00
assert!(!resp.keep_alive().unwrap())
}
#[test]
fn test_content_type() {
2017-11-27 07:31:29 +01:00
let resp = HttpResponse::build(StatusCode::OK)
2018-04-14 01:02:01 +02:00
.content_type("text/plain")
.body(Body::Empty);
2018-04-29 07:55:47 +02:00
assert_eq!(resp.headers().get(CONTENT_TYPE).unwrap(), "text/plain")
2017-10-23 07:54:11 +02:00
}
#[test]
fn test_content_encoding() {
let resp = HttpResponse::build(StatusCode::OK).finish();
assert_eq!(resp.content_encoding(), None);
2018-04-14 01:02:01 +02:00
#[cfg(feature = "brotli")]
{
let resp = HttpResponse::build(StatusCode::OK)
2018-04-14 01:02:01 +02:00
.content_encoding(ContentEncoding::Br)
.finish();
assert_eq!(resp.content_encoding(), Some(ContentEncoding::Br));
}
2017-11-27 07:31:29 +01:00
let resp = HttpResponse::build(StatusCode::OK)
2018-04-14 01:02:01 +02:00
.content_encoding(ContentEncoding::Gzip)
.finish();
assert_eq!(resp.content_encoding(), Some(ContentEncoding::Gzip));
2017-11-07 01:23:58 +01:00
}
2017-11-27 19:39:47 +01:00
#[test]
fn test_json() {
2018-04-14 01:02:01 +02:00
let resp = HttpResponse::build(StatusCode::OK).json(vec!["v1", "v2", "v3"]);
2018-03-06 04:28:42 +01:00
let ct = resp.headers().get(CONTENT_TYPE).unwrap();
2018-03-06 09:43:25 +01:00
assert_eq!(ct, HeaderValue::from_static("application/json"));
2018-04-14 01:02:01 +02:00
assert_eq!(
*resp.body(),
Body::from(Bytes::from_static(b"[\"v1\",\"v2\",\"v3\"]"))
);
2017-11-27 19:39:47 +01:00
}
#[test]
fn test_json_ct() {
let resp = HttpResponse::build(StatusCode::OK)
2018-03-06 04:28:42 +01:00
.header(CONTENT_TYPE, "text/json")
.json(vec!["v1", "v2", "v3"]);
2018-03-06 04:28:42 +01:00
let ct = resp.headers().get(CONTENT_TYPE).unwrap();
2018-03-06 09:43:25 +01:00
assert_eq!(ct, HeaderValue::from_static("text/json"));
2018-04-14 01:02:01 +02:00
assert_eq!(
*resp.body(),
Body::from(Bytes::from_static(b"[\"v1\",\"v2\",\"v3\"]"))
);
2017-11-27 19:39:47 +01:00
}
2017-11-28 22:52:53 +01:00
impl Body {
pub(crate) fn binary(&self) -> Option<&Binary> {
match *self {
Body::Binary(ref bin) => Some(bin),
_ => None,
}
}
}
#[test]
fn test_into_response() {
let req = HttpRequest::default();
2017-11-28 22:52:53 +01:00
let resp: HttpResponse = "test".into();
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("text/plain; charset=utf-8")
);
2017-11-28 22:52:53 +01:00
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().binary().unwrap(), &Binary::from("test"));
2017-12-14 18:43:42 +01:00
let resp: HttpResponse = "test".respond_to(req.clone()).ok().unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("text/plain; charset=utf-8")
);
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().binary().unwrap(), &Binary::from("test"));
2017-11-28 22:52:53 +01:00
let resp: HttpResponse = b"test".as_ref().into();
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("application/octet-stream")
);
2017-11-28 22:52:53 +01:00
assert_eq!(resp.status(), StatusCode::OK);
2018-04-29 07:55:47 +02:00
assert_eq!(resp.body().binary().unwrap(), &Binary::from(b"test".as_ref()));
2017-11-28 22:52:53 +01:00
2017-12-14 18:43:42 +01:00
let resp: HttpResponse = b"test".as_ref().respond_to(req.clone()).ok().unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("application/octet-stream")
);
assert_eq!(resp.status(), StatusCode::OK);
2018-04-29 07:55:47 +02:00
assert_eq!(resp.body().binary().unwrap(), &Binary::from(b"test".as_ref()));
2017-11-28 22:52:53 +01:00
let resp: HttpResponse = "test".to_owned().into();
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("text/plain; charset=utf-8")
);
2017-11-28 22:52:53 +01:00
assert_eq!(resp.status(), StatusCode::OK);
2018-04-29 07:55:47 +02:00
assert_eq!(resp.body().binary().unwrap(), &Binary::from("test".to_owned()));
2018-04-14 01:02:01 +02:00
2018-04-29 07:55:47 +02:00
let resp: HttpResponse = "test".to_owned().respond_to(req.clone()).ok().unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("text/plain; charset=utf-8")
);
assert_eq!(resp.status(), StatusCode::OK);
2018-04-29 07:55:47 +02:00
assert_eq!(resp.body().binary().unwrap(), &Binary::from("test".to_owned()));
2017-11-28 22:52:53 +01:00
let resp: HttpResponse = (&"test".to_owned()).into();
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("text/plain; charset=utf-8")
);
2017-11-28 22:52:53 +01:00
assert_eq!(resp.status(), StatusCode::OK);
2018-04-29 07:55:47 +02:00
assert_eq!(resp.body().binary().unwrap(), &Binary::from(&"test".to_owned()));
2018-04-14 01:02:01 +02:00
2018-04-29 07:55:47 +02:00
let resp: HttpResponse =
(&"test".to_owned()).respond_to(req.clone()).ok().unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("text/plain; charset=utf-8")
);
assert_eq!(resp.status(), StatusCode::OK);
2018-04-29 07:55:47 +02:00
assert_eq!(resp.body().binary().unwrap(), &Binary::from(&"test".to_owned()));
2017-11-28 22:52:53 +01:00
let b = Bytes::from_static(b"test");
let resp: HttpResponse = b.into();
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("application/octet-stream")
);
2017-11-28 22:52:53 +01:00
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.body().binary().unwrap(),
&Binary::from(Bytes::from_static(b"test"))
);
2017-11-28 22:52:53 +01:00
let b = Bytes::from_static(b"test");
2017-12-14 18:43:42 +01:00
let resp: HttpResponse = b.respond_to(req.clone()).ok().unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("application/octet-stream")
);
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.body().binary().unwrap(),
&Binary::from(Bytes::from_static(b"test"))
);
2017-11-28 22:52:53 +01:00
let b = BytesMut::from("test");
let resp: HttpResponse = b.into();
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("application/octet-stream")
);
2017-11-28 22:52:53 +01:00
assert_eq!(resp.status(), StatusCode::OK);
2018-04-29 07:55:47 +02:00
assert_eq!(resp.body().binary().unwrap(), &Binary::from(BytesMut::from("test")));
let b = BytesMut::from("test");
2017-12-14 18:43:42 +01:00
let resp: HttpResponse = b.respond_to(req.clone()).ok().unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2018-04-14 01:02:01 +02:00
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("application/octet-stream")
);
assert_eq!(resp.status(), StatusCode::OK);
2018-04-29 07:55:47 +02:00
assert_eq!(resp.body().binary().unwrap(), &Binary::from(BytesMut::from("test")));
2017-11-28 22:52:53 +01:00
}
#[test]
fn test_into_builder() {
let resp: HttpResponse = "test".into();
assert_eq!(resp.status(), StatusCode::OK);
let mut builder = resp.into_builder();
let resp = builder.status(StatusCode::BAD_REQUEST).finish();
assert_eq!(resp.status(), StatusCode::BAD_REQUEST);
}
2017-10-23 07:54:11 +02:00
}