DATAREDIS-286 - Avoid overflow in JedisConnection#expire / #pexpire.

Workaround Jedis issue https://github.com/xetorthio/jedis/pull/575.
We now delegate calls to expire(...) to pExpireAt(...) in case the seconds parameter is > than Integer.MAX_VALUE.
We query the server time to compute the new expiration date.

Original pull request: #52.
This commit is contained in:
Thomas Darimont
2014-03-21 18:03:08 +01:00
committed by Christoph Strobl
parent f8806c8cfc
commit 2806567acc
2 changed files with 57 additions and 0 deletions

View File

@@ -26,6 +26,7 @@ import java.util.Map;
import java.util.Properties;
import java.util.Queue;
import java.util.Set;
import java.util.concurrent.TimeUnit;
import org.springframework.core.convert.converter.Converter;
import org.springframework.dao.DataAccessException;
@@ -704,6 +705,18 @@ public class JedisConnection implements RedisConnection {
}
public Boolean expire(byte[] key, long seconds) {
/*
* @see DATAREDIS-286 to avoid overflow in Jedis
*
* TODO Remove this workaround when we upgrade to a Jedis version that contains a
* fix for: https://github.com/xetorthio/jedis/pull/575
*/
if (seconds > Integer.MAX_VALUE) {
return pExpireAt(key, time() + TimeUnit.SECONDS.toMillis(seconds));
}
try {
if (isPipelined()) {
pipeline(new JedisResult(pipeline.expire(key, (int) seconds), JedisConverters.longToBoolean()));
@@ -879,6 +892,18 @@ public class JedisConnection implements RedisConnection {
}
public Boolean pExpire(byte[] key, long millis) {
/*
* @see DATAREDIS-286 to avoid overflow in Jedis
*
* TODO Remove this workaround when we upgrade to a Jedis version that contains a
* fix for: https://github.com/xetorthio/jedis/pull/575
*/
if (millis > Integer.MAX_VALUE) {
return pExpireAt(key, time() + millis);
}
try {
if (isPipelined()) {
pipeline(new JedisResult(pipeline.pexpire(key, (int) millis), JedisConverters.longToBoolean()));

View File

@@ -16,6 +16,7 @@
package org.springframework.data.redis.connection.jedis;
import static org.hamcrest.CoreMatchers.*;
import static org.junit.Assert.*;
import java.util.HashSet;
@@ -333,4 +334,35 @@ public class JedisConnectionIntegrationTests extends AbstractConnectionIntegrati
AllOf.allOf(IsInstanceOf.instanceOf(List.class), IsCollectionContaining.hasItems("awesome".getBytes(),
"cool".getBytes(), "supercalifragilisticexpialidocious".getBytes())));
}
/**
* @see DATAREDIS-286
*/
@Test
public void expireShouldSupportExiprationForValuesLargerThanInteger() {
connection.set("expireKey", "foo");
long seconds = ((long) Integer.MAX_VALUE) + 1;
connection.expire("expireKey", seconds);
long ttl = connection.ttl("expireKey");
assertThat(ttl, is(seconds));
}
/**
* @see DATAREDIS-286
*/
@Test
public void pExpireShouldSupportExiprationForValuesLargerThanInteger() {
connection.set("pexpireKey", "foo");
long millis = ((long) Integer.MAX_VALUE) + 10;
connection.pExpire("pexpireKey", millis);
long ttl = connection.pTtl("pexpireKey");
assertTrue(String.format("difference between millis=%s and ttl=%s should not be greater than 20ms but is %s",
millis, ttl, millis - ttl), millis - ttl < 20L);
}
}