cache windfiles on startup

ls1intum · Dec 21, 2023 · 55869e1 · 55869e1
1 parent 173846b
commit 55869e1
Showing 1 changed file with 37 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,41 @@ 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();
+                if (filename == null) {
+                    continue;
+                }
+                String directory = resource.getURL().getPath().split("templates/aeolus/")[1].split("/")[0];
+                String projectType = filename.split("_")[0].replace(".yaml", "");
+                Optional<ProjectType> optionalProjectType = Optional.empty();
+                if (!projectType.equals("default")) {
+                    try {
+                        optionalProjectType = Optional.of(ProjectType.valueOf(projectType.toUpperCase()));
+                    }
+                    catch (IllegalArgumentException e) {
+                        logger.error("Invalid project type {} in file {}", projectType, resource.getFilename());
+                        continue;
+                    }
+                }
+                String uniqueKey = directory + "_" + filename;
+                byte[] fileContent = IOUtils.toByteArray(resource.getInputStream());
+                String script = new String(fileContent, StandardCharsets.UTF_8);
+                Windfile windfile = readWindfile(script);
+                this.addInstanceVariablesToWindfile(windfile, ProgrammingLanguage.valueOf(directory.toUpperCase()), optionalProjectType);
+                templateCache.put(uniqueKey, windfile);
+            }
+            catch (IOException e) {
+                logger.error("Failed to load windfile {}", resource.getFilename(), e);
+            }
+        }
     }
 
     /**
@@ -118,7 +153,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;
     }