update broadcast manager

alibaba · Aug 23, 2024 · 5fc0e0a · 5fc0e0a
1 parent 9f791ae
commit 5fc0e0a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/...redis-lettuce/src/main/java/com/alicp/jetcache/redis/lettuce/LettuceBroadcastManager.java b/...redis-lettuce/src/main/java/com/alicp/jetcache/redis/lettuce/LettuceBroadcastManager.java
@@ -51,6 +51,7 @@ public LettuceBroadcastManager(CacheManager cacheManager, RedisLettuceCacheConfi
         this.channel = config.getBroadcastChannel().getBytes(StandardCharsets.UTF_8);
         this.lettuceConnectionManager = config.getConnectionManager();
         this.lettuceConnectionManager.init(config.getRedisClient(), config.getConnection());
+        this.lettuceConnectionManager.init(config.getRedisClient(), config.getPubSubConnection());
         this.stringAsyncCommands = (BaseRedisAsyncCommands<byte[], byte[]>) lettuceConnectionManager
                 .asyncCommands(config.getRedisClient());
     }

diff --git a/...edis-lettuce/src/main/java/com/alicp/jetcache/redis/lettuce/LettuceConnectionManager.java b/...edis-lettuce/src/main/java/com/alicp/jetcache/redis/lettuce/LettuceConnectionManager.java
@@ -49,11 +49,13 @@ private LettuceObjects getLettuceObjectsFromMap(AbstractRedisClient redisClient)
     }
 
     public void init(AbstractRedisClient redisClient, StatefulConnection connection) {
-        map.computeIfAbsent(redisClient, key -> {
-            LettuceObjects lo = new LettuceObjects();
-            lo.connection = connection;
-            return lo;
-        });
+        LettuceObjects lettuceObjects = map.computeIfAbsent(redisClient, key -> new LettuceObjects());
+        lettuceObjects.connection = connection;
+    }
+
+    public void init(AbstractRedisClient redisClient, StatefulRedisPubSubConnection pubSubConnection) {
+        LettuceObjects lettuceObjects = map.computeIfAbsent(redisClient, key -> new LettuceObjects());
+        lettuceObjects.pubSubConnection = pubSubConnection;
     }
 
     public StatefulConnection connection(AbstractRedisClient redisClient) {