package io.quarkus.cache.runtime.caffeine;

import static io.quarkus.cache.runtime.NullValueConverter.fromCacheValue;
import static io.quarkus.cache.runtime.NullValueConverter.toCacheValue;

import java.time.Duration;
import java.util.concurrent.Callable;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import java.util.function.Supplier;

import com.github.benmanes.caffeine.cache.AsyncCache;
import com.github.benmanes.caffeine.cache.Caffeine;

public class CaffeineCache {

    private AsyncCache<Object, Object> cache;

    private String name;

    private Integer initialCapacity;

    private Long maximumSize;

    private Duration expireAfterWrite;

    private Duration expireAfterAccess;

    public CaffeineCache(CaffeineCacheInfo cacheInfo) {
        this.name = cacheInfo.name;
        Caffeine<Object, Object> builder = Caffeine.newBuilder();
        /*
         * The following line is a workaround for a GraalVM issue: https://github.com/oracle/graal/issues/1524
         * TODO: Remove it as soon as a Quarkus release depends on GraalVM 19.3.0 or higher.
         */
        builder.executor(task -> ForkJoinPool.commonPool().execute(task));
        if (cacheInfo.initialCapacity != null) {
            this.initialCapacity = cacheInfo.initialCapacity;
            builder.initialCapacity(cacheInfo.initialCapacity);
        }
        if (cacheInfo.maximumSize != null) {
            this.maximumSize = cacheInfo.maximumSize;
            builder.maximumSize(cacheInfo.maximumSize);
        }
        if (cacheInfo.expireAfterWrite != null) {
            this.expireAfterWrite = cacheInfo.expireAfterWrite;
            builder.expireAfterWrite(cacheInfo.expireAfterWrite);
        }
        if (cacheInfo.expireAfterAccess != null) {
            this.expireAfterAccess = cacheInfo.expireAfterAccess;
            builder.expireAfterAccess(cacheInfo.expireAfterAccess);
        }
        cache = builder.buildAsync();
    }

    public Object get(Object key, Callable<Object> valueLoader, long lockTimeout) throws Exception {
        if (lockTimeout <= 0) {
            return fromCacheValue(cache.synchronous().get(key, k -> new MappingSupplier(valueLoader).get()));
        }

        // The lock timeout logic starts here.

        /*
         * If the current key is not already associated with a value in the Caffeine cache, there's no way to know if the
         * current thread or another one started the missing value computation. The following variable will be used to
         * determine whether or not a timeout should be triggered during the computation depending on which thread started it.
         */
        boolean[] isCurrentThreadComputation = { false };

        CompletableFuture<Object> future = cache.get(key, (k, executor) -> {
            isCurrentThreadComputation[0] = true;
            return CompletableFuture.supplyAsync(new MappingSupplier(valueLoader), executor);
        });

        if (isCurrentThreadComputation[0]) {
            // The value is missing and its computation was started from the current thread.
            // We'll wait for the result no matter how long it takes.
            return fromCacheValue(future.get());
        } else {
            // The value is either already present in the cache or missing and its computation was started from another thread.
            // We want to retrieve it from the cache within the lock timeout delay.
            try {
                return fromCacheValue(future.get(lockTimeout, TimeUnit.MILLISECONDS));
            } catch (TimeoutException e) {
                // Timeout triggered! We don't want to wait any longer for the value computation and we'll simply invoke the
                // cached method and return its result without caching it.
                // TODO: Add statistics here to monitor the timeout.
                return valueLoader.call();
            }
        }
    }

    public void invalidate(Object key) {
        cache.synchronous().invalidate(key);
    }

    public void invalidateAll() {
        cache.synchronous().invalidateAll();
    }

    public String getName() {
        return name;
    }

    // For testing purposes only.
    public Integer getInitialCapacity() {
        return initialCapacity;
    }

    // For testing purposes only.
    public Long getMaximumSize() {
        return maximumSize;
    }

    // For testing purposes only.
    public Duration getExpireAfterWrite() {
        return expireAfterWrite;
    }

    // For testing purposes only.
    public Duration getExpireAfterAccess() {
        return expireAfterAccess;
    }

    private static class MappingSupplier implements Supplier<Object> {

        private final Callable<?> valueLoader;

        public MappingSupplier(Callable<?> valueLoader) {
            this.valueLoader = valueLoader;
        }

        @Override
        public Object get() {
            try {
                return toCacheValue(valueLoader.call());
            } catch (RuntimeException e) {
                throw e;
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
        }
    }
}

@@ -300,50 +300,8 @@
 re;%0A
-import java.util.concurrent.ForkJoinPool;%0A
 impo
@@ -939,309 +939,8 @@
 ();%0A
-        /*%0A         * The following line is a workaround for a GraalVM issue: https://github.com/oracle/graal/issues/1524%0A         * TODO: Remove it as soon as a Quarkus release depends on GraalVM 19.3.0 or higher.%0A         */%0A        builder.executor(task -%3E ForkJoinPool.commonPool().execute(task));
 %0A   

Remove Caffeine ForkJoinPool#commonPool fix