Class: MockRedis::Zset

Inherits:
Object
  • Object
show all
Extended by:
Forwardable
Includes:
Enumerable
Defined in:
lib/mock_redis/zset.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(enum = nil) ⇒ Zset

Returns a new instance of Zset.



13
14
15
16
17
18
 
# File 'lib/mock_redis/zset.rb', line 13

def initialize(enum = nil)
  @members = Set.new
  @members.merge(enum) if enum

  @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 



26
27
28
29
30
 
# File 'lib/mock_redis/zset.rb', line 26

def add(score, member)
  members.add(member)
  scores[member] = score.to_f
  self
end

#delete?(member) ⇒ Boolean

Returns:

  • (Boolean) 


32
33
34
35
 
# File 'lib/mock_redis/zset.rb', line 32

def delete?(member)
  scores.delete(member)
  members.delete?(member) && self
end

#each ⇒ Object 



37
38
39
 
# File 'lib/mock_redis/zset.rb', line 37

def each
  members.each { |m| yield score(m), m }
end

#in_range(min, max) ⇒ Object 



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
68
69
 
# File 'lib/mock_redis/zset.rb', line 41

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 



20
21
22
23
24
 
# File 'lib/mock_redis/zset.rb', line 20

def initialize_copy(source)
  super
  @members = @members.clone
  @scores  = @scores.clone
end

#intersection(other) ⇒ Object 



71
72
73
74
75
76
77
78
79
80
 
# File 'lib/mock_redis/zset.rb', line 71

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 



82
83
84
 
# File 'lib/mock_redis/zset.rb', line 82

def score(member)
  scores[member]
end

#sorted ⇒ Object 



86
87
88
89
90
 
# File 'lib/mock_redis/zset.rb', line 86

def sorted
  members.map do |m|
    [score(m), m]
  end.sort
end

#sorted_members ⇒ Object 



92
93
94
 
# File 'lib/mock_redis/zset.rb', line 92

def sorted_members
  sorted.map(&:last)
end

#union(other) ⇒ Object 



96
97
98
99
100
101
102
103
104
105
 
# File 'lib/mock_redis/zset.rb', line 96

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