Update app.py

app.py

@@ -537,7 +537,7 @@ class CodeExecutor:
             
             logger.info("Using existing .class files, skipping compilation")
         
-        # Execute Java with optimized settings
+        # Execute Java with multiple fallback strategies
         results = []
         for main_file in main_files:
             # Determine class name
@@ -562,62 +562,104 @@ class CodeExecutor:
                 )
                 continue
             
-            try:
-                start_time = asyncio.get_event_loop().time()
-                
-                # Optimized JVM settings for container environments
-                java_cmd = [
-                    "java",
-                    "-XX:+UseSerialGC",           # Use simple GC for small heap
-                    "-Xmx64m",                     # Max heap 64MB
-                    "-Xms8m",                      # Initial heap 8MB
-                    "-XX:MaxMetaspaceSize=64m",    # Limit metaspace
-                    "-XX:MaxDirectMemorySize=32m", # Limit direct memory
-                    "-XX:-TieredCompilation",      # Disable tiered compilation
-                    "-XX:CICompilerCount=1",       # Single compiler thread
-                    "-Djava.awt.headless=true",    # Headless mode
-                    "-Dfile.encoding=UTF-8",       # UTF-8 encoding
-                    class_name
-                ]
-                
-                stdout, stderr, returncode = await self._execute_with_input(
-                    java_cmd, workspace, input_data
-                )
-                
-                execution_time = asyncio.get_event_loop().time() - start_time
-                
-                results.append(
-                    ExecutionResult(
+            # Try different JVM configurations
+            jvm_configs = [
+                # Config 1: Minimal memory with explicit small code cache
+                [
+                    "-XX:+UseSerialGC",
+                    "-Xmx32m",
+                    "-Xms4m",
+                    "-XX:ReservedCodeCacheSize=16m",
+                    "-XX:InitialCodeCacheSize=4m",
+                    "-XX:MaxMetaspaceSize=32m",
+                    "-XX:CompressedClassSpaceSize=8m",
+                    "-XX:-TieredCompilation",
+                    "-XX:CICompilerCount=1",
+                    "-Xss256k",
+                    "-XX:+UseStringDeduplication",
+                    "-XX:MaxDirectMemorySize=16m",
+                    "-Djava.awt.headless=true",
+                    "-Dfile.encoding=UTF-8",
+                ],
+                # Config 2: Even more minimal
+                [
+                    "-Xmx16m",
+                    "-Xms2m",
+                    "-XX:ReservedCodeCacheSize=8m",
+                    "-XX:MaxMetaspaceSize=16m",
+                    "-XX:-TieredCompilation",
+                    "-Djava.awt.headless=true",
+                ],
+                # Config 3: Absolute minimum
+                [
+                    "-Xmx8m",
+                    "-XX:ReservedCodeCacheSize=4m",
+                    "-Djava.awt.headless=true",
+                ],
+                # Config 4: Let JVM decide with hints
+                [
+                    "-XX:+UseSerialGC",
+                    "-XX:-TieredCompilation",
+                    "-Djava.awt.headless=true",
+                ],
+            ]
+            
+            result = None
+            for config_idx, jvm_opts in enumerate(jvm_configs):
+                try:
+                    logger.info(f"Trying JVM config {config_idx + 1} for {class_name}")
+                    start_time = asyncio.get_event_loop().time()
+                    
+                    java_cmd = ["java"] + jvm_opts + [class_name]
+                    
+                    stdout, stderr, returncode = await self._execute_with_input(
+                        java_cmd, workspace, input_data
+                    )
+                    
+                    execution_time = asyncio.get_event_loop().time() - start_time
+                    
+                    # Check if it's a memory-related error
+                    stderr_text = stderr.decode("utf-8", errors="replace")
+                    if "Could not reserve" in stderr_text or "insufficient memory" in stderr_text:
+                        logger.warning(f"Config {config_idx + 1} failed with memory error, trying next...")
+                        continue
+                    
+                    result = ExecutionResult(
                         success=returncode == 0,
                         stdout=stdout.decode("utf-8", errors="replace"),
-                        stderr=stderr.decode("utf-8", errors="replace"),
+                        stderr=stderr_text,
                         execution_time=execution_time,
                         exit_code=returncode,
                     )
-                )
-                
-            except asyncio.TimeoutError:
-                results.append(
-                    ExecutionResult(
+                    
+                    logger.info(f"Java execution completed with config {config_idx + 1}")
+                    break
+                    
+                except asyncio.TimeoutError:
+                    result = ExecutionResult(
                         success=False,
                         stdout="",
                         stderr="Execution timeout exceeded",
                         execution_time=MAX_EXECUTION_TIME,
                         exit_code=-1,
                     )
+                    break
+                except Exception as e:
+                    logger.error(f"Error with config {config_idx + 1}: {str(e)}")
+                    continue
+            
+            # If all configs failed, return error
+            if result is None:
+                result = ExecutionResult(
+                    success=False,
+                    stdout="",
+                    stderr="Failed to execute Java code with all memory configurations. The container environment may have insufficient memory for JVM.",
+                    execution_time=0,
+                    exit_code=-1,
+                    error="All JVM configurations failed"
                 )
-            except Exception as e:
-                logger.error(f"Java execution error: {str(e)}")
-                results.append(
-                    ExecutionResult(
-                        success=False,
-                        stdout="",
-                        stderr=str(e),
-                        execution_time=0,
-                        exit_code=-1,
-                        error=str(e),
-                    )
-                )
+            
+            results.append(result)
         
         return self._combine_results(results, main_files)