LRUResourceCache.java

package com.murex.booking.sequence.sweeper.utils;

import java.util.LinkedHashMap;
import java.util.Map;
import java.util.function.Consumer;

public class LRUResourceCache<K, V> extends LinkedHashMap<K, V> {
    private final int capacity;
    private final Consumer<V> closeResource;

    public LRUResourceCache(int capacity, Consumer<V> closeResource) {
        super(capacity, 0.75f, true);
        this.capacity = capacity;
        this.closeResource = closeResource;
    }

    @Override
    protected boolean removeEldestEntry(Map.Entry<K, V> oldest) {
        if (size() > capacity) {
            closeResource.accept(oldest.getValue());
            remove(oldest.getKey());
        }
        return false;
    }
}