perf: spread initialisation load across cluster (#239)

redis-rb · Aug 14, 2023 · d9fe424 · d9fe424
2 parents c22ca36 + 5e3d396
commit d9fe424
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/lib/redis_client/cluster/node.rb b/lib/redis_client/cluster/node.rb
@@ -201,6 +201,10 @@ def each(&block)
         @topology.clients.each_value(&block)
       end
 
+      def shuffled_nodes
+        @topology.clients.values.shuffle
+      end
+
       def sample
         @topology.clients.values.sample
       end

diff --git a/lib/redis_client/cluster/router.rb b/lib/redis_client/cluster/router.rb
@@ -23,7 +23,7 @@ def initialize(config, pool: nil, **kwargs)
         @pool = pool
         @client_kwargs = kwargs
         @node = fetch_cluster_info(@config, pool: @pool, **@client_kwargs)
-        @command = ::RedisClient::Cluster::Command.load(@node)
+        @command = ::RedisClient::Cluster::Command.load(@node.shuffled_nodes)
         @mutex = Mutex.new
         @command_builder = @config.command_builder
       end