feat(scheduled_tasks): update for k8s (#734)

crates/libpk/src/_config.rs

@@ -88,6 +88,7 @@ pub struct ScheduledTasksConfig {
     pub set_guild_count: bool,
     pub expected_gateway_count: usize,
     pub gateway_url: String,
+    pub prometheus_url: String,
 }
 
 fn _metrics_default() -> bool {

crates/scheduled_tasks/src/tasks.rs

@@ -92,12 +92,14 @@ pub async fn update_discord_stats(ctx: AppCtx) -> anyhow::Result<()> {
 
     let mut guild_count = 0;
     let mut channel_count = 0;
+    let mut url = cfg.gateway_url.clone();
 
     for idx in 0..cfg.expected_gateway_count {
-        let res = client
+        if url.contains("{clusterid}") {
-            .get(format!("http://cluster{idx}.{}/stats", cfg.gateway_url))
+            url = url.replace("{clusterid}", &idx.to_string());
-            .send()
+        }
-            .await?;
+
+        let res = client.get(&url).send().await?;
 
         let stat: GatewayStatus = res.json().await?;
 
@@ -163,6 +165,11 @@ pub async fn update_stats_api(ctx: AppCtx) -> anyhow::Result<()> {
         .build()
         .expect("error making client");
 
+    let cfg = config
+        .scheduled_tasks
+        .as_ref()
+        .expect("missing scheduled_tasks config");
+
     #[derive(serde::Deserialize, Debug)]
     struct PrometheusResult {
         data: PrometheusResultData,
@@ -178,11 +185,9 @@ pub async fn update_stats_api(ctx: AppCtx) -> anyhow::Result<()> {
 
     macro_rules! prom_instant_query {
         ($t:ty, $q:expr) => {{
+            tracing::info!("Query: {}", $q);
             let resp = client
-                .get(format!(
+                .get(format!("{}/api/v1/query?query={}", cfg.prometheus_url, $q))
-                    "http://vm.svc.pluralkit.net/select/0/prometheus/api/v1/query?query={}",
-                    $q
-                ))
                 .send()
                 .await?;