support previous versions of redis gem
The following table defines how input parameters and result values should be modified for the namespace.
COMMANDS is a hash. Each key is the name of a command and each value is a two element array.
The first element in the value array describes how to modify the arguments passed. It can be one of:
nil Do nothing. :first Add the namespace to the first argument passed, e.g. GET key => GET namespace:key :all Add the namespace to all arguments passed, e.g. MGET key1 key2 => MGET namespace:key1 namespace:key2 :exclude_first Add the namespace to all arguments but the first, e.g. :exclude_last Add the namespace to all arguments but the last, e.g. BLPOP key1 key2 timeout => BLPOP namespace:key1 namespace:key2 timeout :exclude_options Add the namespace to all arguments, except the last argument, if the last argument is a hash of options. ZUNIONSTORE key1 2 key2 key3 WEIGHTS 2 1 => ZUNIONSTORE namespace:key1 2 namespace:key2 namespace:key3 WEIGHTS 2 1 :alternate Add the namespace to every other argument, e.g. MSET key1 value1 key2 value2 => MSET namespace:key1 value1 namespace:key2 value2
The second element in the value array describes how to modify the return value of the Redis call. It can be one of:
nil Do nothing. :all Add the namespace to all elements returned, e.g. key1 key2 => namespace:key1 namespace:key2
# File lib/redis/namespace.rb, line 196 def initialize(namespace, options = {}) @namespace = namespace @redis = options[:redis] || Redis.current @warning = options[:warning] || false end
# File lib/redis/namespace.rb, line 243 def exec method_missing(:exec) end
# File lib/redis/namespace.rb, line 218 def keys(query = nil) query.nil? ? super("*") : super end
# File lib/redis/namespace.rb, line 247 def method_missing(command, *args, &block) handling = COMMANDS[command.to_s] || COMMANDS[ALIASES[command.to_s]] # redis-namespace does not know how to handle this command. # Passing it to @redis as is, where redis-namespace shows # a warning message if @warning is set. if handling.nil? warn("Passing '#{command}' command to redis as is.") if @warning return @redis.send(command, *args, &block) end (before, after) = handling # Add the namespace to any parameters that are keys. case before when :first args[0] = add_namespace(args[0]) if args[0] when :all args = add_namespace(args) when :exclude_first first = args.shift args = add_namespace(args) args.unshift(first) if first when :exclude_last last = args.pop unless args.length == 1 args = add_namespace(args) args.push(last) if last when :exclude_options if args.last.is_a?(Hash) last = args.pop args = add_namespace(args) args.push(last) else args = add_namespace(args) end when :alternate args.each_with_index { |a, i| args[i] = add_namespace(a) if i.even? } when :sort args[0] = add_namespace(args[0]) if args[0] if args[1].is_a?(Hash) [:by, :store].each do |key| args[1][key] = add_namespace(args[1][key]) if args[1][key] end args[1][:get] = Array(args[1][:get]) args[1][:get].each_index do |i| args[1][:get][i] = add_namespace(args[1][:get][i]) unless args[1][:get][i] == "#" end end when :eval_style # redis.eval() and evalsha() can either take the form: # # redis.eval(script, [key1, key2], [argv1, argv2]) # # Or: # # redis.eval(script, :keys => ['k1', 'k2'], :argv => ['arg1', 'arg2']) # # This is a tricky + annoying special case, where we only want the `keys` # argument to be namespaced. if args.last.is_a?(Hash) args.last[:keys] = add_namespace(args.last[:keys]) else args[1] = add_namespace(args[1]) end end # Dispatch the command to Redis and store the result. result = @redis.send(command, *args, &block) # Don't try to remove namespace from a Redis::Future, you can't. return result if result.is_a?(Redis::Future) # Remove the namespace from results that are keys. case after when :all result = rem_namespace(result) when :first result[0] = rem_namespace(result[0]) if result end result end
# File lib/redis/namespace.rb, line 222 def multi(&block) if block_given? namespaced_block(:multi, &block) else method_missing(:multi) end end
# File lib/redis/namespace.rb, line 234 def namespace(desired_namespace = nil) if desired_namespace yield Redis::Namespace.new(desired_namespace, :redis => @redis) end @namespace end
# File lib/redis/namespace.rb, line 230 def pipelined(&block) namespaced_block(:pipelined, &block) end
# File lib/redis/namespace.rb, line 210 def respond_to?(command, include_private=false) if self_respond_to?(command, include_private) true else @redis.respond_to?(command, include_private) end end
Ruby defines a now deprecated type method so we need to override it here since it will never hit #method_missing
# File lib/redis/namespace.rb, line 204 def type(key) method_missing(:type, key) end
# File lib/redis/namespace.rb, line 345 def add_namespace(key) return key unless key && @namespace case key when Array key.map {|k| add_namespace k} when Hash Hash[*key.map {|k, v| [ add_namespace(k), v ]}.flatten] else "#{@namespace}:#{key}" end end
# File lib/redis/namespace.rb, line 335 def namespaced_block(command, &block) original = @redis result = redis.send(command) do |r| @redis = r yield self end @redis = original result end
# File lib/redis/namespace.rb, line 358 def rem_namespace(key) return key unless key && @namespace case key when Array key.map {|k| rem_namespace k} when Hash Hash[*key.map {|k, v| [ rem_namespace(k), v ]}.flatten] else key.to_s.gsub(/^#{@namespace}:/, "") end end