Fix NPE when manually flushing a batch #2444

When manually flushing a batch with a set size, we could potentially run into a NPE if the command queue is empty when flush() is called.

Author: luciopaiva

src/main/java/io/lettuce/core/dynamic/SimpleBatcher.java

@@ -35,6 +35,7 @@
  * </ul>
  *
  * @author Mark Paluch
+ * @author Lucio Paiva
  */
 class SimpleBatcher implements Batcher {
 
@@ -148,12 +149,12 @@ private List<RedisCommand<Object, Object, Object>> prepareForceFlush() {
 
         List<RedisCommand<Object, Object, Object>> batch = new ArrayList<>(Math.max(batchSize, 10));
 
-        do {
+        while (!queue.isEmpty()) {
+
             RedisCommand<Object, Object, Object> poll = queue.poll();
 
-            assert poll != null;
             batch.add(poll);
-        } while (!queue.isEmpty());
+        }
 
         return batch;
     }
@@ -166,7 +167,6 @@ private List<RedisCommand<Object, Object, Object>> prepareDefaultFlush(int consu
 
             RedisCommand<Object, Object, Object> poll = queue.poll();
 
-            assert poll != null;
             batch.add(poll);
         }
 

src/test/java/io/lettuce/core/dynamic/RedisCommandsBatchingIntegrationTests.java

@@ -42,6 +42,7 @@
 
 /**
  * @author Mark Paluch
+ * @author Lucio Paiva
  */
 @ExtendWith(LettuceExtension.class)
 class RedisCommandsBatchingIntegrationTests extends TestSupport {
@@ -99,6 +100,16 @@ void selectiveBatchingShouldHandleErrors() {
         }
     }
 
+    @Test
+    void shouldNotCrashWhenFlushCalledWithEmptyQueue() {
+
+        RedisCommandFactory factory = new RedisCommandFactory(redis.getStatefulConnection());
+
+        SelectiveBatchingWithSize api = factory.getCommands(SelectiveBatchingWithSize.class);
+
+        api.flush();
+    }
+
     @Test
     void shouldExecuteBatchingSynchronously() {
 
@@ -215,4 +226,9 @@ static interface SelectiveBatching extends Commands, BatchExecutor {
 
     }
 
+    @BatchSize(5)
+    interface SelectiveBatchingWithSize extends Commands, BatchExecutor {
+
+    }
+
 }