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 



712
713
714
 
# File 'lib/rediska/driver.rb', line 712

def [](key)
  get(key)
end

#[]=(key, value) ⇒ Object 



716
717
718
 
# File 'lib/rediska/driver.rb', line 716

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

#append(key, value) ⇒ Object 



156
157
158
159
 
# File 'lib/rediska/driver.rb', line 156

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

#auth(password) ⇒ Object 



46
47
48
 
# File 'lib/rediska/driver.rb', line 46

def auth(password)
  'OK'
end

#bgrewriteaof ⇒ Object 



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

def bgrewriteaof
end

#bgsave ⇒ Object 



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

def bgsave
end

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



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

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 



411
412
413
414
415
416
417
418
419
420
 
# File 'lib/rediska/driver.rb', line 411

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 



380
381
382
383
384
385
386
387
388
389
 
# File 'lib/rediska/driver.rb', line 380

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 



398
399
400
401
402
403
 
# File 'lib/rediska/driver.rb', line 398

def brpoplpush(key1, key2, opts = {})
  data_type_check(key1, Array)
  brpop(key1).tap do |elem|
    lpush(key2, elem) unless elem.nil?
  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:



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

def connected?
  true
end

#dbsize ⇒ Object 



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

def dbsize
  data.keys.count
end

#decr(key) ⇒ Object 



805
806
807
808
 
# File 'lib/rediska/driver.rb', line 805

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

#decrby(key, by) ⇒ Object 



810
811
812
813
 
# File 'lib/rediska/driver.rb', line 810

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 



538
539
540
541
542
543
544
545
546
547
 
# File 'lib/rediska/driver.rb', line 538

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 



84
85
86
87
88
 
# File 'lib/rediska/driver.rb', line 84

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

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

#echo(string) ⇒ Object 



245
246
247
 
# File 'lib/rediska/driver.rb', line 245

def echo(string)
  string
end

#exists(key) ⇒ Object 



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

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

#expire(key, ttl) ⇒ Object 



574
575
576
577
578
 
# File 'lib/rediska/driver.rb', line 574

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

#expireat(key, timestamp) ⇒ Object 



604
605
606
607
 
# File 'lib/rediska/driver.rb', line 604

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

#flushall ⇒ Object 



39
40
41
42
43
44
 
# File 'lib/rediska/driver.rb', line 39

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

  'OK'
end

#flushdb ⇒ Object

NOT IMPLEMENTED:

  • subscribe

  • psubscribe

  • publish



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

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

  'OK'
end

#get(key) ⇒ Object 



119
120
121
122
 
# File 'lib/rediska/driver.rb', line 119

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

#getbit(key, offset) ⇒ Object 



124
125
126
127
 
# File 'lib/rediska/driver.rb', line 124

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 



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

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

#getset(key, value) ⇒ Object 



141
142
143
144
145
146
 
# File 'lib/rediska/driver.rb', line 141

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

#hdel(key, field) ⇒ Object 



176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
 
# File 'lib/rediska/driver.rb', line 176

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 



704
705
706
707
708
 
# File 'lib/rediska/driver.rb', line 704

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 



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

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

#hgetall(key) ⇒ Object 



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

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

#hincrby(key, field, increment) ⇒ Object 



682
683
684
685
686
687
688
689
690
691
 
# File 'lib/rediska/driver.rb', line 682

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 



693
694
695
696
697
698
699
700
701
702
 
# File 'lib/rediska/driver.rb', line 693

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 



195
196
197
198
199
 
# File 'lib/rediska/driver.rb', line 195

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

#hlen(key) ⇒ Object 



670
671
672
673
674
 
# File 'lib/rediska/driver.rb', line 670

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

#hmget(key, *fields) ⇒ Object 



656
657
658
659
660
661
662
663
664
665
666
667
668
 
# File 'lib/rediska/driver.rb', line 656

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 



633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
 
# File 'lib/rediska/driver.rb', line 633

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 



201
202
203
204
205
206
207
208
209
210
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
 
# File 'lib/rediska/driver.rb', line 201

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 



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

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 



626
627
628
629
630
631
 
# File 'lib/rediska/driver.rb', line 626

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 



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

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

#incr(key) ⇒ Object 



790
791
792
793
 
# File 'lib/rediska/driver.rb', line 790

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

#incrby(key, by) ⇒ Object 



795
796
797
798
 
# File 'lib/rediska/driver.rb', line 795

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 



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

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

#info ⇒ Object 



57
58
59
60
61
 
# File 'lib/rediska/driver.rb', line 57

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

#keys(pattern = '*') ⇒ Object 



237
238
239
 
# File 'lib/rediska/driver.rb', line 237

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

#lastsave ⇒ Object 



253
254
255
 
# File 'lib/rediska/driver.rb', line 253

def lastsave
  Time.now.to_i
end

#lindex(key, index) ⇒ Object 



295
296
297
298
 
# File 'lib/rediska/driver.rb', line 295

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

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



300
301
302
303
304
305
306
307
308
309
 
# File 'lib/rediska/driver.rb', line 300

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 



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

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

#lpop(key) ⇒ Object 



405
406
407
408
409
 
# File 'lib/rediska/driver.rb', line 405

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

#lpush(key, value) ⇒ Object 



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

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 



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

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 



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

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

#lrem(key, count, value) ⇒ Object 



318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
 
# File 'lib/rediska/driver.rb', line 318

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:



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

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 



281
282
283
284
285
286
287
288
289
290
291
292
293
 
# File 'lib/rediska/driver.rb', line 281

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 



148
149
150
151
152
153
154
 
# File 'lib/rediska/driver.rb', line 148

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 



63
64
 
# File 'lib/rediska/driver.rb', line 63

def monitor
end

#move(key, destination_id) ⇒ Object

Raises:



75
76
77
78
79
80
81
82
 
# File 'lib/rediska/driver.rb', line 75

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 



765
766
767
768
769
770
771
772
773
774
775
776
777
778
 
# File 'lib/rediska/driver.rb', line 765

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 



780
781
782
783
784
785
786
787
788
 
# File 'lib/rediska/driver.rb', line 780

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 



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

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

#pexpire(key, ttl) ⇒ Object 



580
581
582
583
584
585
586
 
# File 'lib/rediska/driver.rb', line 580

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

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

  1
end

#ping ⇒ Object 



249
250
251
 
# File 'lib/rediska/driver.rb', line 249

def ping
  'PONG'
end

#pttl(key) ⇒ Object 



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

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 



826
827
 
# File 'lib/rediska/driver.rb', line 826

def quit
end

#randomkey ⇒ Object 



241
242
243
 
# File 'lib/rediska/driver.rb', line 241

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

#read ⇒ Object 



23
24
25
 
# File 'lib/rediska/driver.rb', line 23

def read
  replies.shift
end

#rename(key, new_key) ⇒ Object 



558
559
560
561
562
563
 
# File 'lib/rediska/driver.rb', line 558

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 



565
566
567
568
569
570
571
572
 
# File 'lib/rediska/driver.rb', line 565

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:



90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
 
# File 'lib/rediska/driver.rb', line 90

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 



374
375
376
377
378
 
# File 'lib/rediska/driver.rb', line 374

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

#rpoplpush(key1, key2) ⇒ Object 



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

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 



336
337
338
339
340
341
342
343
344
345
 
# File 'lib/rediska/driver.rb', line 336

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 



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

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 



434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
 
# File 'lib/rediska/driver.rb', line 434

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 



66
67
 
# File 'lib/rediska/driver.rb', line 66

def save
end

#scan(start_cursor, *args) ⇒ Object 



835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
 
# File 'lib/rediska/driver.rb', line 835

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, Fixnum)

  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 



483
484
485
486
487
 
# File 'lib/rediska/driver.rb', line 483

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

#sdiff(key1, *keys) ⇒ Object 



520
521
522
523
524
525
526
 
# File 'lib/rediska/driver.rb', line 520

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 



528
529
530
531
532
 
# File 'lib/rediska/driver.rb', line 528

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

#select(index) ⇒ Object 



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

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

  'OK'
end

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



720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
 
# File 'lib/rediska/driver.rb', line 720

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 



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

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 



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

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

  'OK'
end

#setnx(key, value) ⇒ Object 



549
550
551
552
553
554
555
556
 
# File 'lib/rediska/driver.rb', line 549

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

#setrange(key, offset, value) ⇒ Object 



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

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

#shutdown ⇒ Object 



829
830
 
# File 'lib/rediska/driver.rb', line 829

def shutdown
end

#sinter(*keys) ⇒ Object 



489
490
491
492
493
494
495
496
497
498
 
# File 'lib/rediska/driver.rb', line 489

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 



500
501
502
503
504
 
# File 'lib/rediska/driver.rb', line 500

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

#sismember(key, value) ⇒ Object 



428
429
430
431
432
 
# File 'lib/rediska/driver.rb', line 428

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 



832
833
 
# File 'lib/rediska/driver.rb', line 832

def slaveof(host, port)
end

#smembers(key) ⇒ Object 



422
423
424
425
426
 
# File 'lib/rediska/driver.rb', line 422

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

#smove(source, destination, value) ⇒ Object 



469
470
471
472
473
474
 
# File 'lib/rediska/driver.rb', line 469

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 



476
477
478
479
480
481
 
# File 'lib/rediska/driver.rb', line 476

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

#srandmember(key, number = nil) ⇒ Object 



534
535
536
 
# File 'lib/rediska/driver.rb', line 534

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

#srem(key, value) ⇒ Object 



453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
 
# File 'lib/rediska/driver.rb', line 453

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

#strlen(key) ⇒ Object 



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

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

#sunion(*keys) ⇒ Object 



506
507
508
509
510
511
512
 
# File 'lib/rediska/driver.rb', line 506

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 



514
515
516
517
518
 
# File 'lib/rediska/driver.rb', line 514

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

#sync ⇒ Object 



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

def sync ; end

#time ⇒ Object 



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

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

#timeout=(usecs) ⇒ Object 



20
21
 
# File 'lib/rediska/driver.rb', line 20

def timeout=(usecs)
end

#ttl(key) ⇒ Object 



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

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 



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

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 



868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
 
# File 'lib/rediska/driver.rb', line 868

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 



911
912
913
914
 
# File 'lib/rediska/driver.rb', line 911

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

#zcount(key, min, max) ⇒ Object 



922
923
924
925
926
 
# File 'lib/rediska/driver.rb', line 922

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

#zincrby(key, num, value) ⇒ Object 



928
929
930
931
932
933
934
 
# File 'lib/rediska/driver.rb', line 928

def zincrby(key, num, value)
  data_type_check(key, ZSet)
  data[key] ||= ZSet.new
  data[key][value.to_s] ||= 0
  data[key].increment(value.to_s, num)
  data[key][value.to_s].to_s
end

#zinterstore(out, *args) ⇒ Object 



1027
1028
1029
1030
1031
1032
 
# File 'lib/rediska/driver.rb', line 1027

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 



950
951
952
953
954
955
956
957
958
959
 
# File 'lib/rediska/driver.rb', line 950

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 



972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
 
# File 'lib/rediska/driver.rb', line 972

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 



936
937
938
939
940
941
 
# File 'lib/rediska/driver.rb', line 936

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 



898
899
900
901
902
903
904
905
906
907
908
909
 
# File 'lib/rediska/driver.rb', line 898

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 



1016
1017
1018
1019
1020
1021
1022
1023
1024
1025
 
# File 'lib/rediska/driver.rb', line 1016

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 



1007
1008
1009
1010
1011
1012
1013
1014
 
# File 'lib/rediska/driver.rb', line 1007

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 



961
962
963
964
965
966
967
968
969
970
 
# File 'lib/rediska/driver.rb', line 961

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 



989
990
991
992
993
994
995
996
997
998
999
1000
1001
1002
1003
1004
1005
 
# File 'lib/rediska/driver.rb', line 989

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 



943
944
945
946
947
948
 
# File 'lib/rediska/driver.rb', line 943

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 



1041
1042
1043
1044
1045
1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
 
# File 'lib/rediska/driver.rb', line 1041

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, Fixnum)

  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



1080
1081
1082
1083
1084
1085
1086
1087
1088
1089
1090
1091
 
# File 'lib/rediska/driver.rb', line 1080

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

#zscore(key, value) ⇒ Object 



916
917
918
919
920
 
# File 'lib/rediska/driver.rb', line 916

def zscore(key, value)
  data_type_check(key, ZSet)
  value = data[key] && data[key][value.to_s]
  value && value.to_s
end

#zunionstore(out, *args) ⇒ Object 



1034
1035
1036
1037
1038
1039
 
# File 'lib/rediska/driver.rb', line 1034

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