com.cybermkd.plugin.redis.RedisPlugin Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ICEREST-PLUGIN Show documentation
ICEREST官方插件集合
The newest version!
package com.cybermkd.plugin.redis;

import com.cybermkd.common.Plugin;
import com.cybermkd.common.util.Stringer;
import com.cybermkd.plugin.redis.serializer.FstSerializer;
import com.cybermkd.plugin.redis.serializer.ISerializer;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;
import redis.clients.jedis.Protocol;

/**
 * RedisPlugin.
 * RedisPlugin 支持多个 Redis 服务端，只需要创建多个 RedisPlugin 对象
 * 对应这多个不同的 Redis 服务端即可。也支持多个 RedisPlugin 对象对应同一
 * Redis 服务的不同 database，具体例子见 jfinal 手册
 */
public class RedisPlugin implements Plugin {

    private String cacheName;

    private String host;
    private Integer port = null;
    private Integer timeout = null;
    private String password = null;
    private Integer database = null;
    private String clientName = null;

    private ISerializer serializer = null;
    private IKeyNamingPolicy keyNamingPolicy = null;
    private JedisPoolConfig jedisPoolConfig = new JedisPoolConfig();

    public RedisPlugin(String cacheName, String host) {
        if (Stringer.isBlank(cacheName))
            throw new IllegalArgumentException("cacheName can not be blank.");
        if (Stringer.isBlank(host))
            throw new IllegalArgumentException("host can not be blank.");
        this.cacheName = cacheName.trim();
        this.host = host;
    }

    public RedisPlugin(String cacheName, String host, int port) {
        this(cacheName, host);
        this.port = port;
    }

    public RedisPlugin(String cacheName, String host, int port, int timeout) {
        this(cacheName, host, port);
        this.timeout = timeout;
    }

    public RedisPlugin(String cacheName, String host, int port, int timeout, String password) {
        this(cacheName, host, port, timeout);
        if (Stringer.isBlank(password))
            throw new IllegalArgumentException("password can not be blank.");
        this.password = password;
    }

    public RedisPlugin(String cacheName, String host, int port, int timeout, String password, int database) {
        this(cacheName, host, port, timeout, password);
        this.database = database;
    }

    public RedisPlugin(String cacheName, String host, int port, int timeout, String password, int database, String clientName) {
        this(cacheName, host, port, timeout, password, database);
        if (Stringer.isBlank(clientName))
            throw new IllegalArgumentException("clientName can not be blank.");
        this.clientName = clientName;
    }

    public RedisPlugin(String cacheName, String host, int port, String password) {
        this(cacheName, host, port, Protocol.DEFAULT_TIMEOUT, password);
    }

    public RedisPlugin(String cacheName, String host, String password) {
        this(cacheName, host, Protocol.DEFAULT_PORT, Protocol.DEFAULT_TIMEOUT, password);
    }

    public boolean start() {
        JedisPool jedisPool;
        if (port != null && timeout != null && password != null && database != null && clientName != null)
            jedisPool = new JedisPool(jedisPoolConfig, host, port, timeout, password, database, clientName);
        else if (port != null && timeout != null && password != null && database != null)
            jedisPool = new JedisPool(jedisPoolConfig, host, port, timeout, password, database);
        else if (port != null && timeout != null && password != null)
            jedisPool = new JedisPool(jedisPoolConfig, host, port, timeout, password);
        else if (port != null && timeout != null)
            jedisPool = new JedisPool(jedisPoolConfig, host, port, timeout);
        else if (port != null)
            jedisPool = new JedisPool(jedisPoolConfig, host, port);
        else
            jedisPool = new JedisPool(jedisPoolConfig, host);

        if (serializer == null)
            serializer = FstSerializer.me;
        if (keyNamingPolicy == null)
            keyNamingPolicy = IKeyNamingPolicy.defaultKeyNamingPolicy;

        Cache cache = new Cache(cacheName, jedisPool, serializer, keyNamingPolicy);
        Redis.addCache(cache);
        return true;
    }

    public boolean stop() {
        Cache cache = Redis.removeCache(cacheName);
        if (cache == Redis.mainCache)
            Redis.mainCache = null;
        cache.jedisPool.destroy();
        return true;
    }

    /**
     * 当RedisPlugin 提供的设置属性仍然无法满足需求时，通过此方法获取到
     * JedisPoolConfig 对象，可对 redis 进行更加细致的配置
     *      * 例如：
     * redisPlugin.getJedisPoolConfig().setMaxTotal(100);
     * 
     */
    public JedisPoolConfig getJedisPoolConfig() {
        return jedisPoolConfig;
    }

    public void setJedisPoolConfig(JedisPoolConfig config) {
        this.jedisPoolConfig = config;
    }

    // ---------

    public void setSerializer(ISerializer serializer) {
        this.serializer = serializer;
    }

    public void setKeyNamingPolicy(IKeyNamingPolicy keyNamingPolicy) {
        this.keyNamingPolicy = keyNamingPolicy;
    }

    // ---------

    public void setTestWhileIdle(boolean testWhileIdle) {
        jedisPoolConfig.setTestWhileIdle(testWhileIdle);
    }

    public void setMinEvictableIdleTimeMillis(int minEvictableIdleTimeMillis) {
        jedisPoolConfig.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);
    }

    public void setTimeBetweenEvictionRunsMillis(int timeBetweenEvictionRunsMillis) {
        jedisPoolConfig.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);
    }

    public void setNumTestsPerEvictionRun(int numTestsPerEvictionRun) {
        jedisPoolConfig.setNumTestsPerEvictionRun(numTestsPerEvictionRun);
    }
}