Support issuing Latency commands (#3729)

* add LATENCY LATEST, LATENCY HISTORY and LATENCY RESET commands

* add LatencyEvent class

src/main/java/redis/clients/jedis/BuilderFactory.java

@@ -997,6 +997,53 @@ public Map<String, CommandInfo> build(Object data) {
     }
   };
 
+  public static final Builder<Map<String, LatencyLatestInfo>> LATENCY_LATEST_RESPONSE = new Builder<Map<String, LatencyLatestInfo>>() {
+    @Override
+    public Map<String, LatencyLatestInfo> build(Object data) {
+      if (data == null) {
+        return null;
+      }
+
+      List<Object> rawList = (List<Object>) data;
+      Map<String, LatencyLatestInfo> map = new HashMap<>(rawList.size());
+
+      for (Object rawLatencyLatestInfo : rawList) {
+        if (rawLatencyLatestInfo == null) {
+          continue;
+        }
+
+        LatencyLatestInfo latestInfo = LatencyLatestInfo.LATENCY_LATEST_BUILDER.build(rawLatencyLatestInfo);
+        String name = latestInfo.getCommand();
+        map.put(name, latestInfo);
+      }
+
+      return map;
+    }
+  };
+
+  public static final Builder<List<LatencyHistoryInfo>> LATENCY_HISTORY_RESPONSE = new Builder<List<LatencyHistoryInfo>>() {
+    @Override
+    public List<LatencyHistoryInfo> build(Object data) {
+      if (data == null) {
+        return null;
+      }
+
+      List<Object> rawList = (List<Object>) data;
+      List<LatencyHistoryInfo> response = new ArrayList<>(rawList.size());
+
+      for (Object rawLatencyHistoryInfo : rawList) {
+        if (rawLatencyHistoryInfo == null) {
+          continue;
+        }
+
+        LatencyHistoryInfo historyInfo = LatencyHistoryInfo.LATENCY_HISTORY_BUILDER.build(rawLatencyHistoryInfo);
+        response.add(historyInfo);
+      }
+
+      return response;
+    }
+  };
+
   private static final Builder<List<List<Long>>> CLUSTER_SHARD_SLOTS_RANGES = new Builder<List<List<Long>>>() {
 
     @Override

src/main/java/redis/clients/jedis/Jedis.java

@@ -14,6 +14,7 @@
 import java.util.Map.Entry;
 import java.util.Set;
 import java.util.stream.Collectors;
+import java.util.Arrays;
 
 import javax.net.ssl.HostnameVerifier;
 import javax.net.ssl.SSLParameters;
@@ -9280,6 +9281,26 @@ public String latencyDoctor() {
     return connection.getBulkReply();
   }
 
+  public Map<String, LatencyLatestInfo> latencyLatest() {
+    checkIsInMultiOrPipeline();
+    connection.sendCommand(LATENCY, LATEST);
+    return BuilderFactory.LATENCY_LATEST_RESPONSE.build(connection.getOne());
+  }
+
+  public List<LatencyHistoryInfo> latencyHistory(LatencyEvent event) {
+    checkIsInMultiOrPipeline();
+    connection.sendCommand(new CommandArguments(LATENCY).add(HISTORY).add(event));
+    return BuilderFactory.LATENCY_HISTORY_RESPONSE.build(connection.getOne());
+  }
+
+  public long latencyReset(LatencyEvent... events) {
+    checkIsInMultiOrPipeline();
+    CommandArguments arguments = new CommandArguments(LATENCY).add(Keyword.RESET);
+    Arrays.stream(events).forEach(arguments::add);
+    connection.sendCommand(arguments);
+    return connection.getIntegerReply();
+  }
+
   @Override
   public StreamEntryID xadd(final String key, final StreamEntryID id, final Map<String, String> hash) {
     checkIsInMultiOrPipeline();

src/main/java/redis/clients/jedis/Protocol.java

@@ -299,7 +299,7 @@ public static enum Keyword implements Rawable {
     REV, WITHCOORD, WITHDIST, WITHHASH, ANY, FROMMEMBER, FROMLONLAT, BYRADIUS, BYBOX, BYLEX, BYSCORE,
     STOREDIST, TO, FORCE, TIMEOUT, DB, UNLOAD, ABORT, IDX, MINMATCHLEN, WITHMATCHLEN, FULL,
     DELETE, LIBRARYNAME, WITHCODE, DESCRIPTION, GETKEYS, GETKEYSANDFLAGS, DOCS, FILTERBY, DUMP,
-    MODULE, ACLCAT, PATTERN, DOCTOR, USAGE, SAMPLES, PURGE, STATS, LOADEX, CONFIG, ARGS, RANK,
+    MODULE, ACLCAT, PATTERN, DOCTOR, LATEST, HISTORY, USAGE, SAMPLES, PURGE, STATS, LOADEX, CONFIG, ARGS, RANK,
     NOW, VERSION, ADDR, SKIPME, USER, LADDR,
     CHANNELS, NUMPAT, NUMSUB, SHARDCHANNELS, SHARDNUMSUB;
 

src/main/java/redis/clients/jedis/args/LatencyEvent.java

@@ -0,0 +1,24 @@
+package redis.clients.jedis.args;
+
+import redis.clients.jedis.util.SafeEncoder;
+
+public enum LatencyEvent implements Rawable {
+
+    ACTIVE_DEFRAG_CYCLE("active-defrag-cycle"), AOF_FSYNC_ALWAYS("aof-fsync-always"), AOF_STAT("aof-stat"),
+    AOF_REWRITE_DIFF_WRITE("aof-rewrite-diff-write"), AOF_RENAME("aof-rename"), AOF_WRITE("aof-write"),
+    AOF_WRITE_ACTIVE_CHILD("aof-write-active-child"), AOF_WRITE_ALONE("aof-write-alone"),
+    AOF_WRITE_PENDING_FSYNC("aof-write-pending-fsync"), COMMAND("command"), EXPIRE_CYCLE("expire-cycle"),
+    EVICTION_CYCLE("eviction-cycle"), EVICTION_DEL("eviction-del"), FAST_COMMAND("fast-command"),
+    FORK("fork"), RDB_UNLINK_TEMP_FILE("rdb-unlink-temp-file");
+
+    private final byte[] raw;
+
+    private LatencyEvent(String s) {
+        raw = SafeEncoder.encode(s);
+    }
+
+    @Override
+    public byte[] getRaw() {
+        return raw;
+    }
+}

src/main/java/redis/clients/jedis/commands/ServerCommands.java

@@ -1,12 +1,18 @@
 package redis.clients.jedis.commands;
 
 import redis.clients.jedis.args.FlushMode;
+import redis.clients.jedis.args.LatencyEvent;
 import redis.clients.jedis.args.SaveMode;
 import redis.clients.jedis.exceptions.JedisException;
 import redis.clients.jedis.params.LolwutParams;
 import redis.clients.jedis.params.ShutdownParams;
+import redis.clients.jedis.resps.LatencyHistoryInfo;
+import redis.clients.jedis.resps.LatencyLatestInfo;
 import redis.clients.jedis.util.KeyValue;
 
+import java.util.List;
+import java.util.Map;
+
 public interface ServerCommands {
 
   /**
@@ -246,4 +252,10 @@ default void shutdown(SaveMode saveMode) throws JedisException {
    * @return the report
    */
   String latencyDoctor();
+
+  Map<String, LatencyLatestInfo> latencyLatest();
+
+  List<LatencyHistoryInfo> latencyHistory(LatencyEvent events);
+
+  long latencyReset(LatencyEvent... events);
 }

src/main/java/redis/clients/jedis/resps/LatencyHistoryInfo.java

@@ -0,0 +1,38 @@
+package redis.clients.jedis.resps;
+
+import redis.clients.jedis.Builder;
+
+import java.util.List;
+
+import static redis.clients.jedis.BuilderFactory.LONG;
+
+public class LatencyHistoryInfo {
+
+    private final long timestamp;
+    private final long latency;
+
+    public LatencyHistoryInfo(long timestamp, long latency) {
+        this.timestamp = timestamp;
+        this.latency = latency;
+    }
+
+    public long getTimestamp() {
+        return timestamp;
+    }
+
+    public long getLatency() {
+        return latency;
+    }
+
+    public static final Builder<LatencyHistoryInfo> LATENCY_HISTORY_BUILDER = new Builder<LatencyHistoryInfo>() {
+        @Override
+        public LatencyHistoryInfo build(Object data) {
+            List<Object> commandData = (List<Object>) data;
+
+            long timestamp = LONG.build(commandData.get(0));
+            long latency = LONG.build(commandData.get(1));
+
+            return new LatencyHistoryInfo(timestamp, latency);
+        }
+    };
+}

src/main/java/redis/clients/jedis/resps/LatencyLatestInfo.java

@@ -0,0 +1,53 @@
+package redis.clients.jedis.resps;
+
+import redis.clients.jedis.Builder;
+
+import java.util.List;
+
+import static redis.clients.jedis.BuilderFactory.LONG;
+import static redis.clients.jedis.BuilderFactory.STRING;
+
+public class LatencyLatestInfo {
+
+    private final String command;
+    private final long timestamp;
+    private final long lastEventLatency;
+    private final long maxEventLatency;
+
+    public LatencyLatestInfo(String command, long timestamp, long lastEventLatency, long maxEventLatency) {
+        this.command = command;
+        this.timestamp = timestamp;
+        this.lastEventLatency = lastEventLatency;
+        this.maxEventLatency = maxEventLatency;
+    }
+
+    public String getCommand() {
+        return command;
+    }
+
+    public long getTimestamp() {
+        return timestamp;
+    }
+
+    public long getLastEventLatency() {
+        return lastEventLatency;
+    }
+
+    public long getMaxEventLatency() {
+        return maxEventLatency;
+    }
+
+    public static final Builder<LatencyLatestInfo> LATENCY_LATEST_BUILDER = new Builder<LatencyLatestInfo>() {
+        @Override
+        public LatencyLatestInfo build(Object data) {
+            List<Object> commandData = (List<Object>) data;
+
+            String command = STRING.build(commandData.get(0));
+            long timestamp = LONG.build(commandData.get(1));
+            long lastEventLatency = LONG.build(commandData.get(2));
+            long maxEventLatency = LONG.build(commandData.get(3));
+
+            return new LatencyLatestInfo(command, timestamp, lastEventLatency, maxEventLatency);
+        }
+    };
+}

src/test/java/redis/clients/jedis/commands/jedis/ControlCommandsTest.java

@@ -30,12 +30,15 @@
 import redis.clients.jedis.JedisMonitor;
 import redis.clients.jedis.Protocol;
 import redis.clients.jedis.args.ClientPauseMode;
+import redis.clients.jedis.args.LatencyEvent;
 import redis.clients.jedis.exceptions.JedisDataException;
 import redis.clients.jedis.HostAndPorts;
 import redis.clients.jedis.params.CommandListFilterByParams;
 import redis.clients.jedis.params.LolwutParams;
 import redis.clients.jedis.resps.CommandDocument;
 import redis.clients.jedis.resps.CommandInfo;
+import redis.clients.jedis.resps.LatencyHistoryInfo;
+import redis.clients.jedis.resps.LatencyLatestInfo;
 import redis.clients.jedis.util.AssertUtil;
 import redis.clients.jedis.util.KeyValue;
 import redis.clients.jedis.util.SafeEncoder;
@@ -440,6 +443,23 @@ public void latencyDoctor() {
     assertNotNull(report);
   }
 
+  @Test
+  public void latencyLatest() {
+    Map<String, LatencyLatestInfo> report = jedis.latencyLatest();
+    assertNotNull(report);
+  }
+
+  @Test
+  public void latencyHistoryFork() {
+    List<LatencyHistoryInfo> report = jedis.latencyHistory(LatencyEvent.FORK);
+    assertNotNull(report);
+  }
+
+  @Test
+  public void latencyReset() {
+    assertTrue(jedis.latencyReset() >= 0);
+  }
+
   @Test
   public void commandCount() {
     assertTrue(jedis.commandCount() > 100);