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

#initializeZset

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

#membersObject (readonly)

Returns the value of attribute members


9
10
11
# File 'lib/mock_redis/zset.rb', line 9

def members
  @members
end

#scoresObject (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

Returns:

  • (Boolean)

30
31
32
33
# File 'lib/mock_redis/zset.rb', line 30

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

#eachObject


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

#sortedObject


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_membersObject


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