feat: rewrite scheduled_tasks in rust

lib/libpk/src/_config.rs

@@ -31,7 +31,8 @@ pub struct DatabaseConfig {
     pub(crate) data_db_uri: String,
     pub(crate) data_db_max_connections: Option<u32>,
     pub(crate) data_db_min_connections: Option<u32>,
-    //    pub(crate) _messages_db_uri: String,
+    pub(crate) messages_db_uri: Option<String>,
+    pub(crate) stats_db_uri: Option<String>,
     pub(crate) db_password: Option<String>,
     pub data_redis_addr: String,
 }
@@ -79,6 +80,13 @@ pub struct S3Config {
     pub endpoint: String,
 }
 
+#[derive(Deserialize, Debug)]
+pub struct ScheduledTasksConfig {
+    pub set_guild_count: bool,
+    pub expected_gateway_count: usize,
+    pub gateway_url: String,
+}
+
 fn _metrics_default() -> bool {
     false
 }
@@ -96,6 +104,8 @@ pub struct PKConfig {
     pub api: Option<ApiConfig>,
     #[serde(default)]
     pub avatars: Option<AvatarsConfig>,
+    #[serde(default)]
+    pub scheduled_tasks: Option<ScheduledTasksConfig>,
 
     #[serde(default = "_metrics_default")]
     pub run_metrics_server: bool,

lib/libpk/src/db/mod.rs

@@ -44,3 +44,53 @@ pub async fn init_data_db() -> anyhow::Result<PgPool> {
 
     Ok(pool.connect_with(options).await?)
 }
+
+pub async fn init_messages_db() -> anyhow::Result<PgPool> {
+    info!("connecting to messages database");
+
+    let mut options = PgConnectOptions::from_str(
+        &crate::config
+            .db
+            .messages_db_uri
+            .as_ref()
+            .expect("missing messages db uri"),
+    )?;
+
+    if let Some(password) = crate::config.db.db_password.clone() {
+        options = options.password(&password);
+    }
+
+    let mut pool = PgPoolOptions::new();
+
+    if let Some(max_conns) = crate::config.db.data_db_max_connections {
+        pool = pool.max_connections(max_conns);
+    }
+
+    if let Some(min_conns) = crate::config.db.data_db_min_connections {
+        pool = pool.min_connections(min_conns);
+    }
+
+    Ok(pool.connect_with(options).await?)
+}
+
+pub async fn init_stats_db() -> anyhow::Result<PgPool> {
+    info!("connecting to stats database");
+
+    let mut options = PgConnectOptions::from_str(
+        &crate::config
+            .db
+            .stats_db_uri
+            .as_ref()
+            .expect("missing messages db uri"),
+    )?;
+
+    if let Some(password) = crate::config.db.db_password.clone() {
+        options = options.password(&password);
+    }
+
+    Ok(PgPoolOptions::new()
+        .max_connections(1)
+        .min_connections(1)
+        .connect_with(options)
+        .await?)
+}

lib/libpk/src/db/repository/stats.rs

@@ -3,6 +3,19 @@ pub async fn get_stats(pool: &sqlx::postgres::PgPool) -> anyhow::Result<Counts>
     Ok(counts)
 }
 
+pub async fn insert_stats(
+    pool: &sqlx::postgres::PgPool,
+    table: &str,
+    value: i64,
+) -> anyhow::Result<()> {
+    // danger sql injection
+    sqlx::query(format!("insert into {table} values (now(), $1)").as_str())
+        .bind(value)
+        .execute(pool)
+        .await?;
+    Ok(())
+}
+
 #[derive(serde::Serialize, sqlx::FromRow)]
 pub struct Counts {
     pub system_count: i64,