1
0
mirror of https://github.com/fafhrd91/actix-web synced 2024-11-25 08:52:42 +01:00
actix-web/src/server/srv.rs

761 lines
26 KiB
Rust
Raw Normal View History

use std::{io, net, thread};
2017-10-07 06:48:14 +02:00
use std::rc::Rc;
2017-12-27 21:58:32 +01:00
use std::sync::{Arc, mpsc as sync_mpsc};
2017-12-14 01:44:35 +01:00
use std::time::Duration;
use std::marker::PhantomData;
2017-12-17 21:35:04 +01:00
use std::collections::HashMap;
2017-10-07 06:48:14 +02:00
2018-01-05 22:30:21 +01:00
use actix::prelude::*;
2018-01-06 01:32:36 +01:00
use actix::actors::signal;
2017-12-29 01:25:47 +01:00
use futures::{Future, Sink, Stream};
use futures::sync::mpsc;
use tokio_io::{AsyncRead, AsyncWrite};
use tokio_core::net::TcpStream;
2017-12-27 20:22:27 +01:00
use mio;
use num_cpus;
2017-12-28 21:38:37 +01:00
use net2::TcpBuilder;
2017-10-07 06:48:14 +02:00
#[cfg(feature="tls")]
use native_tls::TlsAcceptor;
#[cfg(feature="tls")]
2017-12-28 22:07:29 +01:00
use tokio_tls::TlsStream;
2017-11-04 21:24:57 +01:00
#[cfg(feature="alpn")]
use openssl::ssl::{AlpnError, SslAcceptorBuilder};
#[cfg(feature="alpn")]
2017-12-28 22:07:29 +01:00
use tokio_openssl::SslStream;
2017-12-14 07:54:52 +01:00
use helpers;
2018-01-12 03:35:05 +01:00
use super::{HttpHandler, IntoHttpHandler, IoStream};
use super::{PauseServer, ResumeServer, StopServer};
use super::channel::{HttpChannel, WrapperStream};
use super::worker::{Conn, Worker, StreamHandlerType, StopWorker};
use super::settings::{ServerSettings, WorkerSettings};
2017-12-08 07:54:44 +01:00
2017-10-08 23:56:51 +02:00
/// An HTTP Server
///
2017-12-14 08:09:20 +01:00
/// `T` - async stream, anything that implements `AsyncRead` + `AsyncWrite`.
///
/// `A` - peer address
2017-10-22 03:54:24 +02:00
///
/// `H` - request handler
pub struct HttpServer<T, A, H, U>
2018-01-04 07:43:44 +01:00
where H: HttpHandler + 'static
{
2017-12-14 06:38:47 +01:00
h: Option<Rc<WorkerSettings<H>>>,
io: PhantomData<T>,
addr: PhantomData<A>,
threads: usize,
2017-12-17 21:35:04 +01:00
backlog: i32,
2017-12-26 23:36:03 +01:00
host: Option<String>,
2017-12-14 07:36:28 +01:00
keep_alive: Option<u64>,
factory: Arc<Fn() -> U + Send + Sync>,
workers: Vec<SyncAddress<Worker<H>>>,
2017-12-27 20:22:27 +01:00
sockets: HashMap<net::SocketAddr, net::TcpListener>,
2017-12-27 21:58:32 +01:00
accept: Vec<(mio::SetReadiness, sync_mpsc::Sender<Command>)>,
2017-12-28 20:36:20 +01:00
exit: bool,
2017-12-29 01:25:47 +01:00
shutdown_timeout: u16,
2018-01-06 01:32:36 +01:00
signals: Option<SyncAddress<signal::ProcessSignals>>,
no_signals: bool,
2017-10-07 06:48:14 +02:00
}
2018-01-04 07:43:44 +01:00
unsafe impl<T, A, H, U> Sync for HttpServer<T, A, H, U> where H: HttpHandler + 'static {}
unsafe impl<T, A, H, U> Send for HttpServer<T, A, H, U> where H: HttpHandler + 'static {}
2017-12-28 02:49:10 +01:00
2018-01-08 02:13:49 +01:00
impl<T, A, H, U, V> Actor for HttpServer<T, A, H, U>
where A: 'static,
T: IoStream,
H: HttpHandler,
U: IntoIterator<Item=V> + 'static,
V: IntoHttpHandler<Handler=H>,
{
2017-10-07 06:48:14 +02:00
type Context = Context<Self>;
2017-12-14 01:44:35 +01:00
fn started(&mut self, ctx: &mut Self::Context) {
self.update_time(ctx);
}
}
impl<T, A, H, U, V> HttpServer<T, A, H, U>
2017-12-14 01:44:35 +01:00
where A: 'static,
T: IoStream,
2017-12-14 01:44:35 +01:00
H: HttpHandler,
U: IntoIterator<Item=V> + 'static,
V: IntoHttpHandler<Handler=H>,
2017-10-22 03:54:24 +02:00
{
2017-12-14 08:09:20 +01:00
/// Create new http server with application factory
pub fn new<F>(factory: F) -> Self
where F: Sync + Send + 'static + Fn() -> U,
2017-12-06 20:00:39 +01:00
{
2017-12-14 06:38:47 +01:00
HttpServer{ h: None,
2017-10-22 03:54:24 +02:00
io: PhantomData,
addr: PhantomData,
threads: num_cpus::get(),
2017-12-17 21:35:04 +01:00
backlog: 2048,
2017-12-26 23:36:03 +01:00
host: None,
2017-12-14 06:38:47 +01:00
keep_alive: None,
factory: Arc::new(factory),
workers: Vec::new(),
2017-12-17 21:35:04 +01:00
sockets: HashMap::new(),
2017-12-27 21:58:32 +01:00
accept: Vec::new(),
2017-12-28 20:36:20 +01:00
exit: false,
2017-12-29 01:25:47 +01:00
shutdown_timeout: 30,
2018-01-06 01:32:36 +01:00
signals: None,
no_signals: false,
}
}
2018-01-08 02:13:49 +01:00
fn update_time(&self, ctx: &mut Context<Self>) {
helpers::update_date();
ctx.run_later(Duration::new(1, 0), |slf, ctx| slf.update_time(ctx));
}
/// Set number of workers to start.
///
/// By default http server uses number of available logical cpu as threads count.
pub fn threads(mut self, num: usize) -> Self {
self.threads = num;
self
2017-10-07 06:48:14 +02:00
}
2017-12-17 21:35:04 +01:00
/// Set the maximum number of pending connections.
///
/// This refers to the number of clients that can be waiting to be served.
/// Exceeding this number results in the client getting an error when
/// attempting to connect. It should only affect servers under significant load.
///
/// Generally set in the 64-2048 range. Default value is 2048.
2017-12-27 20:22:27 +01:00
///
/// This method should be called before `bind()` method call.
2017-12-17 21:35:04 +01:00
pub fn backlog(mut self, num: i32) -> Self {
self.backlog = num;
self
}
2017-12-14 06:38:47 +01:00
/// Set server keep-alive setting.
///
/// By default keep alive is enabled.
///
/// - `Some(75)` - enable
///
/// - `Some(0)` - disable
///
/// - `None` - use `SO_KEEPALIVE` socket option
2017-12-14 07:36:28 +01:00
pub fn keep_alive(mut self, val: Option<u64>) -> Self {
2017-12-14 06:38:47 +01:00
self.keep_alive = val;
self
}
2017-12-26 23:36:03 +01:00
/// Set server host name.
///
/// Host name is used by application router aa a hostname for url generation.
/// Check [ConnectionInfo](./dev/struct.ConnectionInfo.html#method.host) documentation
/// for more information.
pub fn server_hostname(mut self, val: String) -> Self {
self.host = Some(val);
self
}
2017-12-28 20:36:20 +01:00
/// Send `SystemExit` message to actix system
///
/// `SystemExit` message stops currently running system arbiter and all
/// nested arbiters.
pub fn system_exit(mut self) -> Self {
self.exit = true;
self
}
2018-01-06 01:32:36 +01:00
/// Set alternative address for `ProcessSignals` actor.
pub fn signals(mut self, addr: SyncAddress<signal::ProcessSignals>) -> Self {
self.signals = Some(addr);
self
}
/// Disable signal handling
pub fn disable_signals(mut self) -> Self {
self.no_signals = true;
self
}
2017-12-29 01:25:47 +01:00
/// Timeout for graceful workers shutdown.
///
/// After receiving a stop signal, workers have this much time to finish serving requests.
/// Workers still alive after the timeout are force dropped.
///
/// By default shutdown timeout sets to 30 seconds.
pub fn shutdown_timeout(mut self, sec: u16) -> Self {
self.shutdown_timeout = sec;
self
}
/// Get addresses of bound sockets.
pub fn addrs(&self) -> Vec<net::SocketAddr> {
self.sockets.keys().cloned().collect()
}
2017-12-17 21:35:04 +01:00
/// The socket address to bind
///
/// To mind multiple addresses this method can be call multiple times.
pub fn bind<S: net::ToSocketAddrs>(mut self, addr: S) -> io::Result<Self> {
2017-10-15 23:53:03 +02:00
let mut err = None;
2017-12-17 21:35:04 +01:00
let mut succ = false;
2017-12-28 02:49:10 +01:00
for addr in addr.to_socket_addrs()? {
match create_tcp_listener(addr, self.backlog) {
Ok(lst) => {
succ = true;
self.sockets.insert(lst.local_addr().unwrap(), lst);
},
Err(e) => err = Some(e),
2017-10-15 23:53:03 +02:00
}
}
2017-12-17 21:35:04 +01:00
if !succ {
2017-10-15 23:53:03 +02:00
if let Some(e) = err.take() {
Err(e)
} else {
Err(io::Error::new(io::ErrorKind::Other, "Can not bind to address."))
}
} else {
2017-12-17 21:35:04 +01:00
Ok(self)
2017-10-15 23:53:03 +02:00
}
2017-10-07 06:48:14 +02:00
}
2017-12-13 21:47:07 +01:00
fn start_workers(&mut self, settings: &ServerSettings, handler: &StreamHandlerType)
2017-12-28 21:38:37 +01:00
-> Vec<mpsc::UnboundedSender<Conn<net::TcpStream>>>
{
// start workers
let mut workers = Vec::new();
for _ in 0..self.threads {
let s = settings.clone();
2017-12-28 21:38:37 +01:00
let (tx, rx) = mpsc::unbounded::<Conn<net::TcpStream>>();
2017-12-13 21:47:07 +01:00
let h = handler.clone();
2017-12-14 07:36:28 +01:00
let ka = self.keep_alive;
let factory = Arc::clone(&self.factory);
let addr = Arbiter::start(move |ctx: &mut Context<_>| {
2017-12-29 20:33:04 +01:00
let apps: Vec<_> = (*factory)()
.into_iter()
.map(|h| h.into_handler(s.clone())).collect();
2018-01-05 22:30:21 +01:00
ctx.add_message_stream(rx);
2017-12-14 06:38:47 +01:00
Worker::new(apps, h, ka)
});
workers.push(tx);
self.workers.push(addr);
}
info!("Starting {} http workers", self.threads);
workers
}
2018-01-06 01:32:36 +01:00
// subscribe to os signals
2018-01-08 02:13:49 +01:00
fn subscribe_to_signals(&self) -> Option<SyncAddress<signal::ProcessSignals>> {
if !self.no_signals {
2018-01-06 01:32:36 +01:00
if let Some(ref signals) = self.signals {
2018-01-08 02:13:49 +01:00
Some(signals.clone())
2018-01-06 01:32:36 +01:00
} else {
2018-01-08 02:13:49 +01:00
Some(Arbiter::system_registry().get::<signal::ProcessSignals>())
2018-01-06 01:32:36 +01:00
}
2018-01-08 02:13:49 +01:00
} else {
None
2018-01-06 01:32:36 +01:00
}
}
}
impl<H: HttpHandler, U, V> HttpServer<TcpStream, net::SocketAddr, H, U>
where U: IntoIterator<Item=V> + 'static,
V: IntoHttpHandler<Handler=H>,
{
2018-01-15 22:47:25 +01:00
/// Start listening for incoming connections.
///
2018-01-15 22:47:25 +01:00
/// This method starts number of http handler workers in separate threads.
/// For each address this method starts separate thread which does `accept()` in a loop.
2017-12-19 18:08:36 +01:00
///
/// This methods panics if no socket addresses get bound.
2017-12-26 21:46:27 +01:00
///
/// This method requires to run within properly configured `Actix` system.
///
/// ```rust
/// extern crate actix;
/// extern crate actix_web;
/// use actix_web::*;
///
/// fn main() {
/// let sys = actix::System::new("example"); // <- create Actix system
///
/// HttpServer::new(
/// || Application::new()
/// .resource("/", |r| r.h(httpcodes::HTTPOk)))
2017-12-28 02:49:10 +01:00
/// .bind("127.0.0.1:0").expect("Can not bind to 127.0.0.1:0")
2017-12-26 21:46:27 +01:00
/// .start();
/// # actix::Arbiter::system().send(actix::msgs::SystemExit(0));
///
/// let _ = sys.run(); // <- Run actix system, this method actually starts all async processes
/// }
/// ```
2017-12-19 18:08:36 +01:00
pub fn start(mut self) -> SyncAddress<Self>
{
2017-12-17 21:35:04 +01:00
if self.sockets.is_empty() {
2018-01-15 22:47:25 +01:00
panic!("HttpServer::bind() has to be called before start()");
2017-12-17 21:35:04 +01:00
} else {
2017-12-27 20:22:27 +01:00
let addrs: Vec<(net::SocketAddr, net::TcpListener)> =
self.sockets.drain().collect();
2017-12-26 23:36:03 +01:00
let settings = ServerSettings::new(Some(addrs[0].0), &self.host, false);
2017-12-17 21:35:04 +01:00
let workers = self.start_workers(&settings, &StreamHandlerType::Normal);
// start acceptors threads
for (addr, sock) in addrs {
info!("Starting http server on {}", addr);
2017-12-27 21:58:32 +01:00
self.accept.push(
start_accept_thread(sock, addr, self.backlog, workers.clone()));
2017-12-17 21:35:04 +01:00
}
2017-12-08 18:24:05 +01:00
2017-12-17 21:35:04 +01:00
// start http server actor
2018-01-08 02:13:49 +01:00
let signals = self.subscribe_to_signals();
let addr: SyncAddress<_> = Actor::start(self);
2018-01-27 19:04:56 +01:00
signals.map(|signals| signals.send(
signal::Subscribe(addr.clone().into_subscriber())));
2018-01-08 02:13:49 +01:00
addr
2017-12-17 21:35:04 +01:00
}
}
2017-12-28 02:49:10 +01:00
2018-01-15 22:47:25 +01:00
/// Spawn new thread and start listening for incoming connections.
2017-12-28 02:49:10 +01:00
///
/// This method spawns new thread and starts new actix system. Other than that it is
2018-01-06 01:32:36 +01:00
/// similar to `start()` method. This method blocks.
2017-12-28 02:49:10 +01:00
///
/// This methods panics if no socket addresses get bound.
///
2018-01-06 01:32:36 +01:00
/// ```rust,ignore
2017-12-28 02:49:10 +01:00
/// # extern crate futures;
/// # extern crate actix;
/// # extern crate actix_web;
/// # use futures::Future;
/// use actix_web::*;
///
/// fn main() {
2018-01-06 01:32:36 +01:00
/// HttpServer::new(
2017-12-28 02:49:10 +01:00
/// || Application::new()
/// .resource("/", |r| r.h(httpcodes::HTTPOk)))
/// .bind("127.0.0.1:0").expect("Can not bind to 127.0.0.1:0")
2018-01-06 01:32:36 +01:00
/// .run();
2017-12-28 02:49:10 +01:00
/// }
/// ```
2018-01-06 01:32:36 +01:00
pub fn run(mut self) {
2017-12-28 20:36:20 +01:00
self.exit = true;
2018-01-06 01:32:36 +01:00
self.no_signals = false;
2017-12-28 02:49:10 +01:00
2018-01-06 01:32:36 +01:00
let _ = thread::spawn(move || {
2017-12-28 02:49:10 +01:00
let sys = System::new("http-server");
2018-01-06 01:32:36 +01:00
self.start();
let _ = sys.run();
}).join();
2017-12-28 02:49:10 +01:00
}
}
#[cfg(feature="tls")]
impl<H: HttpHandler, U, V> HttpServer<TlsStream<TcpStream>, net::SocketAddr, H, U>
where U: IntoIterator<Item=V> + 'static,
V: IntoHttpHandler<Handler=H>,
{
2018-01-15 22:47:25 +01:00
/// Start listening for incoming tls connections.
2017-12-18 22:06:41 +01:00
pub fn start_tls(mut self, pkcs12: ::Pkcs12) -> io::Result<SyncAddress<Self>> {
2017-12-17 21:35:04 +01:00
if self.sockets.is_empty() {
Err(io::Error::new(io::ErrorKind::Other, "No socket addresses are bound"))
} else {
2017-12-27 21:58:32 +01:00
let addrs: Vec<(net::SocketAddr, net::TcpListener)> = self.sockets.drain().collect();
2017-12-27 00:17:20 +01:00
let settings = ServerSettings::new(Some(addrs[0].0), &self.host, false);
2017-12-17 21:35:04 +01:00
let acceptor = match TlsAcceptor::builder(pkcs12) {
Ok(builder) => {
match builder.build() {
Ok(acceptor) => acceptor,
Err(err) => return Err(io::Error::new(io::ErrorKind::Other, err))
}
}
2017-12-17 21:35:04 +01:00
Err(err) => return Err(io::Error::new(io::ErrorKind::Other, err))
};
let workers = self.start_workers(&settings, &StreamHandlerType::Tls(acceptor));
// start acceptors threads
for (addr, sock) in addrs {
info!("Starting tls http server on {}", addr);
2017-12-27 21:58:32 +01:00
self.accept.push(
start_accept_thread(sock, addr, self.backlog, workers.clone()));
}
2017-12-17 21:35:04 +01:00
// start http server actor
2018-01-08 02:13:49 +01:00
let signals = self.subscribe_to_signals();
let addr: SyncAddress<_> = Actor::start(self);
2018-01-27 20:15:03 +01:00
signals.map(|signals| signals.send(
signal::Subscribe(addr.clone().into_subscriber())));
2018-01-08 02:13:49 +01:00
Ok(addr)
}
}
}
#[cfg(feature="alpn")]
impl<H: HttpHandler, U, V> HttpServer<SslStream<TcpStream>, net::SocketAddr, H, U>
where U: IntoIterator<Item=V> + 'static,
V: IntoHttpHandler<Handler=H>,
{
2018-01-15 22:47:25 +01:00
/// Start listening for incoming tls connections.
///
2017-12-17 21:35:04 +01:00
/// This method sets alpn protocols to "h2" and "http/1.1"
pub fn start_ssl(mut self, mut builder: SslAcceptorBuilder) -> io::Result<SyncAddress<Self>>
{
2017-12-17 21:35:04 +01:00
if self.sockets.is_empty() {
Err(io::Error::new(io::ErrorKind::Other, "No socket addresses are bound"))
} else {
// alpn support
builder.set_alpn_protos(b"\x02h2\x08http/1.1")?;
builder.set_alpn_select_callback(|_, protos| {
const H2: &[u8] = b"\x02h2";
if protos.windows(3).any(|window| window == H2) {
Ok(b"h2")
} else {
Err(AlpnError::NOACK)
}
});
let acceptor = builder.build();
2017-12-27 21:58:32 +01:00
let addrs: Vec<(net::SocketAddr, net::TcpListener)> = self.sockets.drain().collect();
2017-12-27 00:17:20 +01:00
let settings = ServerSettings::new(Some(addrs[0].0), &self.host, false);
2017-12-17 21:35:04 +01:00
let workers = self.start_workers(&settings, &StreamHandlerType::Alpn(acceptor));
// start acceptors threads
for (addr, sock) in addrs {
info!("Starting tls http server on {}", addr);
2017-12-27 21:58:32 +01:00
self.accept.push(
2017-12-27 22:26:31 +01:00
start_accept_thread(sock, addr, self.backlog, workers.clone()));
2017-12-17 21:35:04 +01:00
}
2017-12-17 21:35:04 +01:00
// start http server actor
2018-01-08 02:13:49 +01:00
let signals = self.subscribe_to_signals();
let addr: SyncAddress<_> = Actor::start(self);
2018-01-27 20:15:03 +01:00
signals.map(|signals| signals.send(
signal::Subscribe(addr.clone().into_subscriber())));
2018-01-08 02:13:49 +01:00
Ok(addr)
}
}
}
impl<T, A, H, U, V> HttpServer<WrapperStream<T>, A, H, U>
2017-12-19 03:56:58 +01:00
where A: 'static,
T: AsyncRead + AsyncWrite + 'static,
H: HttpHandler,
U: IntoIterator<Item=V> + 'static,
V: IntoHttpHandler<Handler=H>,
{
2018-01-15 22:47:25 +01:00
/// Start listening for incoming connections from a stream.
2017-12-19 03:56:58 +01:00
///
/// This method uses only one thread for handling incoming connections.
2017-12-19 18:08:36 +01:00
pub fn start_incoming<S>(mut self, stream: S, secure: bool) -> SyncAddress<Self>
2017-12-19 03:56:58 +01:00
where S: Stream<Item=(T, A), Error=io::Error> + 'static
{
if !self.sockets.is_empty() {
2017-12-27 20:22:27 +01:00
let addrs: Vec<(net::SocketAddr, net::TcpListener)> =
self.sockets.drain().collect();
2017-12-26 23:36:03 +01:00
let settings = ServerSettings::new(Some(addrs[0].0), &self.host, false);
2017-12-19 03:56:58 +01:00
let workers = self.start_workers(&settings, &StreamHandlerType::Normal);
// start acceptors threads
for (addr, sock) in addrs {
info!("Starting http server on {}", addr);
2017-12-27 21:58:32 +01:00
self.accept.push(
start_accept_thread(sock, addr, self.backlog, workers.clone()));
2017-12-19 03:56:58 +01:00
}
}
// set server settings
let addr: net::SocketAddr = "127.0.0.1:8080".parse().unwrap();
2017-12-26 23:36:03 +01:00
let settings = ServerSettings::new(Some(addr), &self.host, secure);
2017-12-29 20:33:04 +01:00
let apps: Vec<_> = (*self.factory)()
.into_iter()
.map(|h| h.into_handler(settings.clone())).collect();
2017-12-19 03:56:58 +01:00
self.h = Some(Rc::new(WorkerSettings::new(apps, self.keep_alive)));
// start server
2018-01-08 02:13:49 +01:00
let signals = self.subscribe_to_signals();
let addr: SyncAddress<_> = HttpServer::create(move |ctx| {
2017-12-19 03:56:58 +01:00
ctx.add_stream(stream.map(
move |(t, _)| Conn{io: WrapperStream::new(t), peer: None, http2: false}));
2017-12-19 03:56:58 +01:00
self
2018-01-08 02:13:49 +01:00
});
2018-01-27 19:04:56 +01:00
signals.map(|signals| signals.send(
signal::Subscribe(addr.clone().into_subscriber())));
2018-01-08 02:13:49 +01:00
addr
2017-12-19 03:56:58 +01:00
}
}
2018-01-05 22:30:21 +01:00
/// Signals support
2017-12-28 20:36:20 +01:00
/// Handle `SIGINT`, `SIGTERM`, `SIGQUIT` signals and send `SystemExit(0)`
/// message to `System` actor.
2018-01-08 02:13:49 +01:00
impl<T, A, H, U, V> Handler<signal::Signal> for HttpServer<T, A, H, U>
where T: IoStream,
2017-12-28 20:36:20 +01:00
H: HttpHandler + 'static,
2018-01-08 02:13:49 +01:00
U: IntoIterator<Item=V> + 'static,
V: IntoHttpHandler<Handler=H>,
2017-12-28 20:36:20 +01:00
A: 'static,
{
2018-01-05 22:30:21 +01:00
type Result = ();
fn handle(&mut self, msg: signal::Signal, ctx: &mut Context<Self>) {
2017-12-28 20:36:20 +01:00
match msg.0 {
signal::SignalType::Int => {
info!("SIGINT received, exiting");
self.exit = true;
Handler::<StopServer>::handle(self, StopServer{graceful: false}, ctx);
}
signal::SignalType::Term => {
info!("SIGTERM received, stopping");
self.exit = true;
Handler::<StopServer>::handle(self, StopServer{graceful: true}, ctx);
}
signal::SignalType::Quit => {
info!("SIGQUIT received, exiting");
self.exit = true;
Handler::<StopServer>::handle(self, StopServer{graceful: false}, ctx);
}
_ => (),
2018-01-05 22:30:21 +01:00
}
2017-12-28 20:36:20 +01:00
}
}
2018-01-08 02:13:49 +01:00
impl<T, A, H, U, V> Handler<io::Result<Conn<T>>> for HttpServer<T, A, H, U>
where T: IoStream,
2017-10-22 03:54:24 +02:00
H: HttpHandler + 'static,
2018-01-08 02:13:49 +01:00
U: IntoIterator<Item=V> + 'static,
V: IntoHttpHandler<Handler=H>,
A: 'static,
{
2018-01-05 22:30:21 +01:00
type Result = ();
fn handle(&mut self, msg: io::Result<Conn<T>>, _: &mut Context<Self>) -> Self::Result {
match msg {
Ok(msg) =>
Arbiter::handle().spawn(
HttpChannel::new(
Rc::clone(self.h.as_ref().unwrap()), msg.io, msg.peer, msg.http2)),
Err(err) =>
debug!("Error handling request: {}", err),
}
2017-10-07 06:48:14 +02:00
}
}
2018-01-08 02:13:49 +01:00
impl<T, A, H, U, V> Handler<PauseServer> for HttpServer<T, A, H, U>
where T: IoStream,
2017-12-27 21:58:32 +01:00
H: HttpHandler + 'static,
2018-01-08 02:13:49 +01:00
U: IntoIterator<Item=V> + 'static,
V: IntoHttpHandler<Handler=H>,
2017-12-27 21:58:32 +01:00
A: 'static,
{
2018-01-05 22:30:21 +01:00
type Result = ();
fn handle(&mut self, _: PauseServer, _: &mut Context<Self>)
2017-12-27 21:58:32 +01:00
{
for item in &self.accept {
let _ = item.1.send(Command::Pause);
let _ = item.0.set_readiness(mio::Ready::readable());
}
}
}
2018-01-08 02:13:49 +01:00
impl<T, A, H, U, V> Handler<ResumeServer> for HttpServer<T, A, H, U>
where T: IoStream,
2017-12-27 21:58:32 +01:00
H: HttpHandler + 'static,
2018-01-08 02:13:49 +01:00
U: IntoIterator<Item=V> + 'static,
V: IntoHttpHandler<Handler=H>,
2017-12-27 21:58:32 +01:00
A: 'static,
{
2018-01-05 22:30:21 +01:00
type Result = ();
fn handle(&mut self, _: ResumeServer, _: &mut Context<Self>) {
2017-12-27 21:58:32 +01:00
for item in &self.accept {
let _ = item.1.send(Command::Resume);
let _ = item.0.set_readiness(mio::Ready::readable());
}
}
}
2018-01-08 02:13:49 +01:00
impl<T, A, H, U, V> Handler<StopServer> for HttpServer<T, A, H, U>
where T: IoStream,
2017-12-27 21:58:32 +01:00
H: HttpHandler + 'static,
2018-01-08 02:13:49 +01:00
U: IntoIterator<Item=V> + 'static,
V: IntoHttpHandler<Handler=H>,
2017-12-27 21:58:32 +01:00
A: 'static,
{
2018-01-05 22:30:21 +01:00
type Result = actix::Response<Self, StopServer>;
fn handle(&mut self, msg: StopServer, ctx: &mut Context<Self>) -> Self::Result {
2017-12-28 22:07:29 +01:00
// stop accept threads
2017-12-27 21:58:32 +01:00
for item in &self.accept {
let _ = item.1.send(Command::Stop);
let _ = item.0.set_readiness(mio::Ready::readable());
}
2017-12-28 02:49:10 +01:00
2017-12-28 22:07:29 +01:00
// stop workers
2017-12-29 01:25:47 +01:00
let (tx, rx) = mpsc::channel(1);
let dur = if msg.graceful {
Some(Duration::new(u64::from(self.shutdown_timeout), 0))
} else {
None
};
2017-12-28 22:07:29 +01:00
for worker in &self.workers {
2017-12-29 01:25:47 +01:00
let tx2 = tx.clone();
let fut = worker.call(self, StopWorker{graceful: dur});
ActorFuture::then(fut, move |_, slf, _| {
slf.workers.pop();
if slf.workers.is_empty() {
let _ = tx2.send(());
// we need to stop system if server was spawned
if slf.exit {
2018-01-05 22:30:21 +01:00
Arbiter::system().send(actix::msgs::SystemExit(0))
2017-12-29 01:25:47 +01:00
}
}
2018-01-05 22:30:21 +01:00
actix::fut::ok(())
2017-12-29 01:25:47 +01:00
}).spawn(ctx);
2017-12-28 22:07:29 +01:00
}
2017-12-29 01:25:47 +01:00
if !self.workers.is_empty() {
Self::async_reply(
rx.into_future().map(|_| ()).map_err(|_| ()).actfuture())
} else {
// we need to stop system if server was spawned
if self.exit {
2018-01-05 22:30:21 +01:00
Arbiter::system().send(actix::msgs::SystemExit(0))
2017-12-29 01:25:47 +01:00
}
2018-01-07 07:59:39 +01:00
Self::reply(Ok(()))
2017-12-28 02:49:10 +01:00
}
2017-12-27 21:58:32 +01:00
}
}
enum Command {
Pause,
Resume,
Stop,
}
fn start_accept_thread(sock: net::TcpListener, addr: net::SocketAddr, backlog: i32,
2017-12-28 21:38:37 +01:00
workers: Vec<mpsc::UnboundedSender<Conn<net::TcpStream>>>)
2017-12-27 21:58:32 +01:00
-> (mio::SetReadiness, sync_mpsc::Sender<Command>)
{
let (tx, rx) = sync_mpsc::channel();
let (reg, readiness) = mio::Registration::new2();
2017-12-27 20:22:27 +01:00
// start accept thread
2017-12-13 21:47:07 +01:00
let _ = thread::Builder::new().name(format!("Accept on {}", addr)).spawn(move || {
2017-12-27 21:58:32 +01:00
const SRV: mio::Token = mio::Token(0);
const CMD: mio::Token = mio::Token(1);
let mut server = Some(
mio::net::TcpListener::from_listener(sock, &addr)
.expect("Can not create mio::net::TcpListener"));
2017-12-27 20:22:27 +01:00
// Create a poll instance
let poll = match mio::Poll::new() {
Ok(poll) => poll,
Err(err) => panic!("Can not create mio::Poll: {}", err),
};
// Start listening for incoming connections
2017-12-27 21:58:32 +01:00
if let Some(ref srv) = server {
if let Err(err) = poll.register(
srv, SRV, mio::Ready::readable(), mio::PollOpt::edge()) {
panic!("Can not register io: {}", err);
}
}
2018-01-15 22:47:25 +01:00
// Start listening for incoming commands
2017-12-27 21:58:32 +01:00
if let Err(err) = poll.register(&reg, CMD,
2017-12-27 20:22:27 +01:00
mio::Ready::readable(), mio::PollOpt::edge()) {
2017-12-27 21:58:32 +01:00
panic!("Can not register Registration: {}", err);
2017-12-27 20:22:27 +01:00
}
// Create storage for events
let mut events = mio::Events::with_capacity(128);
2017-12-27 21:58:32 +01:00
let mut next = 0;
2017-12-13 21:47:07 +01:00
loop {
2017-12-27 20:22:27 +01:00
if let Err(err) = poll.poll(&mut events, None) {
panic!("Poll error: {}", err);
}
for event in events.iter() {
match event.token() {
2017-12-27 21:58:32 +01:00
SRV => {
if let Some(ref server) = server {
loop {
match server.accept_std() {
Ok((sock, addr)) => {
2017-12-28 21:38:37 +01:00
let msg = Conn{
2017-12-27 21:58:32 +01:00
io: sock, peer: Some(addr), http2: false};
workers[next].unbounded_send(msg)
.expect("worker thread died");
next = (next + 1) % workers.len();
},
Err(err) => if err.kind() == io::ErrorKind::WouldBlock {
break
} else {
error!("Error accepting connection: {:?}", err);
return
}
2017-12-27 20:22:27 +01:00
}
}
}
2017-12-27 21:58:32 +01:00
},
CMD => match rx.try_recv() {
Ok(cmd) => match cmd {
Command::Pause => if let Some(server) = server.take() {
if let Err(err) = poll.deregister(&server) {
error!("Can not deregister server socket {}", err);
} else {
info!("Paused accepting connections on {}", addr);
}
},
Command::Resume => {
let lst = create_tcp_listener(addr, backlog)
.expect("Can not create net::TcpListener");
server = Some(
mio::net::TcpListener::from_listener(lst, &addr)
.expect("Can not create mio::net::TcpListener"));
if let Some(ref server) = server {
if let Err(err) = poll.register(
server, SRV, mio::Ready::readable(), mio::PollOpt::edge())
{
error!("Can not resume socket accept process: {}", err);
} else {
info!("Accepting connections on {} has been resumed",
addr);
}
}
},
Command::Stop => return,
},
Err(err) => match err {
sync_mpsc::TryRecvError::Empty => (),
sync_mpsc::TryRecvError::Disconnected => return,
}
2017-12-27 22:26:31 +01:00
},
2017-12-27 20:22:27 +01:00
_ => unreachable!(),
2017-12-13 21:47:07 +01:00
}
}
}
});
2017-12-27 21:58:32 +01:00
(readiness, tx)
}
fn create_tcp_listener(addr: net::SocketAddr, backlog: i32) -> io::Result<net::TcpListener> {
let builder = match addr {
net::SocketAddr::V4(_) => TcpBuilder::new_v4()?,
net::SocketAddr::V6(_) => TcpBuilder::new_v6()?,
};
builder.bind(addr)?;
builder.reuse_address(true)?;
Ok(builder.listen(backlog)?)
2017-12-13 21:47:07 +01:00
}