Cleanup MULTI state if MULTI command fails #1650

We now clear the MULTI state on a best-effort basis when the MULTI command fails (Redis error, timeout, …).
redis · Mar 11, 2021 · 8410593 · 8410593
1 parent 00c4f5d
commit 8410593
Showing 1 changed file with 18 additions and 6 deletions.
diff --git a/src/main/java/io/lettuce/core/StatefulRedisConnectionImpl.java b/src/main/java/io/lettuce/core/StatefulRedisConnectionImpl.java
@@ -158,9 +158,7 @@ public <T> RedisCommand<K, V, T> dispatch(RedisCommand<K, V, T> command) {
 
         RedisCommand<K, V, T> toSend = preProcessCommand(command);
 
-        if (command.getType().name().equals(MULTI.name())) {
-            multi = (multi == null ? new MultiOutput<>(codec) : multi);
-        }
+        potentiallyEnableMulti(command);
 
         return super.dispatch(toSend);
     }
@@ -174,14 +172,28 @@ public <T> RedisCommand<K, V, T> dispatch(RedisCommand<K, V, T> command) {
             RedisCommand<K, V, ?> command = preProcessCommand(o);
 
             sentCommands.add(command);
-            if (command.getType().name().equals(MULTI.name())) {
-                multi = (multi == null ? new MultiOutput<>(codec) : multi);
-            }
+            potentiallyEnableMulti(command);
         });
 
         return super.dispatch(sentCommands);
     }
 
+    private void potentiallyEnableMulti(RedisCommand<K, V, ?> command) {
+
+        if (command.getType().name().equals(MULTI.name())) {
+
+            multi = (multi == null ? new MultiOutput<>(codec) : multi);
+
+            if (command instanceof CompleteableCommand) {
+                ((CompleteableCommand<?>) command).onComplete((ignored, e) -> {
+                    if (e != null) {
+                        multi = null;
+                    }
+                });
+            }
+        }
+    }
+
     protected <T> RedisCommand<K, V, T> preProcessCommand(RedisCommand<K, V, T> command) {
 
         RedisCommand<K, V, T> local = command;