Update jedis sync client with thread pooling (#86)

* Add jedis threadpool for standalone client Signed-off-by: Andrew Carbonetto <[email protected]> --------- Signed-off-by: Andrew Carbonetto <[email protected]>
Bit-Quill · Feb 16, 2024 · 82be845 · 82be845
1 parent 26e7a0b
commit 82be845
Showing 1 changed file with 29 additions and 14 deletions.
diff --git a/java/benchmarks/src/main/java/glide/benchmarks/clients/jedis/JedisClient.java b/java/benchmarks/src/main/java/glide/benchmarks/clients/jedis/JedisClient.java
@@ -6,18 +6,24 @@
 import java.util.Set;
 import redis.clients.jedis.DefaultJedisClientConfig;
 import redis.clients.jedis.HostAndPort;
+import redis.clients.jedis.Jedis;
 import redis.clients.jedis.JedisCluster;
 import redis.clients.jedis.JedisPool;
-import redis.clients.jedis.commands.JedisCommands;
 
 /** A Jedis client with sync capabilities. See: https://github.com/redis/jedis */
 public class JedisClient implements SyncClient {
-
-    private JedisCommands jedis;
+    boolean isClusterMode;
+    private JedisPool jedisPool;
+    private JedisCluster jedisCluster;
 
     @Override
     public void closeConnection() {
-        // nothing to do
+        if (jedisCluster != null) {
+            jedisCluster.close();
+        }
+        if (jedisPool != null) {
+            jedisPool.close();
+        }
     }
 
     @Override
@@ -27,27 +33,36 @@ public String getName() {
 
     @Override
     public void connectToRedis(ConnectionSettings connectionSettings) {
-        if (connectionSettings.clusterMode) {
-            jedis =
+        isClusterMode = connectionSettings.clusterMode;
+        if (isClusterMode) {
+            jedisCluster =
                     new JedisCluster(
                             Set.of(new HostAndPort(connectionSettings.host, connectionSettings.port)),
                             DefaultJedisClientConfig.builder().ssl(connectionSettings.useSsl).build());
-        } else {
-            try (JedisPool pool =
-                    new JedisPool(
-                            connectionSettings.host, connectionSettings.port, connectionSettings.useSsl)) {
-                jedis = pool.getResource();
-            }
         }
+        jedisPool =
+                new JedisPool(connectionSettings.host, connectionSettings.port, connectionSettings.useSsl);
     }
 
     @Override
     public void set(String key, String value) {
-        jedis.set(key, value);
+        if (isClusterMode) {
+            jedisCluster.set(key, value);
+        } else {
+            try (Jedis jedis = jedisPool.getResource()) {
+                jedis.set(key, value);
+            }
+        }
     }
 
     @Override
     public String get(String key) {
-        return jedis.get(key);
+        if (isClusterMode) {
+            return jedisCluster.get(key);
+        } else {
+            try (Jedis jedis = jedisPool.getResource()) {
+                return jedis.get(key);
+            }
+        }
     }
 }