Add default implementation of JredisPool
- Add DefaultJredisPool to pool JRedis connections - Prevent returning the same resource to the pool multiple times on JredisConnection close - Fixes DATAREDIS-154 previous JRedis pool left broken connections in the pool
This commit is contained in:
@@ -0,0 +1,143 @@
|
||||
/*
|
||||
* Copyright 2013 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.springframework.data.redis.connection.jredis;
|
||||
|
||||
import org.jredis.ClientRuntimeException;
|
||||
import org.jredis.JRedis;
|
||||
import org.jredis.RedisException;
|
||||
import org.jredis.connector.ConnectionSpec;
|
||||
import org.jredis.connector.NotConnectedException;
|
||||
import org.jredis.ri.alphazero.connection.DefaultConnectionSpec;
|
||||
import org.junit.After;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.springframework.data.redis.SettingsUtils;
|
||||
import org.apache.commons.pool.impl.GenericObjectPool.Config;
|
||||
import static org.junit.Assert.fail;
|
||||
import static org.junit.Assert.assertNotSame;
|
||||
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
|
||||
/**
|
||||
* Integration test of {@link DefaultJredisPool}
|
||||
*
|
||||
* @author Jennifer Hickey
|
||||
*
|
||||
*/
|
||||
public class DefaultJredisPoolTests {
|
||||
|
||||
private ConnectionSpec connectionSpec;
|
||||
|
||||
private JredisPool pool;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
this.connectionSpec = DefaultConnectionSpec.newSpec(SettingsUtils.getHost(), SettingsUtils.getPort(),
|
||||
0, null);
|
||||
}
|
||||
|
||||
@After
|
||||
public void tearDown() {
|
||||
if (this.pool != null) {
|
||||
this.pool.destroy();
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetResource() throws RedisException {
|
||||
this.pool = new DefaultJredisPool(connectionSpec);
|
||||
JRedis client = pool.getResource();
|
||||
assertNotNull(client);
|
||||
client.ping();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetResourcePoolExhausted() {
|
||||
Config poolConfig = new Config();
|
||||
poolConfig.maxActive = 1;
|
||||
poolConfig.maxWait = 1;
|
||||
this.pool = new DefaultJredisPool(connectionSpec, poolConfig);
|
||||
JRedis client = pool.getResource();
|
||||
assertNotNull(client);
|
||||
try {
|
||||
pool.getResource();
|
||||
fail("ClientRuntimeException should be thrown when pool exhausted");
|
||||
} catch (ClientRuntimeException e) {
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetResourceValidate() {
|
||||
Config poolConfig = new Config();
|
||||
poolConfig.testOnBorrow = true;
|
||||
this.pool = new DefaultJredisPool(connectionSpec, poolConfig);
|
||||
JRedis client = pool.getResource();
|
||||
assertNotNull(client);
|
||||
}
|
||||
|
||||
@Test(expected = ClientRuntimeException.class)
|
||||
public void testGetResourceCreationUnsuccessful() {
|
||||
// Config poolConfig = new Config();
|
||||
// poolConfig.testOnBorrow = true;
|
||||
this.pool = new DefaultJredisPool(DefaultConnectionSpec.newSpec(SettingsUtils.getHost(), 3333, 0,
|
||||
null));
|
||||
pool.getResource();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testReturnResource() throws RedisException {
|
||||
Config poolConfig = new Config();
|
||||
poolConfig.maxActive = 1;
|
||||
poolConfig.maxWait = 1;
|
||||
this.pool = new DefaultJredisPool(connectionSpec);
|
||||
JRedis client = pool.getResource();
|
||||
assertNotNull(client);
|
||||
pool.returnResource(client);
|
||||
assertNotNull(pool.getResource());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testReturnBrokenResource() throws RedisException {
|
||||
Config poolConfig = new Config();
|
||||
poolConfig.maxActive = 1;
|
||||
poolConfig.maxWait = 1;
|
||||
this.pool = new DefaultJredisPool(connectionSpec, poolConfig);
|
||||
JRedis client = pool.getResource();
|
||||
assertNotNull(client);
|
||||
pool.returnBrokenResource(client);
|
||||
JRedis client2 = pool.getResource();
|
||||
assertNotSame(client, client2);
|
||||
try {
|
||||
client.ping();
|
||||
fail("Broken resouce connection should be closed");
|
||||
} catch (NotConnectedException e) {
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateWithHostAndPort() {
|
||||
this.pool = new DefaultJredisPool(SettingsUtils.getHost(), SettingsUtils.getPort());
|
||||
assertNotNull(pool.getResource());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateWithHostPortAndDbIndex() {
|
||||
this.pool = new DefaultJredisPool(SettingsUtils.getHost(), SettingsUtils.getPort(), 1, null, 0);
|
||||
assertNotNull(pool.getResource());
|
||||
}
|
||||
|
||||
}
|
||||
@@ -21,6 +21,7 @@ import static org.junit.Assert.fail;
|
||||
|
||||
import java.util.Arrays;
|
||||
|
||||
import org.jredis.JRedis;
|
||||
import org.jredis.protocol.BulkResponse;
|
||||
import org.junit.After;
|
||||
import org.junit.Ignore;
|
||||
@@ -28,11 +29,14 @@ import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.springframework.dao.DataAccessException;
|
||||
import org.springframework.data.redis.RedisConnectionFailureException;
|
||||
import org.springframework.data.redis.SettingsUtils;
|
||||
import org.springframework.data.redis.connection.AbstractConnectionIntegrationTests;
|
||||
import org.springframework.data.redis.connection.DefaultSortParameters;
|
||||
import org.springframework.data.redis.connection.RedisConnection;
|
||||
import org.springframework.data.redis.connection.SortParameters.Order;
|
||||
import org.springframework.test.context.ContextConfiguration;
|
||||
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
|
||||
import org.apache.commons.pool.impl.GenericObjectPool.Config;
|
||||
|
||||
/**
|
||||
* Integration test of {@link JredisConnection}
|
||||
@@ -85,6 +89,35 @@ public class JRedisConnectionIntegrationTests extends AbstractConnectionIntegrat
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testConnectionStaysOpenWhenPooled() {
|
||||
JredisConnectionFactory factory2 = new JredisConnectionFactory(new DefaultJredisPool(
|
||||
SettingsUtils.getHost(), SettingsUtils.getPort()));
|
||||
RedisConnection conn2 = factory2.getConnection();
|
||||
conn2.close();
|
||||
conn2.ping();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testConnectionNotReturnedOnException() {
|
||||
Config config = new Config();
|
||||
config.maxActive = 1;
|
||||
config.maxWait = 1;
|
||||
JredisConnectionFactory factory2 = new JredisConnectionFactory(new DefaultJredisPool(
|
||||
SettingsUtils.getHost(), SettingsUtils.getPort(), config));
|
||||
RedisConnection conn2 = factory2.getConnection();
|
||||
((JRedis) conn2.getNativeConnection()).quit();
|
||||
try {
|
||||
conn2.ping();
|
||||
fail("Expected RedisConnectionFailureException trying to use a closed connection");
|
||||
} catch (RedisConnectionFailureException e) {
|
||||
}
|
||||
conn2.close();
|
||||
// Verify we get a new connection from the pool and not the broken one
|
||||
RedisConnection conn3 = factory2.getConnection();
|
||||
conn3.ping();
|
||||
}
|
||||
|
||||
@Test(expected = UnsupportedOperationException.class)
|
||||
public void testMultiExec() throws Exception {
|
||||
super.testMultiExec();
|
||||
|
||||
@@ -20,6 +20,7 @@ import java.util.Collection;
|
||||
|
||||
import org.springframework.data.redis.SettingsUtils;
|
||||
import org.springframework.data.redis.connection.jedis.JedisConnectionFactory;
|
||||
import org.springframework.data.redis.connection.jredis.DefaultJredisPool;
|
||||
import org.springframework.data.redis.connection.jredis.JredisConnectionFactory;
|
||||
import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory;
|
||||
import org.springframework.data.redis.connection.srp.SrpConnectionFactory;
|
||||
@@ -39,9 +40,8 @@ public abstract class AtomicCountersParam {
|
||||
jedisConnFactory.afterPropertiesSet();
|
||||
|
||||
// JRedis
|
||||
JredisConnectionFactory jredisConnFactory = new JredisConnectionFactory();
|
||||
jredisConnFactory.setPort(SettingsUtils.getPort());
|
||||
jredisConnFactory.setHostName(SettingsUtils.getHost());
|
||||
JredisConnectionFactory jredisConnFactory = new JredisConnectionFactory(new DefaultJredisPool(
|
||||
SettingsUtils.getHost(), SettingsUtils.getPort()));
|
||||
jredisConnFactory.afterPropertiesSet();
|
||||
|
||||
// Lettuce
|
||||
|
||||
@@ -21,6 +21,7 @@ import java.util.Collection;
|
||||
import org.springframework.data.redis.Person;
|
||||
import org.springframework.data.redis.SettingsUtils;
|
||||
import org.springframework.data.redis.connection.jedis.JedisConnectionFactory;
|
||||
import org.springframework.data.redis.connection.jredis.DefaultJredisPool;
|
||||
import org.springframework.data.redis.connection.jredis.JredisConnectionFactory;
|
||||
import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory;
|
||||
import org.springframework.data.redis.connection.srp.SrpConnectionFactory;
|
||||
@@ -80,11 +81,8 @@ public abstract class CollectionTestParams {
|
||||
jsonPersonTemplate.afterPropertiesSet();
|
||||
|
||||
// jredis
|
||||
JredisConnectionFactory jredisConnFactory = new JredisConnectionFactory();
|
||||
|
||||
jredisConnFactory.setPort(SettingsUtils.getPort());
|
||||
jredisConnFactory.setHostName(SettingsUtils.getHost());
|
||||
|
||||
JredisConnectionFactory jredisConnFactory = new JredisConnectionFactory(new DefaultJredisPool(
|
||||
SettingsUtils.getHost(), SettingsUtils.getPort()));
|
||||
jredisConnFactory.afterPropertiesSet();
|
||||
|
||||
RedisTemplate<String, String> stringTemplateJR = new StringRedisTemplate(jredisConnFactory);
|
||||
|
||||
@@ -22,6 +22,7 @@ import org.junit.runners.Parameterized.Parameters;
|
||||
import org.springframework.data.redis.Person;
|
||||
import org.springframework.data.redis.SettingsUtils;
|
||||
import org.springframework.data.redis.connection.jedis.JedisConnectionFactory;
|
||||
import org.springframework.data.redis.connection.jredis.DefaultJredisPool;
|
||||
import org.springframework.data.redis.connection.jredis.JredisConnectionFactory;
|
||||
import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory;
|
||||
import org.springframework.data.redis.connection.srp.SrpConnectionFactory;
|
||||
@@ -88,9 +89,8 @@ public class RedisMapTests extends AbstractRedisMapTests<Object, Object> {
|
||||
jsonPersonTemplate.afterPropertiesSet();
|
||||
|
||||
// JRedis
|
||||
JredisConnectionFactory jredisConnFactory = new JredisConnectionFactory();
|
||||
jredisConnFactory.setPort(SettingsUtils.getPort());
|
||||
jredisConnFactory.setHostName(SettingsUtils.getHost());
|
||||
JredisConnectionFactory jredisConnFactory = new JredisConnectionFactory(new DefaultJredisPool(
|
||||
SettingsUtils.getHost(), SettingsUtils.getPort()));
|
||||
jredisConnFactory.afterPropertiesSet();
|
||||
|
||||
RedisTemplate genericTemplateJR = new RedisTemplate();
|
||||
|
||||
@@ -38,6 +38,7 @@ import org.junit.runners.Parameterized.Parameters;
|
||||
import org.springframework.data.redis.Person;
|
||||
import org.springframework.data.redis.SettingsUtils;
|
||||
import org.springframework.data.redis.connection.jedis.JedisConnectionFactory;
|
||||
import org.springframework.data.redis.connection.jredis.DefaultJredisPool;
|
||||
import org.springframework.data.redis.connection.jredis.JredisConnectionFactory;
|
||||
import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory;
|
||||
import org.springframework.data.redis.core.RedisTemplate;
|
||||
@@ -258,9 +259,8 @@ public class RedisPropertiesTests extends RedisMapTests {
|
||||
jsonPersonTemplate.afterPropertiesSet();
|
||||
|
||||
// JRedis
|
||||
JredisConnectionFactory jredisConnFactory = new JredisConnectionFactory();
|
||||
jredisConnFactory.setPort(SettingsUtils.getPort());
|
||||
jredisConnFactory.setHostName(SettingsUtils.getHost());
|
||||
JredisConnectionFactory jredisConnFactory = new JredisConnectionFactory(new DefaultJredisPool(
|
||||
SettingsUtils.getHost(), SettingsUtils.getPort()));
|
||||
jredisConnFactory.afterPropertiesSet();
|
||||
|
||||
RedisTemplate<String, String> genericTemplateJR = new StringRedisTemplate(jredisConnFactory);
|
||||
|
||||
Reference in New Issue
Block a user