Module: Rediska::Driver
- Included in:
- Connection
- Defined in:
- lib/rediska/driver.rb
Instance Attribute Summary collapse
-
#buffer ⇒ Object
Returns the value of attribute buffer.
-
#database_id ⇒ Object
Returns the value of attribute database_id.
- #replies ⇒ Object
Instance Method Summary collapse
- #[](key) ⇒ Object
- #[]=(key, value) ⇒ Object
- #append(key, value) ⇒ Object
- #auth(password) ⇒ Object
- #bgrewriteaof ⇒ Object
- #bgsave ⇒ Object
- #connect_unix(path, timeout) ⇒ Object
- #connected? ⇒ Boolean
- #dbsize ⇒ Object
- #decr(key) ⇒ Object
- #decrby(key, by) ⇒ Object
- #del(*keys) ⇒ Object
- #disconnect ⇒ Object
- #echo(string) ⇒ Object
- #exec ⇒ Object
- #exists(key) ⇒ Object
- #expire(key, ttl) ⇒ Object
- #expireat(key, timestamp) ⇒ Object
- #flushall ⇒ Object
-
#flushdb ⇒ Object
NOT IMPLEMENTED: * blpop * brpop * brpoplpush * discard * sort * subscribe * psubscribe * publish.
- #get(key) ⇒ Object
- #getbit(key, offset) ⇒ Object
- #getrange(key, start, ending) ⇒ Object (also: #substr)
- #getset(key, value) ⇒ Object
- #hdel(key, field) ⇒ Object
- #hexists(key, field) ⇒ Object
- #hget(key, field) ⇒ Object
- #hgetall(key) ⇒ Object
- #hincrby(key, field, increment) ⇒ Object
- #hincrbyfloat(key, field, increment) ⇒ Object
- #hkeys(key) ⇒ Object
- #hlen(key) ⇒ Object
- #hmget(key, *fields) ⇒ Object
- #hmset(key, *fields) ⇒ Object
- #hset(key, field, value) ⇒ Object
- #hsetnx(key, field, value) ⇒ Object
- #hvals(key) ⇒ Object
- #incr(key) ⇒ Object
- #incrby(key, by) ⇒ Object
- #info ⇒ Object
- #keys(pattern = '*') ⇒ Object
- #lastsave ⇒ Object
- #lindex(key, index) ⇒ Object
- #linsert(key, where, pivot, value) ⇒ Object
- #llen(key) ⇒ Object
- #lpop(key) ⇒ Object
- #lpush(key, value) ⇒ Object
- #lpushx(key, value) ⇒ Object
- #lrange(key, startidx, endidx) ⇒ Object
- #lrem(key, count, value) ⇒ Object
- #lset(key, index, value) ⇒ Object
- #ltrim(key, start, stop) ⇒ Object
- #mget(*keys) ⇒ Object
- #monitor ⇒ Object
- #move(key, destination_id) ⇒ Object
- #mset(*pairs) ⇒ Object
- #msetnx(*pairs) ⇒ Object
- #multi ⇒ Object
- #persist(key) ⇒ Object
- #ping ⇒ Object
- #quit ⇒ Object
- #randomkey ⇒ Object
- #read ⇒ Object
- #rename(key, new_key) ⇒ Object
- #renamenx(key, new_key) ⇒ Object
- #rpop(key) ⇒ Object
- #rpoplpush(key1, key2) ⇒ Object
- #rpush(key, value) ⇒ Object
- #rpushx(key, value) ⇒ Object
- #sadd(key, value) ⇒ Object
- #save ⇒ Object
- #scan(start_cursor, *args) ⇒ Object
- #scard(key) ⇒ Object
- #sdiff(key1, *keys) ⇒ Object
- #sdiffstore(destination, key1, *keys) ⇒ Object
- #select(index) ⇒ Object
- #set(key, value) ⇒ Object
- #setbit(key, offset, bit) ⇒ Object
- #setex(key, seconds, value) ⇒ Object
- #setnx(key, value) ⇒ Object
- #setrange(key, offset, value) ⇒ Object
- #shutdown ⇒ Object
- #sinter(*keys) ⇒ Object
- #sinterstore(destination, *keys) ⇒ Object
- #sismember(key, value) ⇒ Object
- #slaveof(host, port) ⇒ Object
- #smembers(key) ⇒ Object
- #smove(source, destination, value) ⇒ Object
- #sort(key) ⇒ Object
- #spop(key) ⇒ Object
- #srandmember(key, number = nil) ⇒ Object
- #srem(key, value) ⇒ Object
- #strlen(key) ⇒ Object
- #sunion(*keys) ⇒ Object
- #sunionstore(destination, *keys) ⇒ Object
- #sync ⇒ Object
- #time ⇒ Object
- #timeout=(usecs) ⇒ Object
- #ttl(key) ⇒ Object
- #type(key) ⇒ Object
- #unwatch ⇒ Object
- #watch(_) ⇒ Object
- #write(command) ⇒ Object
- #zadd(key, *args) ⇒ Object
- #zcard(key) ⇒ Object
- #zcount(key, min, max) ⇒ Object
- #zincrby(key, num, value) ⇒ Object
- #zinterstore(out, *args) ⇒ Object
- #zrange(key, start, stop, with_scores = nil) ⇒ Object
- #zrangebyscore(key, min, max, *opts) ⇒ Object
- #zrank(key, value) ⇒ Object
- #zrem(key, value) ⇒ Object
- #zremrangebyrank(key, start, stop) ⇒ Object
- #zremrangebyscore(key, min, max) ⇒ Object
- #zrevrange(key, start, stop, with_scores = nil) ⇒ Object
- #zrevrangebyscore(key, max, min, *opts) ⇒ Object
- #zrevrank(key, value) ⇒ Object
- #zscore(key, value) ⇒ Object
- #zunionstore(out, *args) ⇒ Object
Instance Attribute Details
#buffer ⇒ Object
Returns the value of attribute buffer.
3 4 5 |
# File 'lib/rediska/driver.rb', line 3 def buffer @buffer end |
#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
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 |
#bgrewriteaof ⇒ Object
96 97 |
# File 'lib/rediska/driver.rb', line 96 def bgrewriteaof end |
#bgsave ⇒ Object
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
10 11 12 |
# File 'lib/rediska/driver.rb', line 10 def connected? true end |
#dbsize ⇒ Object
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 |
#disconnect ⇒ Object
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 |
#exec ⇒ Object
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, ) data.expires[key] = Time.at() true end |
#flushall ⇒ Object
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 |
#flushdb ⇒ Object
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 |
#info ⇒ Object
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 |
#lastsave ⇒ Object
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
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 |
#monitor ⇒ Object
87 88 |
# File 'lib/rediska/driver.rb', line 87 def monitor end |
#move(key, destination_id) ⇒ Object
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 |
#multi ⇒ Object
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 |
#ping ⇒ Object
184 185 186 |
# File 'lib/rediska/driver.rb', line 184 def ping 'PONG' end |
#quit ⇒ Object
684 685 |
# File 'lib/rediska/driver.rb', line 684 def quit end |
#randomkey ⇒ Object
176 177 178 |
# File 'lib/rediska/driver.rb', line 176 def randomkey data.keys[rand(dbsize)] end |
#read ⇒ Object
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 |
#save ⇒ Object
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 |
#shutdown ⇒ Object
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 |
#sync ⇒ Object
583 |
# File 'lib/rediska/driver.rb', line 583 def sync ; end |
#time ⇒ Object
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 |
#unwatch ⇒ Object
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 |