lkva/NDOJ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add submission indexes to optimize user_problems

Browse source
This commit is contained in:
cuom1999 2023-03-07 19:59:12 -06:00
parent ec2c2ccf13
commit d2f261acfe
3 changed files with 64 additions and 41 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
judge/migrations/0154_add_submission_indexes.py Normal file
View file

@ -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",
),
),
]

5
judge/models/submission.py
View file

@ -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(

53
judge/performance_points.py
View file

@ -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 (
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 = True AND judge_problem.is_organization_private = False AND judge_submission.points IS NOT NULL AND judge_submission.user_id = %s)
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
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_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
) AS submission
JOIN judge_language
ON submission.language_id = judge_language.id
ORDER BY submission.max_points DESC, submission.sub_date DESC;
LIMIT %s OFFSET %s
""",
(user.id, user.id, end - start + 1, start),
)