[FEATURE] Cache images to dont overload the server

nck974 · Jan 18, 2024 · 61bddc1 · 61bddc1
1 parent 340309a
commit 61bddc1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/diogenes/src/main/java/dev/nichoko/diogenes/controller/FilesController.java b/diogenes/src/main/java/dev/nichoko/diogenes/controller/FilesController.java
@@ -1,7 +1,10 @@
 package dev.nichoko.diogenes.controller;
 
+import java.util.concurrent.TimeUnit;
+
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.core.io.Resource;
+import org.springframework.http.CacheControl;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PathVariable;
@@ -25,8 +28,13 @@ public FilesController(FileStorageService fileStorageService) {
     @GetMapping("/images/{imageName}")
     public ResponseEntity<Resource> serveImage(@PathVariable String imageName) {
         Resource resource = fileStorageService.readItemImage(imageName);
+
+        // Set caching headers
+        CacheControl cacheControl = CacheControl.maxAge(30, TimeUnit.DAYS);
+
         return ResponseEntity.ok()
                 .contentType(FileTypeChecker.guessImageContentType(imageName))
+                .cacheControl(cacheControl)
                 .body(resource);
 
     }