Merge pull request #1337 from hotate29/create-index

Create indexes on the ranking tables

Author: kenkoooo

atcoder-problems-backend/sql-client/src/internal/virtual_contest_manager.rs

@@ -157,7 +157,7 @@ impl VirtualContestManager for PgPool {
     async fn get_own_contests(&self, internal_user_id: &str) -> Result<Vec<VirtualContestInfo>> {
         let contests = sqlx::query_as(
             r"
-            SELECT 
+            SELECT
                 id,
                 title,
                 memo,
@@ -184,7 +184,7 @@ impl VirtualContestManager for PgPool {
     ) -> Result<Vec<VirtualContestInfo>> {
         let contests = sqlx::query_as(
             r"
-            SELECT 
+            SELECT
                 a.id,
                 a.title,
                 a.memo,
@@ -234,7 +234,7 @@ impl VirtualContestManager for PgPool {
     async fn get_single_contest_participants(&self, contest_id: &str) -> Result<Vec<String>> {
         let participants = sqlx::query(
             r"
-            SELECT 
+            SELECT
                 b.atcoder_user_id
             FROM (
                 SELECT internal_user_id
@@ -280,7 +280,7 @@ impl VirtualContestManager for PgPool {
     async fn get_recent_contest_info(&self) -> Result<Vec<VirtualContestInfo>> {
         let contests = sqlx::query_as(
             r"
-            SELECT 
+            SELECT
                 id,
                 title,
                 memo,

atcoder-problems-backend/sql-client/src/language_count.rs

@@ -113,7 +113,7 @@ impl LanguageCountClient for PgPool {
     async fn load_language_count(&self) -> Result<Vec<UserLanguageCount>> {
         let count = sqlx::query_as(
             r"
-            SELECT 
+            SELECT
                 user_id,
                 simplified_language,
                 problem_count
@@ -150,7 +150,7 @@ impl LanguageCountClient for PgPool {
         let count = sqlx::query_as(
             r"
             SELECT user_id, simplified_language, problem_count FROM language_count
-            WHERE user_id = $1
+            WHERE LOWER(user_id) = LOWER($1)
             ORDER BY simplified_language
             ",
         )
@@ -171,7 +171,7 @@ impl LanguageCountClient for PgPool {
                 OVER(PARTITION BY simplified_language ORDER BY problem_count DESC) AS rank
                 FROM language_count
             )
-            AS s2 WHERE user_id = $1
+            AS s2 WHERE LOWER(user_id) = LOWER($1)
             ORDER BY simplified_language
             ",
         )

atcoder-problems-backend/sql-client/src/rated_point_sum.rs

@@ -27,7 +27,7 @@ impl RatedPointSumClient for PgPool {
                 GROUP BY contest_id
             ) AS contest_problem_count
             JOIN contests ON contests.id=contest_problem_count.contest_id
-            WHERE 
+            WHERE
                 contests.start_epoch_second >= $1
                 AND contests.rate_change != $2
                 AND contest_problem_count.problem_count >= 2

atcoder-problems-backend/sql-client/src/simple_client.rs

@@ -114,7 +114,7 @@ impl SimpleClient for PgPool {
     async fn load_contests(&self) -> Result<Vec<Contest>> {
         let contests = sqlx::query_as(
             r"
-                 SELECT 
+                 SELECT
                     id,
                     start_epoch_second,
                     duration_second,

atcoder-problems-backend/sql-client/src/submission_client.rs

@@ -121,7 +121,7 @@ impl SubmissionClient for PgPool {
                 r"
                     SELECT * FROM submissions
                     WHERE result = 'AC'
-                    AND user_id = ANY($1)
+                    AND LOWER(user_id) = ANY($1)
                     ",
             )
             .bind(user_ids)
@@ -136,11 +136,11 @@ impl SubmissionClient for PgPool {
             SubmissionRequest::InvalidResult { from_second } => sqlx::query_as(
                 r"
                     SELECT * FROM submissions
-                    WHERE 
+                    WHERE
                         result != ALL(
                             ARRAY['AC', 'WA', 'TLE', 'CE', 'RE', 'MLE', 'OLE', 'QLE', 'IE', 'NG']
                         )
-                    AND 
+                    AND
                         epoch_second >= $1
                     ORDER BY id DESC
                     ",
@@ -163,7 +163,7 @@ impl SubmissionClient for PgPool {
             } => sqlx::query_as(
                 r"
                     SELECT * FROM submissions
-                    WHERE user_id = ANY($1)
+                    WHERE LOWER(user_id) = ANY($1)
                     AND problem_id = ANY($2)
                     AND epoch_second >= $3
                     AND epoch_second <= $4

config/database-definition.sql

@@ -16,8 +16,6 @@ CREATE TABLE submissions (
   execution_time  INT,
   PRIMARY KEY (id)
 );
-CREATE INDEX ON submissions (user_id);
-CREATE INDEX ON submissions (LOWER(user_id));
 CREATE INDEX ON submissions (epoch_second);
 CREATE INDEX ON submissions (user_id, epoch_second ASC);
 CREATE INDEX ON submissions (LOWER(user_id), epoch_second ASC);
@@ -79,6 +77,8 @@ CREATE TABLE accepted_count (
   problem_count INT           NOT NULL,
   PRIMARY KEY (user_id)
 );
+CREATE INDEX ON accepted_count (LOWER(user_id));
+CREATE INDEX ON accepted_count (problem_count DESC, user_id);
 
 DROP TABLE IF EXISTS points;
 CREATE TABLE points (
@@ -94,6 +94,8 @@ CREATE TABLE rated_point_sum (
   point_sum       BIGINT NOT NULL,
   PRIMARY KEY (user_id)
 );
+CREATE INDEX ON rated_point_sum (LOWER(user_id));
+CREATE INDEX ON rated_point_sum (point_sum DESC, user_id);
 
 DROP TABLE IF EXISTS language_count;
 CREATE TABLE language_count (
@@ -102,6 +104,8 @@ CREATE TABLE language_count (
   problem_count         INT NOT NULL,
   PRIMARY KEY (user_id, simplified_language)
 );
+CREATE INDEX ON language_count (LOWER(user_id));
+CREATE INDEX ON language_count (simplified_language, problem_count DESC, user_id);
 
 DROP TABLE IF EXISTS predicted_rating;
 CREATE TABLE predicted_rating (
@@ -124,6 +128,8 @@ CREATE TABLE max_streaks (
   streak                BIGINT NOT NULL,
   PRIMARY KEY (user_id)
 );
+CREATE INDEX ON max_streaks (LOWER(user_id));
+CREATE INDEX ON max_streaks (streak DESC, user_id);
 
 -- For internal services:
 DROP TABLE IF EXISTS internal_problem_list_items;
@@ -186,7 +192,7 @@ CREATE INDEX ON internal_virtual_contest_items (internal_virtual_contest_id);
 CREATE TABLE internal_virtual_contest_participants (
   internal_virtual_contest_id VARCHAR(255) REFERENCES internal_virtual_contests(id) ON DELETE CASCADE ON UPDATE CASCADE,
   internal_user_id      VARCHAR(255) REFERENCES internal_users ON DELETE CASCADE ON UPDATE CASCADE,
-  PRIMARY KEY (internal_virtual_contest_id, internal_user_id)  
+  PRIMARY KEY (internal_virtual_contest_id, internal_user_id)
 );
 CREATE INDEX ON internal_virtual_contest_participants (internal_user_id);