fix(judge): use lock in testing

Author: zdszero

judge/tasks.py

@@ -1,4 +1,5 @@
 import dramatiq
+import multiprocessing
 
 from account.models import User, UserProfile
 from submission.models import Submission
@@ -8,6 +9,7 @@
 from .dispatcher import JudgeDispatcher
 from .testing import SubmissionTester
 
+lock = multiprocessing.Lock()
 
 @dramatiq.actor(**DRAMATIQ_WORKER_ARGS())
 def judge_task(submission_id, problem_id):
@@ -19,33 +21,41 @@ def judge_task(submission_id, problem_id):
 
 @dramatiq.actor(**DRAMATIQ_WORKER_ARGS())
 def local_judge_task(submission_id, problem_id, user_id):
+    global lock
     submission = Submission.objects.get(id=submission_id)
 
-    problem = Problem.objects.get(id=problem_id)
-    problem.submission_number += 1
+    with lock:
+        problem = Problem.objects.get(id=problem_id)
+        problem.submission_number += 1
+        problem.save()
+    judge_res = False
     try:
-        if SubmissionTester(submission).judge():
-            problem.accepted_number += 1
+        judge_res = SubmissionTester(submission).judge()
     except Exception as e:
         raise
-    problem.save()
+    with lock:
+        problem = Problem.objects.get(id=problem_id)
+        if judge_res:
+            problem.accepted_number += 1
+        problem.save()
 
     score = submission.grade
     user = User.objects.get(id=user_id)
     assert user
-    profile = UserProfile.objects.get(user=user)
-    assert profile
-    profile.total_submissions += 1
-    if problem._id not in profile.problems_status:
-        profile.problems_status[problem._id] = score
-        profile.total_score += score
-        if score == 100:
-            profile.accepted_number += 1
-    else:
-        prev_score = profile.problems_status[problem._id]
-        if score > prev_score:
+    with lock:
+        profile = UserProfile.objects.get(user=user)
+        assert profile
+        profile.total_submissions += 1
+        if problem._id not in profile.problems_status:
             profile.problems_status[problem._id] = score
-            profile.total_score += (score - prev_score)
+            profile.total_score += score
             if score == 100:
                 profile.accepted_number += 1
-    profile.save()
+        else:
+            prev_score = profile.problems_status[problem._id]
+            if score > prev_score:
+                profile.problems_status[problem._id] = score
+                profile.total_score += (score - prev_score)
+                if score == 100:
+                    profile.accepted_number += 1
+        profile.save()

manage

@@ -42,7 +42,8 @@ function run_onl {
   [ -s "$NVM_DIR/nvm.sh" ] && \. "$NVM_DIR/nvm.sh"  # This loads nvm
   nvm use v8.12.0
   workdir=$(pwd)
-  process_num=$(($(nproc) * 2 - 1))
+  process_num=$(nproc)
+  echo "starting ${process_num} procs"
 
   # run onl
   if ! command pgrep -f "$supervisor_cmd" > /dev/null; then