Check result returned by inc/dec protocols is number

hito-asa · Jul 22, 2011 · 72d7d9b · 72d7d9b
1 parent 3a71abb
commit 72d7d9b
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/src/main/java/net/rubyeye/xmemcached/command/text/TextIncrDecrCommand.java b/src/main/java/net/rubyeye/xmemcached/command/text/TextIncrDecrCommand.java
@@ -72,9 +72,14 @@ public final boolean decode(MemcachedTCPSession session, ByteBuffer buffer) {
 				countDownLatch();
 				return true;
 			} else {
-				setResult(Long.valueOf(line.trim()));
-				countDownLatch();
-				return true;
+				String rt = line.trim();
+				if (Character.isDigit(rt.charAt(0))) {
+					setResult(Long.valueOf(rt));
+					countDownLatch();
+					return true;
+				} else {
+					return decodeError(line);
+				}
 			}
 		}
 		return false;
@@ -91,11 +96,11 @@ public final void encode() {
 		}
 		byte[] buf = new byte[size];
 		if (isNoreply()) {
-			ByteUtils.setArguments(buf, 0, cmdBytes, this.keyBytes, this
-					.getDelta(), Constants.NO_REPLY);
+			ByteUtils.setArguments(buf, 0, cmdBytes, this.keyBytes,
+					this.getDelta(), Constants.NO_REPLY);
 		} else {
-			ByteUtils.setArguments(buf, 0, cmdBytes, this.keyBytes, this
-					.getDelta());
+			ByteUtils.setArguments(buf, 0, cmdBytes, this.keyBytes,
+					this.getDelta());
 		}
 		this.ioBuffer = IoBuffer.wrap(buf);
 	}