From d2f261acfe6a3c50f16912b25c70bfc9ab6cadf3 Mon Sep 17 00:00:00 2001
From: cuom1999 <lephuocdinh99@gmail.com>
Date: Tue, 7 Mar 2023 19:59:12 -0600
Subject: [PATCH] Add submission indexes to optimize user_problems

---
 .../migrations/0154_add_submission_indexes.py | 27 +++++++
 judge/models/submission.py                    |  5 ++
 judge/performance_points.py                   | 73 ++++++++-----------
 3 files changed, 64 insertions(+), 41 deletions(-)
 create mode 100644 judge/migrations/0154_add_submission_indexes.py

diff --git a/judge/migrations/0154_add_submission_indexes.py b/judge/migrations/0154_add_submission_indexes.py
new file mode 100644
index 0000000..28c5fec
--- /dev/null
+++ b/judge/migrations/0154_add_submission_indexes.py
@@ -0,0 +1,27 @@
+# Generated by Django 3.2.18 on 2023-03-08 01:55
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("judge", "0153_drop_comment_page"),
+    ]
+
+    operations = [
+        migrations.AddIndex(
+            model_name="submission",
+            index=models.Index(
+                fields=["problem", "user", "-points"],
+                name="judge_submi_problem_5687ea_idx",
+            ),
+        ),
+        migrations.AddIndex(
+            model_name="submission",
+            index=models.Index(
+                fields=["contest_object", "problem", "user", "-points"],
+                name="judge_submi_contest_31cdbb_idx",
+            ),
+        ),
+    ]
diff --git a/judge/models/submission.py b/judge/models/submission.py
index 16b079b..fea9b64 100644
--- a/judge/models/submission.py
+++ b/judge/models/submission.py
@@ -233,6 +233,11 @@ class Submission(models.Model):
         verbose_name = _("submission")
         verbose_name_plural = _("submissions")
 
+        indexes = [
+            models.Index(fields=["problem", "user", "-points"]),
+            models.Index(fields=["contest_object", "problem", "user", "-points"]),
+        ]
+
 
 class SubmissionSource(models.Model):
     submission = models.OneToOneField(
diff --git a/judge/performance_points.py b/judge/performance_points.py
index bf2a320..c806081 100644
--- a/judge/performance_points.py
+++ b/judge/performance_points.py
@@ -21,49 +21,40 @@ PPBreakdown = namedtuple(
 def get_pp_breakdown(user, start=0, end=settings.DMOJ_PP_ENTRIES):
     with connection.cursor() as cursor:
         cursor.execute(
-            """
-            SELECT submission.problem_code,
-                   submission.problem_name,
-                   submission.max_points,
-                   submission.sub_id,
-                   submission.sub_date,
-                   submission.case_points,
-                   submission.case_total,
-                   submission.result,
+            f"""
+            SELECT max_points_table.problem_code,
+                   max_points_table.problem_name,
+                   max_points_table.max_points,
+                   judge_submission.id,
+                   judge_submission.date,
+                   judge_submission.case_points,
+                   judge_submission.case_total,
+                   judge_submission.result,
                    judge_language.short_name,
                    judge_language.key
-            FROM
-              (SELECT max_points_table.problem_code problem_code,
-                     max_points_table.problem_name problem_name,
-                     max_points_table.max_points max_points,
-                     judge_submission.id sub_id,
-                     judge_submission.date sub_date,
-                     judge_submission.case_points case_points,
-                     judge_submission.case_total case_total,
-                     judge_submission.result result,
-                     judge_submission.language_id language_id
-                FROM judge_submission
-                JOIN (
-                      SELECT judge_problem.id problem_id,
-                             judge_problem.name problem_name,
-                             judge_problem.code problem_code,
-                             MAX(judge_submission.points) AS max_points
-                        FROM judge_problem
-                       INNER JOIN judge_submission
-                          ON (judge_problem.id = judge_submission.problem_id)
-                       WHERE (judge_problem.is_public = True AND judge_problem.is_organization_private = False AND judge_submission.points IS NOT NULL AND judge_submission.user_id = %s)
-                       GROUP BY judge_problem.id
-                      HAVING MAX(judge_submission.points) > 0.0
-                     ) AS max_points_table
-                  ON (judge_submission.problem_id = max_points_table.problem_id AND judge_submission.points = max_points_table.max_points AND judge_submission.user_id = %s)
-               GROUP BY max_points_table.problem_id
-               ORDER BY max_points DESC, judge_submission.date DESC
-               LIMIT %s
-               OFFSET %s
-            ) AS submission
-            JOIN judge_language
-              ON submission.language_id = judge_language.id
-            ORDER BY submission.max_points DESC, submission.sub_date DESC;
+            FROM (
+                SELECT judge_problem.id problem_id,
+                       judge_problem.name problem_name,
+                       judge_problem.code problem_code,
+                       MAX(judge_submission.points) AS max_points
+                FROM judge_problem
+                INNER JOIN judge_submission ON (judge_problem.id = judge_submission.problem_id)
+                WHERE (judge_problem.is_public AND
+                       NOT judge_problem.is_organization_private AND
+                       judge_submission.points IS NOT NULL AND
+                       judge_submission.user_id = %s)
+                GROUP BY judge_problem.id
+                HAVING MAX(judge_submission.points) > 0.0
+            ) AS max_points_table
+            INNER JOIN judge_submission ON (
+                judge_submission.problem_id = max_points_table.problem_id AND
+                judge_submission.points = max_points_table.max_points AND
+                judge_submission.user_id = %s
+            )
+            INNER JOIN judge_language ON (judge_submission.language_id = judge_language.id)
+            GROUP BY max_points_table.problem_id
+            ORDER BY max_points DESC, judge_submission.date DESC
+            LIMIT %s OFFSET %s
         """,
             (user.id, user.id, end - start + 1, start),
         )