redis-cluster-client 0.7.8 → 0.7.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 41d87ce45a108ff8ec7d3c17805276d5ba7df11d577de3f7d6b1176bdd297ed0
-  data.tar.gz: 7118539907bb755225ba908cf160ad8af73aeb5204820682ca3fc3a67813f3cb
+  metadata.gz: b591c92886bf877147317dd04695d2d22354958443f83e386b061b45b0407b28
+  data.tar.gz: 40a46369e32c60a3c165aff646d3dbd53c3ef427fef6220d4bacc17fc6a97232
 SHA512:
-  metadata.gz: 96e3c0f9c2860002642e079b756065e32f04891cd793dce9efcb74bb87a13683b14fee13c039693d99599549fee7ace68d54561b8c730d5e9eb7bb937a82c57d
-  data.tar.gz: 2e512b774b5778e3f7a2a3557bb45d97971d209c787494a19d81e0d8146f5d51bbef7940f37fa38f7587e1037185e23fa27ca8d48a46225ced6640be8425198e
+  metadata.gz: 0e8eda475d22314feb84d16c2d5d75dfe714abecf2c6482a7600c012d02705110c40a0ed7717029a7e8ad154040c6cd44978ffd6b9a2c0350ec33e45d1e1b81e
+  data.tar.gz: a8fc2bb1febdac9e44c23aaab93ddce567d9ea1fe4ff09f74aa3ebac7eba56efceff44b5f13fea0284d4bec8b2677bd58dfee0eedfebe9cdc4ed9e7d4f6cb6df

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,46 +11,31 @@ 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)
+            yield(c, slot)
+          rescue StandardError
             c.call('UNWATCH')
-            reply
+            raise
           end
         end
       end
 
       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

@@ -8,6 +8,8 @@ require 'redis_client/cluster/key_slot_converter'
 require 'redis_client/cluster/node'
 require 'redis_client/cluster/node_key'
 require 'redis_client/cluster/normalized_cmd_name'
+require 'redis_client/cluster/transaction'
+require 'redis_client/cluster/optimistic_locking'
 
 class RedisClient
   class Cluster
@@ -44,6 +46,7 @@ class RedisClient
         when 'memory'   then send_memory_command(method, command, args, &block)
         when 'script'   then send_script_command(method, command, args, &block)
         when 'pubsub'   then send_pubsub_command(method, command, args, &block)
+        when 'watch'    then send_watch_command(command, &block)
         when 'acl', 'auth', 'bgrewriteaof', 'bgsave', 'quit', 'save'
           @node.call_all(method, command, args).first.then(&TSF.call(block))
         when 'flushall', 'flushdb'
@@ -179,6 +182,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
@@ -298,6 +311,17 @@ class RedisClient
         end
       end
 
+      # for redis-rb
+      def send_watch_command(command)
+        ::RedisClient::Cluster::OptimisticLocking.new(self).watch(command[1..]) do |c, slot|
+          transaction = ::RedisClient::Cluster::Transaction.new(
+            self, @command_builder, node: c, slot: slot
+          )
+          yield transaction
+          transaction.execute
+        end
+      end
+
       def update_cluster_info!
         @node.reload!
       end

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,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: redis-cluster-client
 version: !ruby/object:Gem::Version
-  version: 0.7.8
+  version: 0.7.10
 platform: ruby
 authors:
 - Taishi Kasuga
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-02-18 00:00:00.000000000 Z
+date: 2024-02-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redis-client