From 32b19a6c7ec8c9d9240dfd132e01105a8ad43f07 Mon Sep 17 00:00:00 2001
From: Nikita Koksharov <nkoksharov@redisson.pro>
Date: Sat, 6 Jan 2024 15:20:28 +0300
Subject: [PATCH] refactoring

---
 .../redisson/pubsub/PubSubConnectionEntry.java    | 15 ++++++++++-----
 .../redisson/pubsub/PublishSubscribeService.java  |  8 --------
 2 files changed, 10 insertions(+), 13 deletions(-)

diff --git a/redisson/src/main/java/org/redisson/pubsub/PubSubConnectionEntry.java b/redisson/src/main/java/org/redisson/pubsub/PubSubConnectionEntry.java
index d99b96107..10d2831c2 100644
--- a/redisson/src/main/java/org/redisson/pubsub/PubSubConnectionEntry.java
+++ b/redisson/src/main/java/org/redisson/pubsub/PubSubConnectionEntry.java
@@ -28,10 +28,7 @@ import org.redisson.connection.ServiceManager;
 import org.redisson.misc.AsyncSemaphore;
 import org.redisson.misc.WrappedLock;
 
-import java.util.EventListener;
-import java.util.LinkedList;
-import java.util.Map;
-import java.util.Queue;
+import java.util.*;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.ConcurrentLinkedQueue;
@@ -58,6 +55,14 @@ public class PubSubConnectionEntry {
     private final ServiceManager serviceManager;
     private final PublishSubscribeService subscribeService;
 
+    private static final Map<PubSubType, PubSubType> SUBSCRIBE2UNSUBSCRIBE = new HashMap<>();
+
+    static {
+        SUBSCRIBE2UNSUBSCRIBE.put(PubSubType.SUBSCRIBE, PubSubType.UNSUBSCRIBE);
+        SUBSCRIBE2UNSUBSCRIBE.put(PubSubType.SSUBSCRIBE, PubSubType.SUNSUBSCRIBE);
+        SUBSCRIBE2UNSUBSCRIBE.put(PubSubType.PSUBSCRIBE, PubSubType.PUNSUBSCRIBE);
+    }
+
     public PubSubConnectionEntry(RedisPubSubConnection conn, ConnectionManager connectionManager) {
         super();
         this.conn = conn;
@@ -170,7 +175,7 @@ public class PubSubConnectionEntry {
         CompletableFuture<PubSubConnectionEntry> pp = new CompletableFuture<>();
         pp.whenComplete((r, e) -> {
             if (e != null) {
-                PubSubType unsubscribeType = PublishSubscribeService.SUBSCRIBE2UNSUBSCRIBE.get(type);
+                PubSubType unsubscribeType = SUBSCRIBE2UNSUBSCRIBE.get(type);
                 CompletableFuture<Codec> f = subscribeService.unsubscribe(channelName, unsubscribeType);
                 f.whenComplete((rr, ee) -> {
                     pm.completeExceptionally(e);
diff --git a/redisson/src/main/java/org/redisson/pubsub/PublishSubscribeService.java b/redisson/src/main/java/org/redisson/pubsub/PublishSubscribeService.java
index a18162aa0..ac6c13673 100644
--- a/redisson/src/main/java/org/redisson/pubsub/PublishSubscribeService.java
+++ b/redisson/src/main/java/org/redisson/pubsub/PublishSubscribeService.java
@@ -115,14 +115,6 @@ public class PublishSubscribeService {
 
     private boolean shardingSupported = false;
 
-    public static final Map<PubSubType, PubSubType> SUBSCRIBE2UNSUBSCRIBE = new HashMap<>();
-
-    static {
-        SUBSCRIBE2UNSUBSCRIBE.put(PubSubType.SUBSCRIBE, PubSubType.UNSUBSCRIBE);
-        SUBSCRIBE2UNSUBSCRIBE.put(PubSubType.SSUBSCRIBE, PubSubType.SUNSUBSCRIBE);
-        SUBSCRIBE2UNSUBSCRIBE.put(PubSubType.PSUBSCRIBE, PubSubType.PUNSUBSCRIBE);
-    }
-
     public PublishSubscribeService(ConnectionManager connectionManager) {
         super();
         this.connectionManager = connectionManager;