All Downloads are FREE. Search and download functionalities are using the official Maven repository.

net.sf.ehcache.constructs.refreshahead.RefreshAheadCache Maven / Gradle / Ivy

Go to download

Internal ehcache-core module. This artifact is not meant to be used directly for jdk 1.5

The newest version!
/**
 *  Copyright Terracotta, Inc.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package net.sf.ehcache.constructs.refreshahead;

import net.sf.ehcache.Cache;
import net.sf.ehcache.CacheException;
import net.sf.ehcache.Ehcache;
import net.sf.ehcache.Element;
import net.sf.ehcache.Status;
import net.sf.ehcache.config.CacheConfiguration;
import net.sf.ehcache.config.PersistenceConfiguration;
import net.sf.ehcache.config.PersistenceConfiguration.Strategy;
import net.sf.ehcache.config.TerracottaConfiguration;
import net.sf.ehcache.config.TerracottaConfiguration.Consistency;
import net.sf.ehcache.constructs.EhcacheDecoratorAdapter;
import net.sf.ehcache.constructs.refreshahead.ThreadedWorkQueue.BatchWorker;
import net.sf.ehcache.extension.CacheExtension;
import net.sf.ehcache.loader.CacheLoader;
import net.sf.ehcache.statistics.extended.ExtendedStatistics;
import net.sf.ehcache.store.MemoryStoreEvictionPolicy;
import net.sf.ehcache.util.VmUtils;

import java.io.Serializable;
import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicLong;

/**
 * A cache decorator which implements read ahead refreshing. Read ahead occurs
 * when a cache entry is accessed prior to its expiration, and triggers a reload of
 * the value in the background.
 * 

 * A significant attempt is made to ensure only one node of the cache works on a specific key at a time. There is no guarantee that every
 * triggered refresh ahead case will be processed. As the maximum number of backlog entries is reached, refresh ahead requests will be
 * dropped silently.
 * 

 * Provided the Number of threads per node * Number of nodes < the maximum backlog, only one node in the cluster will refresha given key at
 * a time.
 *
 * @author cschanck
 *
 */
public class RefreshAheadCache extends EhcacheDecoratorAdapter {

    private static final Object REFRESH_VALUE = Boolean.TRUE;
    private static final int DEFAULT_SUPPORT_TTL_SECONDS = (int)TimeUnit.SECONDS.convert(10, TimeUnit.MINUTES);
    private final AtomicLong refreshSuccessCount = new AtomicLong();
    private final RefreshAheadCacheConfiguration refreshAheadConfig;
    private CacheConfiguration supportConfig;

    private volatile Ehcache supportCache;
    private volatile ThreadedWorkQueue refreshWorkQueue;

    /**
     * Create a Refresh Ahead Cache Adaptor with the specified configuration. An auxiliary EhCache
     * Cache will be created for the purposes of synchronization, so only one node
     * in a clustered environment will refresh a key at a given time.
     *
     * @param adaptedCache
     * @param refreshConfig
     */
    public RefreshAheadCache(Ehcache adaptedCache, RefreshAheadCacheConfiguration refreshConfig) {

        super(adaptedCache);
        this.refreshAheadConfig = refreshConfig;

        // XA transactions cannot actually refresh sensibly. At least not
        // reasonably. GAE doesn't support threads. Other conditions around?
        boolean refreshAllowed = !underlyingCache.getCacheConfiguration().isXaStrictTransactional();
        refreshAllowed = refreshAllowed && !underlyingCache.getCacheConfiguration().isXaTransactional();
        refreshAllowed = refreshAllowed && !underlyingCache.getCacheConfiguration().isLocalTransactional();
        refreshAllowed = refreshAllowed && !VmUtils.isInGoogleAppEngine();

        if (refreshAllowed) {
            initSupportCache();
            initWorkQueue();
        } else {
            throw new UnsupportedOperationException("refresh-ahead not supported under transactions or with GAE");
        }
    }

    private void initSupportCache() {
        // create the support cache
        // make this cache clustered in the same way as the underlying cache,
        this.supportConfig = new CacheConfiguration();
        supportConfig.name(underlyingCache.getName() + "_" + getClass().getName() + "_refreshAheadSupport");
        supportConfig = supportConfig.persistence(new PersistenceConfiguration().strategy(Strategy.NONE));
        int activeSize = 2 * refreshAheadConfig.getBatchSize() * refreshAheadConfig.getNumberOfThreads();
        supportConfig = supportConfig.maxEntriesLocalHeap(activeSize);
        supportConfig = supportConfig.memoryStoreEvictionPolicy(MemoryStoreEvictionPolicy.LRU);
        supportConfig = supportConfig.timeToLiveSeconds(DEFAULT_SUPPORT_TTL_SECONDS);

        // TC stuff
        if (underlyingCache.getCacheConfiguration().isTerracottaClustered()) {
            supportConfig = supportConfig.persistence(new PersistenceConfiguration().strategy(Strategy.DISTRIBUTED));

            TerracottaConfiguration newTerracottaConfig = new TerracottaConfiguration().clustered(true);

            newTerracottaConfig.consistency(Consistency.STRONG);

            supportConfig.addTerracotta(newTerracottaConfig);
        } else { supportConfig.setMaxElementsOnDisk(activeSize); }

        // here we try to create the support cache.
        this.supportCache = new Cache(supportConfig);

        Ehcache prior = underlyingCache.getCacheManager().addCacheIfAbsent(supportCache);
        if (prior != supportCache) {
            throw new IllegalStateException("Unable to add refresh ahead support cache due to name collision: "
                    + refreshAheadConfig.getName());
        }

        // wipe it on startup. might wobble in a clustered case, but clears out orphans.
        prior.removeAll();

        // catch the dispose. not sure this is the best way to do it at all.
        // we could register a listener alternatively
        underlyingCache.registerCacheExtension(new CacheExtension() {

            @Override
            public void init() {
            }

            @Override
            public Status getStatus() {
                return underlyingCache.getStatus();
            }

            @Override
            public void dispose() throws CacheException {
                RefreshAheadCache.this.localDispose();
            }

            @Override
            public CacheExtension clone(Ehcache cache) throws CloneNotSupportedException {
                throw new CloneNotSupportedException();
            }
        });
    }

    private void initWorkQueue() {
        BatchWorker batchWorker = new BatchWorker() {

            @Override
            public void process(Collection collection) {

                // only fetch this once for each process() call
                long accessTime = System.currentTimeMillis();

                HashSet keysToProcess = new HashSet();
                for (Object key : collection) {

                    // check if it was loaded by someone else in the meantime -- does it still qualify for refresh ahead?
                    Element quickTest = underlyingCache.getQuiet(key);
                    if (quickTest == null || checkForRefresh(quickTest, accessTime, refreshAheadConfig.getTimeToRefreshMillis())) {
                        final Element ersatz = new Element(key, REFRESH_VALUE);

                        if (supportCache.putIfAbsent(ersatz) == null) {
                            // work, work, work
                            keysToProcess.add(key);
                        }
                    }
                }
                try {
                    // iterate through the loaders
                    for (CacheLoader loader : underlyingCache.getRegisteredCacheLoaders()) {
                        // if we are out of keys, punt
                        if (keysToProcess.isEmpty()) {
                            break;
                        }

                        // try and load them all
                        Map values = loader.loadAll(keysToProcess);
                        // subtract the ones that were loaded
                        keysToProcess.removeAll(values.keySet());
                        try {
                            for (Map.Entry entry : values.entrySet()) {
                                Element newElement = new Element(entry.getKey(), entry.getValue());
                                underlyingCache.put(newElement);
                                refreshSuccessCount.incrementAndGet();
                            }
                        } finally {
                            // subtract from the support cache
                            supportCache.removeAll(values.keySet());
                        }
                    }
                    // assume we got here ok, now evict any that don't evict
                    if (refreshAheadConfig.isEvictOnLoadMiss() && !keysToProcess.isEmpty()) {
                        underlyingCache.removeAll(keysToProcess);
                    }
                } finally {
                    // this is utterly paranoid. but still.
                    supportCache.removeAll(keysToProcess);
                }

            }
        };

        this.refreshWorkQueue = new ThreadedWorkQueue(batchWorker, refreshAheadConfig.getNumberOfThreads(), new ThreadFactory() {

            @Override
            public Thread newThread(Runnable r) {
                Thread t = new Thread(r);
                t.setDaemon(true);
                return t;
            }
        }, refreshAheadConfig.getMaximumRefreshBacklogItems(), refreshAheadConfig.getBatchSize());
    }

    private boolean checkForRefresh(Element elem, long accessTime, long timeToRefreshMillis) {
        if (elem == null) {
            return false;
        }

        long minAccessForRefreshTime = elem.getCreationTime() + timeToRefreshMillis;

        return (accessTime >= minAccessForRefreshTime);
    }

    private void possiblyTriggerRefresh(Element elem, long timeToRefreshMillis) {
        if (checkForRefresh(elem, System.currentTimeMillis(), timeToRefreshMillis)) {
            // now add the key to the queue. smallest overhead we could get.
            refreshWorkQueue.offer(elem.getObjectKey());
        }
    }

    @Override
    public Element get(Object key) throws IllegalStateException, CacheException {
        Element elem = super.get(key);
        possiblyTriggerRefresh(elem, refreshAheadConfig.getTimeToRefreshMillis());
        return elem;
    }

    @Override
    public Element get(Serializable key) throws IllegalStateException, CacheException {
        Element elem = super.get(key);
        possiblyTriggerRefresh(elem, refreshAheadConfig.getTimeToRefreshMillis());
        return elem;
    }

    /**
     * number of refreshes processed locally.
     */
    @org.terracotta.statistics.Statistic(name = "refreshed", tags = "refreshahead")
    public long getRefreshSuccessCount() {
        return refreshSuccessCount.get();
    }

    private void localDispose() throws IllegalStateException {
        synchronized (this) {
            if (refreshWorkQueue != null) {
                refreshWorkQueue.shutdown();
                refreshWorkQueue = null;
            }
            if (supportCache != null) {
                try {
                    supportCache.getCacheManager().removeCache(getName());
                } catch (Throwable t) {
                }
                supportCache = null;
            }
        }
    }

    @Override
    public String getName() {
        if (refreshAheadConfig.getName() != null) {
            return refreshAheadConfig.getName();
        }
        return super.getName();
    }

    /**
     * Gets offer count.
     *
     * @return the offer count
     */
    @org.terracotta.statistics.Statistic(name = "offered", tags = "refreshahead")
    public long getOfferCount() {
        return refreshWorkQueue.getOfferedCount();
    }

    /**
     * Gets dropped count.
     *
     * @return the dropped count
     */
    @org.terracotta.statistics.Statistic(name = "dropped", tags = "refreshahead")
    public long getDroppedCount() {
        return refreshWorkQueue.getDroppedCount();
    }

    /**
     * Gets processed count.
     *
     * @return the processed count
     */
    @org.terracotta.statistics.Statistic(name = "processed", tags = "refreshahead")
    public long getProcessedCount() {
        return refreshWorkQueue.getProcessedCount();
    }

    /**
     * Gets backlog count.
     *
     * @return the backlog count
     */
    @org.terracotta.statistics.Statistic(name = "backlog", tags = "refreshahead")
    public long getBacklogCount() {
        return refreshWorkQueue.getBacklogCount();
    }

    /**
     * Find refreshed counter statistic.
     *
     * @param cache the cache this statistic is attached to.
     * @return the set
     */
    public static Set> findRefreshedStatistic(Ehcache cache) {
        return cache.getStatistics().getExtended().passthru("refreshed",
            Collections.singletonMap("refreshahead", null).keySet());
    }

    /**
     * Find offer statistic.
     *
     * @param cache the cache this statistic is attached to.
     * @return the set
     */
    public static Set> findOfferStatistic(Ehcache cache) {
        return cache.getStatistics().getExtended().passthru("offered",
            Collections.singletonMap("refreshahead", null).keySet());
    }

    /**
     * Find dropped statistic.
     *
     * @param cache the cache
     * @return the set
     */
    public static Set> findDroppedStatistic(Ehcache cache) {
        return cache.getStatistics().getExtended().passthru("dropped",
            Collections.singletonMap("refreshahead", null).keySet());
    }

    /**
     * Find processed statistic.
     *
     * @param cache the cache
     * @return the set
     */
    public static Set> findProcessedStatistic(Ehcache cache) {
        return cache.getStatistics().getExtended().passthru("processed",
            Collections.singletonMap("refreshahead", null).keySet());
    }

    /**
     * Find backlog statistic.
     *
     * @param cache the cache
     * @return the set
     */
    public static Set> findBacklogStatistic(Ehcache cache) {
        return cache.getStatistics().getExtended().passthru("backlog",
            Collections.singletonMap("refreshahead", null).keySet());
    }

}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy