Class: MockRedis::Database

Inherits:
Object
  • Object
show all
Includes:
HashMethods, InfoMethod, ListMethods, SetMethods, SortMethod, StringMethods, ZsetMethods
Defined in:
lib/mock_redis/database.rb

Constant Summary

Constants included from InfoMethod

InfoMethod::ALL_INFO, InfoMethod::CLIENTS_INFO, InfoMethod::COMMAND_STATS_COMBINED_INFO, InfoMethod::COMMAND_STATS_SOLO_INFO, InfoMethod::CPU_INFO, InfoMethod::DEFAULT_INFO, InfoMethod::KEYSPACE_INFO, InfoMethod::MEMORY_INFO, InfoMethod::PERSISTENCE_INFO, InfoMethod::REPLICATION_INFO, InfoMethod::SERVER_INFO, InfoMethod::STATS_INFO

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from InfoMethod

#info

Methods included from SortMethod

#sort

Methods included from ZsetMethods

#zadd, #zcard, #zcount, #zincrby, #zinterstore, #zrange, #zrangebyscore, #zrank, #zrem, #zremrangebyrank, #zremrangebyscore, #zrevrange, #zrevrangebyscore, #zrevrank, #zscore, #zunionstore

Methods included from StringMethods

#[], #[]=, #append, #decr, #decrby, #get, #getbit, #getrange, #getset, #incr, #incrby, #incrbyfloat, #mget, #mset, #msetnx, #set, #setbit, #setex, #setnx, #setrange, #strlen

Methods included from SetMethods

#sadd, #scard, #sdiff, #sdiffstore, #sinter, #sinterstore, #sismember, #smembers, #smove, #spop, #srandmember, #srem, #sunion, #sunionstore

Methods included from ListMethods

#blpop, #brpop, #brpoplpush, #lindex, #linsert, #llen, #lpop, #lpush, #lpushx, #lrange, #lrem, #lset, #ltrim, #rpop, #rpoplpush, #rpush, #rpushx

Methods included from HashMethods

#hdel, #hexists, #hget, #hgetall, #hincrby, #hkeys, #hlen, #hmget, #hmset, #hset, #hsetnx, #hvals, #mapped_hmget, #mapped_hmset

Constructor Details

#initialize(base, *args) ⇒ Database

Returns a new instance of Database.



24
25
26
27
28
 
# File 'lib/mock_redis/database.rb', line 24

def initialize(base, *args)
  @base = base
  @data = MockRedis::IndifferentHash.new
  @expire_times = []
end

Instance Attribute Details

#data ⇒ Object (readonly)

Returns the value of attribute data.



22
23
24
 
# File 'lib/mock_redis/database.rb', line 22

def data
  @data
end

#expire_times ⇒ Object (readonly)

Returns the value of attribute expire_times.



22
23
24
 
# File 'lib/mock_redis/database.rb', line 22

def expire_times
  @expire_times
end

Instance Method Details

#auth(_) ⇒ Object

Redis commands go below this line and above ‘private’



38
 
# File 'lib/mock_redis/database.rb', line 38

def auth(_) 'OK' end

#bgrewriteaof ⇒ Object 



40
 
# File 'lib/mock_redis/database.rb', line 40

def bgrewriteaof() "Background append only file rewriting started" end

#bgsave ⇒ Object 



42
 
# File 'lib/mock_redis/database.rb', line 42

def bgsave() "Background saving started" end

#connected? ⇒ Boolean

Returns:

  • (Boolean) 


48
49
50
 
# File 'lib/mock_redis/database.rb', line 48

def connected?
  true
end

#dbsize ⇒ Object 



52
53
54
 
# File 'lib/mock_redis/database.rb', line 52

def dbsize
  data.keys.length
end

#del(*keys) ⇒ Object 



56
57
58
59
60
61
62
 
# File 'lib/mock_redis/database.rb', line 56

def del(*keys)
  keys.
    find_all{|key| data[key]}.
    each {|k| persist(k)}.
    each {|k| data.delete(k)}.
    length
end

#disconnect ⇒ Object 



44
45
46
 
# File 'lib/mock_redis/database.rb', line 44

def disconnect
  nil
end

#echo(msg) ⇒ Object 



64
65
66
 
# File 'lib/mock_redis/database.rb', line 64

def echo(msg)
  msg.to_s
end

#exists(key) ⇒ Object 



85
86
87
 
# File 'lib/mock_redis/database.rb', line 85

def exists(key)
  data.has_key?(key)
end

#expire(key, seconds) ⇒ Object 



68
69
70
 
# File 'lib/mock_redis/database.rb', line 68

def expire(key, seconds)
  expireat(key, @base.now.to_i + seconds.to_i)
end

#expire_keys ⇒ Object

This method isn’t private, but it also isn’t a Redis command, so it doesn’t belong up above with all the Redis commands.



247
248
249
250
251
252
253
254
255
256
257
258
259
 
# File 'lib/mock_redis/database.rb', line 247

def expire_keys
  now = @base.now

  to_delete = expire_times.take_while do |(time, key)|
    time <= now
  end

  to_delete.each do |(time, key)|
    del(key)
  end

  expire_times.slice!(0, to_delete.length)
end

#expireat(key, timestamp) ⇒ Object 



72
73
74
75
76
77
78
79
80
81
82
83
 
# File 'lib/mock_redis/database.rb', line 72

def expireat(key, timestamp)
  unless looks_like_integer?(timestamp.to_s)
    raise Redis::CommandError, "ERR value is not an integer or out of range"
  end

  if exists(key)
    set_expiration(key, @base.time_at(timestamp.to_i))
    true
  else
    false
  end
end

#flushdb ⇒ Object 



89
90
91
92
 
# File 'lib/mock_redis/database.rb', line 89

def flushdb
  data.keys.each {|k| del(k)}
  'OK'
end

#initialize_copy(source) ⇒ Object 



30
31
32
33
34
 
# File 'lib/mock_redis/database.rb', line 30

def initialize_copy(source)
  @data = @data.clone
  @data.keys.each {|k| @data[k] = @data[k].clone}
  @expire_times = @expire_times.map{|x| x.clone}
end

#keys(format = '*') ⇒ Object 



94
95
96
 
# File 'lib/mock_redis/database.rb', line 94

def keys(format = '*')
  data.keys.grep(redis_pattern_to_ruby_regex(format))
end

#lastsave ⇒ Object 



98
99
100
 
# File 'lib/mock_redis/database.rb', line 98

def lastsave
  @base.now.to_i
end

#persist(key) ⇒ Object 



102
103
104
105
106
107
108
109
 
# File 'lib/mock_redis/database.rb', line 102

def persist(key)
  if exists(key) && has_expiration?(key)
    remove_expiration(key)
    true
  else
    false
  end
end

#ping ⇒ Object 



111
112
113
 
# File 'lib/mock_redis/database.rb', line 111

def ping
  'PONG'
end

#quit ⇒ Object 



115
116
117
 
# File 'lib/mock_redis/database.rb', line 115

def quit
  'OK'
end

#randomkey ⇒ Object 



119
120
121
 
# File 'lib/mock_redis/database.rb', line 119

def randomkey
  data.keys[rand(data.length)]
end

#rename(key, newkey) ⇒ Object 



123
124
125
126
127
128
129
130
131
 
# File 'lib/mock_redis/database.rb', line 123

def rename(key, newkey)
  if !data.include?(key)
    raise Redis::CommandError, "ERR no such key"
  elsif key == newkey
    raise Redis::CommandError, "ERR source and destination objects are the same"
  end
  data[newkey] = data.delete(key)
  'OK'
end

#renamenx(key, newkey) ⇒ Object 



133
134
135
136
137
138
139
140
141
142
143
144
145
 
# File 'lib/mock_redis/database.rb', line 133

def renamenx(key, newkey)
  if !data.include?(key)
    raise Redis::CommandError, "ERR no such key"
  elsif key == newkey
    raise Redis::CommandError, "ERR source and destination objects are the same"
  end
  if exists(newkey)
    false
  else
    rename(key, newkey)
    true
  end
end

#save ⇒ Object 



147
148
149
 
# File 'lib/mock_redis/database.rb', line 147

def save
  'OK'
end

#ttl(key) ⇒ Object 



151
152
153
154
155
156
157
 
# File 'lib/mock_redis/database.rb', line 151

def ttl(key)
  if has_expiration?(key)
    expiration(key).to_i - @base.now.to_i
  else
    -1
  end
end

#type(key) ⇒ Object 



159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
 
# File 'lib/mock_redis/database.rb', line 159

def type(key)
  if !exists(key)
    'none'
  elsif hashy?(key)
    'hash'
  elsif stringy?(key)
    'string'
  elsif listy?(key)
    'list'
  elsif sety?(key)
    'set'
  elsif zsety?(key)
    'zset'
  else
    raise ArgumentError, "Not sure how #{data[key].inspect} got in here"
  end
end