org.infinispan.quarkus.hibernate.cache.VersionedEntry Maven / Gradle / Ivy

Go to download
package org.infinispan.quarkus.hibernate.cache;

import org.hibernate.cache.spi.entry.CacheEntry;
import org.hibernate.cache.spi.entry.StructuredCacheEntry;
import org.jboss.logging.Logger;

import java.time.Duration;
import java.util.Comparator;
import java.util.Map;
import java.util.function.BiFunction;
import java.util.function.Predicate;

final class VersionedEntry {

    static final Logger log = Logger.getLogger(VersionedEntry.class);
    static final boolean trace = log.isTraceEnabled();

    public static final Duration TOMBSTONE_LIFESPAN = Duration.ofSeconds(60);

    static final ExcludeEmptyFilter EXCLUDE_EMPTY_VERSIONED_ENTRY = new ExcludeEmptyFilter();

    final Object value;
    final Object version;
    final long timestamp;
    Duration lifespan = Duration.ofSeconds(Long.MAX_VALUE);

    VersionedEntry(Object value, Object version, long timestamp) {
        this.value = value;
        this.version = version;
        this.timestamp = timestamp;
    }

    VersionedEntry(long timestamp) {
        this(null, null, timestamp);
    }

    VersionedEntry(long timestamp, Duration lifespan) {
        this(null, null, timestamp);
        this.lifespan = lifespan;
    }

    Object getVersion() {
        return version;
    }

    long getTimestamp() {
        return timestamp;
    }

    long getLifespanNanos() {
        return lifespan.toNanos();
    }

    Object getValue() {
        return value;
    }

    static final class ComputeFn implements BiFunction {

        final VersionedEntry entry;
        final InternalRegion region;

        ComputeFn(VersionedEntry entry, InternalRegion region) {
            this.entry = entry;
            this.region = region;
        }

        @Override
        public Object apply(Object key, Object oldValue) {
            if (trace) {
                log.tracef("Applying %s to %s", this, oldValue);
            }
            if (entry.version == null) {
                // eviction or post-commit removal: we'll store it with given timestamp
                entry.lifespan = TOMBSTONE_LIFESPAN;
                return entry;
            }

            Object oldVersion;
            long oldTimestamp = Long.MIN_VALUE;
            if (oldValue instanceof VersionedEntry) {
                VersionedEntry oldVersionedEntry = (VersionedEntry) oldValue;
                oldVersion = oldVersionedEntry.version;
                oldTimestamp = oldVersionedEntry.timestamp;
                oldValue = oldVersionedEntry.value;
            } else {
                oldVersion = findVersion(oldValue);
            }

            if (oldVersion == null) {
                assert oldValue == null || oldTimestamp != Long.MIN_VALUE : oldValue;
                if (entry.timestamp <= oldTimestamp) {
                    // either putFromLoad or regular update/insert - in either case this update might come
                    // when it was evicted/region-invalidated. In both cases, with old timestamp we'll leave
                    // the invalid value
                    assert oldValue == null;
                    return null;
                } else {
                    return entry.value instanceof CacheEntry ? entry.value : entry;
                }
            } else {
                Comparator