Align exceptions when invoking commands #119

Throw RedisException instead of future-transport on disconnected state when reconnect is disabled.
redis · Aug 19, 2015 · 949964f · 949964f
1 parent 5c1aae2
commit 949964f
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 7 deletions.
diff --git a/src/main/java/com/lambdaworks/redis/cluster/ClusterDistributionChannelWriter.java b/src/main/java/com/lambdaworks/redis/cluster/ClusterDistributionChannelWriter.java
@@ -38,6 +38,8 @@ public ClusterDistributionChannelWriter(RedisChannelWriter<K, V> defaultWriter)
     @Override
     public <T> RedisCommand<K, V, T> write(RedisCommand<K, V, T> command) {
 
+        checkArgument(command != null, "command must not be null");
+
         RedisCommand<K, V, T> commandToSend = command;
         CommandArgs<K, V> args = command.getArgs();
         RedisChannelWriter<K, V> channelWriter = null;

diff --git a/src/main/java/com/lambdaworks/redis/protocol/CommandHandler.java b/src/main/java/com/lambdaworks/redis/protocol/CommandHandler.java
@@ -2,6 +2,8 @@
 
 package com.lambdaworks.redis.protocol;
 
+import static com.google.common.base.Preconditions.checkArgument;
+
 import java.nio.channels.ClosedChannelException;
 import java.nio.charset.Charset;
 import java.util.ArrayDeque;
@@ -164,15 +166,15 @@ protected void decode(ChannelHandlerContext ctx, ByteBuf buffer) throws Interrup
     @Override
     public <T> RedisCommand<K, V, T> write(RedisCommand<K, V, T> command) {
 
+        checkArgument(command != null, "command must not be null");
+
         if (lifecycleState == LifecycleState.CLOSED) {
             throw new RedisException("Connection is closed");
         }
 
         if ((channel == null || !isConnected()) && !clientOptions.isAutoReconnect()) {
-            command.setException(new RedisException(
-                    "Connection is in a disconnected state and reconnect is disabled. Commands are not accepted."));
-            command.complete();
-            return command;
+            throw new RedisException(
+                    "Connection is in a disconnected state and reconnect is disabled. Commands are not accepted.");
         }
 
         try {

diff --git a/src/test/java/com/lambdaworks/redis/ClientTest.java b/src/test/java/com/lambdaworks/redis/ClientTest.java
@@ -70,9 +70,9 @@ public void disconnectedConnectionWithoutReconnect() throws Exception {
         connection.quit();
         Thread.sleep(500);
         try {
-            connection.get(key).get();
-        } catch (Exception e) {
-            assertThat(e).hasRootCauseInstanceOf(RedisException.class).hasMessageContaining(
+            connection.get(key);
+        } catch (RedisException e) {
+            assertThat(e).hasMessageContaining(
                     "Connection is in a disconnected state and reconnect is disabled");
         } finally {
             connection.close();