Update app.py

app.py

@@ -447,32 +447,49 @@ class CodeExecutor:
 
         return self._combine_results(results, main_files)
 
+# Thay thế method _execute_java trong CodeExecutor class
+
     async def _execute_java(
         self,
         main_files: List[str],
         workspace: str,
         input_data: Optional[List[str]] = None,
     ) -> ExecutionResult:
-        """Compile and execute Java code with input support"""
-
+        """Compile and execute Java code with input support and memory optimization"""
+    
         # Check if we have .java files to compile
         java_files = list(Path(workspace).glob("*.java"))
         needs_compilation = len(java_files) > 0
-
+    
+        # JVM options for container environment
+        jvm_opts = [
+            "-Xms16m",           # Initial heap size
+            "-Xmx128m",          # Maximum heap size
+            "-XX:ReservedCodeCacheSize=16m",  # Code cache size
+            "-XX:InitialCodeCacheSize=2m",    # Initial code cache
+            "-XX:+UseSerialGC",              # Simple GC for small apps
+            "-XX:+TieredCompilation",        # Enable tiered compilation
+            "-XX:TieredStopAtLevel=1",       # Stop at level 1 (faster startup)
+            "-XX:+UseStringDeduplication",   # Save memory on strings
+            "-XX:MaxRAMPercentage=10.0",     # Limit RAM usage to 10% of container
+        ]
+    
         # If we have .java files, compile them
         if needs_compilation:
             logger.info(f"Found {len(java_files)} Java source files, compiling...")
-
+    
+            # Compile with memory-optimized javac
+            compile_cmd = ["javac", "-J-Xmx64m"] + [str(f) for f in java_files]
+            
             compile_process = await asyncio.create_subprocess_exec(
-                "javac",
-                *[str(f) for f in java_files],
+                *compile_cmd,
                 cwd=workspace,
                 stdout=asyncio.subprocess.PIPE,
                 stderr=asyncio.subprocess.PIPE,
             )
-
+    
             stdout, stderr = await compile_process.communicate()
-
+    
             if compile_process.returncode != 0:
                 return ExecutionResult(
                     success=False,
@@ -481,7 +498,7 @@ class CodeExecutor:
                     execution_time=0,
                     exit_code=compile_process.returncode,
                 )
-
+    
             logger.info("Java compilation successful")
         else:
             # Check if we have .class files for the main files
@@ -491,10 +508,10 @@ class CodeExecutor:
                     class_file_path = os.path.join(workspace, main_file)
                 else:
                     class_file_path = os.path.join(workspace, f"{main_file}.class")
-
+    
                 if not os.path.exists(class_file_path):
                     class_files_missing.append(main_file)
-
+    
             if class_files_missing:
                 return ExecutionResult(
                     success=False,
@@ -503,9 +520,9 @@ class CodeExecutor:
                     execution_time=0,
                     exit_code=-1,
                 )
-
+    
             logger.info("Using existing .class files, skipping compilation")
-
+    
         # Execute main files
         results = []
         for main_file in main_files:
@@ -516,7 +533,7 @@ class CodeExecutor:
                 class_name = main_file.replace(".java", "")
             else:
                 class_name = main_file
-
+    
             # Verify the .class file exists
             class_file_path = os.path.join(workspace, f"{class_name}.class")
             if not os.path.exists(class_file_path):
@@ -530,16 +547,19 @@ class CodeExecutor:
                     )
                 )
                 continue
-
+    
             try:
                 start_time = asyncio.get_event_loop().time()
-
+    
+                # Build java command with memory optimization
+                java_cmd = ["java"] + jvm_opts + [class_name]
+                
                 stdout, stderr, returncode = await self._execute_with_input(
-                    ["java", class_name], workspace, input_data
+                    java_cmd, workspace, input_data
                 )
-
+    
                 execution_time = asyncio.get_event_loop().time() - start_time
-
+    
                 results.append(
                     ExecutionResult(
                         success=returncode == 0,
@@ -549,7 +569,7 @@ class CodeExecutor:
                         exit_code=returncode,
                     )
                 )
-
+    
             except asyncio.TimeoutError:
                 results.append(
                     ExecutionResult(
@@ -571,7 +591,7 @@ class CodeExecutor:
                         error=str(e),
                     )
                 )
-
+    
         return self._combine_results(results, main_files)
 
     async def _execute_c_cpp(