1
0
mirror of https://github.com/actix/actix-extras.git synced 2024-11-28 01:32:57 +01:00

migrate to tokio

This commit is contained in:
Nikolay Kim 2018-05-24 21:03:16 -07:00
parent 565bcfb561
commit 690169db89
22 changed files with 162 additions and 162 deletions

View File

@ -8,7 +8,7 @@ cache:
matrix: matrix:
include: include:
- rust: 1.24.0 - rust: 1.25.0
- rust: stable - rust: stable
- rust: beta - rust: beta
- rust: nightly - rust: nightly
@ -31,12 +31,12 @@ before_script:
script: script:
- | - |
if [[ "$TRAVIS_RUST_VERSION" != "1.24.0" ]]; then if [[ "$TRAVIS_RUST_VERSION" != "1.25.0" ]]; then
cargo clean cargo clean
cargo test --features="alpn,tls" -- --nocapture cargo test --features="alpn,tls" -- --nocapture
fi fi
- | - |
if [[ "$TRAVIS_RUST_VERSION" == "1.24.0" ]]; then if [[ "$TRAVIS_RUST_VERSION" == "1.25.0" ]]; then
bash <(curl https://raw.githubusercontent.com/xd009642/tarpaulin/master/travis-install.sh) bash <(curl https://raw.githubusercontent.com/xd009642/tarpaulin/master/travis-install.sh)
USE_SKEPTIC=1 cargo tarpaulin --out Xml --no-count USE_SKEPTIC=1 cargo tarpaulin --out Xml --no-count
bash <(curl -s https://codecov.io/bash) bash <(curl -s https://codecov.io/bash)

View File

@ -2,6 +2,10 @@
## [0.7.0] - 2018-xx-xx ## [0.7.0] - 2018-xx-xx
### Changed
* Migrate to tokio
## [0.6.10] - 2018-05-24 ## [0.6.10] - 2018-05-24

View File

@ -47,7 +47,8 @@ flate2-c = ["flate2/miniz-sys"]
flate2-rust = ["flate2/rust_backend"] flate2-rust = ["flate2/rust_backend"]
[dependencies] [dependencies]
actix = "^0.5.5" #actix = "^0.6.0"
actix = { git="https://github.com/actix/actix.git" }
base64 = "0.9" base64 = "0.9"
bitflags = "1.0" bitflags = "1.0"
@ -62,7 +63,7 @@ mime = "0.3"
mime_guess = "2.0.0-alpha" mime_guess = "2.0.0-alpha"
num_cpus = "1.0" num_cpus = "1.0"
percent-encoding = "1.0" percent-encoding = "1.0"
rand = "0.4" rand = "0.5"
regex = "1.0" regex = "1.0"
serde = "1.0" serde = "1.0"
serde_json = "1.0" serde_json = "1.0"
@ -86,8 +87,11 @@ byteorder = "1"
futures = "0.1" futures = "0.1"
futures-cpupool = "0.1" futures-cpupool = "0.1"
slab = "0.4" slab = "0.4"
tokio = "0.1"
tokio-io = "0.1" tokio-io = "0.1"
tokio-core = "0.1" tokio-tcp = "0.1"
tokio-timer = "0.2"
tokio-reactor = "0.1"
# native-tls # native-tls
native-tls = { version="0.1", optional = true } native-tls = { version="0.1", optional = true }

View File

@ -28,7 +28,7 @@ Actix web is a simple, pragmatic and extremely fast web framework for Rust.
* [API Documentation (Releases)](https://docs.rs/actix-web/) * [API Documentation (Releases)](https://docs.rs/actix-web/)
* [Chat on gitter](https://gitter.im/actix/actix) * [Chat on gitter](https://gitter.im/actix/actix)
* Cargo package: [actix-web](https://crates.io/crates/actix-web) * Cargo package: [actix-web](https://crates.io/crates/actix-web)
* Minimum supported Rust version: 1.24 or later * Minimum supported Rust version: 1.25 or later
## Example ## Example

View File

@ -9,16 +9,16 @@ use actix::actors::{Connect as ResolveConnect, Connector, ConnectorError};
use actix::fut::WrapFuture; use actix::fut::WrapFuture;
use actix::registry::ArbiterService; use actix::registry::ArbiterService;
use actix::{ use actix::{
fut, Actor, ActorFuture, ActorResponse, Arbiter, AsyncContext, Context, fut, Actor, ActorFuture, ActorResponse, AsyncContext, Context, ContextFutureSpawner,
ContextFutureSpawner, Handler, Message, Recipient, Supervised, Syn, Handler, Message, Recipient, Supervised, Syn,
}; };
use futures::task::{current as current_task, Task}; use futures::task::{current as current_task, Task};
use futures::unsync::oneshot; use futures::unsync::oneshot;
use futures::{Async, Future, Poll}; use futures::{Async, Future, Poll};
use http::{Error as HttpError, HttpTryFrom, Uri}; use http::{Error as HttpError, HttpTryFrom, Uri};
use tokio_core::reactor::Timeout;
use tokio_io::{AsyncRead, AsyncWrite}; use tokio_io::{AsyncRead, AsyncWrite};
use tokio_timer::Delay;
#[cfg(feature = "alpn")] #[cfg(feature = "alpn")]
use openssl::ssl::{Error as OpensslError, SslConnector, SslMethod}; use openssl::ssl::{Error as OpensslError, SslConnector, SslMethod};
@ -190,8 +190,8 @@ pub struct ClientConnector {
available: HashMap<Key, VecDeque<Conn>>, available: HashMap<Key, VecDeque<Conn>>,
to_close: Vec<Connection>, to_close: Vec<Connection>,
waiters: HashMap<Key, VecDeque<Waiter>>, waiters: HashMap<Key, VecDeque<Waiter>>,
wait_timeout: Option<(Instant, Timeout)>, wait_timeout: Option<(Instant, Delay)>,
paused: Option<Option<(Instant, Timeout)>>, paused: Option<Option<(Instant, Delay)>>,
} }
impl Actor for ClientConnector { impl Actor for ClientConnector {
@ -563,8 +563,7 @@ impl ClientConnector {
} }
} }
let mut timeout = let mut timeout = Delay::new(time);
Timeout::new(time - Instant::now(), Arbiter::handle()).unwrap();
let _ = timeout.poll(); let _ = timeout.poll();
self.wait_timeout = Some((time, timeout)); self.wait_timeout = Some((time, timeout));
} }
@ -597,7 +596,7 @@ impl Handler<Pause> for ClientConnector {
fn handle(&mut self, msg: Pause, _: &mut Self::Context) { fn handle(&mut self, msg: Pause, _: &mut Self::Context) {
if let Some(time) = msg.time { if let Some(time) = msg.time {
let when = Instant::now() + time; let when = Instant::now() + time;
let mut timeout = Timeout::new(time, Arbiter::handle()).unwrap(); let mut timeout = Delay::new(when);
let _ = timeout.poll(); let _ = timeout.poll();
self.paused = Some(Some((when, timeout))); self.paused = Some(Some((when, timeout)));
} else if self.paused.is_none() { } else if self.paused.is_none() {

View File

@ -10,7 +10,7 @@
//! fn main() { //! fn main() {
//! let sys = actix::System::new("test"); //! let sys = actix::System::new("test");
//! //!
//! actix::Arbiter::handle().spawn({ //! actix::Arbiter::spawn({
//! client::get("http://www.rust-lang.org") // <- Create request builder //! client::get("http://www.rust-lang.org") // <- Create request builder
//! .header("User-Agent", "Actix-web") //! .header("User-Agent", "Actix-web")
//! .finish().unwrap() //! .finish().unwrap()
@ -70,7 +70,7 @@ impl ResponseError for SendRequestError {
/// fn main() { /// fn main() {
/// let sys = actix::System::new("test"); /// let sys = actix::System::new("test");
/// ///
/// actix::Arbiter::handle().spawn({ /// actix::Arbiter::spawn({
/// client::get("http://www.rust-lang.org") // <- Create request builder /// client::get("http://www.rust-lang.org") // <- Create request builder
/// .header("User-Agent", "Actix-web") /// .header("User-Agent", "Actix-web")
/// .finish().unwrap() /// .finish().unwrap()

View File

@ -2,9 +2,9 @@ use bytes::{Bytes, BytesMut};
use futures::unsync::oneshot; use futures::unsync::oneshot;
use futures::{Async, Future, Poll}; use futures::{Async, Future, Poll};
use http::header::CONTENT_ENCODING; use http::header::CONTENT_ENCODING;
use std::time::Duration; use std::time::{Duration, Instant};
use std::{io, mem}; use std::{io, mem};
use tokio_core::reactor::Timeout; use tokio_timer::Delay;
use actix::prelude::*; use actix::prelude::*;
@ -71,7 +71,7 @@ pub struct SendRequest {
conn: Addr<Unsync, ClientConnector>, conn: Addr<Unsync, ClientConnector>,
conn_timeout: Duration, conn_timeout: Duration,
wait_timeout: Duration, wait_timeout: Duration,
timeout: Option<Timeout>, timeout: Option<Delay>,
} }
impl SendRequest { impl SendRequest {
@ -108,7 +108,7 @@ impl SendRequest {
/// Request timeout is the total time before a response must be received. /// Request timeout is the total time before a response must be received.
/// Default value is 5 seconds. /// Default value is 5 seconds.
pub fn timeout(mut self, timeout: Duration) -> Self { pub fn timeout(mut self, timeout: Duration) -> Self {
self.timeout = Some(Timeout::new(timeout, Arbiter::handle()).unwrap()); self.timeout = Some(Delay::new(Instant::now() + timeout));
self self
} }
@ -174,7 +174,7 @@ impl Future for SendRequest {
}; };
let timeout = self.timeout.take().unwrap_or_else(|| { let timeout = self.timeout.take().unwrap_or_else(|| {
Timeout::new(Duration::from_secs(5), Arbiter::handle()).unwrap() Delay::new(Instant::now() + Duration::from_secs(5))
}); });
let pl = Box::new(Pipeline { let pl = Box::new(Pipeline {
@ -229,7 +229,7 @@ pub(crate) struct Pipeline {
decompress: Option<PayloadStream>, decompress: Option<PayloadStream>,
should_decompress: bool, should_decompress: bool,
write_state: RunningState, write_state: RunningState,
timeout: Option<Timeout>, timeout: Option<Delay>,
} }
enum IoBody { enum IoBody {

View File

@ -34,7 +34,7 @@ use httprequest::HttpRequest;
/// fn main() { /// fn main() {
/// let sys = actix::System::new("test"); /// let sys = actix::System::new("test");
/// ///
/// actix::Arbiter::handle().spawn({ /// actix::Arbiter::spawn({
/// ClientRequest::get("http://www.rust-lang.org") // <- Create request builder /// ClientRequest::get("http://www.rust-lang.org") // <- Create request builder
/// .header("User-Agent", "Actix-web") /// .header("User-Agent", "Actix-web")
/// .finish().unwrap() /// .finish().unwrap()

View File

@ -16,6 +16,7 @@ use http_range::HttpRangeParseError;
use httparse; use httparse;
use serde::de::value::Error as DeError; use serde::de::value::Error as DeError;
use serde_json::error::Error as JsonError; use serde_json::error::Error as JsonError;
use tokio_timer::Error as TimerError;
pub use url::ParseError as UrlParseError; pub use url::ParseError as UrlParseError;
// re-exports // re-exports
@ -126,6 +127,9 @@ impl From<failure::Error> for Error {
/// `InternalServerError` for `JsonError` /// `InternalServerError` for `JsonError`
impl ResponseError for JsonError {} impl ResponseError for JsonError {}
/// `InternalServerError` for `TimerError`
impl ResponseError for TimerError {}
/// `InternalServerError` for `UrlParseError` /// `InternalServerError` for `UrlParseError`
impl ResponseError for UrlParseError {} impl ResponseError for UrlParseError {}

View File

@ -112,8 +112,11 @@ extern crate mio;
extern crate net2; extern crate net2;
extern crate rand; extern crate rand;
extern crate slab; extern crate slab;
extern crate tokio_core; extern crate tokio;
extern crate tokio_io; extern crate tokio_io;
extern crate tokio_reactor;
extern crate tokio_tcp;
extern crate tokio_timer;
extern crate url; extern crate url;
#[macro_use] #[macro_use]
extern crate serde; extern crate serde;

View File

@ -663,7 +663,7 @@ mod tests {
use bytes::Bytes; use bytes::Bytes;
use futures::future::{lazy, result}; use futures::future::{lazy, result};
use payload::{Payload, PayloadWriter}; use payload::{Payload, PayloadWriter};
use tokio_core::reactor::Core; use tokio::runtime::current_thread::Runtime;
#[test] #[test]
fn test_boundary() { fn test_boundary() {
@ -710,9 +710,9 @@ mod tests {
#[test] #[test]
fn test_multipart() { fn test_multipart() {
Core::new() Runtime::new()
.unwrap() .unwrap()
.run(lazy(|| { .block_on(lazy(|| {
let (mut sender, payload) = Payload::new(false); let (mut sender, payload) = Payload::new(false);
let bytes = Bytes::from( let bytes = Bytes::from(

View File

@ -524,7 +524,7 @@ mod tests {
use failure::Fail; use failure::Fail;
use futures::future::{lazy, result}; use futures::future::{lazy, result};
use std::io; use std::io;
use tokio_core::reactor::Core; use tokio::runtime::current_thread::Runtime;
#[test] #[test]
fn test_error() { fn test_error() {
@ -542,9 +542,9 @@ mod tests {
#[test] #[test]
fn test_basic() { fn test_basic() {
Core::new() Runtime::new()
.unwrap() .unwrap()
.run(lazy(|| { .block_on(lazy(|| {
let (_, payload) = Payload::new(false); let (_, payload) = Payload::new(false);
let mut payload = PayloadHelper::new(payload); let mut payload = PayloadHelper::new(payload);
@ -559,9 +559,9 @@ mod tests {
#[test] #[test]
fn test_eof() { fn test_eof() {
Core::new() Runtime::new()
.unwrap() .unwrap()
.run(lazy(|| { .block_on(lazy(|| {
let (mut sender, payload) = Payload::new(false); let (mut sender, payload) = Payload::new(false);
let mut payload = PayloadHelper::new(payload); let mut payload = PayloadHelper::new(payload);
@ -584,9 +584,9 @@ mod tests {
#[test] #[test]
fn test_err() { fn test_err() {
Core::new() Runtime::new()
.unwrap() .unwrap()
.run(lazy(|| { .block_on(lazy(|| {
let (mut sender, payload) = Payload::new(false); let (mut sender, payload) = Payload::new(false);
let mut payload = PayloadHelper::new(payload); let mut payload = PayloadHelper::new(payload);
@ -602,9 +602,9 @@ mod tests {
#[test] #[test]
fn test_readany() { fn test_readany() {
Core::new() Runtime::new()
.unwrap() .unwrap()
.run(lazy(|| { .block_on(lazy(|| {
let (mut sender, payload) = Payload::new(false); let (mut sender, payload) = Payload::new(false);
let mut payload = PayloadHelper::new(payload); let mut payload = PayloadHelper::new(payload);
@ -631,9 +631,9 @@ mod tests {
#[test] #[test]
fn test_readexactly() { fn test_readexactly() {
Core::new() Runtime::new()
.unwrap() .unwrap()
.run(lazy(|| { .block_on(lazy(|| {
let (mut sender, payload) = Payload::new(false); let (mut sender, payload) = Payload::new(false);
let mut payload = PayloadHelper::new(payload); let mut payload = PayloadHelper::new(payload);
@ -665,9 +665,9 @@ mod tests {
#[test] #[test]
fn test_readuntil() { fn test_readuntil() {
Core::new() Runtime::new()
.unwrap() .unwrap()
.run(lazy(|| { .block_on(lazy(|| {
let (mut sender, payload) = Payload::new(false); let (mut sender, payload) = Payload::new(false);
let mut payload = PayloadHelper::new(payload); let mut payload = PayloadHelper::new(payload);
@ -699,9 +699,9 @@ mod tests {
#[test] #[test]
fn test_unread_data() { fn test_unread_data() {
Core::new() Runtime::new()
.unwrap() .unwrap()
.run(lazy(|| { .block_on(lazy(|| {
let (_, mut payload) = Payload::new(false); let (_, mut payload) = Payload::new(false);
payload.unread_data(Bytes::from("data")); payload.unread_data(Bytes::from("data"));

View File

@ -770,7 +770,7 @@ mod tests {
use actix::*; use actix::*;
use context::HttpContext; use context::HttpContext;
use futures::future::{lazy, result}; use futures::future::{lazy, result};
use tokio_core::reactor::Core; use tokio::runtime::current_thread::Runtime;
impl<S, H> PipelineState<S, H> { impl<S, H> PipelineState<S, H> {
fn is_none(&self) -> Option<bool> { fn is_none(&self) -> Option<bool> {
@ -796,9 +796,9 @@ mod tests {
#[test] #[test]
fn test_completed() { fn test_completed() {
Core::new() Runtime::new()
.unwrap() .unwrap()
.run(lazy(|| { .block_on(lazy(|| {
let mut info = PipelineInfo::new(HttpRequest::default()); let mut info = PipelineInfo::new(HttpRequest::default());
Completed::<(), Inner<()>>::init(&mut info) Completed::<(), Inner<()>>::init(&mut info)
.is_none() .is_none()

View File

@ -2,12 +2,11 @@ use std::collections::VecDeque;
use std::io; use std::io;
use std::net::SocketAddr; use std::net::SocketAddr;
use std::rc::Rc; use std::rc::Rc;
use std::time::Duration; use std::time::{Duration, Instant};
use actix::Arbiter;
use bytes::{BufMut, BytesMut}; use bytes::{BufMut, BytesMut};
use futures::{Async, Future, Poll}; use futures::{Async, Future, Poll};
use tokio_core::reactor::Timeout; use tokio_timer::Delay;
use error::PayloadError; use error::PayloadError;
use httprequest::HttpRequest; use httprequest::HttpRequest;
@ -53,7 +52,7 @@ pub(crate) struct Http1<T: IoStream, H: 'static> {
payload: Option<PayloadType>, payload: Option<PayloadType>,
buf: BytesMut, buf: BytesMut,
tasks: VecDeque<Entry>, tasks: VecDeque<Entry>,
keepalive_timer: Option<Timeout>, keepalive_timer: Option<Delay>,
} }
struct Entry { struct Entry {
@ -295,8 +294,7 @@ where
if self.keepalive_timer.is_none() && keep_alive > 0 { if self.keepalive_timer.is_none() && keep_alive > 0 {
trace!("Start keep-alive timer"); trace!("Start keep-alive timer");
let mut timer = let mut timer =
Timeout::new(Duration::new(keep_alive, 0), Arbiter::handle()) Delay::new(Instant::now() + Duration::new(keep_alive, 0));
.unwrap();
// register timer // register timer
let _ = timer.poll(); let _ = timer.poll();
self.keepalive_timer = Some(timer); self.keepalive_timer = Some(timer);

View File

@ -4,17 +4,16 @@ use std::collections::VecDeque;
use std::io::{Read, Write}; use std::io::{Read, Write};
use std::net::SocketAddr; use std::net::SocketAddr;
use std::rc::Rc; use std::rc::Rc;
use std::time::Duration; use std::time::{Duration, Instant};
use std::{cmp, io, mem}; use std::{cmp, io, mem};
use actix::Arbiter;
use bytes::{Buf, Bytes}; use bytes::{Buf, Bytes};
use futures::{Async, Future, Poll, Stream}; use futures::{Async, Future, Poll, Stream};
use http2::server::{self, Connection, Handshake, SendResponse}; use http2::server::{self, Connection, Handshake, SendResponse};
use http2::{Reason, RecvStream}; use http2::{Reason, RecvStream};
use modhttp::request::Parts; use modhttp::request::Parts;
use tokio_core::reactor::Timeout;
use tokio_io::{AsyncRead, AsyncWrite}; use tokio_io::{AsyncRead, AsyncWrite};
use tokio_timer::Delay;
use error::PayloadError; use error::PayloadError;
use httpmessage::HttpMessage; use httpmessage::HttpMessage;
@ -46,7 +45,7 @@ where
addr: Option<SocketAddr>, addr: Option<SocketAddr>,
state: State<IoWrapper<T>>, state: State<IoWrapper<T>>,
tasks: VecDeque<Entry<H>>, tasks: VecDeque<Entry<H>>,
keepalive_timer: Option<Timeout>, keepalive_timer: Option<Delay>,
} }
enum State<T: AsyncRead + AsyncWrite> { enum State<T: AsyncRead + AsyncWrite> {
@ -218,9 +217,10 @@ where
let keep_alive = self.settings.keep_alive(); let keep_alive = self.settings.keep_alive();
if keep_alive > 0 && self.keepalive_timer.is_none() { if keep_alive > 0 && self.keepalive_timer.is_none() {
trace!("Start keep-alive timer"); trace!("Start keep-alive timer");
let mut timeout = Timeout::new( let mut timeout = Delay::new(
Duration::new(keep_alive, 0), Instant::now()
Arbiter::handle()).unwrap(); + Duration::new(keep_alive, 0),
);
// register timeout // register timeout
let _ = timeout.poll(); let _ = timeout.poll();
self.keepalive_timer = Some(timeout); self.keepalive_timer = Some(timeout);

View File

@ -5,8 +5,8 @@ use std::{io, time};
use actix; use actix;
use bytes::BytesMut; use bytes::BytesMut;
use futures::{Async, Poll}; use futures::{Async, Poll};
use tokio_core::net::TcpStream;
use tokio_io::{AsyncRead, AsyncWrite}; use tokio_io::{AsyncRead, AsyncWrite};
use tokio_tcp::TcpStream;
mod channel; mod channel;
pub(crate) mod encoding; pub(crate) mod encoding;

View File

@ -563,11 +563,10 @@ impl<H: IntoHttpHandler> HttpServer<H> {
/// Start listening for incoming connections from a stream. /// Start listening for incoming connections from a stream.
/// ///
/// This method uses only one thread for handling incoming connections. /// This method uses only one thread for handling incoming connections.
pub fn start_incoming<T, A, S>(mut self, stream: S, secure: bool) -> Addr<Syn, Self> pub fn start_incoming<T, S>(mut self, stream: S, secure: bool) -> Addr<Syn, Self>
where where
S: Stream<Item = (T, A), Error = io::Error> + 'static, S: Stream<Item = T, Error = io::Error> + 'static,
T: AsyncRead + AsyncWrite + 'static, T: AsyncRead + AsyncWrite + 'static,
A: 'static,
{ {
// set server settings // set server settings
let addr: net::SocketAddr = "127.0.0.1:8080".parse().unwrap(); let addr: net::SocketAddr = "127.0.0.1:8080".parse().unwrap();
@ -581,7 +580,7 @@ impl<H: IntoHttpHandler> HttpServer<H> {
// start server // start server
let signals = self.subscribe_to_signals(); let signals = self.subscribe_to_signals();
let addr: Addr<Syn, _> = HttpServer::create(move |ctx| { let addr: Addr<Syn, _> = HttpServer::create(move |ctx| {
ctx.add_message_stream(stream.map_err(|_| ()).map(move |(t, _)| Conn { ctx.add_message_stream(stream.map_err(|_| ()).map(move |t| Conn {
io: WrapperStream::new(t), io: WrapperStream::new(t),
token: 0, token: 0,
peer: None, peer: None,
@ -687,7 +686,7 @@ where
type Result = (); type Result = ();
fn handle(&mut self, msg: Conn<T>, _: &mut Context<Self>) -> Self::Result { fn handle(&mut self, msg: Conn<T>, _: &mut Context<Self>) -> Self::Result {
Arbiter::handle().spawn(HttpChannel::new( Arbiter::spawn(HttpChannel::new(
Rc::clone(self.h.as_ref().unwrap()), Rc::clone(self.h.as_ref().unwrap()),
msg.io, msg.io,
msg.peer, msg.peer,

View File

@ -4,8 +4,8 @@ use net2::TcpStreamExt;
use slab::Slab; use slab::Slab;
use std::rc::Rc; use std::rc::Rc;
use std::{net, time}; use std::{net, time};
use tokio_core::net::TcpStream; use tokio_reactor::Handle;
use tokio_core::reactor::Handle; use tokio_tcp::TcpStream;
#[cfg(any(feature = "tls", feature = "alpn"))] #[cfg(any(feature = "tls", feature = "alpn"))]
use futures::future; use futures::future;
@ -60,7 +60,6 @@ where
H: HttpHandler + 'static, H: HttpHandler + 'static,
{ {
settings: Rc<WorkerSettings<H>>, settings: Rc<WorkerSettings<H>>,
hnd: Handle,
socks: Slab<SocketInfo>, socks: Slab<SocketInfo>,
tcp_ka: Option<time::Duration>, tcp_ka: Option<time::Duration>,
} }
@ -77,7 +76,6 @@ impl<H: HttpHandler + 'static> Worker<H> {
Worker { Worker {
settings: Rc::new(WorkerSettings::new(h, keep_alive)), settings: Rc::new(WorkerSettings::new(h, keep_alive)),
hnd: Arbiter::handle().clone(),
socks, socks,
tcp_ka, tcp_ka,
} }
@ -130,11 +128,11 @@ where
if self.tcp_ka.is_some() && msg.io.set_keepalive(self.tcp_ka).is_err() { if self.tcp_ka.is_some() && msg.io.set_keepalive(self.tcp_ka).is_err() {
error!("Can not set socket keep-alive option"); error!("Can not set socket keep-alive option");
} }
self.socks.get_mut(msg.token).unwrap().htype.handle( self.socks
Rc::clone(&self.settings), .get_mut(msg.token)
&self.hnd, .unwrap()
msg, .htype
); .handle(Rc::clone(&self.settings), msg);
} }
} }
@ -174,15 +172,15 @@ pub(crate) enum StreamHandlerType {
impl StreamHandlerType { impl StreamHandlerType {
fn handle<H: HttpHandler>( fn handle<H: HttpHandler>(
&mut self, h: Rc<WorkerSettings<H>>, hnd: &Handle, msg: Conn<net::TcpStream>, &mut self, h: Rc<WorkerSettings<H>>, msg: Conn<net::TcpStream>,
) { ) {
match *self { match *self {
StreamHandlerType::Normal => { StreamHandlerType::Normal => {
let _ = msg.io.set_nodelay(true); let _ = msg.io.set_nodelay(true);
let io = TcpStream::from_stream(msg.io, hnd) let io = TcpStream::from_std(msg.io, &Handle::default())
.expect("failed to associate TCP stream"); .expect("failed to associate TCP stream");
hnd.spawn(HttpChannel::new(h, io, msg.peer, msg.http2)); Arbiter::spawn(HttpChannel::new(h, io, msg.peer, msg.http2));
} }
#[cfg(feature = "tls")] #[cfg(feature = "tls")]
StreamHandlerType::Tls(ref acceptor) => { StreamHandlerType::Tls(ref acceptor) => {
@ -190,29 +188,32 @@ impl StreamHandlerType {
io, peer, http2, .. io, peer, http2, ..
} = msg; } = msg;
let _ = io.set_nodelay(true); let _ = io.set_nodelay(true);
let io = TcpStream::from_stream(io, hnd) let io = TcpStream::from_std(io, &Handle::default())
.expect("failed to associate TCP stream"); .expect("failed to associate TCP stream");
hnd.spawn(TlsAcceptorExt::accept_async(acceptor, io).then(move |res| { Arbiter::spawn(TlsAcceptorExt::accept_async(acceptor, io).then(
move |res| {
match res { match res {
Ok(io) => { Ok(io) => {
Arbiter::handle().spawn(HttpChannel::new(h, io, peer, http2)) Arbiter::spawn(HttpChannel::new(h, io, peer, http2))
} }
Err(err) => { Err(err) => {
trace!("Error during handling tls connection: {}", err) trace!("Error during handling tls connection: {}", err)
} }
}; };
future::result(Ok(())) future::result(Ok(()))
})); },
));
} }
#[cfg(feature = "alpn")] #[cfg(feature = "alpn")]
StreamHandlerType::Alpn(ref acceptor) => { StreamHandlerType::Alpn(ref acceptor) => {
let Conn { io, peer, .. } = msg; let Conn { io, peer, .. } = msg;
let _ = io.set_nodelay(true); let _ = io.set_nodelay(true);
let io = TcpStream::from_stream(io, hnd) let io = TcpStream::from_std(io, &Handle::default())
.expect("failed to associate TCP stream"); .expect("failed to associate TCP stream");
hnd.spawn(SslAcceptorExt::accept_async(acceptor, io).then(move |res| { Arbiter::spawn(SslAcceptorExt::accept_async(acceptor, io).then(
move |res| {
match res { match res {
Ok(io) => { Ok(io) => {
let http2 = if let Some(p) = let http2 = if let Some(p) =
@ -222,15 +223,15 @@ impl StreamHandlerType {
} else { } else {
false false
}; };
Arbiter::handle() Arbiter::spawn(HttpChannel::new(h, io, peer, http2));
.spawn(HttpChannel::new(h, io, peer, http2));
} }
Err(err) => { Err(err) => {
trace!("Error during handling tls connection: {}", err) trace!("Error during handling tls connection: {}", err)
} }
}; };
future::result(Ok(())) future::result(Ok(()))
})); },
));
} }
} }
} }

View File

@ -11,8 +11,9 @@ use futures::Future;
use http::header::HeaderName; use http::header::HeaderName;
use http::{HeaderMap, HttpTryFrom, Method, Uri, Version}; use http::{HeaderMap, HttpTryFrom, Method, Uri, Version};
use net2::TcpBuilder; use net2::TcpBuilder;
use tokio_core::net::TcpListener; use tokio::runtime::current_thread::Runtime;
use tokio_core::reactor::Core; use tokio_reactor::Handle;
use tokio_tcp::TcpListener;
#[cfg(feature = "alpn")] #[cfg(feature = "alpn")]
use openssl::ssl::SslAcceptor; use openssl::ssl::SslAcceptor;
@ -112,8 +113,7 @@ impl TestServer {
let sys = System::new("actix-test-server"); let sys = System::new("actix-test-server");
let tcp = net::TcpListener::bind("127.0.0.1:0").unwrap(); let tcp = net::TcpListener::bind("127.0.0.1:0").unwrap();
let local_addr = tcp.local_addr().unwrap(); let local_addr = tcp.local_addr().unwrap();
let tcp = let tcp = TcpListener::from_std(tcp, &Handle::default()).unwrap();
TcpListener::from_listener(tcp, &local_addr, Arbiter::handle()).unwrap();
HttpServer::new(factory) HttpServer::new(factory)
.disable_signals() .disable_signals()
@ -289,8 +289,7 @@ impl<S: 'static> TestServerBuilder<S> {
let tcp = net::TcpListener::bind("127.0.0.1:0").unwrap(); let tcp = net::TcpListener::bind("127.0.0.1:0").unwrap();
let local_addr = tcp.local_addr().unwrap(); let local_addr = tcp.local_addr().unwrap();
let tcp = let tcp = TcpListener::from_std(tcp, &Handle::default()).unwrap();
TcpListener::from_listener(tcp, &local_addr, Arbiter::handle()).unwrap();
let state = self.state; let state = self.state;
@ -309,10 +308,9 @@ impl<S: 'static> TestServerBuilder<S> {
let ssl = self.ssl.take(); let ssl = self.ssl.take();
if let Some(ssl) = ssl { if let Some(ssl) = ssl {
srv.start_incoming( srv.start_incoming(
tcp.incoming().and_then(move |(sock, addr)| { tcp.incoming().and_then(move |sock| {
ssl.accept_async(sock) ssl.accept_async(sock)
.map_err(|e| io::Error::new(io::ErrorKind::Other, e)) .map_err(|e| io::Error::new(io::ErrorKind::Other, e))
.map(move |s| (s, addr))
}), }),
false, false,
); );
@ -616,8 +614,8 @@ impl<S: 'static> TestRequest<S> {
let req = self.finish(); let req = self.finish();
let fut = h(req.clone()); let fut = h(req.clone());
let mut core = Core::new().unwrap(); let mut core = Runtime::new().unwrap();
match core.run(fut) { match core.block_on(fut) {
Ok(r) => match r.respond_to(&req) { Ok(r) => match r.respond_to(&req) {
Ok(reply) => match reply.into().into() { Ok(reply) => match reply.into().into() {
AsyncResultItem::Ok(resp) => Ok(resp), AsyncResultItem::Ok(resp) => Ok(resp),

View File

@ -4,21 +4,20 @@ extern crate bytes;
extern crate futures; extern crate futures;
extern crate h2; extern crate h2;
extern crate http; extern crate http;
extern crate tokio_core; extern crate tokio_timer;
#[macro_use] #[macro_use]
extern crate serde_derive; extern crate serde_derive;
extern crate serde_json; extern crate serde_json;
use std::io; use std::io;
use std::time::Duration; use std::time::{Duration, Instant};
use actix::*;
use actix_web::*; use actix_web::*;
use bytes::Bytes; use bytes::Bytes;
use futures::Future; use futures::Future;
use http::StatusCode; use http::StatusCode;
use serde_json::Value; use serde_json::Value;
use tokio_core::reactor::Timeout; use tokio_timer::Delay;
#[derive(Deserialize)] #[derive(Deserialize)]
struct PParam { struct PParam {
@ -75,8 +74,7 @@ fn test_async_extractor_async() {
let mut srv = test::TestServer::new(|app| { let mut srv = test::TestServer::new(|app| {
app.resource("/{username}/index.html", |r| { app.resource("/{username}/index.html", |r| {
r.route().with(|data: Json<Value>| { r.route().with(|data: Json<Value>| {
Timeout::new(Duration::from_millis(10), &Arbiter::handle()) Delay::new(Instant::now() + Duration::from_millis(10))
.unwrap()
.and_then(move |_| Ok(format!("{}", data.0))) .and_then(move |_| Ok(format!("{}", data.0)))
.responder() .responder()
}) })
@ -171,8 +169,7 @@ fn test_path_and_query_extractor2_async() {
app.resource("/{username}/index.html", |r| { app.resource("/{username}/index.html", |r| {
r.route() r.route()
.with3(|p: Path<PParam>, _: Query<PParam>, data: Json<Value>| { .with3(|p: Path<PParam>, _: Query<PParam>, data: Json<Value>| {
Timeout::new(Duration::from_millis(10), &Arbiter::handle()) Delay::new(Instant::now() + Duration::from_millis(10))
.unwrap()
.and_then(move |_| { .and_then(move |_| {
Ok(format!("Welcome {} - {}!", p.username, data.0)) Ok(format!("Welcome {} - {}!", p.username, data.0))
}) })
@ -201,8 +198,7 @@ fn test_path_and_query_extractor3_async() {
let mut srv = test::TestServer::new(|app| { let mut srv = test::TestServer::new(|app| {
app.resource("/{username}/index.html", |r| { app.resource("/{username}/index.html", |r| {
r.route().with2(|p: Path<PParam>, data: Json<Value>| { r.route().with2(|p: Path<PParam>, data: Json<Value>| {
Timeout::new(Duration::from_millis(10), &Arbiter::handle()) Delay::new(Instant::now() + Duration::from_millis(10))
.unwrap()
.and_then(move |_| { .and_then(move |_| {
Ok(format!("Welcome {} - {}!", p.username, data.0)) Ok(format!("Welcome {} - {}!", p.username, data.0))
}) })
@ -227,8 +223,7 @@ fn test_path_and_query_extractor4_async() {
let mut srv = test::TestServer::new(|app| { let mut srv = test::TestServer::new(|app| {
app.resource("/{username}/index.html", |r| { app.resource("/{username}/index.html", |r| {
r.route().with2(|data: Json<Value>, p: Path<PParam>| { r.route().with2(|data: Json<Value>, p: Path<PParam>| {
Timeout::new(Duration::from_millis(10), &Arbiter::handle()) Delay::new(Instant::now() + Duration::from_millis(10))
.unwrap()
.and_then(move |_| { .and_then(move |_| {
Ok(format!("Welcome {} - {}!", p.username, data.0)) Ok(format!("Welcome {} - {}!", p.username, data.0))
}) })
@ -254,8 +249,7 @@ fn test_path_and_query_extractor2_async2() {
app.resource("/{username}/index.html", |r| { app.resource("/{username}/index.html", |r| {
r.route() r.route()
.with3(|p: Path<PParam>, data: Json<Value>, _: Query<PParam>| { .with3(|p: Path<PParam>, data: Json<Value>, _: Query<PParam>| {
Timeout::new(Duration::from_millis(10), &Arbiter::handle()) Delay::new(Instant::now() + Duration::from_millis(10))
.unwrap()
.and_then(move |_| { .and_then(move |_| {
Ok(format!("Welcome {} - {}!", p.username, data.0)) Ok(format!("Welcome {} - {}!", p.username, data.0))
}) })
@ -294,8 +288,7 @@ fn test_path_and_query_extractor2_async3() {
app.resource("/{username}/index.html", |r| { app.resource("/{username}/index.html", |r| {
r.route() r.route()
.with3(|data: Json<Value>, p: Path<PParam>, _: Query<PParam>| { .with3(|data: Json<Value>, p: Path<PParam>, _: Query<PParam>| {
Timeout::new(Duration::from_millis(10), &Arbiter::handle()) Delay::new(Instant::now() + Duration::from_millis(10))
.unwrap()
.and_then(move |_| { .and_then(move |_| {
Ok(format!("Welcome {} - {}!", p.username, data.0)) Ok(format!("Welcome {} - {}!", p.username, data.0))
}) })
@ -334,8 +327,7 @@ fn test_path_and_query_extractor2_async4() {
app.resource("/{username}/index.html", |r| { app.resource("/{username}/index.html", |r| {
r.route() r.route()
.with(|data: (Json<Value>, Path<PParam>, Query<PParam>)| { .with(|data: (Json<Value>, Path<PParam>, Query<PParam>)| {
Timeout::new(Duration::from_millis(10), &Arbiter::handle()) Delay::new(Instant::now() + Duration::from_millis(10))
.unwrap()
.and_then(move |_| { .and_then(move |_| {
Ok(format!("Welcome {} - {}!", data.1.username, (data.0).0)) Ok(format!("Welcome {} - {}!", data.1.username, (data.0).0))
}) })
@ -443,8 +435,8 @@ fn test_nested_scope_and_path_extractor() {
fn test_impl_trait( fn test_impl_trait(
data: (Json<Value>, Path<PParam>, Query<PParam>), data: (Json<Value>, Path<PParam>, Query<PParam>),
) -> impl Future<Item = String, Error = io::Error> { ) -> impl Future<Item = String, Error = io::Error> {
Timeout::new(Duration::from_millis(10), &Arbiter::handle()) Delay::new(Instant::now() + Duration::from_millis(10))
.unwrap() .map_err(|_| io::Error::new(io::ErrorKind::Other, "timeout"))
.and_then(move |_| Ok(format!("Welcome {} - {}!", data.1.username, (data.0).0))) .and_then(move |_| Ok(format!("Welcome {} - {}!", data.1.username, (data.0).0)))
} }
@ -452,8 +444,8 @@ fn test_impl_trait(
fn test_impl_trait_err( fn test_impl_trait_err(
_data: (Json<Value>, Path<PParam>, Query<PParam>), _data: (Json<Value>, Path<PParam>, Query<PParam>),
) -> impl Future<Item = String, Error = io::Error> { ) -> impl Future<Item = String, Error = io::Error> {
Timeout::new(Duration::from_millis(10), &Arbiter::handle()) Delay::new(Instant::now() + Duration::from_millis(10))
.unwrap() .map_err(|_| io::Error::new(io::ErrorKind::Other, "timeout"))
.and_then(move |_| Err(io::Error::new(io::ErrorKind::Other, "other"))) .and_then(move |_| Err(io::Error::new(io::ErrorKind::Other, "other")))
} }

View File

@ -1,17 +1,16 @@
extern crate actix; extern crate actix;
extern crate actix_web; extern crate actix_web;
extern crate futures; extern crate futures;
extern crate tokio_core; extern crate tokio_timer;
use std::sync::atomic::{AtomicUsize, Ordering}; use std::sync::atomic::{AtomicUsize, Ordering};
use std::sync::Arc; use std::sync::Arc;
use std::thread; use std::thread;
use std::time::Duration; use std::time::{Duration, Instant};
use actix::*;
use actix_web::*; use actix_web::*;
use futures::{future, Future}; use futures::{future, Future};
use tokio_core::reactor::Timeout; use tokio_timer::Delay;
struct MiddlewareTest { struct MiddlewareTest {
start: Arc<AtomicUsize>, start: Arc<AtomicUsize>,
@ -245,8 +244,7 @@ fn test_middleware_async_handler() {
}) })
.resource("/", |r| { .resource("/", |r| {
r.route().a(|_| { r.route().a(|_| {
Timeout::new(Duration::from_millis(10), &Arbiter::handle()) Delay::new(Instant::now() + Duration::from_millis(10))
.unwrap()
.and_then(|_| Ok(HttpResponse::Ok())) .and_then(|_| Ok(HttpResponse::Ok()))
}) })
}) })
@ -281,8 +279,7 @@ fn test_resource_middleware_async_handler() {
App::new().resource("/test", |r| { App::new().resource("/test", |r| {
r.middleware(mw); r.middleware(mw);
r.route().a(|_| { r.route().a(|_| {
Timeout::new(Duration::from_millis(10), &Arbiter::handle()) Delay::new(Instant::now() + Duration::from_millis(10))
.unwrap()
.and_then(|_| Ok(HttpResponse::Ok())) .and_then(|_| Ok(HttpResponse::Ok()))
}) })
}) })
@ -317,8 +314,7 @@ fn test_scope_middleware_async_handler() {
}) })
.resource("/test", |r| { .resource("/test", |r| {
r.route().a(|_| { r.route().a(|_| {
Timeout::new(Duration::from_millis(10), &Arbiter::handle()) Delay::new(Instant::now() + Duration::from_millis(10))
.unwrap()
.and_then(|_| Ok(HttpResponse::Ok())) .and_then(|_| Ok(HttpResponse::Ok()))
}) })
}) })
@ -436,7 +432,7 @@ struct MiddlewareAsyncTest {
impl<S> middleware::Middleware<S> for MiddlewareAsyncTest { impl<S> middleware::Middleware<S> for MiddlewareAsyncTest {
fn start(&self, _: &mut HttpRequest<S>) -> Result<middleware::Started> { fn start(&self, _: &mut HttpRequest<S>) -> Result<middleware::Started> {
let to = Timeout::new(Duration::from_millis(10), &Arbiter::handle()).unwrap(); let to = Delay::new(Instant::now() + Duration::from_millis(10));
let start = Arc::clone(&self.start); let start = Arc::clone(&self.start);
Ok(middleware::Started::Future(Box::new( Ok(middleware::Started::Future(Box::new(
@ -450,7 +446,7 @@ impl<S> middleware::Middleware<S> for MiddlewareAsyncTest {
fn response( fn response(
&self, _: &mut HttpRequest<S>, resp: HttpResponse, &self, _: &mut HttpRequest<S>, resp: HttpResponse,
) -> Result<middleware::Response> { ) -> Result<middleware::Response> {
let to = Timeout::new(Duration::from_millis(10), &Arbiter::handle()).unwrap(); let to = Delay::new(Instant::now() + Duration::from_millis(10));
let response = Arc::clone(&self.response); let response = Arc::clone(&self.response);
Ok(middleware::Response::Future(Box::new( Ok(middleware::Response::Future(Box::new(
@ -462,7 +458,7 @@ impl<S> middleware::Middleware<S> for MiddlewareAsyncTest {
} }
fn finish(&self, _: &mut HttpRequest<S>, _: &HttpResponse) -> middleware::Finished { fn finish(&self, _: &mut HttpRequest<S>, _: &HttpResponse) -> middleware::Finished {
let to = Timeout::new(Duration::from_millis(10), &Arbiter::handle()).unwrap(); let to = Delay::new(Instant::now() + Duration::from_millis(10));
let finish = Arc::clone(&self.finish); let finish = Arc::clone(&self.finish);
middleware::Finished::Future(Box::new(to.from_err().and_then(move |_| { middleware::Finished::Future(Box::new(to.from_err().and_then(move |_| {

View File

@ -6,7 +6,9 @@ extern crate futures;
extern crate h2; extern crate h2;
extern crate http as modhttp; extern crate http as modhttp;
extern crate rand; extern crate rand;
extern crate tokio_core; extern crate tokio;
extern crate tokio_reactor;
extern crate tokio_tcp;
#[cfg(feature = "brotli")] #[cfg(feature = "brotli")]
extern crate brotli2; extern crate brotli2;
@ -25,8 +27,9 @@ use rand::Rng;
use std::io::{Read, Write}; use std::io::{Read, Write};
use std::sync::{mpsc, Arc}; use std::sync::{mpsc, Arc};
use std::{net, thread, time}; use std::{net, thread, time};
use tokio_core::net::TcpStream; use tokio::executor::current_thread;
use tokio_core::reactor::Core; use tokio::runtime::current_thread::Runtime;
use tokio_tcp::TcpStream;
use actix::System; use actix::System;
use actix_web::*; use actix_web::*;
@ -790,9 +793,8 @@ fn test_h2() {
let srv = test::TestServer::new(|app| app.handler(|_| HttpResponse::Ok().body(STR))); let srv = test::TestServer::new(|app| app.handler(|_| HttpResponse::Ok().body(STR)));
let addr = srv.addr(); let addr = srv.addr();
let mut core = Core::new().unwrap(); let mut core = Runtime::new().unwrap();
let handle = core.handle(); let tcp = TcpStream::connect(&addr);
let tcp = TcpStream::connect(&addr, &handle);
let tcp = tcp let tcp = tcp
.then(|res| h2client::handshake(res.unwrap())) .then(|res| h2client::handshake(res.unwrap()))
@ -806,7 +808,7 @@ fn test_h2() {
let (response, _) = client.send_request(request, false).unwrap(); let (response, _) = client.send_request(request, false).unwrap();
// Spawn a task to run the conn... // Spawn a task to run the conn...
handle.spawn(h2.map_err(|e| println!("GOT ERR={:?}", e))); current_thread::spawn(h2.map_err(|e| println!("GOT ERR={:?}", e)));
response.and_then(|response| { response.and_then(|response| {
assert_eq!(response.status(), http::StatusCode::OK); assert_eq!(response.status(), http::StatusCode::OK);
@ -819,7 +821,7 @@ fn test_h2() {
}) })
}) })
}); });
let _res = core.run(tcp); let _res = core.block_on(tcp);
// assert_eq!(_res.unwrap(), Bytes::from_static(STR.as_ref())); // assert_eq!(_res.unwrap(), Bytes::from_static(STR.as_ref()));
} }