Method: Redis::Connection::Memory#zscan

Defined in:

lib/redis/connection/memory.rb

#zscan(key, start_cursor, *args) ⇒ Object

# File 'lib/redis/connection/memory.rb', line 1443

def zscan(key, start_cursor, *args)
  data_type_check(key, ZSet)
  return [] unless data[key]

  match = "*"
  count = 10

  if args.size.odd?
    raise_argument_error('zscan')
  end

  if idx = args.index("MATCH")
    match = args[idx + 1]
  end

  if idx = args.index("COUNT")
    count = args[idx + 1]
  end

  start_cursor = start_cursor.to_i
  data_type_check(start_cursor, Integer)

  cursor = start_cursor
  next_keys = []

  sorted_keys = sort_keys(data[key])

  if start_cursor + count >= sorted_keys.length
    next_keys = sorted_keys.to_a.select { |k| File.fnmatch(match, k[0]) } [start_cursor..-1]
    cursor = 0
  else
    cursor = start_cursor + count
    next_keys = sorted_keys.to_a.select { |k| File.fnmatch(match, k[0]) } [start_cursor..cursor-1]
  end
  return "#{cursor}", next_keys.flatten.map(&:to_s)
end