File: ring_buffer.rb

package info (click to toggle)
ruby-celluloid 0.18.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 848 kB
  • sloc: ruby: 7,579; makefile: 10
file content (64 lines) | stat: -rw-r--r-- 1,058 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
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
module Celluloid
  class RingBuffer
    def initialize(size)
      @size = size
      @start = 0
      @count = 0
      @buffer = Array.new(size)
      @mutex = Mutex.new
    end

    def full?
      @count == @size
    end

    def empty?
      @count == 0
    end

    def push(value)
      @mutex.synchronize do
        stop = (@start + @count) % @size
        @buffer[stop] = value
        if full?
          @start = (@start + 1) % @size
        else
          @count += 1
        end
        value
      end
    end
    alias << push

    def shift
      @mutex.synchronize do
        remove_element
      end
    end

    def flush
      values = []
      @mutex.synchronize do
        values << remove_element until empty?
      end
      values
    end

    def clear
      @buffer = Array.new(@size)
      @start = 0
      @count = 0
    end

    private

    def remove_element
      return nil if empty?
      value = @buffer[@start]
      @buffer[@start] = nil
      @start = (@start + 1) % @size
      @count -= 1
      value
    end
  end
end