DATAREDIS-1222 - Fix limit parameter in DefaultStringRedisConnection#zRangeByLex(String,Range,Limit)

Original Pull Request: #564

Author: anshlykov

src/main/java/org/springframework/data/redis/connection/DefaultStringRedisConnection.java

@@ -62,6 +62,7 @@
  * @author Thomas Darimont
  * @author Mark Paluch
  * @author Ninad Divadkar
+ * @author Andrey Shlykov
  */
 public class DefaultStringRedisConnection implements StringRedisConnection, DecoratedRedisConnection {
 
@@ -3601,7 +3602,7 @@ public Set<String> zRangeByLex(String key) {
 	 */
 	@Override
 	public Set<String> zRangeByLex(String key, Range range) {
-		return zRangeByLex(key, range, null);
+		return zRangeByLex(key, range, Limit.unlimited());
 	}
 
 	/*
@@ -3610,7 +3611,7 @@ public Set<String> zRangeByLex(String key, Range range) {
 	 */
 	@Override
 	public Set<String> zRangeByLex(String key, Range range, Limit limit) {
-		return convertAndReturn(delegate.zRangeByLex(serialize(key), range), byteSetToStringSet);
+		return convertAndReturn(delegate.zRangeByLex(serialize(key), range, limit), byteSetToStringSet);
 	}
 
 	/*

src/main/java/org/springframework/data/redis/connection/jedis/JedisZSetCommands.java

@@ -36,6 +36,7 @@
  * @author Christoph Strobl
  * @author Clement Ong
  * @author Mark Paluch
+ * @author Andrey Shlykov
  * @since 2.0
  */
 @RequiredArgsConstructor
@@ -862,7 +863,7 @@ public Set<byte[]> zRangeByLex(byte[] key, Range range, @Nullable Limit limit) {
 
 		try {
 			if (isPipelined()) {
-				if (limit != null) {
+				if (limit != null && !limit.isUnlimited()) {
 					pipeline(connection.newJedisResult(
 							connection.getRequiredPipeline().zrangeByLex(key, min, max, limit.getOffset(), limit.getCount())));
 				} else {
@@ -872,7 +873,7 @@ public Set<byte[]> zRangeByLex(byte[] key, Range range, @Nullable Limit limit) {
 			}
 
 			if (isQueueing()) {
-				if (limit != null) {
+				if (limit != null && !limit.isUnlimited()) {
 					transaction(connection.newJedisResult(
 							connection.getRequiredTransaction().zrangeByLex(key, min, max, limit.getOffset(), limit.getCount())));
 				} else {

src/test/java/org/springframework/data/redis/connection/AbstractConnectionIntegrationTests.java

@@ -61,6 +61,7 @@
 import org.springframework.data.redis.connection.RedisStringCommands.BitOperation;
 import org.springframework.data.redis.connection.RedisStringCommands.SetOption;
 import org.springframework.data.redis.connection.RedisZSetCommands.Aggregate;
+import org.springframework.data.redis.connection.RedisZSetCommands.Limit;
 import org.springframework.data.redis.connection.RedisZSetCommands.Range;
 import org.springframework.data.redis.connection.RedisZSetCommands.Tuple;
 import org.springframework.data.redis.connection.SortParameters.Order;
@@ -93,6 +94,7 @@
  * @author Thomas Darimont
  * @author Mark Paluch
  * @author Dejan Jankov
+ * @author Andrey Shlykov
  */
 @ProfileValueSourceConfiguration(RedisTestProfileValueSource.class)
 public abstract class AbstractConnectionIntegrationTests {
@@ -2298,7 +2300,7 @@ public void pfCountWithNullKeysShouldThrowIllegalArgumentException() {
 	}
 
 	@SuppressWarnings("unchecked")
-	@Test // DATAREDIS-378
+	@Test // DATAREDIS-378, DATAREDIS-1222
 	@IfProfileValue(name = "redisVersion", value = "2.9.0+")
 	@WithRedisDriver({ RedisDriver.JEDIS, RedisDriver.LETTUCE })
 	public void zRangeByLexTest() {
@@ -2316,6 +2318,10 @@ public void zRangeByLexTest() {
 		actual.add(connection.zRangeByLex("myzset", Range.range().gte("aaa").lt("g")));
 		actual.add(connection.zRangeByLex("myzset", Range.range().gte("e")));
 
+		actual.add(connection.zRangeByLex("myzset", Range.range().lte("c"), Limit.unlimited()));
+		actual.add(connection.zRangeByLex("myzset", Range.range().lte("c"), Limit.limit().count(1)));
+		actual.add(connection.zRangeByLex("myzset", Range.range().lte("c"), Limit.limit().count(1).offset(1)));
+
 		List<Object> results = getResults();
 
 		Set<String> values = (Set<String>) results.get(7);
@@ -2334,6 +2340,18 @@ public void zRangeByLexTest() {
 		values = (Set<String>) results.get(10);
 		assertThat(values).contains("e", "f", "g");
 		assertThat(values).doesNotContain("a", "b", "c", "d");
+
+		values = (Set<String>) results.get(11);
+		assertThat(values).contains("a", "b", "c");
+		assertThat(values).doesNotContain("d", "e", "f", "g");
+
+		values = (Set<String>) results.get(12);
+		assertThat(values).contains("a");
+		assertThat(values).doesNotContain("b", "c", "d", "e", "f", "g");
+
+		values = (Set<String>) results.get(13);
+		assertThat(values).contains("b");
+		assertThat(values).doesNotContain("a", "c", "d", "e", "f", "g");
 	}
 
 	@Test(expected = IllegalArgumentException.class) // DATAREDIS-316, DATAREDIS-692