Updated politics worker

src/worker/sql.rs

@@ -1251,7 +1251,7 @@ pub const QUERY_SELECT_ELECTIONS_NEEDING_CANDIDATES: &str = r#"
                     );
 "#;
 
-pub const QUERY_PROCESS_EXPIRED_AND_FILL: &str = r#"
+pub const QUERY_PROCESS_EXPIRED_AND_SCHEDULE: &str = r#"
         WITH
         doomed AS (
                 SELECT po.id,
@@ -1275,144 +1275,45 @@ pub const QUERY_PROCESS_EXPIRED_AND_FILL: &str = r#"
                 RETURNING po.office_type_id AS office_type_id,
                           po.region_id AS region_id
         ),
-        distinct_types AS (
-                SELECT DISTINCT office_type_id, region_id FROM expired_offices
-        ),
-        votes_per_candidate AS (
-                SELECT
-                        dt.office_type_id,
-                        dt.region_id,
-                        c.character_id,
-                        COUNT(v.id) AS vote_count
-                FROM distinct_types AS dt
-                JOIN falukant_data.election AS e
-                    ON e.office_type_id = dt.office_type_id
-                JOIN falukant_data.vote AS v
-                    ON v.election_id = e.id
-                JOIN falukant_data.candidate AS c
-                    ON c.election_id = e.id
-                 AND c.id           = v.candidate_id
-                WHERE e.date >= (NOW() - INTERVAL '30 days')
-                GROUP BY dt.office_type_id, dt.region_id, c.character_id
-        ),
-        ranked_winners AS (
-                SELECT
-                        vpc.office_type_id,
-                        vpc.region_id,
-                        vpc.character_id,
-                        ROW_NUMBER() OVER (
-                                PARTITION BY vpc.office_type_id, vpc.region_id
-                                ORDER BY vpc.vote_count DESC
-                        ) AS rn
-                FROM votes_per_candidate AS vpc
-        ),
-        selected_winners AS (
-                SELECT
-                        rw.office_type_id,
-                        rw.region_id,
-                        rw.character_id
-                FROM ranked_winners AS rw
-                JOIN falukant_type.political_office_type AS pot
-                    ON pot.id = rw.office_type_id
-                WHERE rw.rn <= pot.seats_per_region
-        ),
-        insert_winners AS (
-                INSERT INTO falukant_data.political_office
-                        (office_type_id, character_id, created_at, updated_at, region_id)
-                SELECT
-                        sw.office_type_id,
-                        sw.character_id,
-                        NOW(),
-                        NOW(),
-                        sw.region_id
-                FROM selected_winners AS sw
-                RETURNING id AS new_office_id, office_type_id, character_id, region_id
-        ),
-        count_inserted AS (
+        gaps_per_region AS (
                 SELECT
                         office_type_id,
                         region_id,
-                        COUNT(*) AS inserted_count
-                FROM insert_winners
+                        COUNT(*) AS gaps
+                FROM expired_offices
                 GROUP BY office_type_id, region_id
         ),
-        needed_to_fill AS (
+        to_schedule AS (
                 SELECT
-                        dt.office_type_id,
-                        dt.region_id,
-                        (pot.seats_per_region - COALESCE(ci.inserted_count, 0)) AS gaps
-                FROM distinct_types AS dt
-                JOIN falukant_type.political_office_type AS pot
-                    ON pot.id = dt.office_type_id
-                LEFT JOIN count_inserted AS ci
-                    ON ci.office_type_id = dt.office_type_id
-                 AND ci.region_id      = dt.region_id
-                WHERE (pot.seats_per_region - COALESCE(ci.inserted_count, 0)) > 0
-        ),
-        random_candidates AS (
-                SELECT
-                        rtf.office_type_id,
-                        rtf.region_id,
-                        ch.id AS character_id,
-                        ROW_NUMBER() OVER (
-                                PARTITION BY rtf.office_type_id, rtf.region_id
-                                ORDER BY RANDOM()
-                        ) AS rn
-                FROM needed_to_fill AS rtf
-                JOIN falukant_data.character AS ch
-                    ON ch.region_id IN (
-                        WITH RECURSIVE region_tree AS (
-                            SELECT id FROM falukant_data.region WHERE id = rtf.region_id
-                            UNION ALL
-                            SELECT r2.id FROM falukant_data.region r2
-                            JOIN region_tree rt ON r2.parent_id = rt.id
-                        )
-                        SELECT id FROM region_tree
+                        g.office_type_id,
+                        g.region_id,
+                        g.gaps,
+                        (CURRENT_DATE + INTERVAL '2 days')::date AS election_date
+                FROM gaps_per_region AS g
+                WHERE NOT EXISTS (
+                        SELECT 1
+                            FROM falukant_data.election AS e
+                         WHERE e.office_type_id = g.office_type_id
+                             AND e.region_id      = g.region_id
+                             AND e.date::date    >= (CURRENT_DATE + INTERVAL '2 days')::date
                     )
-                 AND ch.user_id IS NULL
-                 AND ch.birthdate <= NOW() - INTERVAL '21 days'
-                 AND ch.title_of_nobility IN (
-                             SELECT id FROM falukant_type.title WHERE label_tr != 'noncivil'
-                     )
-                 AND NOT EXISTS (
-                             SELECT 1
-                                 FROM falukant_data.political_office AS po2
-                                 JOIN falukant_type.political_office_type AS pot2
-                                     ON pot2.id = po2.office_type_id
-                                WHERE po2.character_id = ch.id
-                                    AND (po2.created_at + (pot2.term_length * INTERVAL '1 day')) >
-                                                NOW() + INTERVAL '2 days'
-                     )
         ),
-        insert_random AS (
-                INSERT INTO falukant_data.political_office
-                        (office_type_id, character_id, created_at, updated_at, region_id)
+        scheduled AS (
+                INSERT INTO falukant_data.election
+                        (office_type_id, date, posts_to_fill, created_at, updated_at, region_id)
                 SELECT
-                        rc.office_type_id,
-                        rc.character_id,
+                        ts.office_type_id,
+                        ts.election_date,
+                        ts.gaps,
                         NOW(),
                         NOW(),
-                        rc.region_id
-                FROM random_candidates AS rc
-                JOIN needed_to_fill AS rtf
-                    ON rtf.office_type_id = rc.office_type_id
-                 AND rtf.region_id      = rc.region_id
-                WHERE rc.rn <= rtf.gaps
-                RETURNING id AS new_office_id, office_type_id, character_id, region_id
+                        ts.region_id
+                FROM to_schedule AS ts
+                RETURNING id
         )
         SELECT
-                new_office_id AS office_id,
-                office_type_id,
-                character_id,
-                region_id
-            FROM insert_winners
-        UNION ALL
-        SELECT
-                new_office_id AS office_id,
-                office_type_id,
-                character_id,
-                region_id
-            FROM insert_random;
+                COUNT(*) AS scheduled_count
+            FROM scheduled;
 "#;
 
 pub const QUERY_USERS_IN_CITIES_OF_REGIONS: &str = r#"
@@ -4229,4 +4130,3 @@ pub const QUERY_GET_FALUKANT_USER_CERT_AND_EVENT: &str = r#"
       FROM falukant_data.falukant_user
      WHERE id = $1::int;
 "#;
-