Module: MockRedis::ListMethods
- Includes:
- Assertions, UtilityMethods
- Included in:
- Database
- Defined in:
- lib/mock_redis/list_methods.rb
Instance Method Summary collapse
- #blpop(*args) ⇒ Object
- #brpop(*args) ⇒ Object
- #brpoplpush(source, destination, options = {}) ⇒ Object
- #lindex(key, index) ⇒ Object
- #linsert(key, position, pivot, value) ⇒ Object
- #llen(key) ⇒ Object
- #lpop(key) ⇒ Object
- #lpush(key, values) ⇒ Object
- #lpushx(key, value) ⇒ Object
- #lrange(key, start, stop) ⇒ Object
- #lrem(key, count, value) ⇒ Object
- #lset(key, index, value) ⇒ Object
- #ltrim(key, start, stop) ⇒ Object
- #rpop(key) ⇒ Object
- #rpoplpush(source, destination) ⇒ Object
- #rpush(key, values) ⇒ Object
- #rpushx(key, value) ⇒ Object
Instance Method Details
#blpop(*args) ⇒ Object
9 10 11 12 13 14 15 16 17 18 19 20 |
# File 'lib/mock_redis/list_methods.rb', line 9 def blpop(*args) lists, timeout = extract_timeout(args) nonempty_list = first_nonempty_list(lists) if nonempty_list [nonempty_list, lpop(nonempty_list)] elsif timeout > 0 nil else raise MockRedis::WouldBlock, "Can't block forever" end end |
#brpop(*args) ⇒ Object
22 23 24 25 26 27 28 29 30 31 32 33 |
# File 'lib/mock_redis/list_methods.rb', line 22 def brpop(*args) lists, timeout = extract_timeout(args) nonempty_list = first_nonempty_list(lists) if nonempty_list [nonempty_list, rpop(nonempty_list)] elsif timeout > 0 nil else raise MockRedis::WouldBlock, "Can't block forever" end end |
#brpoplpush(source, destination, options = {}) ⇒ Object
35 36 37 38 39 40 41 42 43 44 45 46 47 |
# File 'lib/mock_redis/list_methods.rb', line 35 def brpoplpush(source, destination, = {}) = { :timeout => } if .is_a?(Integer) timeout = .is_a?(Hash) && [:timeout] || 0 assert_valid_timeout(timeout) if llen(source) > 0 rpoplpush(source, destination) elsif timeout > 0 nil else raise MockRedis::WouldBlock, "Can't block forever" end end |
#lindex(key, index) ⇒ Object
49 50 51 |
# File 'lib/mock_redis/list_methods.rb', line 49 def lindex(key, index) with_list_at(key) { |l| l[index.to_i] } end |
#linsert(key, position, pivot, value) ⇒ Object
53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
# File 'lib/mock_redis/list_methods.rb', line 53 def linsert(key, position, pivot, value) unless %w[before after].include?(position.to_s) raise Redis::CommandError, 'ERR syntax error' end assert_listy(key) return 0 unless data[key] pivot_position = (0..llen(key) - 1).find do |i| data[key][i] == pivot.to_s end return -1 unless pivot_position insertion_index = if position.to_s == 'before' pivot_position else pivot_position + 1 end data[key].insert(insertion_index, value.to_s) llen(key) end |
#llen(key) ⇒ Object
77 78 79 |
# File 'lib/mock_redis/list_methods.rb', line 77 def llen(key) with_list_at(key, &:length) end |
#lpop(key) ⇒ Object
81 82 83 |
# File 'lib/mock_redis/list_methods.rb', line 81 def lpop(key) with_list_at(key, &:shift) end |
#lpush(key, values) ⇒ Object
85 86 87 88 89 90 |
# File 'lib/mock_redis/list_methods.rb', line 85 def lpush(key, values) values = [values] unless values.is_a?(Array) assert_has_args(values, 'lpush') with_list_at(key) { |l| values.each { |v| l.unshift(v.to_s) } } llen(key) end |
#lpushx(key, value) ⇒ Object
92 93 94 95 96 97 98 99 100 |
# File 'lib/mock_redis/list_methods.rb', line 92 def lpushx(key, value) value = [value] unless value.is_a?(Array) if value.empty? raise Redis::CommandError, "ERR wrong number of arguments for 'lpushx' command" end assert_listy(key) return 0 unless list_at?(key) lpush(key, value) end |
#lrange(key, start, stop) ⇒ Object
102 103 104 105 |
# File 'lib/mock_redis/list_methods.rb', line 102 def lrange(key, start, stop) start = start.to_i with_list_at(key) { |l| start < l.size ? l[[start, -l.length].max..stop.to_i] : [] } end |
#lrem(key, count, value) ⇒ Object
107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 |
# File 'lib/mock_redis/list_methods.rb', line 107 def lrem(key, count, value) unless looks_like_integer?(count.to_s) raise Redis::CommandError, 'ERR value is not an integer or out of range' end count = count.to_i value = value.to_s with_list_at(key) do |list| indices_with_value = (0..(llen(key) - 1)).find_all do |i| list[i] == value end indices_to_delete = if count == 0 indices_with_value.reverse elsif count > 0 indices_with_value.take(count).reverse else indices_with_value.reverse.take(-count) end indices_to_delete.each { |i| list.delete_at(i) }.length end end |
#lset(key, index, value) ⇒ Object
131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 |
# File 'lib/mock_redis/list_methods.rb', line 131 def lset(key, index, value) assert_listy(key) unless list_at?(key) raise Redis::CommandError, 'ERR no such key' end index = index.to_i unless (0...llen(key)).cover?(index) raise Redis::CommandError, 'ERR index out of range' end data[key][index] = value.to_s 'OK' end |
#ltrim(key, start, stop) ⇒ Object
147 148 149 150 151 152 |
# File 'lib/mock_redis/list_methods.rb', line 147 def ltrim(key, start, stop) with_list_at(key) do |list| list&.replace(list[[start.to_i, -list.length].max..stop.to_i] || []) 'OK' end end |
#rpop(key) ⇒ Object
154 155 156 |
# File 'lib/mock_redis/list_methods.rb', line 154 def rpop(key) with_list_at(key) { |list| list&.pop } end |
#rpoplpush(source, destination) ⇒ Object
158 159 160 161 162 |
# File 'lib/mock_redis/list_methods.rb', line 158 def rpoplpush(source, destination) value = rpop(source) lpush(destination, value) unless value.nil? value end |
#rpush(key, values) ⇒ Object
164 165 166 167 168 169 |
# File 'lib/mock_redis/list_methods.rb', line 164 def rpush(key, values) values = [values] unless values.is_a?(Array) assert_has_args(values, 'rpush') with_list_at(key) { |l| values.each { |v| l.push(v.to_s) } } llen(key) end |
#rpushx(key, value) ⇒ Object
171 172 173 174 175 176 177 178 179 |
# File 'lib/mock_redis/list_methods.rb', line 171 def rpushx(key, value) value = [value] unless value.is_a?(Array) if value.empty? raise Redis::CommandError, "ERR wrong number of arguments for 'rpushx' command" end assert_listy(key) return 0 unless list_at?(key) rpush(key, value) end |