module EventLoop - murasaki-0.2.2 Documentation

add_timer(timer) click to toggle source

Add timer in event loop @param [EventLoop::Timer] timer timer to insert @return [nil] nil

# File lib/murasaki/event_loop.rb, line 32
def add_timer(timer)
  timer.start_time = Time.now.to_f + timer.time
  @timers << timer
  nil
end

config(selector = NIO::Selector.new) click to toggle source

Config EvnetLoop, call by default if any other methods called @param [NIO::Selector] selector an event selector

# File lib/murasaki/event_loop.rb, line 7
def config(selector = NIO::Selector.new)
  # Raw NIO Selector
  @selector = selector
  # Array of active timers
  @timers = []
  # Hash of io and its callback
  @ios = Hash.new
  # IO queue
  @queue = Hash.new
  # Root fiber
  @root_fiber = Fiber.current
  # Thread for timer
  @timer_thread = nil
  nil
end

deregister(io) click to toggle source

Deregister I/O event @param [IO] io io to deregister @return [nil] nil

# File lib/murasaki/event_loop.rb, line 71
def deregister(io)
  @selector.deregister(io)
  @ios.delete(io)
  unless io.closed?
    # If the I/O closed accidentally, it should manually release the queue
    # Otherwise, there would be a memory leak
    fd = io.fileno
    next_register = @queue[fd].shift
    next_register.nil? ? @queue.delete(fd) : register_raw(*next_register)
  end
  nil
end

register(io, interest=(:rw), &callback) click to toggle source

Register I/O event with queue protection @param [IO] io io to register @param [Symbol] interest :r for read only, :w for write only, and :rw for both @yield what to run when io callbacks @return [nil] nil

# File lib/murasaki/event_loop.rb, line 47
def register(io, interest=(:rw), &callback)
  if @queue[io.to_i].nil?
    @queue[io.to_i] = Array.new
    register_raw(io, interest, callback)
  else
    @queue[io.to_i] << [io, interest, callback]
  end
  nil
end

register_raw(io, interest=(:rw), callback) click to toggle source

Register I/O event directly, without any queue protection @param [IO] io io to register @param [Symbol] interest :r for read only, :w for write only, and :rw for both @param [Proc] callback what to run when io callbacks @return [nil] nil

# File lib/murasaki/event_loop.rb, line 62
def register_raw(io, interest=(:rw), callback)
  @selector.register(io, interest)
  @ios[io] = { callback: callback }
  nil
end

release_queue(io) click to toggle source

Manually release queues of closed I/O if the I/O has accidentally closed @param [IO] io io to release @return [nil] nil

# File lib/murasaki/event_loop.rb, line 88
def release_queue(io)
  fd = io.fileno
  @queue.delete(fd)
  nil
end

remove_timer(timer) click to toggle source

# File lib/murasaki/event_loop.rb, line 38
def remove_timer(timer)
  @timers.delete(timer)
end

root_fiber() click to toggle source

# File lib/murasaki/event_loop.rb, line 144
def root_fiber
  @root_fiber
end

run_once() click to toggle source

Run I/O selector once @return [nil] nil

# File lib/murasaki/event_loop.rb, line 96
def run_once
  @selector.select(0.2) do |monitor| # Timeout for 0.2 secs
    @ios[monitor.io][:callback].call(monitor)
  end
  nil
end

running?() click to toggle source

Detect the stop flag @return [Boolean] return if eventloop is set to be stopped

# File lib/murasaki/event_loop.rb, line 139
def running?
  @stop = true if @stop.nil?
  !@stop
end

start() click to toggle source

Start the event loop @return [nil] nil

# File lib/murasaki/event_loop.rb, line 119
def start
  return if running?
  @stop = false
  @timer_thread = Thread.new { loop { sleep 0.5; timer_once }}
  until @stop
    run_once
  end
  @stop = nil
end

stop() click to toggle source

Set the stop flag @return [nil] nil

# File lib/murasaki/event_loop.rb, line 131
def stop
  @stop = true
  Thread.kill(@timer_thread)
  nil
end

timer_once() click to toggle source

Run timer once @return [nil] nil

# File lib/murasaki/event_loop.rb, line 105
def timer_once
  now_time = Time.now.to_f
  callbacks = []
  @timers.reject! do |timer|
    ticked = timer.start_time < now_time
    callbacks << timer.callback if ticked
    ticked
  end
  callbacks.each { |c| c.call }
  nil
end