Added ram and cpus configuration

Author: Alathreon

JShellAPI/src/main/java/org/togetherjava/jshellapi/Config.java

@@ -8,13 +8,17 @@ public record Config(
         long oneTimeSessionTimeoutSeconds,
         long evalTimeoutSeconds,
         long maxAliveSessions,
+        int dockerMaxRamMegaBytes,
+        double dockerCPUsUsage,
         int schedulerThreadCount,
         long schedulerSessionKillScanRate) {
     public Config {
         if(regularSessionTimeoutSeconds <= 0) throw new RuntimeException("Invalid value " + regularSessionTimeoutSeconds);
         if(oneTimeSessionTimeoutSeconds <= 0) throw new RuntimeException("Invalid value " + oneTimeSessionTimeoutSeconds);
         if(evalTimeoutSeconds <= 0) throw new RuntimeException("Invalid value " + evalTimeoutSeconds);
         if(maxAliveSessions <= 0) throw new RuntimeException("Invalid value " + maxAliveSessions);
+        if(dockerMaxRamMegaBytes <= 0) throw new RuntimeException("Invalid value " + dockerMaxRamMegaBytes);
+        if(dockerCPUsUsage <= 0) throw new RuntimeException("Invalid value " + dockerCPUsUsage);
         if(schedulerThreadCount <= 0) throw new RuntimeException("Invalid value " + schedulerThreadCount);
         if(schedulerSessionKillScanRate <= 0) throw new RuntimeException("Invalid value " + schedulerSessionKillScanRate);
     }

JShellAPI/src/main/java/org/togetherjava/jshellapi/service/JShellService.java

@@ -29,7 +29,7 @@ public class JShellService implements Closeable {
     private final boolean renewable;
     private boolean doingOperation;
 
-    public JShellService(JShellSessionService sessionService, String id, long timeout, boolean renewable, long evalTimeout) throws DockerException {
+    public JShellService(JShellSessionService sessionService, String id, long timeout, boolean renewable, long evalTimeout, int maxMemory, double cpus) throws DockerException {
         this.sessionService = sessionService;
         this.id = id;
         this.timeout = timeout;
@@ -52,6 +52,8 @@ public JShellService(JShellSessionService sessionService, String id, long timeou
                     "--pids-limit=2000",
                     "--memory=500M",
                     "--read-only",
+                    "--memory=" + maxMemory + "m",
+                    "--cpus=" + cpus,
                     "--name", containerName(),
                     "jshellwrapper",
                     "java", "-DevalTimeoutSeconds=%d".formatted(evalTimeout), "-jar", "JShellWrapper.jar")

JShellAPI/src/main/java/org/togetherjava/jshellapi/service/JShellSessionService.java

@@ -68,7 +68,7 @@ private synchronized JShellService createSession(String id, long sessionTimeout,
         if(jshellSessions.size() >= config.maxAliveSessions()) {
             throw new ResponseStatusException(HttpStatus.TOO_MANY_REQUESTS, "Too many sessions, try again later :(.");
         }
-        JShellService service = new JShellService(this, id, sessionTimeout, renewable, evalTimeout);
+        JShellService service = new JShellService(this, id, sessionTimeout, renewable, evalTimeout, config.dockerMaxRamMegaBytes(), config.dockerCPUsUsage());
         jshellSessions.put(id, service);
         return service;
     }

JShellAPI/src/main/resources/application.properties

@@ -5,6 +5,10 @@ jshellapi.oneTimeSessionTimeoutSeconds=30
 jshellapi.evalTimeoutSeconds=15
 jshellapi.maxAliveSessions=10
 
+# Docker limits config
+jshellapi.dockerMaxRamMegaBytes=100
+jshellapi.dockerCPUsUsage=0.5
+
 # Internal config
 jshellapi.schedulerThreadCount=1
 jshellapi.schedulerSessionKillScanRate=60