org.darkphoenixs.pool.redis.RedisClusterConnPool Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of connectionpool-client Show documentation
A simple multi-purpose connection pool client (Kafka & Hbase & Redis & RMDB & Socket & Http)
The newest version!
package org.darkphoenixs.pool.redis;

import org.darkphoenixs.pool.ConnectionPool;
import org.darkphoenixs.pool.PoolConfig;
import redis.clients.jedis.HostAndPort;
import redis.clients.jedis.JedisCluster;

import java.io.IOException;
import java.util.HashSet;
import java.util.Properties;
import java.util.Set;

/**
 * Title: RedisClusterConnPool
 * Description: Redis集群连接池
 *
 * @author Victor
 * @version 1.2.4
 * @see ConnectionPool
 * @since 2016 /11/21
 */
public class RedisClusterConnPool implements ConnectionPool {

    private final JedisCluster jedisCluster;

    /**
     * Instantiates a new Redis cluster conn pool.
     *
     * @param clusterNodes the jedis cluster nodes
     */
    public RedisClusterConnPool(final Set clusterNodes) {

        this(clusterNodes, RedisConfig.DEFAULT_TIMEOUT);
    }

    /**
     * Instantiates a new Redis cluster conn pool.
     *
     * @param clusterNodes the cluster nodes
     * @param timeout      the timeout
     */
    public RedisClusterConnPool(final Set clusterNodes,
                                final int timeout) {

        this(clusterNodes, timeout, timeout);
    }

    /**
     * Instantiates a new Redis cluster conn pool.
     *
     * @param clusterNodes      the jedis cluster nodes
     * @param connectionTimeout the connection timeout
     * @param soTimeout         the so timeout
     */
    public RedisClusterConnPool(final Set clusterNodes,
                                final int connectionTimeout,
                                final int soTimeout) {

        this(clusterNodes, connectionTimeout, soTimeout, RedisConfig.DEFAULT_MAXATTE);
    }

    /**
     * Instantiates a new Redis cluster conn pool.
     *
     * @param clusterNodes      the jedis cluster nodes
     * @param connectionTimeout the connection timeout
     * @param soTimeout         the so timeout
     * @param maxAttempts       the max attempts
     */
    public RedisClusterConnPool(final Set clusterNodes,
                                final int connectionTimeout,
                                final int soTimeout,
                                final int maxAttempts) {

        this(clusterNodes, connectionTimeout, soTimeout, maxAttempts, new PoolConfig());
    }

    /**
     * Instantiates a new Redis cluster conn pool.
     *
     * @param properties the properties
     */
    public RedisClusterConnPool(final Properties properties) {

        this(new PoolConfig(), properties);
    }

    /**
     * Instantiates a new Redis cluster conn pool.
     *
     * @param poolConfig the pool config
     * @param properties the properties
     */
    public RedisClusterConnPool(final PoolConfig poolConfig, final Properties properties) {

        Set jedisClusterNodes = new HashSet();

        for (String hostAndPort : properties.getProperty(RedisConfig.CLUSTER_PROPERTY).split(","))

            jedisClusterNodes.add(new HostAndPort(hostAndPort.split(":")[0], Integer.valueOf(hostAndPort.split(":")[1])));

        int timeout = Integer.parseInt(properties.getProperty(RedisConfig.TIMEOUT_PROPERTY, String.valueOf(RedisConfig.DEFAULT_TIMEOUT)));

        int maxAttempts = Integer.valueOf(properties.getProperty(RedisConfig.MAXATTE_PROPERTY, String.valueOf(RedisConfig.DEFAULT_MAXATTE)));

        String password = properties.getProperty(RedisConfig.PASSWORD_PROPERTY, RedisConfig.DEFAULT_PASSWORD);

        jedisCluster = new JedisCluster(jedisClusterNodes, timeout, timeout, maxAttempts, password, poolConfig);
    }

    /**
     * Instantiates a new Redis cluster conn pool.
     *
     * @param poolConfig   the pool config
     * @param clusterNodes the cluster nodes
     */
    public RedisClusterConnPool(final PoolConfig poolConfig, final Set clusterNodes) {

        this(poolConfig, clusterNodes, RedisConfig.DEFAULT_PASSWORD);
    }

    /**
     * Instantiates a new Redis cluster conn pool.
     *
     * @param poolConfig   the pool config
     * @param clusterNodes the cluster nodes
     * @param password     the password
     */
    public RedisClusterConnPool(final PoolConfig poolConfig, final Set clusterNodes, final String password) {

        this(poolConfig, clusterNodes, password, RedisConfig.DEFAULT_TIMEOUT);
    }

    /**
     * Instantiates a new Redis cluster conn pool.
     *
     * @param poolConfig   the pool config
     * @param clusterNodes the cluster nodes
     * @param password     the password
     * @param timeout      the timeout
     */
    public RedisClusterConnPool(final PoolConfig poolConfig, final Set clusterNodes, final String password, final int timeout) {

        this(clusterNodes, timeout, timeout, RedisConfig.DEFAULT_MAXATTE, password, poolConfig);
    }

    /**
     * Instantiates a new Redis cluster conn pool.
     *
     * @param clusterNodes the cluster nodes
     * @param timeout      the timeout
     * @param maxAttempts  the max attempts
     * @param poolConfig   the pool config
     */
    public RedisClusterConnPool(final Set clusterNodes,
                                final int timeout,
                                final int maxAttempts,
                                final PoolConfig poolConfig) {

        this(clusterNodes, timeout, timeout, maxAttempts, RedisConfig.DEFAULT_PASSWORD, poolConfig);
    }

    /**
     * Instantiates a new Redis cluster conn pool.
     *
     * @param clusterNodes      the jedis cluster nodes
     * @param connectionTimeout the connection timeout
     * @param soTimeout         the so timeout
     * @param maxAttempts       the max attempts
     * @param poolConfig        the pool config
     */
    public RedisClusterConnPool(final Set clusterNodes,
                                final int connectionTimeout,
                                final int soTimeout,
                                final int maxAttempts,
                                final PoolConfig poolConfig) {

        this(clusterNodes, connectionTimeout, soTimeout, maxAttempts, RedisConfig.DEFAULT_PASSWORD, poolConfig);
    }

    /**
     * Instantiates a new Redis cluster conn pool.
     *
     * @param clusterNodes      the cluster nodes
     * @param connectionTimeout the connection timeout
     * @param soTimeout         the so timeout
     * @param maxAttempts       the max attempts
     * @param password          the password
     * @param poolConfig        the pool config
     */
    public RedisClusterConnPool(final Set clusterNodes,
                                final int connectionTimeout,
                                final int soTimeout,
                                final int maxAttempts,
                                final String password,
                                final PoolConfig poolConfig) {

        jedisCluster = new JedisCluster(clusterNodes, connectionTimeout, soTimeout, maxAttempts, password, poolConfig);
    }

    @Override
    public JedisCluster getConnection() {

        return jedisCluster;
    }

    @Override
    public void returnConnection(JedisCluster conn) {
        // nothing to do...
    }

    @Override
    public void invalidateConnection(JedisCluster conn) {
        // nothing to do...
    }

    /**
     * Close.
     *
     * @throws IOException the io exception
     */
    public void close() throws IOException {

        jedisCluster.close();
    }
}