redis-cluster-client 0.7.8 → 0.7.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 41d87ce45a108ff8ec7d3c17805276d5ba7df11d577de3f7d6b1176bdd297ed0
-  data.tar.gz: 7118539907bb755225ba908cf160ad8af73aeb5204820682ca3fc3a67813f3cb
+  metadata.gz: 5aa02a0e3934e8246e948260f2a4457b324de2377e7ec2e00fe9bf03a52f5ee4
+  data.tar.gz: 9f97a22af58eb64c64a6b73a79c2947bad06318e17bca235a5aa59fae35aad1d
 SHA512:
-  metadata.gz: 96e3c0f9c2860002642e079b756065e32f04891cd793dce9efcb74bb87a13683b14fee13c039693d99599549fee7ace68d54561b8c730d5e9eb7bb937a82c57d
-  data.tar.gz: 2e512b774b5778e3f7a2a3557bb45d97971d209c787494a19d81e0d8146f5d51bbef7940f37fa38f7587e1037185e23fa27ca8d48a46225ced6640be8425198e
+  metadata.gz: 2e8fcf1be08ba5c2582ad112455b02815410a81834b55bfc0035e0971556ff7b443e60a48a4cc2f5c94fdd81d09a5d03d3c261b4e0afb13ac9f38da5d4f0ab64
+  data.tar.gz: 77f17d9e003384f34e5719f0e84f70c8a04f7de196974fc5af830e4825827b01d7df94b4eb887d4a1a4a77f43caa989cfdcbb1e874d807503bf9ef254ea95a0d

data/lib/redis_client/cluster/optimistic_locking.rb

@@ -1,7 +1,6 @@
 # frozen_string_literal: true
 
 require 'redis_client'
-require 'redis_client/cluster/key_slot_converter'
 require 'redis_client/cluster/transaction'
 
 class RedisClient
@@ -12,15 +11,14 @@ class RedisClient
       end
 
       def watch(keys)
-        ensure_safe_keys(keys)
-        node = find_node(keys)
-        cnt = 0 # We assume redirects occurred when incrementing it.
+        slot = find_slot(keys)
+        raise ::RedisClient::Cluster::Transaction::ConsistencyError, "unsafe watch: #{keys.join(' ')}" if slot.nil?
 
+        node = @router.find_primary_node_by_slot(slot)
         @router.handle_redirection(node, retry_count: 1) do |nd|
-          cnt += 1
           nd.with do |c|
             c.call('WATCH', *keys)
-            reply = yield(c, cnt > 1)
+            reply = yield(c, slot)
             c.call('UNWATCH')
             reply
           end
@@ -29,29 +27,14 @@ class RedisClient
 
       private
 
-      def ensure_safe_keys(keys)
-        return if safe?(keys)
+      def find_slot(keys)
+        return if keys.empty?
+        return if keys.any? { |k| k.nil? || k.empty? }
 
-        raise ::RedisClient::Cluster::Transaction::ConsistencyError, "unsafe watch: #{keys.join(' ')}"
-      end
-
-      def safe?(keys)
-        return false if keys.empty?
-
-        slots = keys.map do |k|
-          return false if k.nil? || k.empty?
-
-          ::RedisClient::Cluster::KeySlotConverter.convert(k)
-        end
-
-        slots.uniq.size == 1
-      end
-
-      def find_node(keys)
-        node_key = @router.find_primary_node_key(['WATCH', *keys])
-        return @router.find_node(node_key) unless node_key.nil?
+        slots = keys.map { |k| @router.find_slot_by_key(k) }
+        return if slots.uniq.size != 1
 
-        raise ::RedisClient::Cluster::Transaction::ConsistencyError, "couldn't determine the node"
+        slots.first
       end
     end
   end

data/lib/redis_client/cluster/router.rb

@@ -179,6 +179,16 @@ class RedisClient
         find_node_key_by_key(key, primary: true)
       end
 
+      def find_slot(command)
+        find_slot_by_key(@command.extract_first_key(command))
+      end
+
+      def find_slot_by_key(key)
+        return if key.empty?
+
+        ::RedisClient::Cluster::KeySlotConverter.convert(key)
+      end
+
       def find_node(node_key, retry_count: 3)
         @node.find_by(node_key)
       rescue ::RedisClient::Cluster::Node::ReloadNeeded

data/lib/redis_client/cluster/transaction.rb

@@ -2,7 +2,6 @@
 
 require 'redis_client'
 require 'redis_client/cluster/pipeline'
-require 'redis_client/cluster/node_key'
 
 class RedisClient
   class Cluster
@@ -10,7 +9,7 @@ class RedisClient
       ConsistencyError = Class.new(::RedisClient::Error)
       MAX_REDIRECTION = 2
 
-      def initialize(router, command_builder, node: nil, resharding: false)
+      def initialize(router, command_builder, node: nil, slot: nil)
         @router = router
         @command_builder = command_builder
         @retryable = true
@@ -18,7 +17,7 @@ class RedisClient
         @pending_commands = []
         @node = node
         prepare_tx unless @node.nil?
-        @resharding_state = resharding
+        @watching_slot = slot
       end
 
       def call(*command, **kwargs, &block)
@@ -111,7 +110,7 @@ class RedisClient
           client.middlewares.call_pipelined(commands, client.config) do
             connection.call_pipelined(commands, nil)
           rescue ::RedisClient::CommandError => e
-            return handle_command_error!(client, commands, e, redirect: redirect) unless redirect.zero?
+            return handle_command_error!(commands, e, redirect: redirect) unless redirect.zero?
 
             raise
           end
@@ -140,15 +139,15 @@ class RedisClient
         results
       end
 
-      def handle_command_error!(client, commands, err, redirect:) # rubocop:disable Metrics/AbcSize
+      def handle_command_error!(commands, err, redirect:) # rubocop:disable Metrics/AbcSize
         if err.message.start_with?('CROSSSLOT')
           raise ConsistencyError, "#{err.message}: #{err.command}"
         elsif err.message.start_with?('MOVED')
-          ensure_the_same_node!(client, commands)
+          ensure_the_same_slot!(commands)
           node = @router.assign_redirection_node(err.message)
           send_transaction(node, redirect: redirect - 1)
         elsif err.message.start_with?('ASK')
-          ensure_the_same_node!(client, commands)
+          ensure_the_same_slot!(commands)
           node = @router.assign_asking_node(err.message)
           try_asking(node) ? send_transaction(node, redirect: redirect - 1) : err
         else
@@ -156,12 +155,10 @@ class RedisClient
         end
       end
 
-      def ensure_the_same_node!(client, commands)
-        node_keys = commands.map { |command| @router.find_primary_node_key(command) }.compact.uniq
-        expected_node_key = ::RedisClient::Cluster::NodeKey.build_from_client(client)
-
-        return if !@resharding_state && node_keys.size == 1 && node_keys.first == expected_node_key
-        return if @resharding_state && node_keys.size == 1
+      def ensure_the_same_slot!(commands)
+        slots = commands.map { |command| @router.find_slot(command) }.compact.uniq
+        return if slots.size == 1 && @watching_slot.nil?
+        return if slots.size == 1 && @watching_slot == slots.first
 
         raise(ConsistencyError, "the transaction should be executed to a slot in a node: #{commands}")
       end

data/lib/redis_client/cluster.rb

@@ -98,9 +98,9 @@ class RedisClient
         return transaction.execute
       end
 
-      ::RedisClient::Cluster::OptimisticLocking.new(@router).watch(watch) do |c, resharding|
+      ::RedisClient::Cluster::OptimisticLocking.new(@router).watch(watch) do |c, slot|
         transaction = ::RedisClient::Cluster::Transaction.new(
-          @router, @command_builder, node: c, resharding: resharding
+          @router, @command_builder, node: c, slot: slot
         )
         yield transaction
         transaction.execute

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: redis-cluster-client
 version: !ruby/object:Gem::Version
-  version: 0.7.8
+  version: 0.7.9
 platform: ruby
 authors:
 - Taishi Kasuga