feat: implement status update breaks

Author: hrideshmg

migrations/20251013082516_implement_status_breaks.sql

@@ -0,0 +1,8 @@
+-- Add migration script here
+CREATE TABLE statusbreaks (
+    id SERIAL PRIMARY KEY,
+    start_date DATE NOT NULL,
+    end_date DATE NOT NULL,
+    year INT NOT NULL,
+    reason TEXT
+);

src/graphql/mutations/status_mutations.rs

@@ -3,7 +3,7 @@ use chrono_tz::Asia::Kolkata;
 use sqlx::PgPool;
 use std::sync::Arc;
 
-use crate::models::status_update::StatusUpdateRecord;
+use crate::models::status_update::{CreateStatusBreakInput, StatusBreakRecord, StatusUpdateRecord};
 
 #[derive(Default)]
 pub struct StatusMutations;
@@ -38,4 +38,26 @@ impl StatusMutations {
 
         Ok(status)
     }
+
+    async fn create_status_break(
+        &self,
+        ctx: &Context<'_>,
+        input: CreateStatusBreakInput,
+    ) -> Result<StatusBreakRecord> {
+        let pool = ctx.data::<Arc<PgPool>>().expect("Pool must be in context");
+        let status = sqlx::query_as::<_, StatusBreakRecord>(
+            "INSERT INTO StatusBreaks (start_date, end_date, year, reason)
+             VALUES ($1, $2, $3, $4)
+             RETURNING *
+            ",
+        )
+        .bind(input.start_date)
+        .bind(input.end_date)
+        .bind(input.year)
+        .bind(&input.reason)
+        .fetch_one(pool.as_ref())
+        .await?;
+
+        Ok(status)
+    }
 }

src/graphql/queries/member_queries.rs

@@ -168,13 +168,20 @@ impl StatusInfo {
     async fn consecutive_misses(&self, ctx: &Context<'_>) -> Result<Option<i64>> {
         let pool = ctx.data::<Arc<PgPool>>().expect("Pool must be in context.");
 
+        // We measure the miss streak by finding the distance to the last sent update.
+        // The inner most subquery here is used for filtering out dates which occur during a break.
         let result: Option<i64> = sqlx::query_scalar(
             "
             SELECT distance
             FROM (
               SELECT is_sent, ROW_NUMBER() OVER (ORDER BY date DESC) - 1 AS distance
-              FROM StatusUpdateHistory
+              FROM StatusUpdateHistory suh
               WHERE member_id = $1
+              AND NOT EXISTS (
+                SELECT * FROM StatusBreaks sb
+                WHERE year = (SELECT year from Member where member_id=$1)
+                AND suh.date BETWEEN sb.start_date AND sb.end_date
+              )
             )
             WHERE is_sent = TRUE
             ORDER BY distance ASC

src/models/status_update.rs

@@ -1,4 +1,4 @@
-use async_graphql::SimpleObject;
+use async_graphql::{InputObject, SimpleObject};
 use chrono::NaiveDate;
 use sqlx::FromRow;
 
@@ -15,3 +15,20 @@ pub struct StatusUpdateStreakRecord {
     pub current_streak: Option<i64>,
     pub max_streak: Option<i64>,
 }
+
+#[derive(SimpleObject, FromRow)]
+pub struct StatusBreakRecord {
+    pub id: i32,
+    pub start_date: NaiveDate,
+    pub end_date: NaiveDate,
+    pub year: i32,
+    pub reason: Option<String>,
+}
+
+#[derive(InputObject)]
+pub struct CreateStatusBreakInput {
+    pub start_date: NaiveDate,
+    pub end_date: NaiveDate,
+    pub year: i32,
+    pub reason: Option<String>,
+}