sim: add missing indexes for fast ranking generation

varqox · Nov 1, 2024 · f66103b · f66103b
1 parent 870f914
commit f66103b
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/subprojects/sim/src/sim/db/schema.cc b/subprojects/sim/src/sim/db/schema.cc
@@ -322,6 +322,13 @@ const DbSchema& get_schema() {
                         "  KEY `for_contest_initial_problem_final_by_score_and_full_status` (`initial_final_candidate`,`user_id`,`contest_problem_id`,`score` DESC,`full_status`,`id` DESC),"
                         // Needed to efficiently query contest view coloring
                         "  KEY `initial_final` (`user_id`,`contest_problem_id`,`contest_problem_initial_final`),"
+                        // For ranking
+                        "  KEY `contest_final` (`contest_id`,`contest_problem_final`),"
+                        "  KEY `contest_initial_final` (`contest_id`,`contest_problem_initial_final`),"
+                        "  KEY `contest_round_final` (`contest_round_id`,`contest_problem_final`),"
+                        "  KEY `contest_initial_round_final` (`contest_round_id`,`contest_problem_initial_final`),"
+                        "  KEY `contest_problem_final` (`contest_problem_id`,`contest_problem_final`),"
+                        "  KEY `contest_initial_problem_final` (`contest_problem_id`,`contest_problem_initial_final`),"
                         // For foreign keys
                         "  KEY `file_id` (`file_id`),"
                         "  CONSTRAINT `submissions_ibfk_1` FOREIGN KEY (`file_id`) REFERENCES `internal_files` (`id`) ON UPDATE CASCADE,"

diff --git a/subprojects/sim/src/sim_upgrader.cc b/subprojects/sim/src/sim_upgrader.cc
@@ -40,17 +40,27 @@ run_command(const vector<string>& args, const Spawner::Options& options = {}) {
 
 // Update the below hash and body of the function do_perform_upgrade()
 constexpr StringView NORMALIZED_SCHEMA_HASH_BEFORE_UPGRADE =
-    "bffaa77f52297395cf099de1afdc67b919d37a5e0143a64d6f046583037915d8";
+    "935264658c3a283d37e125526d67102dd03f4596fdfb704d970cd227c48d8bda";
 
 static void do_perform_upgrade(
     [[maybe_unused]] const string& sim_dir, [[maybe_unused]] sim::mysql::Connection& mysql
 ) {
     STACK_UNWINDING_MARK;
 
     // Upgrade here
-    mysql.execute("ALTER TABLE add_problem_jobs DROP COLUMN added_problem_id");
-    mysql.execute("ALTER TABLE reupload_problem_jobs DROP COLUMN problem_id");
-    mysql.execute("ALTER TABLE problem_tags MODIFY is_hidden tinyint(1) NOT NULL AFTER problem_id");
+    mysql.execute(
+        "ALTER TABLE submissions ADD KEY `contest_final` (`contest_id`,`contest_problem_final`)"
+    );
+    mysql.execute("ALTER TABLE submissions ADD KEY `contest_initial_final` (`contest_id`,"
+                  "`contest_problem_initial_final`)");
+    mysql.execute("ALTER TABLE submissions ADD KEY `contest_round_final` (`contest_round_id`,"
+                  "`contest_problem_final`)");
+    mysql.execute("ALTER TABLE submissions ADD KEY `contest_initial_round_final` "
+                  "(`contest_round_id`,`contest_problem_initial_final`)");
+    mysql.execute("ALTER TABLE submissions ADD KEY `contest_problem_final` (`contest_problem_id`,"
+                  "`contest_problem_final`)");
+    mysql.execute("ALTER TABLE submissions ADD KEY `contest_initial_problem_final` "
+                  "(`contest_problem_id`,`contest_problem_initial_final`)");
 }
 
 enum class LockKind {