fix: improve rejudge

internal/models/result.go

@@ -6,11 +6,12 @@ type Result int32
 
 const (
     NotTested               Result = 0
-    Accepted                Result = 1
+    Testing                 Result = 1
-    TimeLimitExceeded       Result = 2
+    Accepted                Result = 2
-    MemoryLimitExceeded     Result = 3
+    TimeLimitExceeded       Result = 3
-    CompilationError        Result = 4
+    MemoryLimitExceeded     Result = 4
-    SystemFailDuringTesting Result = 5
+    CompilationError        Result = 5
+    SystemFailDuringTesting Result = 6
 )
 
 func (result Result) Valid() error {

internal/storage/solution.go

@@ -70,14 +70,18 @@ func (storage *SolutionStorage) ReadSolutionById(ctx context.Context, id int32)
 }
 
 func (storage *SolutionStorage) RejudgeSolution(ctx context.Context, id int32) error {
-	query := storage.db.Rebind("UPDATE solutions SET result = ? WHERE id = ?")
+    tx,err := storage.db.Begin()
-	_, err := storage.db.ExecContext(ctx, query, models.NotTested, id)
-	query := storage.db.Rebind("UPDATE testruns SET result = ? WHERE solution_id = ?")
-	_, err := storage.db.ExecContext(ctx, query, models.NotTested, id)
-    //FIXME: update result
 	if err != nil {
 		return storage.HandlePgErr(err)
 	}
+    tx.Exec("UPDATE solutions SET result = ? WHERE id = ?",models.NotTested, id)
+    tx.Exec("UPDATE testgroupruns SET result = ? WHERE solution_id = ?",models.NotTested, id)
+    tx.Exec("UPDATE testruns SET result = ? WHERE testgrouprun_id IN (SELECT id FROM tesgrouprun WHERE solution_id = ?)", models.NotTested, id)
+    err = tx.Commit()
+	if err != nil {
+		return storage.HandlePgErr(err)
+	}
+    //FIXME: update result
 
 	return nil
 }