Add Java

execute.py

@@ -42,6 +42,8 @@ def check_correctness(check_program, timeout, task_id, completion_id, language):
         p = multiprocessing.Process(target=unsafe_execute, args=(check_program, result, timeout))
     elif language == "cpp":
         p = multiprocessing.Process(target=unsafe_execute_cpp, args=(check_program, result, timeout))
+    elif language == "java":
+        p = multiprocessing.Process(target=unsafe_execute_java, args=(check_program, result, timeout))
     elif language == "javascript":
         p = multiprocessing.Process(target=unsafe_execute_js, args=(check_program, result, timeout))
     else:
@@ -135,6 +137,44 @@ def unsafe_execute_cpp(check_program, result, timeout):
             except TimeoutException:
                 result.append("timed out")            
 
+def unsafe_execute_java(check_program, result, timeout):
+
+    with create_tempdir():
+
+        open(f"Main.java", 'w').write(check_program)
+
+        res = "failed: unknown error"
+        compile_returncode = -1
+        for _ in range(5):
+            try:
+                compilation_result = subprocess.run(['javac', "Main.java"], timeout=5,
+                                                    capture_output=True)
+                compile_returncode = compilation_result.returncode
+                break
+            except subprocess.TimeoutExpired as e:
+                continue
+        if compile_returncode != 0:
+            res = "failed: compilation error"
+        else:
+            try:
+                with time_limit(timeout):
+                    exec_result = subprocess.run([f'java', '-cp', tmp_dir, 'Main'], timeout=timeout, capture_output=True)
+                if exec_result.returncode == 0:
+                    res = "passed"
+                elif exec_result.returncode == 1:
+                    if "AssertionError" in exec_result.stderr.decode('unicode-escape'):
+                        res = "failed: wrong answer"
+                    else:
+                        res = f"failed: {exec_result.stderr.decode()}"
+                except subprocess.TimeoutExpired as e:
+                    res = "timed out"
+                except BaseException as e:
+                    res = f"failed: {e}"
+                result.append(res)             
+
+            except TimeoutException:
+                result.append("timed out")
+
 def unsafe_execute_js(check_program, result, timeout):
 
     with create_tempdir():