org.nutz.plugins.cache.impl.redis.RedisCache Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nutz-plugins-cache Show documentation
There is a newer version: 1.r.69.v20220215
package org.nutz.plugins.cache.impl.redis;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Set;

import org.apache.shiro.cache.Cache;
import org.nutz.lang.Streams;
import org.nutz.log.Log;
import org.nutz.log.Logs;
import org.nutz.plugins.cache.CacheSerializer;
import org.nutz.plugins.cache.impl.lcache.LCacheManager;
import org.nutz.plugins.cache.serializer.DefaultJdkSerializer;

import redis.clients.jedis.Jedis;

@SuppressWarnings("unchecked")
public class RedisCache implements Cache {

    private static final Log log = Logs.get();

    @Deprecated
    public static boolean DEBUG = false;

    private String name;
    private byte[] nameByteArray;
    
    protected CacheSerializer serializer = new DefaultJdkSerializer();
    
    protected boolean debug;
    
    protected int ttl;

    public RedisCache setName(String name) {
        this.name = name;
        this.nameByteArray = name.getBytes();
        return this;
    }

    @Override
    public V get(K key) {
        if (debug)
            log.debugf("HGET name=%s key=%s", name, key);
        Jedis jedis = null;
        byte[] buf = null;
        try {
            jedis = jedis();
            buf = jedis.hget(nameByteArray, genKey(key));
            if (buf == null)
                return null;
            return (V) serializer.toObject(buf);
        } finally {
            Streams.safeClose(jedis);
        }
    }

    @Override
    public V put(K key, V value) {
        if (debug)
            log.debugf("HSET name=%s key=%s", name, key);
        Jedis jedis = null;
        try {
            jedis = jedis();
            jedis.hset(nameByteArray, genKey(key), (byte[])serializer.fromObject(value));
            return null;
        } finally {
            Streams.safeClose(jedis);
        }
    }

    @Override
    public V remove(K key) {
        if (debug)
            log.debugf("HDEL name=%s key=%s", name, key);
        Jedis jedis = null;
        try {
            jedis = jedis();
            jedis.hdel(nameByteArray, genKey(key));
            return null;
        } finally {
            Streams.safeClose(jedis);
        }
    }

    @Override
    public void clear() {
        if (debug)
            log.debugf("DEL name=%s", name);
        Jedis jedis = null;
        try {
            jedis = jedis();
            jedis.del(nameByteArray);
        } finally {
            Streams.safeClose(jedis);
        }
    }

    public int size() {
        if (debug)
            log.debugf("HLEN name=%s", name);
        Jedis jedis = null;
        try {
            jedis = jedis();
            return jedis.hlen(nameByteArray).intValue();
        } finally {
            Streams.safeClose(jedis);
        }
    }

    public Set keys() {
        if (debug)
            log.debugf("HKEYS name=%s", name);
        Jedis jedis = null;
        try {
            jedis = jedis();
            return (Set) jedis.hkeys(name);
        } finally {
            Streams.safeClose(jedis);
        }
    }

    @Override
    public Collection values() {
        if (debug)
            log.debugf("HVALES name=%s", name);
        Jedis jedis = null;
        try {
            jedis = jedis();
            Collection vals = jedis.hvals(nameByteArray);
            List list = new ArrayList();
            for (byte[] buf : vals)
                list.add((V) serializer.fromObject(buf));
            return list;
        } finally {
            Streams.safeClose(jedis);
        }
    }
    
    protected byte[] genKey(Object key) {
        return key.toString().getBytes();
    }
    
    protected Jedis jedis() {
        return LCacheManager.me().jedis();
    }
    
    public RedisCache setDebug(boolean debug) {
        this.debug = debug;
        return this;
    }
    
    public boolean isDebug() {
        return debug;
    }
    
    public RedisCache setSerializer(CacheSerializer serializer) {
        this.serializer = serializer;
        return this;
    }
    
    public void setTtl(int ttl) {
        this.ttl = ttl;
    }
}