Added notification for REMOVED case in ehcache event listener

peteralfonsi · Dec 1, 2023 · c6a93a4 · c6a93a4
1 parent 01bcad7
commit c6a93a4
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/server/src/main/java/org/opensearch/common/cache/tier/EhCacheDiskCachingTier.java b/server/src/main/java/org/opensearch/common/cache/tier/EhCacheDiskCachingTier.java
@@ -320,6 +320,16 @@ public void onEvent(CacheEvent<? extends K, ? extends byte[]> event) {
                     assert event.getNewValue() == null;
                     break;
                 case REMOVED:
+                    this.removalListener.ifPresent(
+                        listener -> listener.onRemoval(
+                            new RemovalNotification<>(
+                                event.getKey(),
+                                valueSerializer.deserialize(event.getOldValue()),
+                                RemovalReason.INVALIDATED,
+                                TierType.DISK
+                            )
+                        )
+                    );
                     count.dec();
                     assert event.getNewValue() == null;
                     break;