valkey-rb 0.3.5

data/lib/valkey/commands/generic_commands.rb

@@ -0,0 +1,454 @@
+# frozen_string_literal: true
+
+class Valkey
+  module Commands
+    # this module contains generic commands that are not specific to any data type
+    #
+    # @see https://valkey.io/commands/#generic
+    #
+    module GenericCommands
+      # Scan the keyspace
+      #
+      # @example Retrieve the first batch of keys
+      #   valkey.scan(0)
+      #     # => ["4", ["key:21", "key:47", "key:42"]]
+      # @example Retrieve a batch of keys matching a pattern
+      #   valkey.scan(4, :match => "key:1?")
+      #     # => ["92", ["key:13", "key:18"]]
+      # @example Retrieve a batch of keys of a certain type
+      #   valkey.scan(92, :type => "zset")
+      #     # => ["173", ["sortedset:14", "sortedset:78"]]
+      #
+      # @param [String, Integer] cursor the cursor of the iteration
+      # @param [Hash] options
+      #   - `:match => String`: only return keys matching the pattern
+      #   - `:count => Integer`: return count keys at most per iteration
+      #   - `:type => String`: return keys only of the given type
+      #
+      # @return [String, Array<String>] the next cursor and all found keys
+      #
+      def scan(cursor, **options)
+        _scan(RequestType::SCAN, cursor, [], **options)
+      end
+
+      # Scan the keyspace
+      #
+      # @example Retrieve all of the keys (with possible duplicates)
+      #   valkey.scan_each.to_a
+      #     # => ["key:21", "key:47", "key:42"]
+      # @example Execute block for each key matching a pattern
+      #   valkey.scan_each(:match => "key:1?") {|key| puts key}
+      #     # => key:13
+      #     # => key:18
+      # @example Execute block for each key of a type
+      #   valkey.scan_each(:type => "hash") {|key| puts valkey.type(key)}
+      #     # => "hash"
+      #     # => "hash"
+      #
+      # @param [Hash] options
+      #   - `:match => String`: only return keys matching the pattern
+      #   - `:count => Integer`: return count keys at most per iteration
+      #   - `:type => String`: return keys only of the given type
+      #
+      # @return [Enumerator] an enumerator for all found keys
+      #
+      # def scan_each(**options, &block)
+      #   return to_enum(:scan_each, **options) unless block_given?
+      #
+      #   cursor = 0
+      #   loop do
+      #     cursor, keys = scan(cursor, **options)
+      #     keys.each(&block)
+      #     break if cursor == "0"
+      #   end
+      # end
+
+      # Remove the expiration from a key.
+      #
+      # @param [String] key
+      # @return [Boolean] whether the timeout was removed or not
+      def persist(key)
+        send_command(RequestType::PERSIST, [key])
+      end
+
+      # Set a key's time to live in seconds.
+      #
+      # @param [String] key
+      # @param [Integer] seconds time to live
+      # @param [Hash] options
+      #   - `:nx => true`: Set expiry only when the key has no expiry.
+      #   - `:xx => true`: Set expiry only when the key has an existing expiry.
+      #   - `:gt => true`: Set expiry only when the new expiry is greater than current one.
+      #   - `:lt => true`: Set expiry only when the new expiry is less than current one.
+      # @return [Boolean] whether the timeout was set or not
+      def expire(key, seconds, nx: nil, xx: nil, gt: nil, lt: nil)
+        args = [key, Integer(seconds)]
+        args << "NX" if nx
+        args << "XX" if xx
+        args << "GT" if gt
+        args << "LT" if lt
+
+        send_command(RequestType::EXPIRE, args)
+      end
+
+      # Set the expiration for a key as a UNIX timestamp.
+      #
+      # @param [String] key
+      # @param [Integer] unix_time expiry time specified as a UNIX timestamp
+      # @param [Hash] options
+      #   - `:nx => true`: Set expiry only when the key has no expiry.
+      #   - `:xx => true`: Set expiry only when the key has an existing expiry.
+      #   - `:gt => true`: Set expiry only when the new expiry is greater than current one.
+      #   - `:lt => true`: Set expiry only when the new expiry is less than current one.
+      # @return [Boolean] whether the timeout was set or not
+      def expireat(key, unix_time, nx: nil, xx: nil, gt: nil, lt: nil)
+        args = [key, Integer(unix_time)]
+        args << "NX" if nx
+        args << "XX" if xx
+        args << "GT" if gt
+        args << "LT" if lt
+
+        send_command(RequestType::EXPIRE_AT, args)
+      end
+
+      # Get a key's expiry time specified as number of seconds from UNIX Epoch
+      #
+      # @param  [String] key
+      # @return [Integer] expiry time specified as number of seconds from UNIX Epoch
+      def expiretime(key)
+        send_command(RequestType::EXPIRE_TIME, [key])
+      end
+
+      # Get the time to live (in seconds) for a key.
+      #
+      # @param [String] key
+      # @return [Integer] remaining time to live in seconds.
+      #
+      # In valkey 2.6 or older the command returns -1 if the key does not exist or if
+      # the key exist but has no associated expire.
+      #
+      # Starting with valkey 2.8 the return value in case of error changed:
+      #
+      #     - The command returns -2 if the key does not exist.
+      #     - The command returns -1 if the key exists but has no associated expire.
+      def ttl(key)
+        send_command(RequestType::TTL, [key])
+      end
+
+      # Set a key's time to live in milliseconds.
+      #
+      # @param [String] key
+      # @param [Integer] milliseconds time to live
+      # @param [Hash] options
+      #   - `:nx => true`: Set expiry only when the key has no expiry.
+      #   - `:xx => true`: Set expiry only when the key has an existing expiry.
+      #   - `:gt => true`: Set expiry only when the new expiry is greater than current one.
+      #   - `:lt => true`: Set expiry only when the new expiry is less than current one.
+      # @return [Boolean] whether the timeout was set or not
+      def pexpire(key, milliseconds, nx: nil, xx: nil, gt: nil, lt: nil)
+        args = [key, Integer(milliseconds)]
+        args << "NX" if nx
+        args << "XX" if xx
+        args << "GT" if gt
+        args << "LT" if lt
+
+        send_command(RequestType::PEXPIRE, args)
+      end
+
+      # Set the expiration for a key as number of milliseconds from UNIX Epoch.
+      #
+      # @param [String] key
+      # @param [Integer] ms_unix_time expiry time specified as number of milliseconds from UNIX Epoch.
+      # @param [Hash] options
+      #   - `:nx => true`: Set expiry only when the key has no expiry.
+      #   - `:xx => true`: Set expiry only when the key has an existing expiry.
+      #   - `:gt => true`: Set expiry only when the new expiry is greater than current one.
+      #   - `:lt => true`: Set expiry only when the new expiry is less than current one.
+      # @return [Boolean] whether the timeout was set or not
+      def pexpireat(key, ms_unix_time, nx: nil, xx: nil, gt: nil, lt: nil)
+        args = [key, Integer(ms_unix_time)]
+        args << "NX" if nx
+        args << "XX" if xx
+        args << "GT" if gt
+        args << "LT" if lt
+
+        send_command(RequestType::PEXPIRE_AT, args)
+      end
+
+      # Get a key's expiry time specified as number of milliseconds from UNIX Epoch
+      #
+      # @param  [String] key
+      # @return [Integer] expiry time specified as number of milliseconds from UNIX Epoch
+      def pexpiretime(key)
+        send_command(RequestType::PEXPIRE_TIME, [key])
+      end
+
+      # Get the time to live (in milliseconds) for a key.
+      #
+      # @param [String] key
+      # @return [Integer] remaining time to live in milliseconds
+      # In valkey 2.6 or older the command returns -1 if the key does not exist or if
+      # the key exist but has no associated expire.
+      #
+      # Starting with valkey 2.8 the return value in case of error changed:
+      #
+      #     - The command returns -2 if the key does not exist.
+      #     - The command returns -1 if the key exists but has no associated expire.
+      def pttl(key)
+        send_command(RequestType::PTTL, [key])
+      end
+
+      # Return a serialized version of the value stored at a key.
+      #
+      # @param [String] key
+      # @return [String] serialized_value
+      def dump(key)
+        send_command(RequestType::DUMP, [key])
+      end
+
+      # Create a key using the serialized value, previously obtained using DUMP.
+      #
+      # @param [String] key
+      # @param [String] ttl
+      # @param [String] serialized_value
+      # @param [Hash] options
+      #   - `:replace => Boolean`: if false, raises an error if key already exists
+      # @raise [valkey::CommandError]
+      # @return [String] `"OK"`
+      def restore(key, ttl, serialized_value, replace: nil)
+        args = [key, ttl, serialized_value]
+        args << 'REPLACE' if replace
+
+        send_command(RequestType::RESTORE, args)
+      end
+
+      # Transfer a key from the connected instance to another instance.
+      #
+      # @param [String, Array<String>] key
+      # @param [Hash] options
+      #   - `:host => String`: host of instance to migrate to
+      #   - `:port => Integer`: port of instance to migrate to
+      #   - `:db => Integer`: database to migrate to (default: same as source)
+      #   - `:timeout => Integer`: timeout (default: same as connection timeout)
+      #   - `:copy => Boolean`: Do not remove the key from the local instance.
+      #   - `:replace => Boolean`: Replace existing key on the remote instance.
+      # @return [String] `"OK"`
+      # def migrate(key, options)
+      #   args = []
+      #   args << (options[:host] || raise(':host not specified'))
+      #   args << (options[:port] || raise(':port not specified'))
+      #   args << (key.is_a?(String) ? key : '')
+      #   args << (options[:db] || @client.db).to_i
+      #   args << (options[:timeout] || @client.timeout).to_i
+      #   args << 'COPY' if options[:copy]
+      #   args << 'REPLACE' if options[:replace]
+      #   args += ['KEYS', *key] if key.is_a?(Array)
+      #
+      #   send_command(RequestType::MIGRATE, args)
+      # end
+
+      # Delete one or more keys.
+      #
+      # @param [String, Array<String>] keys
+      # @return [Integer] number of keys that were deleted
+      def del(*keys)
+        keys.flatten!(1)
+        return 0 if keys.empty?
+
+        send_command(RequestType::DEL, keys)
+      end
+
+      # Unlink one or more keys.
+      #
+      # @param [String, Array<String>] keys
+      # @return [Integer] number of keys that were unlinked
+      def unlink(*keys)
+        send_command(RequestType::UNLINK, keys.flatten)
+      end
+
+      # Determine if a key exists.
+      # This method returns a boolean for compatibility with redis-rb v3.3.5 behavior.
+      #
+      # @param [String] key
+      # @return [Boolean]
+      def exists(key)
+        send_command(RequestType::EXISTS, [key], &:positive?)
+      end
+
+      # Move a key to another database.
+      #
+      # @example Move a key to another database
+      #   valkey.set "foo", "bar"
+      #     # => "OK"
+      #   valkey.move "foo", 2
+      #     # => true
+      #   valkey.exists "foo"
+      #     # => false
+      #   valkey.select 2
+      #     # => "OK"
+      #   valkey.exists "foo"
+      #     # => true
+      #   valkey.get "foo"
+      #     # => "bar"
+      #
+      # @param [String] key
+      # @param [Integer] db
+      # @return [Boolean] whether the key was moved or not
+      def move(key, db)
+        send_command(RequestType::MOVE, [key, db])
+      end
+
+      # Copy a value from one key to another.
+      #
+      # @example Copy a value to another key
+      #   valkey.set "foo", "value"
+      #     # => "OK"
+      #   valkey.copy "foo", "bar"
+      #     # => true
+      #   valkey.get "bar"
+      #     # => "value"
+      #
+      # @example Copy a value to a key in another database
+      #   valkey.set "foo", "value"
+      #     # => "OK"
+      #   valkey.copy "foo", "bar", db: 2
+      #     # => true
+      #   valkey.select 2
+      #     # => "OK"
+      #   valkey.get "bar"
+      #     # => "value"
+      #
+      # @param [String] source
+      # @param [String] destination
+      # @param [Integer] db
+      # @param [Boolean] replace removes the `destination` key before copying value to it
+      # @return [Integer] 1 if the key was copied, 0 otherwise
+      def copy(source, destination, db: nil, replace: false)
+        args = [source, destination]
+        args << "DB" << db if db
+        args << "REPLACE" if replace
+
+        result = send_command(RequestType::COPY, args)
+        result ? 1 : 0
+      end
+
+      def object(subcommand, *args)
+        map = {
+          refcount: RequestType::OBJECT_REF_COUNT,
+          encoding: RequestType::OBJECT_ENCODING,
+          idletime: RequestType::OBJECT_IDLE_TIME,
+          freq: RequestType::OBJECT_FREQ
+        }
+
+        send_command(map[subcommand.to_sym], args.flatten)
+      end
+
+      # Return a random key from the keyspace.
+      #
+      # @return [String]
+      def randomkey
+        send_command(RequestType::RANDOM_KEY)
+      end
+
+      # Rename a key. If the new key already exists it is overwritten.
+      #
+      # @param [String] old_name
+      # @param [String] new_name
+      # @return [String] `OK`
+      def rename(old_name, new_name)
+        send_command(RequestType::RENAME, [old_name, new_name])
+      end
+
+      # Rename a key, only if the new key does not exist.
+      #
+      # @param [String] old_name
+      # @param [String] new_name
+      # @return [Boolean] whether the key was renamed or not
+      def renamenx(old_name, new_name)
+        send_command(RequestType::RENAME_NX, [old_name, new_name])
+      end
+
+      # Sort the elements in a list, set or sorted set.
+      #
+      # @example Retrieve the first 2 elements from an alphabetically sorted "list"
+      #   valkey.sort("list", :order => "alpha", :limit => [0, 2])
+      #     # => ["a", "b"]
+      # @example Store an alphabetically descending list in "target"
+      #   valkey.sort("list", :order => "desc alpha", :store => "target")
+      #     # => 26
+      #
+      # @param [String] key
+      # @param [Hash] options
+      #   - `:by => String`: use external key to sort elements by
+      #   - `:limit => [offset, count]`: skip `offset` elements, return a maximum
+      #   of `count` elements
+      #   - `:get => [String, Array<String>]`: single key or array of keys to
+      #   retrieve per element in the result
+      #   - `:order => String`: combination of `ASC`, `DESC` and optionally `ALPHA`
+      #   - `:store => String`: key to store the result at
+      #
+      # @return [Array<String>, Array<Array<String>>, Integer]
+      #   - when `:get` is not specified, or holds a single element, an array of elements
+      #   - when `:get` is specified, and holds more than one element, an array of
+      #   elements where every element is an array with the result for every
+      #   element specified in `:get`
+      #   - when `:store` is specified, the number of elements in the stored result
+      def sort(key, by: nil, limit: nil, get: nil, order: nil, store: nil)
+        args = [key]
+        args << "BY" << by if by
+
+        if limit
+          args << "LIMIT"
+          args.concat(limit)
+        end
+
+        get = Array(get)
+        get.each do |item|
+          args << "GET" << item
+        end
+
+        args.concat(order.split(" ")) if order
+        args << "STORE" << store if store
+
+        send_command(RequestType::SORT, args) do |reply|
+          if get.size > 1 && !store
+            reply.each_slice(get.size).to_a if reply
+          else
+            reply
+          end
+        end
+      end
+
+      def touch(*keys)
+        send_command(RequestType::TOUCH, keys.flatten)
+      end
+
+      def wait(*keys)
+        send_command(RequestType::WAIT, keys.flatten)
+      end
+
+      def waitof(*keys)
+        send_command(RequestType::WAIT_AOF, keys.flatten)
+      end
+
+      # Determine the type stored at key.
+      #
+      # @param [String] key
+      # @return [String] `string`, `list`, `set`, `zset`, `hash` or `none`
+      def type(key)
+        send_command(RequestType::TYPE, [key])
+      end
+
+      def _scan(command, cursor, args, match: nil, count: nil, type: nil, &block)
+        # SSCAN/ZSCAN/HSCAN already prepend the key to +args+.
+
+        args << cursor
+        args << "MATCH" << match if match
+        args << "COUNT" << Integer(count) if count
+        args << "TYPE" << type if type
+
+        send_command(command, args, &block)
+      end
+    end
+  end
+end

data/lib/valkey/commands/geo_commands.rb

@@ -0,0 +1,87 @@
+# frozen_string_literal: true
+
+class Valkey
+  module Commands
+    # This module contains commands on geospatial operations.
+    #
+    # @see https://valkey.io/commands/#geo
+    #
+    module GeoCommands
+      # Add one or more geospatial items (longitude, latitude, name) to a key.
+      #
+      # @example
+      #   valkey.geoadd("locations", 13.361389, 38.115556, "Palermo", 15.087269, 37.502669, "Catania")
+      #     # => Integer (number of elements added)
+      #
+      # @param [String] key the name of the key
+      # @param [Array<String, Float>] members one or more longitude, latitude, and name triplets
+      # @return [Integer] the number of elements added
+      def geoadd(key, *members)
+        send_command(RequestType::GEO_ADD, [key, *members])
+      end
+
+      # Retrieve the positions (longitude, latitude) of one or more elements.
+      #
+      # @example
+      #   valkey.geopos("locations", "Palermo", "Catania")
+      #     # => [[13.361389, 38.115556], [15.087269, 37.502669]]
+      #
+      # @param [String] key the name of the key
+      # @param [Array<String>] members one or more member names to get positions for
+      # @return [Array<Array<Float, Float>, nil>] list of positions or nil for missing members
+      def geopos(key, *members)
+        send_command(RequestType::GEO_POS, [key, *members])
+      end
+
+      # Returns geohash string representing position for specified members of the specified key.
+      #
+      # @param [String] key
+      # @param [String, Array<String>] member one member or array of members
+      # @return [Array<String, nil>] returns array containg geohash string if member is present, nil otherwise
+      def geohash(key, *member)
+        send_command(RequestType::GEO_HASH, [key, *member])
+      end
+
+      # Returns the distance between two members of a geospatial index
+      #
+      # @param [String] key
+      # @param [Array<String>] members
+      # @param ['m', 'km', 'mi', 'ft'] unit
+      # @return [String, nil] returns distance in specified unit if both members present, nil otherwise.
+      def geodist(key, member1, member2, unit = 'm')
+        send_command(RequestType::GEO_DIST, [key, member1, member2, unit])
+      end
+
+      # Perform raw GEOSEARCH command with direct arguments like Redis
+      #
+      # @example
+      #   valkey.geosearch("places", "FROMMEMBER", "berlin", "BYRADIUS", 1000, "km", "WITHDIST")
+      #
+      # @param [Array<String>] args full argument list for GEOSEARCH
+      # @return [Array] raw result from server
+      def geosearch(*args)
+        send_command(RequestType::GEO_SEARCH, args)
+      end
+
+      # Store the result of a GEOSEARCH query into a new sorted set key.
+      #
+      # @example
+      #   valkey.geosearchstore(
+      #     "nearby:berlin",       # destination key
+      #     "Places",               # source key
+      #     "FROMMEMBER", "Berlin",
+      #     "BYRADIUS", 200, "km",
+      #     "ASC", "COUNT", 10
+      #   )
+      #   # => 2 (number of items stored)
+      #
+      # @param [String] destination the name of the key where results will be stored
+      # @param [String] source the name of the source geo key to search from
+      # @param [Array<String, Integer>] args full argument list like GEOSEARCH (e.g., FROMMEMBER, BYRADIUS, COUNT, etc.)
+      # @return [Integer] the number of items stored in the destination key
+      def geosearchstore(destination, source, *args)
+        send_command(RequestType::GEO_SEARCH_STORE, [destination, source, *args])
+      end
+    end
+  end
+end