use std::{fmt, net}; use std::rc::Rc; use std::sync::Arc; use std::cell::{Cell, RefCell, RefMut, UnsafeCell}; use futures_cpupool::{Builder, CpuPool}; use helpers; use super::KeepAlive; use super::channel::Node; use super::shared::{SharedBytes, SharedBytesPool}; /// Various server settings #[derive(Debug, Clone)] pub struct ServerSettings { addr: Option, secure: bool, host: String, cpu_pool: Arc, } struct InnerCpuPool { cpu_pool: UnsafeCell>, } impl fmt::Debug for InnerCpuPool { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { write!(f, "CpuPool") } } impl InnerCpuPool { fn new() -> Self { InnerCpuPool { cpu_pool: UnsafeCell::new(None), } } fn cpu_pool(&self) -> &CpuPool { unsafe { let val = &mut *self.cpu_pool.get(); if val.is_none() { *val = Some(Builder::new().create()); } val.as_ref().unwrap() } } } unsafe impl Sync for InnerCpuPool {} impl Default for ServerSettings { fn default() -> Self { ServerSettings { addr: None, secure: false, host: "localhost:8080".to_owned(), cpu_pool: Arc::new(InnerCpuPool::new()), } } } impl ServerSettings { /// Crate server settings instance pub(crate) fn new(addr: Option, host: &Option, secure: bool) -> ServerSettings { let host = if let Some(ref host) = *host { host.clone() } else if let Some(ref addr) = addr { format!("{}", addr) } else { "localhost".to_owned() }; let cpu_pool = Arc::new(InnerCpuPool::new()); ServerSettings { addr, secure, host, cpu_pool } } /// Returns the socket address of the local half of this TCP connection pub fn local_addr(&self) -> Option { self.addr } /// Returns true if connection is secure(https) pub fn secure(&self) -> bool { self.secure } /// Returns host header value pub fn host(&self) -> &str { &self.host } /// Returns default `CpuPool` for server pub fn cpu_pool(&self) -> &CpuPool { self.cpu_pool.cpu_pool() } } pub(crate) struct WorkerSettings { h: RefCell>, keep_alive: u64, ka_enabled: bool, bytes: Rc, messages: Rc, channels: Cell, node: Box>, } impl WorkerSettings { pub(crate) fn new(h: Vec, keep_alive: KeepAlive) -> WorkerSettings { let (keep_alive, ka_enabled) = match keep_alive { KeepAlive::Timeout(val) => (val as u64, true), KeepAlive::Os | KeepAlive::Tcp(_) => (0, true), KeepAlive::Disabled => (0, false), }; WorkerSettings { keep_alive, ka_enabled, h: RefCell::new(h), bytes: Rc::new(SharedBytesPool::new()), messages: Rc::new(helpers::SharedMessagePool::new()), channels: Cell::new(0), node: Box::new(Node::head()), } } pub fn num_channels(&self) -> usize { self.channels.get() } pub fn head(&self) -> &Node<()> { &self.node } pub fn handlers(&self) -> RefMut> { self.h.borrow_mut() } pub fn keep_alive(&self) -> u64 { self.keep_alive } pub fn keep_alive_enabled(&self) -> bool { self.ka_enabled } pub fn get_shared_bytes(&self) -> SharedBytes { SharedBytes::new(self.bytes.get_bytes(), Rc::clone(&self.bytes)) } pub fn get_http_message(&self) -> helpers::SharedHttpInnerMessage { helpers::SharedHttpInnerMessage::new(self.messages.get(), Rc::clone(&self.messages)) } pub fn add_channel(&self) { self.channels.set(self.channels.get() + 1); } pub fn remove_channel(&self) { let num = self.channels.get(); if num > 0 { self.channels.set(num-1); } else { error!("Number of removed channels is bigger than added channel. Bug in actix-web"); } } }