demo-tabledata-redis/src/main/java/com/fr/plugin/db/redis/core/RedisPool.java

package com.fr.plugin.db.redis.core;

import com.fanruan.api.conf.ConfigurationKit;
import com.fanruan.api.generic.Matcher;
import com.fanruan.api.generic.Runner;
import com.fanruan.api.log.LogKit;
import com.fanruan.api.util.StringKit;
import com.fr.config.Configuration;
import com.fr.plugin.db.redis.core.emb.Redis;
import com.fr.plugin.db.redis.core.emb.impl.ProxyRedis;
import com.fr.plugin.db.redis.core.emb.impl.SimpleRedis;
import com.fr.plugin.db.redis.core.pool.RedisConnectionPoolConfig;
import com.fr.plugin.db.redis.core.pool.RedisConnectionProxyConfig;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @author richie
 * @version 10.0
 * Created by richie on 2019-03-18
 */
public class RedisPool {

    private static RedisPool pool = new RedisPool();

    static {
        ConfigurationKit.listenCacheChange(new Matcher<Class<? extends Configuration>>() {
            @Override
            public boolean match(Class clazz) {
                return RedisConnectionPoolConfig.class == clazz;
            }
        }, new Runner() {
            @Override
            public void run() {
                pool.clearAll();
            }
        });
    }

    public static RedisPool getPool() {
        return pool;
    }

    private Map<String, JedisPool> jedisPoolMap = new ConcurrentHashMap<String, JedisPool>();

    private void clearAll() {
        jedisPoolMap.clear();
    }

    public Redis getResource(String host, int port, String password) {
        if (RedisConnectionProxyConfig.getInstance().isOpen()) {
            try {
                return createJedisBySSH(host, port, password);
            } catch (JSchException e) {
                LogKit.error(e.getMessage(), e);
            }
        } else {
            String feature = host + ":" + port + "@" + password;
            JedisPool jedisPool = jedisPoolMap.get(feature);
            if (jedisPool == null) {
                jedisPool = createJedisPool(host, port, password);
                jedisPoolMap.put(feature, jedisPool);
            }
            return new SimpleRedis(jedisPool.getResource());
        }
        Jedis jedis = new Jedis(host, port);
        jedis.auth(password);
        return new SimpleRedis(jedis);
    }

    private JedisPool createJedisPool(String host, int port, String password) {
        RedisConnectionPoolConfig poolConfig = RedisConnectionPoolConfig.getInstance();
        JedisPoolConfig config = new JedisPoolConfig();
        config.setMaxTotal(poolConfig.getMaxTotal());
        //连接耗尽时是否阻塞, false报异常,true阻塞直到超时, 默认true
        config.setBlockWhenExhausted(poolConfig.getBlockWhenExhausted());
        //是否启用pool的jmx管理功能, 默认true
        config.setJmxEnabled(true);
        //MBean ObjectName = new ObjectName("org.apache.commons.pool2:type=GenericObjectPool,name=" + "pool" + i); 默认为"pool", JMX不熟,具体不知道是干啥的...默认就好.
        config.setJmxNamePrefix("pool");
        //是否启用后进先出, 默认true
        config.setLifo(poolConfig.getLifo());
        //最大空闲连接数
        config.setMaxIdle(poolConfig.getMaxIdle());
        //最大连接数，可配置
        //获取连接时的最大等待毫秒数(如果设置为阻塞时BlockWhenExhausted),如果超时就抛异常, 小于零:阻塞不确定的时间,  默认-1
        config.setMaxWaitMillis(poolConfig.getMaxWait());
        //逐出连接的最小空闲时间 默认1800000毫秒(30分钟)
        config.setMinEvictableIdleTimeMillis(1800000);
        //最小空闲连接数, 默认0
        config.setMinIdle(0);
        //每次逐出检查时 逐出的最大数目 如果为负数就是 : 1/abs(n), 默认3
        config.setNumTestsPerEvictionRun(3);
        //对象空闲多久后逐出, 当空闲时间>该值 且 空闲连接>最大空闲数 时直接逐出,不再根据MinEvictableIdleTimeMillis判断  (默认逐出策略)
        config.setSoftMinEvictableIdleTimeMillis(1800000);
        //在获取连接的时候检查有效性, 默认false
        config.setTestOnBorrow(false);
        //在空闲时检查有效性, 默认false
        config.setTestWhileIdle(false);
        //逐出扫描的时间间隔(毫秒) 如果为负数,则不运行逐出线程, 默认-1
        config.setTimeBetweenEvictionRunsMillis(-1);
        if (StringKit.isNotBlank(password)) {
            return new JedisPool(config, host, port, poolConfig.getTimeout(), password);
        } else {
            return new JedisPool(config, host, port, poolConfig.getTimeout());
        }
    }

    private Redis createJedisBySSH(String host, int port, String password) throws JSchException {
        RedisConnectionProxyConfig proxyConfig = RedisConnectionProxyConfig.getInstance();
        JSch jsch = new JSch();
        if (StringKit.isNotBlank(proxyConfig.getPrivateKeyPath())) {
            jsch.addIdentity(proxyConfig.getPrivateKeyPath());
        }
        Session session = jsch.getSession(proxyConfig.getUsername(), proxyConfig.getHost(), proxyConfig.getPort());
        if (StringKit.isNotBlank(proxyConfig.getPassword())) {
            session.setPassword(proxyConfig.getPassword());
        }
        session.setConfig("StrictHostKeyChecking", "no");
        session.connect();

        int newPort = session.setPortForwardingL(0, host, port);

        Jedis jedis = new Jedis(host, newPort);

        if (StringKit.isNotBlank(password)) {
            jedis.auth(password);
        }
        return new ProxyRedis(session, jedis);

    }
}