112 lines
3.0 KiB
Java
112 lines
3.0 KiB
Java
package redis.clients.jedis;
|
|
|
|
import org.apache.commons.pool.BasePoolableObjectFactory;
|
|
import org.apache.commons.pool.impl.GenericObjectPool;
|
|
|
|
public class JedisPool {
|
|
private final GenericObjectPool internalPool;
|
|
|
|
public JedisPool(final GenericObjectPool.Config poolConfig, final String host) {
|
|
this(poolConfig, host, Protocol.DEFAULT_PORT, Protocol.DEFAULT_TIMEOUT, null);
|
|
}
|
|
|
|
public JedisPool(final GenericObjectPool.Config poolConfig,final String host, final int port) {
|
|
this(poolConfig, host, port, Protocol.DEFAULT_TIMEOUT, null);
|
|
}
|
|
|
|
public JedisPool(final GenericObjectPool.Config poolConfig,final String host, final int port, final int timeout) {
|
|
this(poolConfig, host, port, timeout, null);
|
|
}
|
|
|
|
public JedisPool(final GenericObjectPool.Config poolConfig,final String host, final int port, final int timeout, final String password) {
|
|
final String lhost;
|
|
final int lport;
|
|
final int ltimeout;
|
|
final String lpassword;
|
|
lhost = host;
|
|
lport = port;
|
|
ltimeout = (timeout > 0) ? timeout : Protocol.DEFAULT_TIMEOUT;
|
|
lpassword = password;
|
|
|
|
final JedisFactory factory = new JedisFactory(lhost, lport, ltimeout, lpassword);
|
|
this.internalPool = new GenericObjectPool(factory, poolConfig);
|
|
}
|
|
|
|
public JedisPool(final GenericObjectPool.Config poolConfig, final JedisShardInfo shardInfo) {
|
|
this(poolConfig, shardInfo.getHost(), shardInfo.getPort(), shardInfo.getTimeout(), shardInfo.getPassword());
|
|
}
|
|
|
|
public Jedis getResource() throws Exception {
|
|
return (Jedis) internalPool.borrowObject();
|
|
}
|
|
|
|
public void returnResource(final Jedis jedis) throws Exception {
|
|
internalPool.returnObject(jedis);
|
|
}
|
|
|
|
/**
|
|
* PoolableObjectFactory custom impl.
|
|
*/
|
|
private static class JedisFactory extends BasePoolableObjectFactory {
|
|
private final String host;
|
|
private final int port;
|
|
private final int timeout;
|
|
private final String password;
|
|
|
|
public JedisFactory(final String host, final int port, final int timeout, final String password) {
|
|
super();
|
|
this.host = host;
|
|
this.port = port;
|
|
this.timeout = (timeout > 0) ? timeout : -1;
|
|
this.password = password;
|
|
}
|
|
|
|
@Override
|
|
public Object makeObject() throws Exception {
|
|
final Jedis jedis;
|
|
if (timeout > 0) {
|
|
jedis = new Jedis(this.host, this.port, this.timeout);
|
|
} else {
|
|
jedis = new Jedis(this.host, this.port);
|
|
}
|
|
|
|
jedis.connect();
|
|
if (null != this.password) {
|
|
jedis.auth(this.password);
|
|
}
|
|
return jedis;
|
|
}
|
|
|
|
@Override
|
|
public void destroyObject(final Object obj) throws Exception {
|
|
if(obj instanceof Jedis) {
|
|
final Jedis jedis = (Jedis) obj;
|
|
if (jedis.isConnected()) {
|
|
try {
|
|
jedis.quit();
|
|
jedis.disconnect();
|
|
} catch (Exception e) {
|
|
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public boolean validateObject(final Object obj) {
|
|
if(obj instanceof Jedis) {
|
|
final Jedis jedis = (Jedis) obj;
|
|
try {
|
|
return jedis.isConnected() && jedis.ping().equals("PONG");
|
|
} catch (final Exception e) {
|
|
return false;
|
|
}
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
}
|