Module: Rediska::Driver

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

Instance Attribute Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#bufferObject

Returns the value of attribute buffer.



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

def buffer
  @buffer
end

#database_idObject

Returns the value of attribute database_id.



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

def database_id
  @database_id
end

#repliesObject



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

def replies
  @replies ||= []
end

Instance Method Details

#[](key) ⇒ Object



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

def [](key)
  get(key)
end

#[]=(key, value) ⇒ Object



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

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

#append(key, value) ⇒ Object



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

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

#auth(password) ⇒ Object



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

def auth(password)
  'OK'
end

#bgrewriteaofObject



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

def bgrewriteaof
end

#bgsaveObject



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

def bgsave
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

#dbsizeObject



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

def dbsize
  data.keys.count
end

#decr(key) ⇒ Object



663
664
665
666
# File 'lib/rediska/driver.rb', line 663

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

#decrby(key, by) ⇒ Object



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

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



427
428
429
430
431
432
433
434
435
436
# File 'lib/rediska/driver.rb', line 427

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

#disconnectObject



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

def disconnect
end

#echo(string) ⇒ Object



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

def echo(string)
  string
end

#execObject



693
694
695
# File 'lib/rediska/driver.rb', line 693

def exec
  buffer.tap {|x| self.buffer = nil }
end

#exists(key) ⇒ Object



201
202
203
# File 'lib/rediska/driver.rb', line 201

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

#expire(key, ttl) ⇒ Object



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

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

#expireat(key, timestamp) ⇒ Object



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

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

#flushallObject



63
64
65
66
67
68
# File 'lib/rediska/driver.rb', line 63

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

  'OK'
end

#flushdbObject

NOT IMPLEMENTED:

  • blpop

  • brpop

  • brpoplpush

  • discard

  • sort

  • subscribe

  • psubscribe

  • publish



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

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

  'OK'
end

#get(key) ⇒ Object



108
109
110
111
# File 'lib/rediska/driver.rb', line 108

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

#getbit(key, offset) ⇒ Object



113
114
115
116
# File 'lib/rediska/driver.rb', line 113

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



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

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

#getset(key, value) ⇒ Object



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

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

#hdel(key, field) ⇒ Object



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

def hdel(key, field)
  field = field.to_s
  data_type_check(key, Hash)
  data[key] && data[key].delete(field)
  remove_key_for_empty_collection(key)
end

#hexists(key, field) ⇒ Object



577
578
579
580
581
# File 'lib/rediska/driver.rb', line 577

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



154
155
156
157
# File 'lib/rediska/driver.rb', line 154

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

#hgetall(key) ⇒ Object



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

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

#hincrby(key, field, increment) ⇒ Object



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

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



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

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



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

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

#hlen(key) ⇒ Object



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

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

#hmget(key, *fields) ⇒ Object



529
530
531
532
533
534
535
536
537
538
539
540
541
# File 'lib/rediska/driver.rb', line 529

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



506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
# File 'lib/rediska/driver.rb', line 506

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



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

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



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

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



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

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

#incr(key) ⇒ Object



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

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

#incrby(key, by) ⇒ Object



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

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

#infoObject



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

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

#keys(pattern = '*') ⇒ Object



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

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

#lastsaveObject



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

def lastsave
  Time.now.to_i
end

#lindex(key, index) ⇒ Object



230
231
232
233
# File 'lib/rediska/driver.rb', line 230

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

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



235
236
237
238
239
240
241
242
243
244
# File 'lib/rediska/driver.rb', line 235

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



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

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

#lpop(key) ⇒ Object



314
315
316
317
318
# File 'lib/rediska/driver.rb', line 314

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

#lpush(key, value) ⇒ Object



286
287
288
289
290
291
292
293
# File 'lib/rediska/driver.rb', line 286

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



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

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

#lrange(key, startidx, endidx) ⇒ Object



211
212
213
214
# File 'lib/rediska/driver.rb', line 211

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

#lrem(key, count, value) ⇒ Object



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

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)


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

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



216
217
218
219
220
221
222
223
224
225
226
227
228
# File 'lib/rediska/driver.rb', line 216

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

  if start < 0 && data[key].count < start.abs
    # 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.
    data[key]
  else
    data[key] = data[key][start..stop]
  end
end

#mget(*keys) ⇒ Object



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

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

#monitorObject



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

def monitor
end

#move(key, destination_id) ⇒ Object

Raises:

  • (Redis::CommandError)


99
100
101
102
103
104
105
106
# File 'lib/rediska/driver.rb', line 99

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



624
625
626
627
628
629
630
631
632
633
634
635
636
637
# File 'lib/rediska/driver.rb', line 624

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



639
640
641
642
643
644
645
646
647
# File 'lib/rediska/driver.rb', line 639

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

#multiObject



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

def multi
  self.buffer = []
  yield if block_given?

  'OK'
end

#persist(key) ⇒ Object



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

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

#pingObject



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

def ping
  'PONG'
end

#quitObject



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

def quit
end

#randomkeyObject



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

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

#readObject



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

def read
  replies.shift
end

#rename(key, new_key) ⇒ Object



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

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



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

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

#rpop(key) ⇒ Object



301
302
303
304
305
# File 'lib/rediska/driver.rb', line 301

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

#rpoplpush(key1, key2) ⇒ Object



307
308
309
310
311
312
# File 'lib/rediska/driver.rb', line 307

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

#rpush(key, value) ⇒ Object



271
272
273
274
275
276
277
278
# File 'lib/rediska/driver.rb', line 271

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



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

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

#sadd(key, value) ⇒ Object



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

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

#saveObject



90
91
# File 'lib/rediska/driver.rb', line 90

def save
end

#scan(start_cursor, *args) ⇒ Object



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
741
742
743
# File 'lib/rediska/driver.rb', line 712

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



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

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

#sdiff(key1, *keys) ⇒ Object



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

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



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

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

#select(index) ⇒ Object



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

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

  'OK'
end

#set(key, value) ⇒ Object



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

def set(key, value)
  data[key] = value.to_s

  'OK'
end

#setbit(key, offset, bit) ⇒ Object



599
600
601
602
603
604
605
606
607
608
609
# File 'lib/rediska/driver.rb', line 599

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



611
612
613
614
615
616
# File 'lib/rediska/driver.rb', line 611

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

  'OK'
end

#setnx(key, value) ⇒ Object



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

def setnx(key, value)
  if exists(key)
    false
  else
    set(key, value)
    true
  end
end

#setrange(key, offset, value) ⇒ Object



618
619
620
621
622
# File 'lib/rediska/driver.rb', line 618

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

#shutdownObject



687
688
# File 'lib/rediska/driver.rb', line 687

def shutdown
end

#sinter(*keys) ⇒ Object



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

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



389
390
391
392
393
# File 'lib/rediska/driver.rb', line 389

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

#sismember(key, value) ⇒ Object



326
327
328
329
330
# File 'lib/rediska/driver.rb', line 326

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



690
691
# File 'lib/rediska/driver.rb', line 690

def slaveof(host, port)
end

#smembers(key) ⇒ Object



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

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

#smove(source, destination, value) ⇒ Object



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

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

#sort(key) ⇒ Object



649
650
651
# File 'lib/rediska/driver.rb', line 649

def sort(key)
  # TODO: Implement
end

#spop(key) ⇒ Object



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

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

#srandmember(key, number = nil) ⇒ Object



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

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

#srem(key, value) ⇒ Object



351
352
353
354
355
356
# File 'lib/rediska/driver.rb', line 351

def srem(key, value)
  data_type_check(key, ::Set)
  deleted = !!(data[key] && data[key].delete?(value.to_s))
  remove_key_for_empty_collection(key)
  deleted
end

#strlen(key) ⇒ Object



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

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

#sunion(*keys) ⇒ Object



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

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



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

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

#syncObject



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

def sync ; end

#timeObject



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

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



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

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



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

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

#unwatchObject



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

def unwatch
  'OK'
end

#watch(_) ⇒ Object



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

def watch(_)
  'OK'
end

#write(command) ⇒ Object



23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# File 'lib/rediska/driver.rb', line 23

def write(command)
  meffod = command.shift.to_s.downcase.to_sym
  if respond_to?(meffod)
    reply = send(meffod, *command)
  else
    raise Redis::CommandError, "ERR unknown command #{meffod.upcase}"
  end

  if reply == true
    reply = 1
  elsif reply == false
    reply = 0
  end

  replies << reply
  buffer << reply if buffer && meffod != :multi
  nil
end

#zadd(key, *args) ⇒ Object



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
771
772
773
# File 'lib/rediska/driver.rb', line 745

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



788
789
790
791
# File 'lib/rediska/driver.rb', line 788

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

#zcount(key, min, max) ⇒ Object



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

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



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

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



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

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



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

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



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

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



813
814
815
816
817
818
# File 'lib/rediska/driver.rb', line 813

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



775
776
777
778
779
780
781
782
783
784
785
786
# File 'lib/rediska/driver.rb', line 775

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



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

def zremrangebyrank(key, start, stop)
  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



890
891
892
893
894
895
896
897
# File 'lib/rediska/driver.rb', line 890

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



844
845
846
847
848
849
850
851
852
853
# File 'lib/rediska/driver.rb', line 844

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



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

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



820
821
822
823
824
825
# File 'lib/rediska/driver.rb', line 820

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



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

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



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

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