mirror of https://github.com/mastodon/flodgatt
57 lines
1.7 KiB
Rust
57 lines
1.7 KiB
Rust
|
use crate::{err, maybe_update};
|
||
|
|
use url::Url;
|
||
|
|
|
||
|
|
fn none_if_empty(item: &str) -> Option<String> {
|
||
|
|
if item.is_empty() {
|
||
|
|
None
|
||
|
|
} else {
|
||
|
|
Some(item.to_string())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug)]
|
||
|
|
pub struct RedisConfig {
|
||
|
|
pub user: Option<String>,
|
||
|
|
pub password: Option<String>,
|
||
|
|
pub port: String,
|
||
|
|
pub host: String,
|
||
|
|
pub db: Option<String>,
|
||
|
|
pub namespace: Option<String>,
|
||
|
|
}
|
||
|
|
impl Default for RedisConfig {
|
||
|
|
fn default() -> Self {
|
||
|
|
Self {
|
||
|
|
user: None,
|
||
|
|
password: None,
|
||
|
|
db: None,
|
||
|
|
port: "6379".to_string(),
|
||
|
|
host: "127.0.0.1".to_string(),
|
||
|
|
namespace: None,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
impl RedisConfig {
|
||
|
|
pub fn from_url(url: Url) -> Self {
|
||
|
|
let mut password = url.password().as_ref().map(|str| str.to_string());
|
||
|
|
let mut db = none_if_empty(&url.path()[1..]);
|
||
|
|
for (k, v) in url.query_pairs() {
|
||
|
|
match k.to_string().as_str() {
|
||
|
|
"password" => { password = Some(v.to_string());},
|
||
|
|
"db" => { db = Some(v.to_string())},
|
||
|
|
_ => { err::die_with_msg(format!("Unsupported parameter {} in REDIS_URL.\n Flodgatt supports only `password` and `db` parameters.", k))}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
RedisConfig {
|
||
|
|
user: none_if_empty(url.username()),
|
||
|
|
host: err::unwrap_or_die(url.host_str(), "Missing or invalid host in REDIS_URL"),
|
||
|
|
port: err::unwrap_or_die(url.port(), "Missing or invalid port in REDIS_URL"),
|
||
|
|
namespace: None,
|
||
|
|
password,
|
||
|
|
db,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
maybe_update!(maybe_update_host; host);
|
||
|
|
maybe_update!(maybe_update_port; port);
|
||
|
|
maybe_update!(maybe_update_namespace; Some(namespace));
|
||
|
|
}
|