package com.cjs.cms.util.redis;

import java.io.Serializable;
import java.util.concurrent.locks.ReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock;

import org.apache.commons.lang3.SerializationUtils;
import org.apache.ibatis.cache.Cache;

import com.cjs.cms.util.redis.JedisTemplate.JedisAction;

import redis.clients.jedis.Jedis;

/**
 * 自定义Mybatis整合Redis缓存
 * 
 * @author tongyufu
 * 
 */
public final class RedisCache implements Cache {

    private final ReadWriteLock  readWriteLock = new ReentrantReadWriteLock();
    private String               id;
    private static JedisTemplate jedisTemplate = new JedisTemplate();

    public RedisCache(String id) {
        this.id = id;
    }

    @Override
    public void clear() {
        this.getCache().flushDB();
    }

    @Override
    public String getId() {
        return this.id;
    }

    @Override
    public void putObject(Object key, Object value) {
        this.getCache().set(SerializationUtils.serialize((Serializable) key),
            SerializationUtils.serialize((Serializable) value), 3600);
    }

    @Override
    public Object getObject(Object key) {
        byte[] bitKey = SerializationUtils.serialize((Serializable) key);
        byte[] value = this.getCache().get(bitKey);
        return value == null ? null : SerializationUtils.deserialize(value);
    }

    @Override
    public ReadWriteLock getReadWriteLock() {
        return readWriteLock;
    }

    @Override
    public int getSize() {
        return this.getCache().execute(new JedisAction<Integer>() {

            @Override
            public Integer action(Jedis jedis) {
                return Integer.parseInt(jedis.dbSize().toString());
            }
        });
    }

    @Override
    public Object removeObject(final Object key) {
        return this.getCache().execute(new JedisAction<Object>() {

            @Override
            public Object action(Jedis jedis) {
                return jedis.expire(SerializationUtils.serialize((Serializable) key), 0);
            }
        });
    }

    private JedisTemplate getCache() {
        return jedisTemplate;
    }
}