Sha256: 8b560576a4d2ec8af4a391e863cd464afb34d561f3e44ea0b7de4ab3f50f480b

Contents?: true

Size: 1.53 KB

Versions: 8

Compression:

Stored size: 1.53 KB

Contents

module Workers
  class Scheduler
    include Workers::Helpers

    def initialize(options = {})
      @logger = Workers::LogProxy.new(options[:logger])
      @pool = options[:pool] || Workers.pool
      @schedule = SortedSet.new
      @mutex = Mutex.new
      @thread = Thread.new { start_loop }

      return nil
    end

    def schedule(timer)
      @mutex.synchronize do
        @schedule << timer
      end

      wakeup

      return nil
    end

    def unschedule(timer)
      @mutex.synchronize do
        @schedule.delete(timer)
      end

      return true
    end

    def wakeup
      @thread.wakeup

      return nil
    end

    def dispose
      @mutex.synchronize do
        @pool.shutdown
        @pool.join
        @thread.kill
      end

      return nil
    end

    def alive?
      return @thread && @thread.alive?
    end

    private

    def start_loop
      while true
        delay = nil

        @mutex.synchronize do
          process_overdue
          delay = next_delay
        end

        delay ? sleep(delay) : sleep
      end

      return nil
    end

    def process_overdue
      overdue = []

      while @schedule.first && @schedule.first.overdue?
        overdue << @schedule.first
        @schedule.delete(@schedule.first)
      end

      overdue.each do |timer|
        @pool.perform do
          timer.fire
        end

        timer.reset
        @schedule << timer if timer.repeat
      end

      return nil
    end

    def next_delay
      @schedule.first ? @schedule.first.sec_remaining : nil
    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version Path
workers-0.2.2 lib/workers/scheduler.rb
workers-0.2.1 lib/workers/scheduler.rb
workers-0.2.0 lib/workers/scheduler.rb
workers-0.1.4 lib/workers/scheduler.rb
workers-0.1.3 lib/workers/scheduler.rb
workers-0.1.2 lib/workers/scheduler.rb
workers-0.1.1 lib/workers/scheduler.rb
workers-0.1.0 lib/workers/scheduler.rb