feat: Ops for T Digest

Author: raphaeldelio

redis-om-spring/src/main/java/com/redis/om/spring/client/RedisModulesClient.java

@@ -17,6 +17,7 @@
 import redis.clients.jedis.bloom.commands.TopKFilterCommands;
 import redis.clients.jedis.json.commands.RedisJsonCommands;
 import redis.clients.jedis.search.RediSearchCommands;
+import redis.clients.jedis.bloom.commands.TDigestSketchCommands;
 
 import java.util.Objects;
 import java.util.Optional;
@@ -62,6 +63,10 @@ public TopKFilterCommands clientForTopK() {
     return unifiedJedis;
   }
 
+  public TDigestSketchCommands clientForTDigest() {
+    return unifiedJedis;
+  }
+
   private UnifiedJedis getUnifiedJedis() {
 
     var sentinelConfiguration = jedisConnectionFactory.getSentinelConfiguration();

redis-om-spring/src/main/java/com/redis/om/spring/ops/pds/TDigestOperations.java

@@ -0,0 +1,149 @@
+package com.redis.om.spring.ops.pds;
+
+import redis.clients.jedis.bloom.TDigestMergeParams;
+
+import java.util.List;
+import java.util.Map;
+
+public interface TDigestOperations<K> {
+  /**
+   * Create a new T-Digest sketch.
+   *
+   * @param key The key of the sketch
+   * @return Status string reply
+   */
+  String create(K key);
+
+  /**
+   * Create a new T-Digest sketch with specified compression.
+   *
+   * @param key         The key of the sketch
+   * @param compression The compression parameter
+   * @return Status string reply
+   */
+  String create(K key, int compression);
+
+  /**
+   * Reset the sketch.
+   *
+   * @param key The key of the sketch
+   * @return Status string reply
+   */
+  String reset(K key);
+
+  /**
+   * Merge multiple sketches into one.
+   *
+   * @param key       The key of the destination sketch
+   * @param sourceKeys The keys of the source sketches
+   * @return Status string reply
+   */
+  String merge(K key, K... sourceKeys);
+
+  /**
+   * Merge multiple sketches into one with parameters.
+   *
+   * @param params     The merge parameters
+   * @param key        The key of the destination sketch
+   * @param sourceKeys The keys of the source sketches
+   * @return Status string reply
+   */
+  String merge(TDigestMergeParams params, K key, K... sourceKeys);
+
+  /**
+   * Get information about the sketch.
+   *
+   * @param key The key of the sketch
+   * @return Map of information about the sketch
+   */
+  Map<String, Object> info(K key);
+
+  /**
+   * Add values to the sketch.
+   *
+   * @param key    The key of the sketch
+   * @param values The values to add
+   * @return Status string reply
+   */
+  String add(K key, double... values);
+
+  /**
+   * Get the CDF (Cumulative Distribution Function) for values.
+   *
+   * @param key    The key of the sketch
+   * @param values The values to get CDF for
+   * @return List of CDF values
+   */
+  List<Double> cdf(K key, double... values);
+
+  /**
+   * Get the quantile for values.
+   *
+   * @param key    The key of the sketch
+   * @param values The values to get quantile for
+   * @return List of quantile values
+   */
+  List<Double> quantile(K key, double... values);
+
+  /**
+   * Get the minimum value in the sketch.
+   *
+   * @param key The key of the sketch
+   * @return The minimum value
+   */
+  double min(K key);
+
+  /**
+   * Get the maximum value in the sketch.
+   *
+   * @param key The key of the sketch
+   * @return The maximum value
+   */
+  double max(K key);
+
+  /**
+   * Get the trimmed mean of the sketch.
+   *
+   * @param key     The key of the sketch
+   * @param lowCut  The low cut quantile
+   * @param highCut The high cut quantile
+   * @return The trimmed mean
+   */
+  double trimmedMean(K key, double lowCut, double highCut);
+
+  /**
+   * Get the rank of values.
+   *
+   * @param key    The key of the sketch
+   * @param values The values to get rank for
+   * @return List of ranks
+   */
+  List<Long> rank(K key, double... values);
+
+  /**
+   * Get the reverse rank of values.
+   *
+   * @param key    The key of the sketch
+   * @param values The values to get reverse rank for
+   * @return List of reverse ranks
+   */
+  List<Long> revRank(K key, double... values);
+
+  /**
+   * Get the value by rank.
+   *
+   * @param key   The key of the sketch
+   * @param ranks The ranks to get values for
+   * @return List of values
+   */
+  List<Double> byRank(K key, long... ranks);
+
+  /**
+   * Get the value by reverse rank.
+   *
+   * @param key   The key of the sketch
+   * @param ranks The reverse ranks to get values for
+   * @return List of values
+   */
+  List<Double> byRevRank(K key, long... ranks);
+} 

redis-om-spring/src/main/java/com/redis/om/spring/ops/pds/TDigestOperationsImpl.java

@@ -0,0 +1,103 @@
+package com.redis.om.spring.ops.pds;
+
+import com.redis.om.spring.client.RedisModulesClient;
+import redis.clients.jedis.bloom.TDigestMergeParams;
+
+import java.util.List;
+import java.util.Map;
+
+public class TDigestOperationsImpl<K> implements TDigestOperations<K> {
+  final RedisModulesClient client;
+
+  public TDigestOperationsImpl(RedisModulesClient client) {
+    this.client = client;
+  }
+
+  @Override
+  public String create(K key) {
+    return client.clientForTDigest().tdigestCreate(key.toString());
+  }
+
+  @Override
+  public String create(K key, int compression) {
+    return client.clientForTDigest().tdigestCreate(key.toString(), compression);
+  }
+
+  @Override
+  public String reset(K key) {
+    return client.clientForTDigest().tdigestReset(key.toString());
+  }
+
+  @Override
+  public String merge(K key, K... sourceKeys) {
+    String[] sourceKeyStrings = new String[sourceKeys.length];
+    for (int i = 0; i < sourceKeys.length; i++) {
+      sourceKeyStrings[i] = sourceKeys[i].toString();
+    }
+    return client.clientForTDigest().tdigestMerge(key.toString(), sourceKeyStrings);
+  }
+
+  @Override
+  public String merge(TDigestMergeParams params, K key, K... sourceKeys) {
+    String[] sourceKeyStrings = new String[sourceKeys.length];
+    for (int i = 0; i < sourceKeys.length; i++) {
+      sourceKeyStrings[i] = sourceKeys[i].toString();
+    }
+    return client.clientForTDigest().tdigestMerge(params, key.toString(), sourceKeyStrings);
+  }
+
+  @Override
+  public Map<String, Object> info(K key) {
+    return client.clientForTDigest().tdigestInfo(key.toString());
+  }
+
+  @Override
+  public String add(K key, double... values) {
+    return client.clientForTDigest().tdigestAdd(key.toString(), values);
+  }
+
+  @Override
+  public List<Double> cdf(K key, double... values) {
+    return client.clientForTDigest().tdigestCDF(key.toString(), values);
+  }
+
+  @Override
+  public List<Double> quantile(K key, double... values) {
+    return client.clientForTDigest().tdigestQuantile(key.toString(), values);
+  }
+
+  @Override
+  public double min(K key) {
+    return client.clientForTDigest().tdigestMin(key.toString());
+  }
+
+  @Override
+  public double max(K key) {
+    return client.clientForTDigest().tdigestMax(key.toString());
+  }
+
+  @Override
+  public double trimmedMean(K key, double lowCut, double highCut) {
+    return client.clientForTDigest().tdigestTrimmedMean(key.toString(), lowCut, highCut);
+  }
+
+  @Override
+  public List<Long> rank(K key, double... values) {
+    return client.clientForTDigest().tdigestRank(key.toString(), values);
+  }
+
+  @Override
+  public List<Long> revRank(K key, double... values) {
+    return client.clientForTDigest().tdigestRevRank(key.toString(), values);
+  }
+
+  @Override
+  public List<Double> byRank(K key, long... ranks) {
+    return client.clientForTDigest().tdigestByRank(key.toString(), ranks);
+  }
+
+  @Override
+  public List<Double> byRevRank(K key, long... ranks) {
+    return client.clientForTDigest().tdigestByRevRank(key.toString(), ranks);
+  }
+}