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 ⇒ 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
29
30
31
32
33
 
# File 'lib/mock_redis/zset.rb', line 24

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

#delete?(member) ⇒ Boolean

Returns:

  • (Boolean) 


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

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

#each ⇒ Object 



40
41
42
 
# File 'lib/mock_redis/zset.rb', line 40

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

#in_range(min, max) ⇒ Object 



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
70
71
72
 
# File 'lib/mock_redis/zset.rb', line 44

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 



74
75
76
77
78
79
80
81
82
83
 
# File 'lib/mock_redis/zset.rb', line 74

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 



85
86
87
 
# File 'lib/mock_redis/zset.rb', line 85

def score(member)
  scores[member]
end

#sorted ⇒ Object 



89
90
91
92
93
 
# File 'lib/mock_redis/zset.rb', line 89

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

#sorted_members ⇒ Object 



95
96
97
 
# File 'lib/mock_redis/zset.rb', line 95

def sorted_members
  sorted.map(&:last)
end

#union(other) ⇒ Object 



99
100
101
102
103
104
105
106
107
108
 
# File 'lib/mock_redis/zset.rb', line 99

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