Merge pull request #97 from setronica-dev/fastRemove-args

RedissonMap.fastRemove and getAll behavior corrections

src/main/java/org/redisson/RedissonMap.java

@@ -19,13 +19,7 @@ import io.netty.util.concurrent.Future;
 import io.netty.util.concurrent.Promise;
 
 import java.math.BigDecimal;
-import java.util.Collection;
+import java.util.*;
-import java.util.HashMap;
-import java.util.Iterator;
-import java.util.List;
-import java.util.Map;
-import java.util.NoSuchElementException;
-import java.util.Set;
 
 import org.redisson.async.AsyncOperation;
 import org.redisson.async.OperationListener;
@@ -93,6 +87,9 @@ public class RedissonMap<K, V> extends RedissonExpirable implements RMap<K, V> {
 
     @Override
     public Map<K, V> getAll(final Set<K> keys) {
+        if (keys.size() == 0) {
+            return Collections.emptyMap();
+        }
         List<V> list = connectionManager.read(new ResultOperation<List<V>, V>() {
             @Override
             protected Future<List<V>> execute(RedisAsyncConnection<Object, V> async) {
@@ -104,6 +101,9 @@ public class RedissonMap<K, V> extends RedissonExpirable implements RMap<K, V> {
         int index = 0;
         for (K key : keys) {
             V value = list.get(index);
+            if (value == null) {
+                continue;
+            }
             result.put(key, value);
             index++;
         }
@@ -442,12 +442,16 @@ public class RedissonMap<K, V> extends RedissonExpirable implements RMap<K, V> {
 
     @Override
     public Future<Long> fastRemoveAsync(final K ... keys) {
-        return connectionManager.writeAsync(new ResultOperation<Long, V>() {
+        if (keys != null && keys.length > 0) {
-            @Override
+            return connectionManager.writeAsync(new ResultOperation<Long, V>() {
-            public Future<Long> execute(RedisAsyncConnection<Object, V> async) {
+                @Override
-                return async.hdel(getName(), keys);
+                public Future<Long> execute(RedisAsyncConnection<Object, V> async) {
-            }
+                    return async.hdel(getName(), keys);
-        });
+                }
+            });
+        } else {
+            return connectionManager.getGroup().next().newSucceededFuture(0L);
+        }
     }
 
     @Override

src/test/java/org/redisson/RedissonMapTest.java

@@ -136,7 +136,7 @@ public class RedissonMapTest extends BaseTest {
         map.put(3, 300);
         map.put(4, 400);
 
-        Map<Integer, Integer> filtered = map.getAll(new HashSet<Integer>(Arrays.asList(2, 3)));
+        Map<Integer, Integer> filtered = map.getAll(new HashSet<Integer>(Arrays.asList(2, 3, 5)));
 
         Map<Integer, Integer> expectedMap = new HashMap<Integer, Integer>();
         expectedMap.put(2, 200);
@@ -431,4 +431,11 @@ public class RedissonMapTest extends BaseTest {
         Assert.assertEquals(1, map.size());
     }
 
+    @Test
+    public void testFastRemoveEmpty() throws Exception {
+        RMap<Integer, Integer> map = redisson.getMap("simple");
+        map.put(1, 3);
+        Assert.assertEquals(0, map.fastRemove());
+        Assert.assertEquals(1, map.size());
+    }
 }