Use command flags

Author: uglide

src/main/java/redis/clients/jedis/executors/ClusterCommandExecutor.java

@@ -102,11 +102,18 @@ public final <T> T executeKeylessCommand(CommandObject<T> commandObject) {
     int consecutiveConnectionFailures = 0;
     Exception lastException = null;
 
+    RequiredConnectionType connectionType;
+    if (commandObject.getFlags().contains(CommandObject.CommandFlag.READONLY)) {
+        connectionType = RequiredConnectionType.REPLICA;
+    } else {
+        connectionType = RequiredConnectionType.PRIMARY;
+    }
+
     for (int attemptsLeft = this.maxAttempts; attemptsLeft > 0; attemptsLeft--) {
       Connection connection = null;
       try {
         // Use round-robin distribution for keyless commands
-        connection = getNextConnection();
+        connection = getNextConnection(connectionType);
         return execute(connection, commandObject);
 
       } catch (JedisConnectionException jce) {
@@ -209,23 +216,20 @@ private <T> T doExecuteCommand(CommandObject<T> commandObject, boolean toReplica
     throw maxAttemptsException;
   }
 
+    private enum RequiredConnectionType {
+        PRIMARY,
+        REPLICA
+    }
+  
   /**
    * Gets a connection using round-robin distribution across all cluster nodes.
    * This ensures even distribution of keyless commands across the cluster.
    *
    * @return Connection from the next node in round-robin sequence
    * @throws JedisClusterOperationException if no cluster nodes are available
    */
-  private Connection getNextConnection() {
-    Map<String, ConnectionPool> connectionMap = provider.getConnectionMap();
-
-    if (connectionMap.isEmpty()) {
-      throw new JedisClusterOperationException("No cluster nodes available.");
-    }
-
-    // Convert connection map to list for round-robin access
-    List<Map.Entry<String, ConnectionPool>> nodeList = new ArrayList<>(connectionMap.entrySet());
-
+  private Connection getNextConnection(RequiredConnectionType connectionType) {
+      List<Map.Entry<String, ConnectionPool>> nodeList = selectNextConnectionPool(connectionType);
     // Select node using round-robin distribution for true unified distribution
     // Use modulo directly on the node list size to create a circular counter
     int roundRobinIndex = roundRobinCounter.getAndUpdate(current -> (current + 1) % nodeList.size());
@@ -235,7 +239,25 @@ private Connection getNextConnection() {
     return pool.getResource();
   }
 
-  /**
+    private List<Map.Entry<String, ConnectionPool>> selectNextConnectionPool(RequiredConnectionType connectionType) {
+        Map<String, ConnectionPool> connectionMap;
+
+        // NOTE(imalinovskyi): If we need to connect to replica, we use all nodes, otherwise we use only primary nodes
+        if (connectionType == RequiredConnectionType.REPLICA) {
+            connectionMap = provider.getConnectionMap();
+        } else {
+            connectionMap = provider.getPrimaryNodesConnectionMap();
+        }
+
+        if (connectionMap.isEmpty()) {
+          throw new JedisClusterOperationException("No cluster nodes available.");
+        }
+
+        // Convert connection map to list for round-robin access
+        return new ArrayList<>(connectionMap.entrySet());
+    }
+
+    /**
    * WARNING: This method is accessible for the purpose of testing.
    * This should not be used or overriden.
    */