Contents

require 'timeout'

module EventLoop
  class << self
    SELECTOR = NIO::Selector.new
    TIMERS = []
    IOS = Hash.new
    QUEUE = Hash.new

    def add_timer(timer)
      timer.start_time = Time.now.to_f + timer.time
      TIMERS << timer
    end

    def register(io, interest=(:rw), timeout=nil, resolve=nil, &callback)
      if QUEUE[io.to_i].nil?
        QUEUE[io.to_i] = Array.new
        register_raw(io, interest, timeout, resolve, callback)
      else
        QUEUE[io.to_i] << [io, interest, timeout, resolve, callback]
      end
    end

    def register_raw(io, interest=(:rw), timeout=nil, resolve=nil, callback)
      SELECTOR.register(io, interest)
      timer = nil
      unless timeout.nil?
        timer = EventLoop::Timer.new(timeout) do
          EventLoop.deregister(io)
          resolve.call(TimeoutError) unless resolve.nil?
        end
        EventLoop.add_timer(timer)
      end

      IOS[io] = {
          timer: timer ? timer : nil,
          callback: callback,
      }
    end

    def deregister(io)
      fd = io.to_i
      SELECTOR.deregister(io)
      IOS.delete(io)
      next_register = QUEUE[fd].shift
      next_register.nil? ? QUEUE.delete(fd) : register_raw(*next_register)
    end

    def run_once
      SELECTOR.select(0.2) do |monitor| # Timeout for 0.2 secs
        TIMERS.delete(IOS[monitor.io][:timer]) unless IOS[monitor.io][:timer].nil?
        IOS[monitor.io][:callback].call(monitor)
      end
      EventLoop.timer_once
    end

    def timer_once
      now_time = Time.now.to_f
      TIMERS.delete_if do |timer|
        if timer.start_time < now_time
          timer.callback.call
          true
        end
      end
    end

    def start
      return if running?
      @stop = false
      until @stop
        run_once
      end
      @stop = nil
    end

    def stop
      @stop = true
    end

    def running?
      @stop = true if @stop.nil?
      !@stop
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
em-midori-0.2.0	lib/midori/core_ext/event_loop.rb