Module: Rediska::Driver

Included in:
Connection
Defined in:
lib/rediska/driver.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#database_id ⇒ Object

Returns the value of attribute database_id.



3
4
5
 
# File 'lib/rediska/driver.rb', line 3

def database_id
  @database_id
end

#replies ⇒ Object 



6
7
8
 
# File 'lib/rediska/driver.rb', line 6

def replies
  @replies ||= []
end

Instance Method Details

#[](key) ⇒ Object 



760
761
762
 
# File 'lib/rediska/driver.rb', line 760

def [](key)
  get(key)
end

#[]=(key, value) ⇒ Object 



764
765
766
 
# File 'lib/rediska/driver.rb', line 764

def []=(key, value)
  set(key, value)
end

#append(key, value) ⇒ Object 



166
167
168
169
 
# File 'lib/rediska/driver.rb', line 166

def append(key, value)
  data[key] = (data[key] || '')
  data[key] = data[key] + value.to_s
end

#auth(*args) ⇒ Object 



56
57
58
 
# File 'lib/rediska/driver.rb', line 56

def auth(*args)
  'OK'
end

#bgrewriteaof ⇒ Object 



82
83
 
# File 'lib/rediska/driver.rb', line 82

def bgrewriteaof
end

#bgsave ⇒ Object 



79
80
 
# File 'lib/rediska/driver.rb', line 79

def bgsave
end

#bitcount(key, start_index = 0, end_index = -1)) ⇒ Object 



139
140
141
142
143
 
# File 'lib/rediska/driver.rb', line 139

def bitcount(key, start_index = 0, end_index = -1)
  return 0 unless data[key]

  data[key][start_index..end_index].unpack('B*')[0].count('1')
end

#blpop(keys, timeout = 0) ⇒ Object 



421
422
423
424
425
426
427
428
429
430
 
# File 'lib/rediska/driver.rb', line 421

def blpop(keys, timeout = 0)
  keys = Array(keys)
  keys.each do |key|
    if data[key] && data[key].size > 0
      return [key, data[key].shift]
    end
  end
  sleep(timeout.to_f)
  nil
end

#brpop(keys, timeout = 0) ⇒ Object 



390
391
392
393
394
395
396
397
398
399
 
# File 'lib/rediska/driver.rb', line 390

def brpop(keys, timeout = 0)
  keys = Array(keys)
  keys.each do |key|
    if data[key] && data[key].size > 0
      return [key, data[key].pop]
    end
  end
  sleep(timeout.to_f)
  nil
end

#brpoplpush(key1, key2, opts = {}) ⇒ Object 



408
409
410
411
412
413
 
# File 'lib/rediska/driver.rb', line 408

def brpoplpush(key1, key2, opts = {})
  data_type_check(key1, Array)
  brpop(key1).tap do |elem|
    lpush(key2, elem) unless elem.nil?
  end
end

#client(command, _options = {}) ⇒ Object 



20
21
22
23
24
25
26
27
28
 
# File 'lib/rediska/driver.rb', line 20

def client(command, _options = {})
  case command
  when :setname then true
  when :getname then nil
  when :client then true
  else
    raise Redis::CommandError, "ERR unknown command '#{command}'"
  end
end

#connect_unix(path, timeout) ⇒ Object 



14
15
 
# File 'lib/rediska/driver.rb', line 14

def connect_unix(path, timeout)
end

#connected? ⇒ Boolean

Returns:

  • (Boolean) 


10
11
12
 
# File 'lib/rediska/driver.rb', line 10

def connected?
  true
end

#dbsize ⇒ Object 



272
273
274
 
# File 'lib/rediska/driver.rb', line 272

def dbsize
  data.keys.count
end

#decr(key) ⇒ Object 



853
854
855
856
 
# File 'lib/rediska/driver.rb', line 853

def decr(key)
  data.merge!({ key => (data[key].to_i - 1).to_s || '-1'})
  data[key].to_i
end

#decrby(key, by) ⇒ Object 



858
859
860
861
 
# File 'lib/rediska/driver.rb', line 858

def decrby(key, by)
  data.merge!({ key => ((data[key].to_i - by.to_i) || (by.to_i * -1)).to_s })
  data[key].to_i
end

#del(*keys) ⇒ Object 



586
587
588
589
590
591
592
593
594
595
 
# File 'lib/rediska/driver.rb', line 586

def del(*keys)
  keys = keys.flatten(1)
  raise_argument_error('del') if keys.empty?

  old_count = data.keys.size
  keys.each do |key|
    data.delete(key)
  end
  old_count - data.keys.size
end

#disconnect ⇒ Object 



17
18
 
# File 'lib/rediska/driver.rb', line 17

def disconnect
end

#dump(key) ⇒ Object 



94
95
96
97
98
 
# File 'lib/rediska/driver.rb', line 94

def dump(key)
  return nil unless exists(key)

  Marshal.dump(value: data[key], version: Rediska::VERSION)
end

#echo(string) ⇒ Object 



255
256
257
 
# File 'lib/rediska/driver.rb', line 255

def echo(string)
  string
end

#exists(key) ⇒ Object 



276
277
278
 
# File 'lib/rediska/driver.rb', line 276

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

#expire(key, ttl) ⇒ Object 



622
623
624
625
626
 
# File 'lib/rediska/driver.rb', line 622

def expire(key, ttl)
  return 0 unless data[key]
  data.expires[key] = Time.now + ttl
  1
end

#expireat(key, timestamp) ⇒ Object 



652
653
654
655
 
# File 'lib/rediska/driver.rb', line 652

def expireat(key, timestamp)
  data.expires[key] = Time.at(timestamp)
  true
end

#flushall ⇒ Object 



49
50
51
52
53
54
 
# File 'lib/rediska/driver.rb', line 49

def flushall
  db_class.flushall(database_instance_key)
  self.class.databases[database_instance_key] = []

  'OK'
end

#flushdb ⇒ Object

NOT IMPLEMENTED:

  • subscribe

  • psubscribe

  • publish



42
43
44
45
46
47
 
# File 'lib/rediska/driver.rb', line 42

def flushdb
  db_class.flushdb(database_instance_key, database_id)
  databases.delete_at(database_id)

  'OK'
end

#get(key) ⇒ Object 



129
130
131
132
 
# File 'lib/rediska/driver.rb', line 129

def get(key)
  data_type_check(key, String)
  data[key]
end

#getbit(key, offset) ⇒ Object 



134
135
136
137
 
# File 'lib/rediska/driver.rb', line 134

def getbit(key, offset)
  return unless data[key]
  data[key].unpack('B*')[0].split('')[offset].to_i
end

#getrange(key, start, ending) ⇒ Object Also known as: substr 



145
146
147
148
 
# File 'lib/rediska/driver.rb', line 145

def getrange(key, start, ending)
  return unless data[key]
  data[key][start..ending]
end

#getset(key, value) ⇒ Object 



151
152
153
154
155
156
 
# File 'lib/rediska/driver.rb', line 151

def getset(key, value)
  data_type_check(key, String)
  data[key].tap do
    set(key, value)
  end
end

#hdel(key, field) ⇒ Object 



186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
 
# File 'lib/rediska/driver.rb', line 186

def hdel(key, field)
  data_type_check(key, Hash)
  return 0 unless data[key]

  if field.is_a?(Array)
    old_keys_count = data[key].size
    fields = field.map(&:to_s)

    data[key].delete_if { |k, v| fields.include? k }
    deleted = old_keys_count - data[key].size
  else
    field = field.to_s
    deleted = data[key].delete(field) ? 1 : 0
  end

  remove_key_for_empty_collection(key)
  deleted
end

#hexists(key, field) ⇒ Object 



752
753
754
755
756
 
# File 'lib/rediska/driver.rb', line 752

def hexists(key, field)
  data_type_check(key, Hash)
  return false unless data[key]
  data[key].key?(field.to_s)
end

#hget(key, field) ⇒ Object 



181
182
183
184
 
# File 'lib/rediska/driver.rb', line 181

def hget(key, field)
  data_type_check(key, Hash)
  data[key] && data[key][field.to_s]
end

#hgetall(key) ⇒ Object 



176
177
178
179
 
# File 'lib/rediska/driver.rb', line 176

def hgetall(key)
  data_type_check(key, Hash)
  data[key].to_a.flatten || {}
end

#hincrby(key, field, increment) ⇒ Object 



730
731
732
733
734
735
736
737
738
739
 
# File 'lib/rediska/driver.rb', line 730

def hincrby(key, field, increment)
  data_type_check(key, Hash)
  field = field.to_s
  if data[key]
    data[key][field] = (data[key][field].to_i + increment.to_i).to_s
  else
    data[key] = { field => increment.to_s }
  end
  data[key][field].to_i
end

#hincrbyfloat(key, field, increment) ⇒ Object 



741
742
743
744
745
746
747
748
749
750
 
# File 'lib/rediska/driver.rb', line 741

def hincrbyfloat(key, field, increment)
  data_type_check(key, Hash)
  field = field.to_s
  if data[key]
    data[key][field] = (data[key][field].to_f + increment.to_f).to_s
  else
    data[key] = { field => increment.to_s }
  end
  data[key][field]
end

#hkeys(key) ⇒ Object 



205
206
207
208
209
 
# File 'lib/rediska/driver.rb', line 205

def hkeys(key)
  data_type_check(key, Hash)
  return [] if data[key].nil?
  data[key].keys
end

#hlen(key) ⇒ Object 



718
719
720
721
722
 
# File 'lib/rediska/driver.rb', line 718

def hlen(key)
  data_type_check(key, Hash)
  return 0 unless data[key]
  data[key].size
end

#hmget(key, *fields) ⇒ Object 



704
705
706
707
708
709
710
711
712
713
714
715
716
 
# File 'lib/rediska/driver.rb', line 704

def hmget(key, *fields)
  raise_argument_error('hmget') if fields.empty?

  data_type_check(key, Hash)
  fields.flatten.map do |field|
    field = field.to_s
    if data[key]
      data[key][field]
    else
      nil
    end
  end
end

#hmset(key, *fields) ⇒ Object 



681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
 
# File 'lib/rediska/driver.rb', line 681

def hmset(key, *fields)
  fields = fields[0] if mapped_param?(fields)
  raise_argument_error('hmset') if fields.empty?

  is_list_of_arrays = fields.all?{|field| field.instance_of?(Array)}

  raise_argument_error('hmset') if fields.size.odd? and !is_list_of_arrays
  raise_argument_error('hmset') if is_list_of_arrays and !fields.all?{|field| field.length == 2}

  data_type_check(key, Hash)
  data[key] ||= {}

  if is_list_of_arrays
    fields.each do |pair|
      data[key][pair[0].to_s] = pair[1].to_s
    end
  else
    fields.each_slice(2) do |field|
      data[key][field[0].to_s] = field[1].to_s
    end
  end
end

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



211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
 
# File 'lib/rediska/driver.rb', line 211

def hscan(key, start_cursor, *args)
  data_type_check(key, Hash)
  return ['0', []] unless data[key]

  match = '*'
  count = 10

  raise_argument_error('hscan') if args.size.odd?

  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

  cursor = start_cursor
  next_keys = []

  if start_cursor + count >= data[key].length
    next_keys = (data[key].to_a)[start_cursor..-1]
    cursor = 0
  else
    cursor = start_cursor + count
    next_keys = (data[key].to_a)[start_cursor..cursor-1]
  end

  filtered_next_keys = next_keys.select{|k,v| File.fnmatch(match, k)}
  result = filtered_next_keys.flatten.map(&:to_s)

  return ["#{cursor}", result]
end

#hset(key, field, value) ⇒ Object 



661
662
663
664
665
666
667
668
669
670
671
672
 
# File 'lib/rediska/driver.rb', line 661

def hset(key, field, value)
  data_type_check(key, Hash)
  field = field.to_s
  if data[key]
    result = !data[key].include?(field)
    data[key][field] = value.to_s
    result
  else
    data[key] = { field => value.to_s }
    true
  end
end

#hsetnx(key, field, value) ⇒ Object 



674
675
676
677
678
679
 
# File 'lib/rediska/driver.rb', line 674

def hsetnx(key, field, value)
  data_type_check(key, Hash)
  field = field.to_s
  return false if data[key] && data[key][field]
  hset(key, field, value)
end

#hvals(key) ⇒ Object 



724
725
726
727
728
 
# File 'lib/rediska/driver.rb', line 724

def hvals(key)
  data_type_check(key, Hash)
  return [] unless data[key]
  data[key].values
end

#incr(key) ⇒ Object 



838
839
840
841
 
# File 'lib/rediska/driver.rb', line 838

def incr(key)
  data.merge!({ key => (data[key].to_i + 1).to_s || '1'})
  data[key].to_i
end

#incrby(key, by) ⇒ Object 



843
844
845
846
 
# File 'lib/rediska/driver.rb', line 843

def incrby(key, by)
  data.merge!({ key => (data[key].to_i + by.to_i).to_s || by })
  data[key].to_i
end

#incrbyfloat(key, by) ⇒ Object 



848
849
850
851
 
# File 'lib/rediska/driver.rb', line 848

def incrbyfloat(key, by)
  data.merge!({ key => (data[key].to_f + by.to_f).to_s || by })
  data[key]
end

#info ⇒ Object 



67
68
69
70
71
 
# File 'lib/rediska/driver.rb', line 67

def info
  {
    'redis_version' => '3.0.5',
  }
end

#keys(pattern = '*') ⇒ Object 



247
248
249
 
# File 'lib/rediska/driver.rb', line 247

def keys(pattern = '*')
  data.keys.select { |key| File.fnmatch(pattern, key) }
end

#lastsave ⇒ Object 



263
264
265
 
# File 'lib/rediska/driver.rb', line 263

def lastsave
  Time.now.to_i
end

#lindex(key, index) ⇒ Object 



305
306
307
308
 
# File 'lib/rediska/driver.rb', line 305

def lindex(key, index)
  data_type_check(key, Array)
  data[key] && data[key][index]
end

#linsert(key, where, pivot, value) ⇒ Object 



310
311
312
313
314
315
316
317
318
319
 
# File 'lib/rediska/driver.rb', line 310

def linsert(key, where, pivot, value)
  data_type_check(key, Array)
  return unless data[key]
  index = data[key].index(pivot)
  case where
    when :before then data[key].insert(index, value)
    when :after  then data[key].insert(index + 1, value)
    else raise_syntax_error
  end
end

#llen(key) ⇒ Object 



280
281
282
283
284
 
# File 'lib/rediska/driver.rb', line 280

def llen(key)
  data_type_check(key, Array)
  return 0 unless data[key]
  data[key].size
end

#lpop(key) ⇒ Object 



415
416
417
418
419
 
# File 'lib/rediska/driver.rb', line 415

def lpop(key)
  data_type_check(key, Array)
  return unless data[key]
  data[key].shift
end

#lpush(key, value) ⇒ Object 



365
366
367
368
369
370
371
372
373
374
 
# File 'lib/rediska/driver.rb', line 365

def lpush(key, value)
  raise_argument_error('lpush') if value.respond_to?(:each) && value.empty?

  data_type_check(key, Array)
  data[key] ||= []
  [value].flatten.each do |val|
    data[key].unshift(val.to_s)
  end
  data[key].size
end

#lpushx(key, value) ⇒ Object 



376
377
378
379
380
381
382
 
# File 'lib/rediska/driver.rb', line 376

def lpushx(key, value)
  raise_argument_error('lpushx') if value.respond_to?(:each) && value.empty?

  data_type_check(key, Array)
  return unless data[key]
  lpush(key, value)
end

#lrange(key, startidx, endidx) ⇒ Object 



286
287
288
289
 
# File 'lib/rediska/driver.rb', line 286

def lrange(key, startidx, endidx)
  data_type_check(key, Array)
  (data[key] && data[key][startidx..endidx]) || []
end

#lrem(key, count, value) ⇒ Object 



328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
 
# File 'lib/rediska/driver.rb', line 328

def lrem(key, count, value)
  data_type_check(key, Array)
  return 0 unless data[key]
  old_size = data[key].size
  diff =
    if count == 0
      data[key].delete(value)
      old_size - data[key].size
    else
      array = count > 0 ? data[key].dup : data[key].reverse
      count.abs.times{ array.delete_at(array.index(value) || array.length) }
      data[key] = count > 0 ? array.dup : array.reverse
      old_size - data[key].size
    end
  remove_key_for_empty_collection(key)
  diff
end

#lset(key, index, value) ⇒ Object

Raises:

  • (Redis::CommandError) 


321
322
323
324
325
326
 
# File 'lib/rediska/driver.rb', line 321

def lset(key, index, value)
  data_type_check(key, Array)
  return unless data[key]
  raise Redis::CommandError, 'ERR index out of range' if index >= data[key].size
  data[key][index] = value
end

#ltrim(key, start, stop) ⇒ Object 



291
292
293
294
295
296
297
298
299
300
301
302
303
 
# File 'lib/rediska/driver.rb', line 291

def ltrim(key, start, stop)
  data_type_check(key, Array)
  return unless data[key]

  # Example: we have a list of 3 elements and we give it a ltrim list, -5, -1. This means it
  # should trim to a max of 5. Since 3 < 5 we should not touch the list. This is consistent
  # with behavior of real Redis's ltrim with a negative start argument.
  unless start < 0 && data[key].count < start.abs
    data[key] = data[key][start..stop]
  end

  'OK'
end

#mget(*keys) ⇒ Object 



158
159
160
161
162
163
164
 
# File 'lib/rediska/driver.rb', line 158

def mget(*keys)
  raise_argument_error('mget') if keys.empty?

  # We work with either an array, or list of arguments
  keys = keys.first if keys.size == 1
  data.values_at(*keys)
end

#monitor ⇒ Object 



73
74
 
# File 'lib/rediska/driver.rb', line 73

def monitor
end

#move(key, destination_id) ⇒ Object

Raises:

  • (Redis::CommandError) 


85
86
87
88
89
90
91
92
 
# File 'lib/rediska/driver.rb', line 85

def move(key, destination_id)
  raise Redis::CommandError, 'ERR source and destination objects are the same' if destination_id == database_id
  destination = find_database(destination_id)
  return false unless data.has_key?(key)
  return false if destination.has_key?(key)
  destination[key] = data.delete(key)
  true
end

#mset(*pairs) ⇒ Object 



813
814
815
816
817
818
819
820
821
822
823
824
825
826
 
# File 'lib/rediska/driver.rb', line 813

def mset(*pairs)
  # Handle pairs for mapped_mset command
  pairs = pairs[0] if mapped_param?(pairs)
  raise_argument_error('mset') if pairs.empty? || pairs.size == 1
  # We have to reply with a different error message here to be consistent with
  # redis-rb 3.0.6 / redis-server 2.8.1.
  raise_argument_error('mset', 'mset_odd') if pairs.size.odd?

  pairs.each_slice(2) do |pair|
    data[pair[0].to_s] = pair[1].to_s
  end

  'OK'
end

#msetnx(*pairs) ⇒ Object 



828
829
830
831
832
833
834
835
836
 
# File 'lib/rediska/driver.rb', line 828

def msetnx(*pairs)
  # Handle pairs for mapped_msetnx command
  pairs = pairs[0] if mapped_param?(pairs)
  keys = []
  pairs.each_with_index{|item, index| keys << item.to_s if index % 2 == 0}
  return false if keys.any?{|key| data.key?(key) }
  mset(*pairs)
  true
end

#persist(key) ⇒ Object 



657
658
659
 
# File 'lib/rediska/driver.rb', line 657

def persist(key)
  !!data.expires.delete(key)
end

#pexpire(key, ttl) ⇒ Object 



628
629
630
631
632
633
634
 
# File 'lib/rediska/driver.rb', line 628

def pexpire(key, ttl)
  return 0 unless data[key]

  data.expires[key] = Time.now + (ttl / 1000.0)

  1
end

#ping ⇒ Object 



259
260
261
 
# File 'lib/rediska/driver.rb', line 259

def ping
  'PONG'
end

#pttl(key) ⇒ Object 



644
645
646
647
648
649
650
 
# File 'lib/rediska/driver.rb', line 644

def pttl(key)
  if data.expires.include?(key) && (ttl = data.expires[key].to_f - Time.now.to_f) > 0
    ttl * 1000
  else
    exists(key) ? -1 : -2
  end
end

#quit ⇒ Object 



874
875
 
# File 'lib/rediska/driver.rb', line 874

def quit
end

#randomkey ⇒ Object 



251
252
253
 
# File 'lib/rediska/driver.rb', line 251

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

#read ⇒ Object 



33
34
35
 
# File 'lib/rediska/driver.rb', line 33

def read
  replies.shift
end

#rename(key, new_key) ⇒ Object 



606
607
608
609
610
611
 
# File 'lib/rediska/driver.rb', line 606

def rename(key, new_key)
  return unless data[key]
  data[new_key] = data[key]
  data.expires[new_key] = data.expires[key] if data.expires.include?(key)
  data.delete(key)
end

#renamenx(key, new_key) ⇒ Object 



613
614
615
616
617
618
619
620
 
# File 'lib/rediska/driver.rb', line 613

def renamenx(key, new_key)
  if exists(new_key)
    false
  else
    rename(key, new_key)
    true
  end
end

#restore(key, ttl, serialized_value) ⇒ Object

Raises:

  • (Redis::CommandError) 


100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
 
# File 'lib/rediska/driver.rb', line 100

def restore(key, ttl, serialized_value)
  raise Redis::CommandError, 'ERR Target key name is busy.' if exists(key)

  if serialized_value.nil?
    raise Redis::CommandError, 'ERR DUMP payload version or checksum are wrong'
  end

  parsed_value = begin
    Marshal.load(serialized_value)
  rescue TypeError
    raise Redis::CommandError, 'ERR DUMP payload version or checksum are wrong'
  end

  if parsed_value[:version] != Rediska::VERSION
    raise Redis::CommandError, 'ERR DUMP payload version or checksum are wrong'
  end

  # We could figure out what type the key was and set it with the public API here, or we could
  # just assign the value. If we presume the serialized_value is only ever a return value from
  # `dump` then we've only been given something that was in the internal data structure anyway.
  data[key] = parsed_value[:value]

  # Set a TTL if one has been passed
  ttl = ttl.to_i # Makes nil into 0
  expire(key, ttl / 1000) unless ttl.zero?

  'OK'
end

#rpop(key) ⇒ Object 



384
385
386
387
388
 
# File 'lib/rediska/driver.rb', line 384

def rpop(key)
  data_type_check(key, Array)
  return unless data[key]
  data[key].pop
end

#rpoplpush(key1, key2) ⇒ Object 



401
402
403
404
405
406
 
# File 'lib/rediska/driver.rb', line 401

def rpoplpush(key1, key2)
  data_type_check(key1, Array)
  rpop(key1).tap do |elem|
    lpush(key2, elem) unless elem.nil?
  end
end

#rpush(key, value) ⇒ Object 



346
347
348
349
350
351
352
353
354
355
 
# File 'lib/rediska/driver.rb', line 346

def rpush(key, value)
  raise_argument_error('rpush') if value.respond_to?(:each) && value.empty?

  data_type_check(key, Array)
  data[key] ||= []
  [value].flatten.each do |val|
    data[key].push(val.to_s)
  end
  data[key].size
end

#rpushx(key, value) ⇒ Object 



357
358
359
360
361
362
363
 
# File 'lib/rediska/driver.rb', line 357

def rpushx(key, value)
  raise_argument_error('rpushx') if value.respond_to?(:each) && value.empty?

  data_type_check(key, Array)
  return unless data[key]
  rpush(key, value)
end

#sadd(key, value) ⇒ Object 



444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
 
# File 'lib/rediska/driver.rb', line 444

def sadd(key, value)
  data_type_check(key, ::Set)
  value = Array(value)
  raise_argument_error('sadd') if value.empty?

  result = if data[key]
    old_set = data[key].dup
    data[key].merge(value.map(&:to_s))
    (data[key] - old_set).size
  else
    data[key] = ::Set.new(value.map(&:to_s))
    data[key].size
  end

  # 0 = false, 1 = true, 2+ untouched
  return result == 1 if result < 2
  result
end

#save ⇒ Object 



76
77
 
# File 'lib/rediska/driver.rb', line 76

def save
end

#scan(start_cursor, *args) ⇒ Object 



883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
 
# File 'lib/rediska/driver.rb', line 883

def scan(start_cursor, *args)
  match = '*'
  count = 10

  if args.size.odd?
    raise_argument_error('scan')
  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 = []

  if start_cursor + count >= data.length
    next_keys = keys(match)[start_cursor..-1]
    cursor = 0
  else
    cursor = start_cursor + 10
    next_keys = keys(match)[start_cursor..cursor]
  end

  return "#{cursor}", next_keys
end

#scard(key) ⇒ Object 



493
494
495
496
497
 
# File 'lib/rediska/driver.rb', line 493

def scard(key)
  data_type_check(key, ::Set)
  return 0 unless data[key]
  data[key].size
end

#sdiff(key1, *keys) ⇒ Object 



530
531
532
533
534
535
536
 
# File 'lib/rediska/driver.rb', line 530

def sdiff(key1, *keys)
  [key1, *keys].each { |k| data_type_check(k, ::Set) }
  keys = keys.map { |k| data[k] || ::Set.new }
  keys.inject(data[key1] || Set.new) do |memo, set|
    memo - set
  end.to_a
end

#sdiffstore(destination, key1, *keys) ⇒ Object 



538
539
540
541
542
 
# File 'lib/rediska/driver.rb', line 538

def sdiffstore(destination, key1, *keys)
  data_type_check(destination, ::Set)
  result = sdiff(key1, *keys)
  data[destination] = ::Set.new(result)
end

#select(index) ⇒ Object 



60
61
62
63
64
65
 
# File 'lib/rediska/driver.rb', line 60

def select(index)
  data_type_check(index, Integer)
  self.database_id = index

  'OK'
end

#set(key, value, *array_options) ⇒ Object 



768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
 
# File 'lib/rediska/driver.rb', line 768

def set(key, value, *array_options)
  option_nx = array_options.delete('NX')
  option_xx = array_options.delete('XX')

  return false if option_nx && option_xx

  return false if option_nx && exists(key)
  return false if option_xx && !exists(key)

  data[key] = value.to_s

  options = Hash[array_options.each_slice(2).to_a]
  ttl_in_seconds = options['EX'] if options['EX']
  ttl_in_seconds = options['PX'] / 1000.0 if options['PX']

  expire(key, ttl_in_seconds) if ttl_in_seconds

  'OK'
end

#setbit(key, offset, bit) ⇒ Object 



788
789
790
791
792
793
794
795
796
797
798
 
# File 'lib/rediska/driver.rb', line 788

def setbit(key, offset, bit)
  old_val = data[key] ? data[key].unpack('B*')[0].split('') : []
  size_increment = [((offset/8)+1)*8-old_val.length, 0].max
  old_val += Array.new(size_increment).map{'0'}
  original_val = old_val[offset].to_i
  old_val[offset] = bit.to_s
  new_val = ''
  old_val.each_slice(8){|b| new_val = new_val + b.join('').to_i(2).chr }
  data[key] = new_val
  original_val
end

#setex(key, seconds, value) ⇒ Object 



800
801
802
803
804
805
 
# File 'lib/rediska/driver.rb', line 800

def setex(key, seconds, value)
  data[key] = value.to_s
  expire(key, seconds)

  'OK'
end

#setnx(key, value) ⇒ Object 



597
598
599
600
601
602
603
604
 
# File 'lib/rediska/driver.rb', line 597

def setnx(key, value)
  if exists(key)
    0
  else
    set(key, value)
    1
  end
end

#setrange(key, offset, value) ⇒ Object 



807
808
809
810
811
 
# File 'lib/rediska/driver.rb', line 807

def setrange(key, offset, value)
  return unless data[key]
  s = data[key][offset,value.size]
  data[key][s] = value
end

#shutdown ⇒ Object 



877
878
 
# File 'lib/rediska/driver.rb', line 877

def shutdown
end

#sinter(*keys) ⇒ Object 



499
500
501
502
503
504
505
506
507
508
 
# File 'lib/rediska/driver.rb', line 499

def sinter(*keys)
  raise_argument_error('sinter') if keys.empty?

  keys.each { |k| data_type_check(k, ::Set) }
  return ::Set.new if keys.any? { |k| data[k].nil? }
  keys = keys.map { |k| data[k] || ::Set.new }
  keys.inject do |set, key|
    set & key
  end.to_a
end

#sinterstore(destination, *keys) ⇒ Object 



510
511
512
513
514
 
# File 'lib/rediska/driver.rb', line 510

def sinterstore(destination, *keys)
  data_type_check(destination, ::Set)
  result = sinter(*keys)
  data[destination] = ::Set.new(result)
end

#sismember(key, value) ⇒ Object 



438
439
440
441
442
 
# File 'lib/rediska/driver.rb', line 438

def sismember(key, value)
  data_type_check(key, ::Set)
  return false unless data[key]
  data[key].include?(value.to_s)
end

#slaveof(host, port) ⇒ Object 



880
881
 
# File 'lib/rediska/driver.rb', line 880

def slaveof(host, port)
end

#smembers(key) ⇒ Object 



432
433
434
435
436
 
# File 'lib/rediska/driver.rb', line 432

def smembers(key)
  data_type_check(key, ::Set)
  return [] unless data[key]
  data[key].to_a.reverse
end

#smove(source, destination, value) ⇒ Object 



479
480
481
482
483
484
 
# File 'lib/rediska/driver.rb', line 479

def smove(source, destination, value)
  data_type_check(destination, ::Set)
  result = self.srem(source, value)
  self.sadd(destination, value) if result
  result
end

#spop(key) ⇒ Object 



486
487
488
489
490
491
 
# File 'lib/rediska/driver.rb', line 486

def spop(key)
  data_type_check(key, ::Set)
  elem = srandmember(key)
  srem(key, elem)
  elem
end

#srandmember(key, number = nil) ⇒ Object 



544
545
546
 
# File 'lib/rediska/driver.rb', line 544

def srandmember(key, number = nil)
  number.nil? ? srandmember_single(key) : srandmember_multiple(key, number)
end

#srem(key, value) ⇒ Object 



463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
 
# File 'lib/rediska/driver.rb', line 463

def srem(key, value)
  data_type_check(key, ::Set)
  return false unless data[key]

  if value.is_a?(Array)
    old_size = data[key].size
    value.map(&:to_s).each { |v| data[key].delete(v) }
    deleted = old_size - data[key].size
  else
    deleted = !!data[key].delete?(value.to_s)
  end

  remove_key_for_empty_collection(key)
  deleted
end

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



548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
 
# File 'lib/rediska/driver.rb', line 548

def sscan(key, start_cursor, *args)
  data_type_check(key, ::Set)
  return ['0', []] unless data[key]

  match = '*'
  count = 10

  if args.size.odd?
    raise_argument_error('sscan')
  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

  cursor = start_cursor
  next_keys = []

  if start_cursor + count >= data[key].length
    next_keys = (data[key].to_a)[start_cursor..-1]
    cursor = 0
  else
    cursor = start_cursor + count
    next_keys = (data[key].to_a)[start_cursor..cursor-1]
  end

  filtered_next_keys = next_keys.select{ |k,v| File.fnmatch(match, k)}
  result = filtered_next_keys.flatten.map(&:to_s)

  return ["#{cursor}", result]
end

#strlen(key) ⇒ Object 



171
172
173
174
 
# File 'lib/rediska/driver.rb', line 171

def strlen(key)
  return unless data[key]
  data[key].size
end

#sunion(*keys) ⇒ Object 



516
517
518
519
520
521
522
 
# File 'lib/rediska/driver.rb', line 516

def sunion(*keys)
  keys.each { |k| data_type_check(k, ::Set) }
  keys = keys.map { |k| data[k] || ::Set.new }
  keys.inject(::Set.new) do |set, key|
    set | key
  end.to_a
end

#sunionstore(destination, *keys) ⇒ Object 



524
525
526
527
528
 
# File 'lib/rediska/driver.rb', line 524

def sunionstore(destination, *keys)
  data_type_check(destination, ::Set)
  result = sunion(*keys)
  data[destination] = ::Set.new(result)
end

#sync ⇒ Object 



758
 
# File 'lib/rediska/driver.rb', line 758

def sync ; end

#time ⇒ Object 



267
268
269
270
 
# File 'lib/rediska/driver.rb', line 267

def time
  microseconds = (Time.now.to_f * 1000000).to_i
  [microseconds / 1000000, microseconds % 1000000]
end

#timeout=(usecs) ⇒ Object 



30
31
 
# File 'lib/rediska/driver.rb', line 30

def timeout=(usecs)
end

#ttl(key) ⇒ Object 



636
637
638
639
640
641
642
 
# File 'lib/rediska/driver.rb', line 636

def ttl(key)
  if data.expires.include?(key) && (ttl = data.expires[key].to_i - Time.now.to_i) > 0
    ttl
  else
    exists(key) ? -1 : -2
  end
end

#type(key) ⇒ Object 



863
864
865
866
867
868
869
870
871
872
 
# File 'lib/rediska/driver.rb', line 863

def type(key)
  case data[key]
    when nil then 'none'
    when String then 'string'
    when ZSet then 'zset'
    when Hash then 'hash'
    when Array then 'list'
    when ::Set then 'set'
  end
end

#zadd(key, *args) ⇒ Object 



916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
 
# File 'lib/rediska/driver.rb', line 916

def zadd(key, *args)
  if !args.first.is_a?(Array)
    if args.size < 2
      raise_argument_error('zadd')
    elsif args.size.odd?
      raise_syntax_error
    end
  else
    unless args.all? {|pair| pair.size == 2 }
      raise_syntax_error
    end
  end

  data_type_check(key, ZSet)
  data[key] ||= ZSet.new

  if args.size == 2 && !(Array === args.first)
    score, value = args
    exists = !data[key].key?(value.to_s)
    data[key][value.to_s] = score
  else
    # Turn [1, 2, 3, 4] into [[1, 2], [3, 4]] unless it is already
    args = args.each_slice(2).to_a unless args.first.is_a?(Array)
    exists = args.map(&:last).map { |el| data[key].key?(el.to_s) }.count(false)
    args.each { |s, v| data[key][v.to_s] = s }
  end

  exists
end

#zcard(key) ⇒ Object 



959
960
961
962
 
# File 'lib/rediska/driver.rb', line 959

def zcard(key)
  data_type_check(key, ZSet)
  data[key] ? data[key].size : 0
end

#zcount(key, min, max) ⇒ Object 



964
965
966
967
968
 
# File 'lib/rediska/driver.rb', line 964

def zcount(key, min, max)
  data_type_check(key, ZSet)
  return 0 unless data[key]
  data[key].select_by_score(min, max).size
end

#zinterstore(out, *args) ⇒ Object 



1061
1062
1063
1064
1065
1066
 
# File 'lib/rediska/driver.rb', line 1061

def zinterstore(out, *args)
  data_type_check(out, ZSet)
  args_handler = SortedSetArgumentHandler.new(args)
  data[out] = SortedSetIntersectStore.new(args_handler, data).call
  data[out].size
end

#zrange(key, start, stop, with_scores = nil) ⇒ Object 



984
985
986
987
988
989
990
991
992
993
 
# File 'lib/rediska/driver.rb', line 984

def zrange(key, start, stop, with_scores = nil)
  data_type_check(key, ZSet)
  return [] unless data[key]

  results = sort_keys(data[key])

  # Select just the keys unless we want scores
  results = results.map(&:first) unless with_scores
  results[start..stop].flatten.map(&:to_s)
end

#zrangebyscore(key, min, max, *opts) ⇒ Object 



1006
1007
1008
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
1019
1020
1021
 
# File 'lib/rediska/driver.rb', line 1006

def zrangebyscore(key, min, max, *opts)
  data_type_check(key, ZSet)
  return [] unless data[key]

  range = data[key].select_by_score(min, max)
  vals = if opts.include?('WITHSCORES')
    range.sort_by {|_,v| v }
  else
    range.keys.sort_by {|k| range[k] }
  end

  limit = get_limit(opts, vals)
  vals = vals[*limit] if limit

  vals.flatten.map(&:to_s)
end

#zrank(key, value) ⇒ Object 



970
971
972
973
974
975
 
# File 'lib/rediska/driver.rb', line 970

def zrank(key, value)
  data_type_check(key, ZSet)
  z = data[key]
  return unless z
  z.keys.sort_by {|k| z[k] }.index(value.to_s)
end

#zrem(key, value) ⇒ Object 



946
947
948
949
950
951
952
953
954
955
956
957
 
# File 'lib/rediska/driver.rb', line 946

def zrem(key, value)
  data_type_check(key, ZSet)
  values = Array(value)
  return 0 unless data[key]

  response = values.map do |v|
    data[key].delete(v.to_s) if data[key].has_key?(v.to_s)
  end.compact.size

  remove_key_for_empty_collection(key)
  response
end

#zremrangebyrank(key, start, stop) ⇒ Object 



1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
 
# File 'lib/rediska/driver.rb', line 1050

def zremrangebyrank(key, start, stop)
  data_type_check(key, ZSet)
  return 0 unless data[key]

  sorted_elements = data[key].sort_by { |k, v| v }
  start = sorted_elements.length if start > sorted_elements.length
  elements_to_delete = sorted_elements[start..stop]
  elements_to_delete.each { |elem, rank| data[key].delete(elem) }
  elements_to_delete.size
end

#zremrangebyscore(key, min, max) ⇒ Object 



1041
1042
1043
1044
1045
1046
1047
1048
 
# File 'lib/rediska/driver.rb', line 1041

def zremrangebyscore(key, min, max)
  data_type_check(key, ZSet)
  return 0 unless data[key]

  range = data[key].select_by_score(min, max)
  range.each {|k,_| data[key].delete(k) }
  range.size
end

#zrevrange(key, start, stop, with_scores = nil) ⇒ Object 



995
996
997
998
999
1000
1001
1002
1003
1004
 
# File 'lib/rediska/driver.rb', line 995

def zrevrange(key, start, stop, with_scores = nil)
  data_type_check(key, ZSet)
  return [] unless data[key]

  if with_scores
    data[key].sort_by {|_,v| -v }
  else
    data[key].keys.sort_by {|k| -data[key][k] }
  end[start..stop].flatten.map(&:to_s)
end

#zrevrangebyscore(key, max, min, *opts) ⇒ Object 



1023
1024
1025
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037
1038
1039
 
# File 'lib/rediska/driver.rb', line 1023

def zrevrangebyscore(key, max, min, *opts)
  opts = opts.flatten
  data_type_check(key, ZSet)
  return [] unless data[key]

  range = data[key].select_by_score(min, max)
  vals = if opts.include?('WITHSCORES')
    range.sort_by {|_,v| -v }
  else
    range.keys.sort_by {|k| -range[k] }
  end

  limit = get_limit(opts, vals)
  vals = vals[*limit] if limit

  vals.flatten.map(&:to_s)
end

#zrevrank(key, value) ⇒ Object 



977
978
979
980
981
982
 
# File 'lib/rediska/driver.rb', line 977

def zrevrank(key, value)
  data_type_check(key, ZSet)
  z = data[key]
  return unless z
  z.keys.sort_by {|k| -z[k] }.index(value.to_s)
end

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



1075
1076
1077
1078
1079
1080
1081
1082
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103
1104
1105
1106
1107
1108
1109
1110
1111
 
# File 'lib/rediska/driver.rb', line 1075

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

#zscan_each(key, *args, &block) ⇒ Object

Originally from redis-rb



1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
1125
 
# File 'lib/rediska/driver.rb', line 1114

def zscan_each(key, *args, &block)
  data_type_check(key, ZSet)
  return [] unless data[key]

  return to_enum(:zscan_each, key, options) unless block_given?
  cursor = 0
  loop do
    cursor, values = zscan(key, cursor, options)
    values.each(&block)
    break if cursor == '0'
  end
end

#zunionstore(out, *args) ⇒ Object 



1068
1069
1070
1071
1072
1073
 
# File 'lib/rediska/driver.rb', line 1068

def zunionstore(out, *args)
  data_type_check(out, ZSet)
  args_handler = SortedSetArgumentHandler.new(args)
  data[out] = SortedSetUnionStore.new(args_handler, data).call
  data[out].size
end