garage/src/util/config.rs

136 lines
3.1 KiB
Rust
Raw Normal View History

use std::fmt;
2020-04-23 17:05:46 +00:00
use std::io::Read;
use std::net::SocketAddr;
use std::path::PathBuf;
use serde::{de, Deserialize};
2020-04-23 17:05:46 +00:00
use crate::error::Error;
#[derive(Deserialize, Debug, Clone)]
pub struct Config {
pub metadata_dir: PathBuf,
pub data_dir: PathBuf,
#[serde(deserialize_with = "deserialize_addr")]
2020-04-23 17:05:46 +00:00
pub rpc_bind_addr: SocketAddr,
#[serde(deserialize_with = "deserialize_vec_addr")]
2020-04-23 17:05:46 +00:00
pub bootstrap_peers: Vec<SocketAddr>,
pub consul_host: Option<String>,
pub consul_service_name: Option<String>,
2020-04-23 17:05:46 +00:00
#[serde(default = "default_max_concurrent_rpc_requests")]
pub max_concurrent_rpc_requests: usize,
#[serde(default = "default_block_size")]
pub block_size: usize,
#[serde(default = "default_control_write_max_faults")]
pub control_write_max_faults: usize,
2020-04-23 17:05:46 +00:00
#[serde(default = "default_replication_factor")]
pub meta_replication_factor: usize,
#[serde(default = "default_replication_factor")]
pub data_replication_factor: usize,
pub rpc_tls: Option<TlsConfig>,
2020-04-24 17:46:52 +00:00
pub s3_api: ApiConfig,
2020-10-31 16:28:56 +00:00
pub s3_web: WebConfig,
2020-04-23 17:05:46 +00:00
}
2020-04-24 10:10:01 +00:00
#[derive(Deserialize, Debug, Clone)]
pub struct TlsConfig {
pub ca_cert: String,
pub node_cert: String,
pub node_key: String,
}
2020-04-24 17:46:52 +00:00
#[derive(Deserialize, Debug, Clone)]
pub struct ApiConfig {
#[serde(deserialize_with = "deserialize_addr")]
2020-04-24 17:46:52 +00:00
pub api_bind_addr: SocketAddr,
pub s3_region: String,
}
2020-10-31 16:28:56 +00:00
#[derive(Deserialize, Debug, Clone)]
pub struct WebConfig {
#[serde(deserialize_with = "deserialize_addr")]
2020-11-10 08:57:07 +00:00
pub bind_addr: SocketAddr,
pub root_domain: String,
2020-11-11 18:48:01 +00:00
pub index: String,
2020-10-31 16:28:56 +00:00
}
2020-04-23 17:05:46 +00:00
fn default_max_concurrent_rpc_requests() -> usize {
12
}
fn default_block_size() -> usize {
1048576
}
fn default_replication_factor() -> usize {
3
}
fn default_control_write_max_faults() -> usize {
1
2020-04-23 17:05:46 +00:00
}
pub fn read_config(config_file: PathBuf) -> Result<Config, Error> {
let mut file = std::fs::OpenOptions::new()
.read(true)
.open(config_file.as_path())?;
let mut config = String::new();
file.read_to_string(&mut config)?;
Ok(toml::from_str(&config)?)
}
fn deserialize_addr<'de, D>(deserializer: D) -> Result<SocketAddr, D::Error>
where
D: de::Deserializer<'de>,
{
struct AddrVisitor;
impl<'de> de::Visitor<'de> for AddrVisitor {
type Value = SocketAddr;
fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
formatter.write_str("a string representing a socket address")
}
fn visit_str<E>(self, s: &str) -> Result<Self::Value, E>
where
E: de::Error,
{
use std::net::ToSocketAddrs;
s.to_socket_addrs()
.map_err(|_| de::Error::custom("could not resolve to a socket address"))?
.next()
.ok_or(de::Error::custom("could not resolve to a socket address"))
}
}
deserializer.deserialize_any(AddrVisitor)
}
fn deserialize_vec_addr<'de, D>(deserializer: D) -> Result<Vec<SocketAddr>, D::Error>
where
D: de::Deserializer<'de>,
{
use std::net::ToSocketAddrs;
let mut res = vec![];
for s in <Vec<String>>::deserialize(deserializer)? {
res.push(
s.to_socket_addrs()
.map_err(|_| de::Error::custom("could not resolve to a socket address"))?
.next()
.ok_or(de::Error::custom("could not resolve to a socket address"))?,
);
}
Ok(res)
}