Class: MockRedis::Zset
- Inherits:
-
Object
- Object
- MockRedis::Zset
- Extended by:
- Forwardable
- Includes:
- Enumerable
- Defined in:
- lib/mock_redis/zset.rb
Instance Attribute Summary collapse
-
#members ⇒ Object
readonly
Returns the value of attribute members.
-
#scores ⇒ Object
readonly
Returns the value of attribute scores.
Instance Method Summary collapse
- #add(score, member) ⇒ Object
- #delete?(member) ⇒ Boolean
- #each ⇒ Object
- #in_range(min, max) ⇒ Object
-
#initialize ⇒ Zset
constructor
A new instance of Zset.
- #initialize_copy(source) ⇒ Object
- #intersection(other) ⇒ Object
- #score(member) ⇒ Object
- #sorted ⇒ Object
- #sorted_members ⇒ Object
- #union(other) ⇒ Object
Constructor Details
#initialize ⇒ Zset
Returns a new instance of Zset.
13 14 15 16 |
# File 'lib/mock_redis/zset.rb', line 13 def initialize @members = Set.new @scores = {} end |
Instance Attribute Details
#members ⇒ Object (readonly)
Returns the value of attribute members.
9 10 11 |
# File 'lib/mock_redis/zset.rb', line 9 def members @members end |
#scores ⇒ Object (readonly)
Returns the value of attribute scores.
9 10 11 |
# File 'lib/mock_redis/zset.rb', line 9 def scores @scores end |
Instance Method Details
#add(score, member) ⇒ Object
24 25 26 27 28 |
# File 'lib/mock_redis/zset.rb', line 24 def add(score, member) members.add(member) scores[member] = score.to_f self end |
#delete?(member) ⇒ Boolean
30 31 32 33 |
# File 'lib/mock_redis/zset.rb', line 30 def delete?(member) scores.delete(member) members.delete?(member) && self end |
#each ⇒ Object
35 36 37 |
# File 'lib/mock_redis/zset.rb', line 35 def each members.each { |m| yield score(m), m } end |
#in_range(min, max) ⇒ Object
39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
# File 'lib/mock_redis/zset.rb', line 39 def in_range(min, max) in_from_the_left = case min when '-inf' lambda { |_| true } when '+inf' lambda { |_| false } when /\((.*)$/ left_val = $1.to_f lambda { |x| x.to_f > left_val } else lambda { |x| x.to_f >= min.to_f } end in_from_the_right = case max when '-inf' lambda { |_| false } when '+inf' lambda { |_| true } when /\((.*)$/ right_val = $1.to_f lambda { |x| x.to_f < right_val } else lambda { |x| x.to_f <= max.to_f } end sorted.find_all do |(score, _member)| in_from_the_left[score] && in_from_the_right[score] end end |
#initialize_copy(source) ⇒ Object
18 19 20 21 22 |
# File 'lib/mock_redis/zset.rb', line 18 def initialize_copy(source) super @members = @members.clone @scores = @scores.clone end |
#intersection(other) ⇒ Object
69 70 71 72 73 74 75 76 77 78 |
# File 'lib/mock_redis/zset.rb', line 69 def intersection(other) if !block_given? intersection(other, &:+) else members.intersection(other.members).reduce(self.class.new) do |acc, m| new_score = yield(score(m), other.score(m)) acc.add(new_score, m) end end end |
#score(member) ⇒ Object
80 81 82 |
# File 'lib/mock_redis/zset.rb', line 80 def score(member) scores[member] end |
#sorted ⇒ Object
84 85 86 87 88 |
# File 'lib/mock_redis/zset.rb', line 84 def sorted members.map do |m| [score(m), m] end.sort end |
#sorted_members ⇒ Object
90 91 92 |
# File 'lib/mock_redis/zset.rb', line 90 def sorted_members sorted.map(&:last) end |
#union(other) ⇒ Object
94 95 96 97 98 99 100 101 102 103 |
# File 'lib/mock_redis/zset.rb', line 94 def union(other) if !block_given? union(other, &:+) else members.union(other.members).reduce(self.class.new) do |acc, m| new_score = yield(score(m), other.score(m)) acc.add(new_score, m) end end end |