Speed up JudgeHandler's processing of problem list

Avoid querying unnecessary data
VNOI-Admin · Sep 18, 2023 · 0376c38 · 0376c38
1 parent 2ecdc1b
commit 0376c38
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/judge/bridge/judge_handler.py b/judge/bridge/judge_handler.py
@@ -118,8 +118,8 @@ def _connected(self):
         judge = self.judge = Judge.objects.get(name=self.name)
         judge.start_time = timezone.now()
         judge.online = True
-        judge.problems.set(Problem.objects.filter(code__in=list(self.problems.keys())))
-        judge.runtimes.set(Language.objects.filter(key__in=list(self.executors.keys())))
+        judge.problems.set(Problem.objects.filter(code__in=list(self.problems.keys())).values_list('id', flat=True))
+        judge.runtimes.set(Language.objects.filter(key__in=list(self.executors.keys())).values_list('id', flat=True))
 
         # Cache is_disabled for faster access
         self.is_disabled = judge.is_disabled
@@ -331,7 +331,9 @@ def on_supported_problems(self, packet):
         if not self.working:
             self.judges.update_problems(self)
 
-        self.judge.problems.set(Problem.objects.filter(code__in=list(self.problems.keys())))
+        self.judge.problems.set(
+            Problem.objects.filter(code__in=list(self.problems.keys())).values_list('id', flat=True),
+        )
         json_log.info(self._make_json_log(action='update-problems', count=len(self.problems)))
 
     def on_grading_begin(self, packet):