Support GROUPBY and REDUCE options

Author: sazzad16

src/main/java/com/redislabs/redistimeseries/Keyword.java

@@ -15,7 +15,9 @@ public enum Keyword implements ProtocolCommand {
   UNCOMPRESSED,
   CHUNK_SIZE,
   DUPLICATE_POLICY,
-  ON_DUPLICATE;
+  ON_DUPLICATE,
+  GROUPBY,
+  REDUCE;
 
   private final byte[] raw;
 

src/main/java/com/redislabs/redistimeseries/MultiRangeParams.java

@@ -14,6 +14,8 @@ public class MultiRangeParams {
 
   private boolean withLabels;
 
+  private String[] groupByReduce;
+
   public static MultiRangeParams multiRangeParams() {
     return new MultiRangeParams();
   }
@@ -41,6 +43,11 @@ public MultiRangeParams withLabels(boolean withLabels) {
     return this;
   }
 
+  public MultiRangeParams groupByReduce(String group, String reduce) {
+    this.groupByReduce = new String[] {group, reduce};
+    return this;
+  }
+
   public byte[][] getByteParams(long from, long to, String... filters) {
     List<byte[]> params = new ArrayList<>();
     params.add(Protocol.toByteArray(from));
@@ -66,6 +73,13 @@ public byte[][] getByteParams(long from, long to, String... filters) {
       params.add(SafeEncoder.encode(filter));
     }
 
+    if (groupByReduce != null) {
+      params.add(Keyword.GROUPBY.getRaw());
+      params.add(SafeEncoder.encode(groupByReduce[0]));
+      params.add(Keyword.REDUCE.getRaw());
+      params.add(SafeEncoder.encode(groupByReduce[1]));
+    }
+
     return params.toArray(new byte[params.size()][]);
   }
 }

src/test/java/com/redislabs/redistimeseries/RedisTimeSeriesTest.java

@@ -1,5 +1,6 @@
 package com.redislabs.redistimeseries;
 
+import static org.junit.Assert.assertArrayEquals;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.fail;
@@ -409,6 +410,44 @@ public void testIncrByDecrBy() throws InterruptedException {
     }
   }
 
+  @Test
+  public void groupByReduce() {
+    client.create("ts1", convertMap("metric", "cpu", "metric_name", "system"));
+    client.create("ts2", convertMap("metric", "cpu", "metric_name", "user"));
+
+    client.add("ts1", 1L, 90.0);
+    client.add("ts1", 2L, 45.0);
+    client.add("ts2", 2L, 99.0);
+
+    Range[] range =
+        client.mrange(
+            0L,
+            100L,
+            MultiRangeParams.multiRangeParams().withLabels().groupByReduce("metric_name", "max"),
+            "metric=cpu");
+    assertEquals(2, range.length);
+
+    assertEquals("metric_name=system", range[0].getKey());
+    assertEquals("system", range[0].getLabels().get("metric_name"));
+    assertEquals("max", range[0].getLabels().get("__reducer__"));
+    assertEquals("ts1", range[0].getLabels().get("__source__"));
+    assertArrayEquals(new Value[] {new Value(1, 90), new Value(2, 45)}, range[0].getValues());
+
+    assertEquals("metric_name=user", range[1].getKey());
+    assertEquals("user", range[1].getLabels().get("metric_name"));
+    assertEquals("max", range[1].getLabels().get("__reducer__"));
+    assertEquals("ts2", range[1].getLabels().get("__source__"));
+    assertArrayEquals(new Value[] {new Value(2, 99)}, range[1].getValues());
+  }
+
+  private Map<String, String> convertMap(String... array) {
+    Map<String, String> map = new HashMap<>(array.length / 2);
+    for (int i = 0; i < array.length; i += 2) {
+      map.put(array[i], array[i + 1]);
+    }
+    return map;
+  }
+
   @Test
   public void testGet() {