cache windfiles on startup

ls1intum · Dec 21, 2023 · 94a5643 · 94a5643
1 parent 173846b
commit 94a5643
Showing 1 changed file with 21 additions and 2 deletions.
diff --git a/src/main/java/de/tum/in/www1/artemis/service/connectors/aeolus/AeolusTemplateService.java b/src/main/java/de/tum/in/www1/artemis/service/connectors/aeolus/AeolusTemplateService.java
@@ -33,7 +33,7 @@
 @Profile("aeolus | localci")
 public class AeolusTemplateService {
 
-    private final Logger LOGGER = LoggerFactory.getLogger(AeolusTemplateService.class);
+    private final Logger logger = LoggerFactory.getLogger(AeolusTemplateService.class);
 
     private final ProgrammingLanguageConfiguration programmingLanguageConfiguration;
 
@@ -44,6 +44,25 @@ public class AeolusTemplateService {
     public AeolusTemplateService(ProgrammingLanguageConfiguration programmingLanguageConfiguration, ResourceLoaderService resourceLoaderService) {
         this.programmingLanguageConfiguration = programmingLanguageConfiguration;
         this.resourceLoaderService = resourceLoaderService;
+        // load all scripts into the cache
+        cacheOnBoot();
+    }
+
+    private void cacheOnBoot() {
+        var resources = this.resourceLoaderService.getResources(Path.of("templates", "aeolus"));
+        for (var resource : resources) {
+            try {
+                String filename = resource.getFilename();
+                String directory = resource.getURL().getPath().split("templates/aeolus/")[1].split("/")[0];
+                String uniqueKey = directory + "_" + filename;
+                byte[] fileContent = IOUtils.toByteArray(resource.getInputStream());
+                String script = new String(fileContent, StandardCharsets.UTF_8);
+                templateCache.put(uniqueKey, readWindfile(script));
+            }
+            catch (IOException e) {
+                logger.error("Failed to load script {}", resource.getFilename(), e);
+            }
+        }
     }
 
     /**
@@ -118,7 +137,7 @@ public Windfile getDefaultWindfileFor(ProgrammingExercise exercise) {
                     exercise.hasSequentialTestRuns(), exercise.isTestwiseCoverageEnabled());
         }
         catch (IOException e) {
-            LOGGER.info("No windfile for the settings of exercise {}", exercise.getId(), e);
+            logger.info("No windfile for the settings of exercise {}", exercise.getId(), e);
         }
         return null;
     }