1
0
mirror of https://github.com/fafhrd91/actix-web synced 2024-11-24 00:21:08 +01:00

refactor date generatioin

This commit is contained in:
Nikolay Kim 2018-09-08 09:20:18 -07:00
parent 1907102685
commit c9a52e3197
7 changed files with 64 additions and 67 deletions

View File

@ -1,5 +1,4 @@
use std::net::{Shutdown, SocketAddr};
use std::rc::Rc;
use std::{io, ptr, time};
use bytes::{Buf, BufMut, BytesMut};
@ -15,7 +14,7 @@ const HTTP2_PREFACE: [u8; 14] = *b"PRI * HTTP/2.0";
enum HttpProtocol<T: IoStream, H: HttpHandler + 'static> {
H1(h1::Http1<T, H>),
H2(h2::Http2<T, H>),
Unknown(Rc<WorkerSettings<H>>, Option<SocketAddr>, T, BytesMut),
Unknown(WorkerSettings<H>, Option<SocketAddr>, T, BytesMut),
}
enum ProtocolKind {
@ -40,7 +39,7 @@ where
H: HttpHandler + 'static,
{
pub(crate) fn new(
settings: Rc<WorkerSettings<H>>, io: T, peer: Option<SocketAddr>,
settings: WorkerSettings<H>, io: T, peer: Option<SocketAddr>,
) -> HttpChannel<T, H> {
let ka_timeout = settings.keep_alive_timer();

View File

@ -1,6 +1,5 @@
use std::collections::VecDeque;
use std::net::SocketAddr;
use std::rc::Rc;
use std::time::{Duration, Instant};
use bytes::BytesMut;
@ -43,7 +42,7 @@ bitflags! {
pub(crate) struct Http1<T: IoStream, H: HttpHandler + 'static> {
flags: Flags,
settings: Rc<WorkerSettings<H>>,
settings: WorkerSettings<H>,
addr: Option<SocketAddr>,
stream: H1Writer<T, H>,
decoder: H1Decoder,
@ -90,7 +89,7 @@ where
H: HttpHandler + 'static,
{
pub fn new(
settings: Rc<WorkerSettings<H>>, stream: T, addr: Option<SocketAddr>,
settings: WorkerSettings<H>, stream: T, addr: Option<SocketAddr>,
buf: BytesMut, is_eof: bool, keepalive_timer: Option<Delay>,
) -> Self {
Http1 {
@ -99,7 +98,7 @@ where
} else {
Flags::KEEPALIVE
},
stream: H1Writer::new(stream, Rc::clone(&settings)),
stream: H1Writer::new(stream, settings.clone()),
decoder: H1Decoder::new(),
payload: None,
tasks: VecDeque::new(),
@ -112,7 +111,7 @@ where
#[inline]
pub fn settings(&self) -> &WorkerSettings<H> {
self.settings.as_ref()
&self.settings
}
#[inline]

View File

@ -1,7 +1,6 @@
// #![cfg_attr(feature = "cargo-clippy", allow(redundant_field_names))]
use std::io::{self, Write};
use std::rc::Rc;
use bytes::{BufMut, BytesMut};
use futures::{Async, Poll};
@ -38,11 +37,11 @@ pub(crate) struct H1Writer<T: AsyncWrite, H: 'static> {
headers_size: u32,
buffer: Output,
buffer_capacity: usize,
settings: Rc<WorkerSettings<H>>,
settings: WorkerSettings<H>,
}
impl<T: AsyncWrite, H: 'static> H1Writer<T, H> {
pub fn new(stream: T, settings: Rc<WorkerSettings<H>>) -> H1Writer<T, H> {
pub fn new(stream: T, settings: WorkerSettings<H>) -> H1Writer<T, H> {
H1Writer {
flags: Flags::KEEPALIVE,
written: 0,

View File

@ -38,7 +38,7 @@ where
H: HttpHandler + 'static,
{
flags: Flags,
settings: Rc<WorkerSettings<H>>,
settings: WorkerSettings<H>,
addr: Option<SocketAddr>,
state: State<IoWrapper<T>>,
tasks: VecDeque<Entry<H>>,
@ -58,7 +58,7 @@ where
H: HttpHandler + 'static,
{
pub fn new(
settings: Rc<WorkerSettings<H>>, io: T, addr: Option<SocketAddr>, buf: Bytes,
settings: WorkerSettings<H>, io: T, addr: Option<SocketAddr>, buf: Bytes,
keepalive_timer: Option<Delay>,
) -> Self {
let extensions = io.extensions();
@ -83,7 +83,7 @@ where
}
pub fn settings(&self) -> &WorkerSettings<H> {
self.settings.as_ref()
&self.settings
}
pub fn poll(&mut self) -> Poll<(), ()> {
@ -224,7 +224,7 @@ where
body,
resp,
self.addr,
&self.settings,
self.settings.clone(),
self.extensions.clone(),
));
}
@ -343,7 +343,7 @@ struct Entry<H: HttpHandler + 'static> {
impl<H: HttpHandler + 'static> Entry<H> {
fn new(
parts: Parts, recv: RecvStream, resp: SendResponse<Bytes>,
addr: Option<SocketAddr>, settings: &Rc<WorkerSettings<H>>,
addr: Option<SocketAddr>, settings: WorkerSettings<H>,
extensions: Option<Rc<Extensions>>,
) -> Entry<H>
where
@ -387,7 +387,7 @@ impl<H: HttpHandler + 'static> Entry<H> {
))
}),
payload: psender,
stream: H2Writer::new(resp, Rc::clone(settings)),
stream: H2Writer::new(resp, settings),
flags: EntryFlags::empty(),
recv,
}

View File

@ -1,14 +1,12 @@
#![cfg_attr(feature = "cargo-clippy", allow(redundant_field_names))]
use std::{cmp, io};
use bytes::{Bytes, BytesMut};
use futures::{Async, Poll};
use http2::server::SendResponse;
use http2::{Reason, SendStream};
use modhttp::Response;
use std::rc::Rc;
use std::{cmp, io};
use http::{HttpTryFrom, Method, Version};
use super::helpers;
use super::message::Request;
@ -20,6 +18,7 @@ use header::ContentEncoding;
use http::header::{
HeaderValue, CONNECTION, CONTENT_ENCODING, CONTENT_LENGTH, DATE, TRANSFER_ENCODING,
};
use http::{HttpTryFrom, Method, Version};
use httpresponse::HttpResponse;
const CHUNK_SIZE: usize = 16_384;
@ -40,12 +39,12 @@ pub(crate) struct H2Writer<H: 'static> {
written: u64,
buffer: Output,
buffer_capacity: usize,
settings: Rc<WorkerSettings<H>>,
settings: WorkerSettings<H>,
}
impl<H: 'static> H2Writer<H> {
pub fn new(
respond: SendResponse<Bytes>, settings: Rc<WorkerSettings<H>>,
respond: SendResponse<Bytes>, settings: WorkerSettings<H>,
) -> H2Writer<H> {
H2Writer {
stream: None,

View File

@ -1,5 +1,4 @@
use std::marker::PhantomData;
use std::rc::Rc;
use std::sync::Arc;
use std::{io, mem, net, time};
@ -10,7 +9,7 @@ use futures::{Async, Poll, Stream};
use net2::TcpBuilder;
use num_cpus;
use actix_net::{ssl, NewService, Service, Server};
use actix_net::{ssl, NewService, Server, Service};
//#[cfg(feature = "tls")]
//use native_tls::TlsAcceptor;
@ -603,7 +602,7 @@ where
H: HttpHandler,
Io: IoStream,
{
settings: Rc<WorkerSettings<H>>,
settings: WorkerSettings<H>,
tcp_ka: Option<time::Duration>,
_t: PhantomData<Io>,
}
@ -621,7 +620,7 @@ where
} else {
None
};
let settings = WorkerSettings::create(apps, keep_alive, settings);
let settings = WorkerSettings::new(apps, keep_alive, settings);
HttpServiceHandler {
tcp_ka,
@ -647,7 +646,7 @@ where
fn call(&mut self, mut req: Self::Request) -> Self::Future {
let _ = req.set_nodelay(true);
HttpChannel::new(Rc::clone(&self.settings), req, None)
HttpChannel::new(self.settings.clone(), req, None)
}
// fn shutdown(&self, force: bool) {

View File

@ -2,22 +2,21 @@ use std::cell::{RefCell, RefMut, UnsafeCell};
use std::collections::VecDeque;
use std::fmt::Write;
use std::rc::Rc;
use std::time::{Duration, Instant};
use std::time::Duration;
use std::{env, fmt, net};
use actix::Arbiter;
use bytes::BytesMut;
use futures::Stream;
use futures::{future, Future};
use futures_cpupool::CpuPool;
use http::StatusCode;
use lazycell::LazyCell;
use parking_lot::Mutex;
use time;
use tokio_timer::{Delay, Interval};
use tokio_timer::{sleep, Delay, Interval};
use tokio_current_thread::spawn;
use super::channel::Node;
use super::message::{Request, RequestPool};
// use super::server::{ConnectionRateTag, ConnectionTag, Connections};
use super::KeepAlive;
use body::Body;
use httpresponse::{HttpResponse, HttpResponseBuilder, HttpResponsePool};
@ -134,34 +133,21 @@ impl ServerSettings {
// "Sun, 06 Nov 1994 08:49:37 GMT".len()
const DATE_VALUE_LENGTH: usize = 29;
pub(crate) struct WorkerSettings<H> {
pub(crate) struct WorkerSettings<H>(Rc<Inner<H>>);
struct Inner<H> {
h: Vec<H>,
keep_alive: u64,
ka_enabled: bool,
bytes: Rc<SharedBytesPool>,
messages: &'static RequestPool,
node: RefCell<Node<()>>,
date: UnsafeCell<Date>,
date: UnsafeCell<(bool, Date)>,
}
impl<H: 'static> WorkerSettings<H> {
pub(crate) fn create(
apps: Vec<H>, keep_alive: KeepAlive, settings: ServerSettings,
) -> Rc<WorkerSettings<H>> {
let settings = Rc::new(Self::new(apps, keep_alive, settings));
// periodic date update
let s = settings.clone();
Arbiter::spawn(
Interval::new(Instant::now(), Duration::from_secs(1))
.map_err(|_| ())
.and_then(move |_| {
s.update_date();
Ok(())
}).fold((), |(), _| Ok(())),
);
settings
impl<H> Clone for WorkerSettings<H> {
fn clone(&self) -> Self {
WorkerSettings(self.0.clone())
}
}
@ -175,23 +161,23 @@ impl<H> WorkerSettings<H> {
KeepAlive::Disabled => (0, false),
};
WorkerSettings {
WorkerSettings(Rc::new(Inner {
h,
keep_alive,
ka_enabled,
bytes: Rc::new(SharedBytesPool::new()),
messages: RequestPool::pool(settings),
node: RefCell::new(Node::head()),
date: UnsafeCell::new(Date::new()),
keep_alive,
ka_enabled,
}
date: UnsafeCell::new((false, Date::new())),
}))
}
pub fn head(&self) -> RefMut<Node<()>> {
self.node.borrow_mut()
self.0.node.borrow_mut()
}
pub fn handlers(&self) -> &Vec<H> {
&self.h
&self.0.h
}
pub fn keep_alive_timer(&self) -> Option<Delay> {
@ -205,33 +191,49 @@ impl<H> WorkerSettings<H> {
}
pub fn keep_alive(&self) -> u64 {
self.keep_alive
self.0.keep_alive
}
pub fn keep_alive_enabled(&self) -> bool {
self.ka_enabled
self.0.ka_enabled
}
pub fn get_bytes(&self) -> BytesMut {
self.bytes.get_bytes()
self.0.bytes.get_bytes()
}
pub fn release_bytes(&self, bytes: BytesMut) {
self.bytes.release_bytes(bytes)
self.0.bytes.release_bytes(bytes)
}
pub fn get_request(&self) -> Request {
RequestPool::get(self.messages)
RequestPool::get(self.0.messages)
}
fn update_date(&self) {
// Unsafe: WorkerSetting is !Sync and !Send
unsafe { &mut *self.date.get() }.update();
unsafe { (&mut *self.0.date.get()).0 = false };
}
}
impl<H: 'static> WorkerSettings<H> {
pub fn set_date(&self, dst: &mut BytesMut, full: bool) {
// Unsafe: WorkerSetting is !Sync and !Send
let date_bytes = unsafe { &(*self.date.get()).bytes };
let date_bytes = unsafe {
let date = &mut (*self.0.date.get());
if !date.0 {
date.1.update();
date.0 = true;
// periodic date update
let s = self.clone();
spawn(sleep(Duration::from_secs(1)).then(move |_| {
s.update_date();
future::ok(())
}));
}
&date.1.bytes
};
if full {
let mut buf: [u8; 39] = [0; 39];
buf[..6].copy_from_slice(b"date: ");