server: remove config args, store config path in supervisor
This commit is contained in:
@@ -12,7 +12,7 @@ use crate::{
|
|||||||
DEFAULT_CONFIG_PATH, DEFAULT_SOCKET_PATH, UnixUser, executable_is_suid_or_sgid,
|
DEFAULT_CONFIG_PATH, DEFAULT_SOCKET_PATH, UnixUser, executable_is_suid_or_sgid,
|
||||||
},
|
},
|
||||||
server::{
|
server::{
|
||||||
config::{MysqlConfig, read_config_from_path},
|
config::{MysqlConfig, ServerConfig},
|
||||||
session_handler,
|
session_handler,
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
@@ -244,7 +244,8 @@ fn run_forked_server(
|
|||||||
server_socket: StdUnixStream,
|
server_socket: StdUnixStream,
|
||||||
unix_user: UnixUser,
|
unix_user: UnixUser,
|
||||||
) -> anyhow::Result<()> {
|
) -> anyhow::Result<()> {
|
||||||
let config = read_config_from_path(Some(config_path))?;
|
let config = ServerConfig::read_config_from_path(&config_path)
|
||||||
|
.context("Failed to read server config in forked process")?;
|
||||||
|
|
||||||
let result: anyhow::Result<()> = tokio::runtime::Builder::new_current_thread()
|
let result: anyhow::Result<()> = tokio::runtime::Builder::new_current_thread()
|
||||||
.enable_all()
|
.enable_all()
|
||||||
|
|||||||
@@ -5,23 +5,13 @@ use clap::Parser;
|
|||||||
use clap_verbosity_flag::Verbosity;
|
use clap_verbosity_flag::Verbosity;
|
||||||
use systemd_journal_logger::JournalLog;
|
use systemd_journal_logger::JournalLog;
|
||||||
|
|
||||||
use crate::server::{
|
use crate::{core::common::DEFAULT_CONFIG_PATH, server::supervisor::Supervisor};
|
||||||
config::{ServerConfigArgs, read_config_from_path_with_arg_overrides},
|
|
||||||
supervisor::Supervisor,
|
|
||||||
// server_loop::{
|
|
||||||
// listen_for_incoming_connections_with_socket_path,
|
|
||||||
// listen_for_incoming_connections_with_systemd_socket,
|
|
||||||
// },
|
|
||||||
};
|
|
||||||
|
|
||||||
#[derive(Parser, Debug, Clone)]
|
#[derive(Parser, Debug, Clone)]
|
||||||
pub struct ServerArgs {
|
pub struct ServerArgs {
|
||||||
#[command(subcommand)]
|
#[command(subcommand)]
|
||||||
subcmd: ServerCommand,
|
subcmd: ServerCommand,
|
||||||
|
|
||||||
#[command(flatten)]
|
|
||||||
config_overrides: ServerConfigArgs,
|
|
||||||
|
|
||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
systemd: bool,
|
systemd: bool,
|
||||||
}
|
}
|
||||||
@@ -85,10 +75,15 @@ pub async fn handle_command(
|
|||||||
log::info!("Running in standalone mode");
|
log::info!("Running in standalone mode");
|
||||||
}
|
}
|
||||||
|
|
||||||
let config = read_config_from_path_with_arg_overrides(config_path, args.config_overrides)?;
|
let config_path = config_path.unwrap_or_else(|| PathBuf::from(DEFAULT_CONFIG_PATH));
|
||||||
|
|
||||||
match args.subcmd {
|
match args.subcmd {
|
||||||
ServerCommand::Listen => Supervisor::new(config, systemd_mode).await?.run().await,
|
ServerCommand::Listen => {
|
||||||
|
Supervisor::new(config_path, systemd_mode)
|
||||||
|
.await?
|
||||||
|
.run()
|
||||||
|
.await
|
||||||
|
}
|
||||||
ServerCommand::SocketActivate => {
|
ServerCommand::SocketActivate => {
|
||||||
if !args.systemd {
|
if !args.systemd {
|
||||||
anyhow::bail!(concat!(
|
anyhow::bail!(concat!(
|
||||||
@@ -97,7 +92,10 @@ pub async fn handle_command(
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
Supervisor::new(config, systemd_mode).await?.run().await
|
Supervisor::new(config_path, systemd_mode)
|
||||||
|
.await?
|
||||||
|
.run()
|
||||||
|
.await
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,12 +1,12 @@
|
|||||||
use std::{fs, path::PathBuf};
|
use std::{
|
||||||
|
fs,
|
||||||
|
path::{Path, PathBuf},
|
||||||
|
};
|
||||||
|
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use clap::Parser;
|
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use sqlx::{ConnectOptions, mysql::MySqlConnectOptions};
|
use sqlx::{ConnectOptions, mysql::MySqlConnectOptions};
|
||||||
|
|
||||||
use crate::core::common::DEFAULT_CONFIG_PATH;
|
|
||||||
|
|
||||||
pub const DEFAULT_PORT: u16 = 3306;
|
pub const DEFAULT_PORT: u16 = 3306;
|
||||||
fn default_mysql_port() -> u16 {
|
fn default_mysql_port() -> u16 {
|
||||||
DEFAULT_PORT
|
DEFAULT_PORT
|
||||||
@@ -17,12 +17,6 @@ fn default_mysql_timeout() -> u64 {
|
|||||||
DEFAULT_TIMEOUT
|
DEFAULT_TIMEOUT
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
|
||||||
pub struct ServerConfig {
|
|
||||||
pub socket_path: Option<PathBuf>,
|
|
||||||
pub mysql: MysqlConfig,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||||
#[serde(rename = "mysql")]
|
#[serde(rename = "mysql")]
|
||||||
pub struct MysqlConfig {
|
pub struct MysqlConfig {
|
||||||
@@ -76,94 +70,20 @@ impl MysqlConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Parser, Debug, Clone)]
|
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||||
pub struct ServerConfigArgs {
|
pub struct ServerConfig {
|
||||||
/// Path where the server socket should be created.
|
pub socket_path: Option<PathBuf>,
|
||||||
#[arg(long, value_name = "PATH", global = true)]
|
pub mysql: MysqlConfig,
|
||||||
socket_path: Option<PathBuf>,
|
|
||||||
|
|
||||||
/// Path to the socket of the MySQL server.
|
|
||||||
#[arg(long, value_name = "PATH", global = true)]
|
|
||||||
mysql_socket_path: Option<PathBuf>,
|
|
||||||
|
|
||||||
/// Hostname of the MySQL server.
|
|
||||||
#[arg(
|
|
||||||
long,
|
|
||||||
value_name = "HOST",
|
|
||||||
global = true,
|
|
||||||
conflicts_with = "socket_path"
|
|
||||||
)]
|
|
||||||
mysql_host: Option<String>,
|
|
||||||
|
|
||||||
/// Port of the MySQL server.
|
|
||||||
#[arg(
|
|
||||||
long,
|
|
||||||
value_name = "PORT",
|
|
||||||
global = true,
|
|
||||||
conflicts_with = "socket_path"
|
|
||||||
)]
|
|
||||||
mysql_port: Option<u16>,
|
|
||||||
|
|
||||||
/// Username to use for the MySQL connection.
|
|
||||||
#[arg(long, value_name = "USER", global = true)]
|
|
||||||
mysql_user: Option<String>,
|
|
||||||
|
|
||||||
/// Path to a file containing the MySQL password.
|
|
||||||
#[arg(long, value_name = "PATH", global = true)]
|
|
||||||
mysql_password_file: Option<PathBuf>,
|
|
||||||
|
|
||||||
/// Seconds to wait for the MySQL connection to be established.
|
|
||||||
#[arg(long, value_name = "SECONDS", global = true)]
|
|
||||||
mysql_connect_timeout: Option<u64>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Use the arguments and whichever configuration file which might or might not
|
impl ServerConfig {
|
||||||
/// be found and default values to determine the configuration for the program.
|
/// Reads the server configuration from the specified path, or the default path if none is provided.
|
||||||
pub fn read_config_from_path_with_arg_overrides(
|
pub fn read_config_from_path(config_path: &Path) -> anyhow::Result<Self> {
|
||||||
config_path: Option<PathBuf>,
|
log::debug!("Reading config file at {:?}", config_path);
|
||||||
args: ServerConfigArgs,
|
|
||||||
) -> anyhow::Result<ServerConfig> {
|
|
||||||
let config = read_config_from_path(config_path)?;
|
|
||||||
|
|
||||||
let mysql = config.mysql;
|
fs::read_to_string(config_path)
|
||||||
|
.context(format!("Failed to read config file at {:?}", config_path))
|
||||||
let password = if let Some(path) = &args.mysql_password_file {
|
.and_then(|c| toml::from_str(&c).context("Failed to parse config file"))
|
||||||
Some(
|
.context(format!("Failed to parse config file at {:?}", config_path))
|
||||||
fs::read_to_string(path)
|
}
|
||||||
.context("Failed to read MySQL password file")
|
|
||||||
.map(|s| s.trim().to_owned())?,
|
|
||||||
)
|
|
||||||
} else if let Some(path) = &mysql.password_file {
|
|
||||||
Some(
|
|
||||||
fs::read_to_string(path)
|
|
||||||
.context("Failed to read MySQL password file")
|
|
||||||
.map(|s| s.trim().to_owned())?,
|
|
||||||
)
|
|
||||||
} else {
|
|
||||||
mysql.password.to_owned()
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(ServerConfig {
|
|
||||||
socket_path: args.socket_path.or(config.socket_path),
|
|
||||||
mysql: MysqlConfig {
|
|
||||||
socket_path: args.mysql_socket_path.or(mysql.socket_path),
|
|
||||||
host: args.mysql_host.or(mysql.host),
|
|
||||||
port: args.mysql_port.unwrap_or(mysql.port),
|
|
||||||
username: args.mysql_user.or(mysql.username.to_owned()),
|
|
||||||
password,
|
|
||||||
password_file: args.mysql_password_file.or(mysql.password_file),
|
|
||||||
timeout: args.mysql_connect_timeout.unwrap_or(mysql.timeout),
|
|
||||||
},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn read_config_from_path(config_path: Option<PathBuf>) -> anyhow::Result<ServerConfig> {
|
|
||||||
let config_path = config_path.unwrap_or_else(|| PathBuf::from(DEFAULT_CONFIG_PATH));
|
|
||||||
|
|
||||||
log::debug!("Reading config file at {:?}", &config_path);
|
|
||||||
|
|
||||||
fs::read_to_string(&config_path)
|
|
||||||
.context(format!("Failed to read config file at {:?}", &config_path))
|
|
||||||
.and_then(|c| toml::from_str(&c).context("Failed to parse config file"))
|
|
||||||
.context(format!("Failed to parse config file at {:?}", &config_path))
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -44,6 +44,7 @@ use crate::server::{
|
|||||||
|
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
pub struct Supervisor {
|
pub struct Supervisor {
|
||||||
|
config_path: PathBuf,
|
||||||
config: ServerConfig,
|
config: ServerConfig,
|
||||||
systemd_mode: bool,
|
systemd_mode: bool,
|
||||||
|
|
||||||
@@ -63,13 +64,16 @@ pub struct Supervisor {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl Supervisor {
|
impl Supervisor {
|
||||||
pub async fn new(config: ServerConfig, systemd_mode: bool) -> anyhow::Result<Self> {
|
pub async fn new(config_path: PathBuf, systemd_mode: bool) -> anyhow::Result<Self> {
|
||||||
log::debug!("Starting server supervisor");
|
log::debug!("Starting server supervisor");
|
||||||
log::debug!(
|
log::debug!(
|
||||||
"Running in tokio with {} worker threads",
|
"Running in tokio with {} worker threads",
|
||||||
tokio::runtime::Handle::current().metrics().num_workers()
|
tokio::runtime::Handle::current().metrics().num_workers()
|
||||||
);
|
);
|
||||||
|
|
||||||
|
let config = ServerConfig::read_config_from_path(&config_path)
|
||||||
|
.context("Failed to read server configuration")?;
|
||||||
|
|
||||||
let mut watchdog_duration = None;
|
let mut watchdog_duration = None;
|
||||||
let mut watchdog_micro_seconds = 0;
|
let mut watchdog_micro_seconds = 0;
|
||||||
let watchdog_task =
|
let watchdog_task =
|
||||||
@@ -113,6 +117,7 @@ impl Supervisor {
|
|||||||
// let sigterm_cancel_token = CancellationToken::new();
|
// let sigterm_cancel_token = CancellationToken::new();
|
||||||
|
|
||||||
Ok(Self {
|
Ok(Self {
|
||||||
|
config_path,
|
||||||
config,
|
config,
|
||||||
systemd_mode,
|
systemd_mode,
|
||||||
// sighup_cancel_token,
|
// sighup_cancel_token,
|
||||||
|
|||||||
Reference in New Issue
Block a user