Module: MockRedis::HashMethods

Includes:
Assertions, UtilityMethods
Included in:
Database
Defined in:
lib/mock_redis/hash_methods.rb

Instance Method Summary collapse

Instance Method Details

#hdel(key, field) ⇒ Object



9
10
11
12
13
14
15
16
17
18
19
# File 'lib/mock_redis/hash_methods.rb', line 9

def hdel(key, field)
  with_hash_at(key) do |hash|
    if field.is_a?(Array)
      orig_size = hash.size
      hash.delete_if { |k,v| field.include?(k) }
      orig_size - hash.size
    else
      hash.delete(field.to_s) ? 1 : 0
    end
  end
end

#hexists(key, field) ⇒ Object



21
22
23
# File 'lib/mock_redis/hash_methods.rb', line 21

def hexists(key, field)
  with_hash_at(key) {|h| h.has_key?(field.to_s)}
end

#hget(key, field) ⇒ Object



25
26
27
# File 'lib/mock_redis/hash_methods.rb', line 25

def hget(key, field)
  with_hash_at(key) {|h| h[field.to_s]}
end

#hgetall(key) ⇒ Object



29
30
31
# File 'lib/mock_redis/hash_methods.rb', line 29

def hgetall(key)
  with_hash_at(key) {|h| h}
end

#hincrby(key, field, increment) ⇒ Object



33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# File 'lib/mock_redis/hash_methods.rb', line 33

def hincrby(key, field, increment)
  with_hash_at(key) do |hash|
    field = field.to_s
    unless can_incr?(data[key][field])
      raise Redis::CommandError, "ERR hash value is not an integer"
    end
    unless looks_like_integer?(increment.to_s)
      raise Redis::CommandError, "ERR value is not an integer or out of range"
    end

    new_value = (hash[field] || "0").to_i + increment.to_i
    hash[field] = new_value.to_s
    new_value
  end
end

#hincrbyfloat(key, field, increment) ⇒ Object



49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# File 'lib/mock_redis/hash_methods.rb', line 49

def hincrbyfloat(key, field, increment)
  with_hash_at(key) do |hash|
    field = field.to_s
    unless can_incr_float?(data[key][field])
      raise Redis::CommandError, "ERR hash value is not a valid float"
    end

    unless looks_like_float?(increment.to_s)
      raise Redis::CommandError, "ERR value is not a valid float"
    end

    new_value = (hash[field] || "0").to_f + increment.to_f
    new_value = new_value.to_i if new_value % 1 == 0
    hash[field] = new_value.to_s
    new_value
  end
end

#hkeys(key) ⇒ Object



67
68
69
# File 'lib/mock_redis/hash_methods.rb', line 67

def hkeys(key)
  with_hash_at(key, &:keys)
end

#hlen(key) ⇒ Object



71
72
73
# File 'lib/mock_redis/hash_methods.rb', line 71

def hlen(key)
  hkeys(key).length
end

#hmget(key, *fields) ⇒ Object



75
76
77
78
# File 'lib/mock_redis/hash_methods.rb', line 75

def hmget(key, *fields)
  assert_has_args(fields, 'hmget')
  fields.map{|f| hget(key, f)}
end

#hmset(key, *kvpairs) ⇒ Object



85
86
87
88
89
90
91
92
93
94
95
# File 'lib/mock_redis/hash_methods.rb', line 85

def hmset(key, *kvpairs)
  assert_has_args(kvpairs, 'hmset')
  if kvpairs.length.odd?
    raise Redis::CommandError, "ERR wrong number of arguments for HMSET"
  end

  kvpairs.each_slice(2) do |(k,v)|
    hset(key, k, v)
  end
  'OK'
end

#hset(key, field, value) ⇒ Object



107
108
109
110
# File 'lib/mock_redis/hash_methods.rb', line 107

def hset(key, field, value)
  with_hash_at(key) {|h| h[field.to_s] = value.to_s}
  true
end

#hsetnx(key, field, value) ⇒ Object



112
113
114
115
116
117
118
119
# File 'lib/mock_redis/hash_methods.rb', line 112

def hsetnx(key, field, value)
  if hget(key, field)
    false
  else
    hset(key, field, value)
    true
  end
end

#hvals(key) ⇒ Object



121
122
123
# File 'lib/mock_redis/hash_methods.rb', line 121

def hvals(key)
  with_hash_at(key, &:values)
end

#mapped_hmget(key, *fields) ⇒ Object



80
81
82
83
# File 'lib/mock_redis/hash_methods.rb', line 80

def mapped_hmget(key, *fields)
  reply = hmget(key, *fields)
  Hash[*fields.zip(reply).flatten]
end

#mapped_hmset(key, hash) ⇒ Object



97
98
99
100
101
102
103
104
105
# File 'lib/mock_redis/hash_methods.rb', line 97

def mapped_hmset(key, hash)
  kvpairs = hash.to_a.flatten
  assert_has_args(kvpairs, 'hmset')
  if kvpairs.length.odd?
    raise Redis::CommandError, "ERR wrong number of arguments for 'hmset' command"
  end

  hmset(key, *kvpairs)
end