Enable maximum connections to be configured manually (#182)
* Enable maximum connections to be configured * Add arbiter for handling server startup * Add handling of maxcon for command-line args * Add changelog entrynext
parent
ea1871747d
commit
864c6d461f
@ -0,0 +1,131 @@
|
||||
/*
|
||||
* Created on Sat Jun 26 2021
|
||||
*
|
||||
* This file is a part of Skytable
|
||||
* Skytable (formerly known as TerrabaseDB or Skybase) is a free and open-source
|
||||
* NoSQL database written by Sayan Nandan ("the Author") with the
|
||||
* vision to provide flexibility in data modelling without compromising
|
||||
* on performance, queryability or scalability.
|
||||
*
|
||||
* Copyright (c) 2021, Sayan Nandan <ohsayan@outlook.com>
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*
|
||||
*/
|
||||
|
||||
use crate::config::BGSave;
|
||||
use crate::config::SnapshotConfig;
|
||||
use crate::coredb::CoreDB;
|
||||
use crate::dbnet::{self, Terminator};
|
||||
use crate::diskstore::snapshot::DIR_REMOTE_SNAPSHOT;
|
||||
use crate::services;
|
||||
use crate::PortConfig;
|
||||
use std::fs;
|
||||
use tokio::sync::broadcast;
|
||||
|
||||
#[cfg(unix)]
|
||||
use core::{future::Future, pin::Pin, task::Context, task::Poll};
|
||||
#[cfg(unix)]
|
||||
use tokio::signal::unix::{signal as fnsignal, Signal, SignalKind};
|
||||
#[cfg(unix)]
|
||||
/// Object to bind to unix-specific signals
|
||||
pub struct UnixTerminationSignal {
|
||||
sigterm: Signal,
|
||||
}
|
||||
|
||||
#[cfg(unix)]
|
||||
impl UnixTerminationSignal {
|
||||
pub fn init() -> Result<Self, String> {
|
||||
let sigterm = fnsignal(SignalKind::terminate())
|
||||
.map_err(|e| format!("Failed to bind to signal with: {}", e))?;
|
||||
Ok(Self { sigterm })
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(unix)]
|
||||
impl Future for UnixTerminationSignal {
|
||||
type Output = Option<()>;
|
||||
|
||||
fn poll(mut self: Pin<&mut Self>, ctx: &mut Context<'_>) -> Poll<Self::Output> {
|
||||
self.sigterm.poll_recv(ctx)
|
||||
}
|
||||
}
|
||||
|
||||
/// Start the server waiting for incoming connections or a termsig
|
||||
pub async fn run(
|
||||
ports: PortConfig,
|
||||
bgsave_cfg: BGSave,
|
||||
snapshot_cfg: SnapshotConfig,
|
||||
restore_filepath: Option<String>,
|
||||
maxcon: usize,
|
||||
) -> Result<CoreDB, String> {
|
||||
// Intialize the broadcast channel
|
||||
let (signal, _) = broadcast::channel(1);
|
||||
|
||||
// create the data directories and intialize the coredb
|
||||
fs::create_dir_all(&*DIR_REMOTE_SNAPSHOT)
|
||||
.map_err(|e| format!("Failed to create data directories: '{}'", e))?;
|
||||
let db = CoreDB::new(&snapshot_cfg, restore_filepath)
|
||||
.map_err(|e| format!("Error while initializing database: {}", e))?;
|
||||
|
||||
// initialize the background services
|
||||
let bgsave_handle = tokio::spawn(services::bgsave::bgsave_scheduler(
|
||||
db.clone(),
|
||||
bgsave_cfg,
|
||||
Terminator::new(signal.subscribe()),
|
||||
));
|
||||
let snapshot_handle = tokio::spawn(services::snapshot::snapshot_service(
|
||||
db.clone(),
|
||||
snapshot_cfg,
|
||||
Terminator::new(signal.subscribe()),
|
||||
));
|
||||
|
||||
// bind the ctrlc handler
|
||||
let sig = tokio::signal::ctrl_c();
|
||||
|
||||
// start the server (single or multiple listeners)
|
||||
let mut server = dbnet::connect(ports, maxcon, db.clone(), signal.clone()).await?;
|
||||
|
||||
#[cfg(not(unix))]
|
||||
{
|
||||
// Non-unix, usually Windows specific signal handling.
|
||||
// FIXME(@ohsayan): For now, let's just
|
||||
// bother with ctrl+c, we'll move ahead as users require them
|
||||
tokio::select! {
|
||||
_ = server.run_server() => {}
|
||||
_ = sig => {}
|
||||
}
|
||||
}
|
||||
#[cfg(unix)]
|
||||
{
|
||||
let sigterm = UnixTerminationSignal::init()?;
|
||||
// apart from CTRLC, the only other thing we care about is SIGTERM
|
||||
// FIXME(@ohsayan): Maybe we should respond to SIGHUP too?
|
||||
tokio::select! {
|
||||
_ = server.run_server() => {},
|
||||
_ = sig => {},
|
||||
_ = sigterm => {}
|
||||
}
|
||||
}
|
||||
|
||||
log::info!("Signalling all workers to shut down");
|
||||
// drop the signal and let others exit
|
||||
drop(signal);
|
||||
server.finish_with_termsig().await;
|
||||
|
||||
// wait for the background services to terminate
|
||||
let _ = snapshot_handle.await;
|
||||
let _ = bgsave_handle.await;
|
||||
Ok(db)
|
||||
}
|
Loading…
Reference in New Issue