feat(gateway): get node id from kubernetes

This commit is contained in:
alyssa 2025-04-20 19:25:11 +00:00
parent bfa0071f90
commit 44c5a2d106
2 changed files with 12 additions and 0 deletions

View file

@ -138,10 +138,15 @@ impl PKConfig {
lazy_static! {
#[derive(Debug)]
pub static ref CONFIG: Arc<PKConfig> = {
// hacks
if let Ok(var) = std::env::var("NOMAD_ALLOC_INDEX")
&& std::env::var("pluralkit__discord__cluster__total_nodes").is_ok() {
std::env::set_var("pluralkit__discord__cluster__node_id", var);
}
if let Ok(var) = std::env::var("STATEFULSET_NAME_FOR_INDEX")
&& std::env::var("pluralkit__discord__cluster__total_nodes").is_ok() {
std::env::set_var("pluralkit__discord__cluster__node_id", var.split("-").last().unwrap());
}
Arc::new(Config::builder()
.add_source(config::Environment::with_prefix("pluralkit").separator("__"))