fix: remove image not in database

Shikkanime · Sep 12, 2024 · e648e1e · e648e1e
1 parent ded4ff5
commit e648e1e
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 6 deletions.
diff --git a/src/main/kotlin/fr/shikkanime/controllers/admin/AdminController.kt b/src/main/kotlin/fr/shikkanime/controllers/admin/AdminController.kt
@@ -36,12 +36,6 @@ class AdminController {
     @Inject
     private lateinit var animeService: AnimeService
 
-    @Inject
-    private lateinit var episodeMappingService: EpisodeMappingService
-
-    @Inject
-    private lateinit var episodeVariantService: EpisodeVariantService
-
     @Path
     @Get
     private fun home(@QueryParam("error") error: String?): Response {

diff --git a/src/main/kotlin/fr/shikkanime/services/ImageService.kt b/src/main/kotlin/fr/shikkanime/services/ImageService.kt
@@ -305,6 +305,33 @@ object ImageService {
         get() = toHumanReadable(cache.toList().sumOf { it.size })
 
     fun invalidate() {
+        Constant.injector.getInstance(Database::class.java).entityManager.use {
+            val query = it.createNativeQuery(
+                """
+                SELECT uuid
+                FROM anime
+                UNION
+                SELECT uuid
+                FROM episode_mapping
+                UNION
+                SELECT uuid
+                FROM member
+                """,
+                UUID::class.java
+            )
+
+            val uuids = (query.resultList as List<UUID>) // NOSONAR
+                .map { uuid -> uuid.toString() }
+                .toSet()
+
+            // Calculate the difference between the cache and the UUIDs
+            val difference = cache.filter { img -> img.uuid !in uuids }
+            logger.warning("Removing ${difference.size} images from cache, not found in database")
+            logger.warning("${toHumanReadable(difference.sumOf { img -> img.size })} will be freed")
+
+            cache.removeIf { img -> img.uuid !in uuids }
+        }
+
         addAll(true)
     }