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 



591
592
593
 
# File 'lib/rediska/driver.rb', line 591

def [](key)
  get(key)
end

#[]=(key, value) ⇒ Object 



595
596
597
 
# File 'lib/rediska/driver.rb', line 595

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

#append(key, value) ⇒ Object 



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

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

#auth(password) ⇒ Object 



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

def auth(password)
  'OK'
end

#bgrewriteaof ⇒ Object 



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

def bgrewriteaof
end

#bgsave ⇒ Object 



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

def bgsave
end

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



97
98
99
100
101
 
# File 'lib/rediska/driver.rb', line 97

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

#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 



194
195
196
 
# File 'lib/rediska/driver.rb', line 194

def dbsize
  data.keys.count
end

#decr(key) ⇒ Object 



679
680
681
682
 
# File 'lib/rediska/driver.rb', line 679

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

#decrby(key, by) ⇒ Object 



684
685
686
687
 
# File 'lib/rediska/driver.rb', line 684

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 



433
434
435
436
437
438
439
440
441
442
 
# File 'lib/rediska/driver.rb', line 433

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

#echo(string) ⇒ Object 



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

def echo(string)
  string
end

#exists(key) ⇒ Object 



198
199
200
 
# File 'lib/rediska/driver.rb', line 198

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

#expire(key, ttl) ⇒ Object 



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

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

#expireat(key, timestamp) ⇒ Object 



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

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

#flushall ⇒ Object 



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

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

  'OK'
end

#flushdb ⇒ Object

NOT IMPLEMENTED:

  • blpop

  • brpop

  • brpoplpush

  • subscribe

  • psubscribe

  • publish



35
36
37
38
39
40
 
# File 'lib/rediska/driver.rb', line 35

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

  'OK'
end

#get(key) ⇒ Object 



87
88
89
90
 
# File 'lib/rediska/driver.rb', line 87

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

#getbit(key, offset) ⇒ Object 



92
93
94
95
 
# File 'lib/rediska/driver.rb', line 92

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 



103
104
105
106
 
# File 'lib/rediska/driver.rb', line 103

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

#getset(key, value) ⇒ Object 



109
110
111
112
113
114
 
# File 'lib/rediska/driver.rb', line 109

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

#hdel(key, field) ⇒ Object 



144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
 
# File 'lib/rediska/driver.rb', line 144

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 



583
584
585
586
587
 
# File 'lib/rediska/driver.rb', line 583

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 



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

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

#hgetall(key) ⇒ Object 



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

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

#hincrby(key, field, increment) ⇒ Object 



561
562
563
564
565
566
567
568
569
570
 
# File 'lib/rediska/driver.rb', line 561

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 



572
573
574
575
576
577
578
579
580
581
 
# File 'lib/rediska/driver.rb', line 572

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 



163
164
165
166
167
 
# File 'lib/rediska/driver.rb', line 163

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

#hlen(key) ⇒ Object 



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

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

#hmget(key, *fields) ⇒ Object 



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

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 



512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
 
# File 'lib/rediska/driver.rb', line 512

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

#hset(key, field, value) ⇒ Object 



492
493
494
495
496
497
498
499
500
501
502
503
 
# File 'lib/rediska/driver.rb', line 492

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 



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

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 



555
556
557
558
559
 
# File 'lib/rediska/driver.rb', line 555

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

#incr(key) ⇒ Object 



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

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

#incrby(key, by) ⇒ Object 



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

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

#info ⇒ Object 



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

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

#keys(pattern = '*') ⇒ Object 



169
170
171
 
# File 'lib/rediska/driver.rb', line 169

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

#lastsave ⇒ Object 



185
186
187
 
# File 'lib/rediska/driver.rb', line 185

def lastsave
  Time.now.to_i
end

#lindex(key, index) ⇒ Object 



227
228
229
230
 
# File 'lib/rediska/driver.rb', line 227

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

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



232
233
234
235
236
237
238
239
240
241
 
# File 'lib/rediska/driver.rb', line 232

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 



202
203
204
205
206
 
# File 'lib/rediska/driver.rb', line 202

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

#lpop(key) ⇒ Object 



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

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

#lpush(key, value) ⇒ Object 



283
284
285
286
287
288
289
290
 
# File 'lib/rediska/driver.rb', line 283

def lpush(key, value)
  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 



292
293
294
295
296
 
# File 'lib/rediska/driver.rb', line 292

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

#lrange(key, startidx, endidx) ⇒ Object 



208
209
210
211
 
# File 'lib/rediska/driver.rb', line 208

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

#lrem(key, count, value) ⇒ Object 



250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
 
# File 'lib/rediska/driver.rb', line 250

def lrem(key, count, value)
  data_type_check(key, Array)
  return 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) 


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

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 



213
214
215
216
217
218
219
220
221
222
223
224
225
 
# File 'lib/rediska/driver.rb', line 213

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 



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

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 



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

def monitor
end

#move(key, destination_id) ⇒ Object

Raises:

  • (Redis::CommandError) 


78
79
80
81
82
83
84
85
 
# File 'lib/rediska/driver.rb', line 78

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 



644
645
646
647
648
649
650
651
652
653
654
655
656
657
 
# File 'lib/rediska/driver.rb', line 644

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 



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

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 



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

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

#ping ⇒ Object 



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

def ping
  'PONG'
end

#quit ⇒ Object 



700
701
 
# File 'lib/rediska/driver.rb', line 700

def quit
end

#randomkey ⇒ Object 



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

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 



453
454
455
456
457
458
 
# File 'lib/rediska/driver.rb', line 453

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 



460
461
462
463
464
465
466
467
 
# File 'lib/rediska/driver.rb', line 460

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

#rpop(key) ⇒ Object 



298
299
300
301
302
 
# File 'lib/rediska/driver.rb', line 298

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

#rpoplpush(key1, key2) ⇒ Object 



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

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 



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

def rpush(key, value)
  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 



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

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

#sadd(key, value) ⇒ Object 



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

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 



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

def save
end

#scan(start_cursor, *args) ⇒ Object 



709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
 
# File 'lib/rediska/driver.rb', line 709

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 



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

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

#sdiff(key1, *keys) ⇒ Object 



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

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 



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

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

#select(index) ⇒ Object 



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

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

  'OK'
end

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



599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
 
# File 'lib/rediska/driver.rb', line 599

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 



619
620
621
622
623
624
625
626
627
628
629
 
# File 'lib/rediska/driver.rb', line 619

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 



631
632
633
634
635
636
 
# File 'lib/rediska/driver.rb', line 631

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

  'OK'
end

#setnx(key, value) ⇒ Object 



444
445
446
447
448
449
450
451
 
# File 'lib/rediska/driver.rb', line 444

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

#setrange(key, offset, value) ⇒ Object 



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

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

#shutdown ⇒ Object 



703
704
 
# File 'lib/rediska/driver.rb', line 703

def shutdown
end

#sinter(*keys) ⇒ Object 



384
385
386
387
388
389
390
391
392
393
 
# File 'lib/rediska/driver.rb', line 384

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 



395
396
397
398
399
 
# File 'lib/rediska/driver.rb', line 395

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

#sismember(key, value) ⇒ Object 



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

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 



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

def slaveof(host, port)
end

#smembers(key) ⇒ Object 



317
318
319
320
321
 
# File 'lib/rediska/driver.rb', line 317

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

#smove(source, destination, value) ⇒ Object 



364
365
366
367
368
369
 
# File 'lib/rediska/driver.rb', line 364

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 



371
372
373
374
375
376
 
# File 'lib/rediska/driver.rb', line 371

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

#srandmember(key, number = nil) ⇒ Object 



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

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

#srem(key, value) ⇒ Object 



348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
 
# File 'lib/rediska/driver.rb', line 348

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 



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

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

#sunion(*keys) ⇒ Object 



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

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 



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

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

#sync ⇒ Object 



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

def sync ; end

#time ⇒ Object 



189
190
191
192
 
# File 'lib/rediska/driver.rb', line 189

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 



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

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 



689
690
691
692
693
694
695
696
697
698
 
# File 'lib/rediska/driver.rb', line 689

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 



742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
 
# File 'lib/rediska/driver.rb', line 742

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 



785
786
787
788
 
# File 'lib/rediska/driver.rb', line 785

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

#zcount(key, min, max) ⇒ Object 



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

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 



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

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 



907
908
909
910
911
912
 
# File 'lib/rediska/driver.rb', line 907

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 



824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
 
# File 'lib/rediska/driver.rb', line 824

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

  # Sort by score, or if scores are equal, key alphanum
  results = data[key].sort do |(k1, v1), (k2, v2)|
    if v1 == v2
      k1 <=> k2
    else
      v1 <=> v2
    end
  end
  # 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 



852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
 
# File 'lib/rediska/driver.rb', line 852

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 



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

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 



772
773
774
775
776
777
778
779
780
781
782
783
 
# File 'lib/rediska/driver.rb', line 772

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 



896
897
898
899
900
901
902
903
904
905
 
# File 'lib/rediska/driver.rb', line 896

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 



887
888
889
890
891
892
893
894
 
# File 'lib/rediska/driver.rb', line 887

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 



841
842
843
844
845
846
847
848
849
850
 
# File 'lib/rediska/driver.rb', line 841

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 



869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
 
# File 'lib/rediska/driver.rb', line 869

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 



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

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

#zscore(key, value) ⇒ Object 



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

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 



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

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