grails.plugin.cache.redis.GrailsRedisCacheManager.java Source code

Introduction

Here is the source code for grails.plugin.cache.redis.GrailsRedisCacheManager.java

Source

/* Copyright 2012 SpringSource.
 *
 * 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 grails.plugin.cache.redis;

import grails.plugin.cache.GrailsCacheManager;
import org.springframework.cache.Cache;
import org.springframework.data.redis.cache.DefaultRedisCachePrefix;
import org.springframework.data.redis.cache.RedisCachePrefix;
import org.springframework.data.redis.core.RedisTemplate;

import java.util.Collection;
import java.util.Collections;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

/**
 * Based on org.springframework.data.redis.cache.RedisCacheManager which has all private fields.
 *
 * @author Costin Leau
 * @author Burt Beckwith
 */
public class GrailsRedisCacheManager implements GrailsCacheManager {

    // fast lookup by name map
    protected final ConcurrentMap<String, Cache> caches = new ConcurrentHashMap<String, Cache>();
    protected final Collection<String> names = Collections.unmodifiableSet(caches.keySet());
    @SuppressWarnings("rawtypes")
    protected final RedisTemplate redisTemplate;
    protected boolean usePrefix;
    protected RedisCachePrefix cachePrefix = new DefaultRedisCachePrefix();
    protected Long ttl;

    public GrailsRedisCacheManager(@SuppressWarnings("rawtypes") RedisTemplate template) {
        redisTemplate = template;
    }

    @SuppressWarnings("unchecked")
    public Cache getCache(String name) {
        Cache c = caches.get(name);
        if (c == null) {
            c = new GrailsRedisCache(name, (usePrefix ? cachePrefix.prefix(name) : null), redisTemplate, ttl);
            caches.put(name, c);
        }

        return c;
    }

    public Collection<String> getCacheNames() {
        return names;
    }

    public boolean cacheExists(String name) {
        return getCacheNames().contains(name);
    }

    public boolean destroyCache(String name) {
        Cache cache = caches.remove(name);
        if (cache != null) {
            // TODO remove Redis backing store
            cache.clear();
        }

        return true;
    }

    /**
     * Sets the cache prefix.
     *
     * @param prefix the prefix
     */
    public void setCachePrefix(RedisCachePrefix prefix) {
        cachePrefix = prefix;
    }

    /**
     * Enable the cache prefix.
     */
    public void setUsePrefix(Boolean use) {
        usePrefix = use;
    }

    /**
     * Sets the cache's time to live.
     *
     * @param ttl Time to live in seconds.
     */
    public void setTimeToLive(Long ttl) {
        this.ttl = ttl;
    }

}