use futures::prelude::*;
use log::{error, info};
use std::net::ToSocketAddrs;
use std::sync::Arc;
use tokio::net::TcpListener;
use tokio_rustls::{rustls, TlsAcceptor};
use super::{bind_server, new_runtime, tcp_listener};
use super::handler::NewHandler;
pub mod test;
pub fn start<NH, A>(addr: A, new_handler: NH, tls_config: rustls::ServerConfig)
where
NH: NewHandler + 'static,
A: ToSocketAddrs + 'static + Send,
{
start_with_num_threads(addr, new_handler, tls_config, num_cpus::get())
}
pub fn start_with_num_threads<NH, A>(
addr: A,
new_handler: NH,
tls_config: rustls::ServerConfig,
threads: usize,
) where
NH: NewHandler + 'static,
A: ToSocketAddrs + 'static + Send,
{
let runtime = new_runtime(threads);
let _ = runtime.block_on(async { init_server(addr, new_handler, tls_config).await });
}
pub async fn init_server<NH, A>(
addr: A,
new_handler: NH,
tls_config: rustls::ServerConfig,
) -> Result<(), ()>
where
NH: NewHandler + 'static,
A: ToSocketAddrs + 'static + Send,
{
let listener = tcp_listener(addr).map_err(|_| ()).await?;
let addr = listener.local_addr().unwrap();
info!(
target: "gotham::start",
" Gotham listening on http://{}",
addr
);
bind_server_rustls(listener, new_handler, tls_config)
.map_err(|_| ())
.await
}
async fn bind_server_rustls<NH>(
listener: TcpListener,
new_handler: NH,
tls_config: rustls::ServerConfig,
) -> Result<(), ()>
where
NH: NewHandler + 'static,
{
let tls = TlsAcceptor::from(Arc::new(tls_config));
bind_server(listener, new_handler, move |socket| {
tls.accept(socket).map_err(|e| {
error!(target: "gotham::tls", "TLS handshake error: {:?}", e);
})
})
.await
}