add support for using domain name in configuration #46

Merged
lx merged 4 commits from trinity-1686a/garage:bind-domain into master 2021-03-18 20:26:35 +00:00
Showing only changes of commit ae3b7029a9 - Show all commits

View file

@ -1,8 +1,9 @@
use std::fmt;
use std::io::Read; use std::io::Read;
use std::net::SocketAddr; use std::net::SocketAddr;
use std::path::PathBuf; use std::path::PathBuf;
use serde::Deserialize; use serde::{de, Deserialize};
use crate::error::Error; use crate::error::Error;
@ -11,8 +12,10 @@ pub struct Config {
pub metadata_dir: PathBuf, pub metadata_dir: PathBuf,
pub data_dir: PathBuf, pub data_dir: PathBuf,
#[serde(deserialize_with = "deserialize_addr")]
pub rpc_bind_addr: SocketAddr, pub rpc_bind_addr: SocketAddr,
#[serde(deserialize_with = "deserialize_vec_addr")]
pub bootstrap_peers: Vec<SocketAddr>, pub bootstrap_peers: Vec<SocketAddr>,
pub consul_host: Option<String>, pub consul_host: Option<String>,
pub consul_service_name: Option<String>, pub consul_service_name: Option<String>,
@ -48,12 +51,14 @@ pub struct TlsConfig {
#[derive(Deserialize, Debug, Clone)] #[derive(Deserialize, Debug, Clone)]
pub struct ApiConfig { pub struct ApiConfig {
#[serde(deserialize_with = "deserialize_addr")]
pub api_bind_addr: SocketAddr, pub api_bind_addr: SocketAddr,
pub s3_region: String, pub s3_region: String,
} }
#[derive(Deserialize, Debug, Clone)] #[derive(Deserialize, Debug, Clone)]
pub struct WebConfig { pub struct WebConfig {
#[serde(deserialize_with = "deserialize_addr")]
pub bind_addr: SocketAddr, pub bind_addr: SocketAddr,
pub root_domain: String, pub root_domain: String,
pub index: String, pub index: String,
@ -82,3 +87,49 @@ pub fn read_config(config_file: PathBuf) -> Result<Config, Error> {
Ok(toml::from_str(&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)
}