Method: NIO::Selector#select

Defined in:
lib/nio/selector.rb,
ext/nio4r/selector.c
more...

#select(timeout = nil) ⇒ Object

Select which monitors are ready

[View source]

96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
# File 'lib/nio/selector.rb', line 96

def select(timeout = nil)
  selected_monitors = Set.new

  @lock.synchronize do
    readers = [@wakeup]
    writers = []

    @selectables.each do |io, monitor|
      readers << io if monitor.interests == :r || monitor.interests == :rw
      writers << io if monitor.interests == :w || monitor.interests == :rw
      monitor.readiness = nil
    end

    ready_readers, ready_writers = Kernel.select(readers, writers, [], timeout)
    return unless ready_readers # timeout

    ready_readers.each do |io|
      if io == @wakeup
        # Clear all wakeup signals we've received by reading them
        # Wakeups should have level triggered behavior
        @wakeup.read(@wakeup.stat.size)
      else
        monitor = @selectables[io]
        monitor.readiness = :r
        selected_monitors << monitor
      end
    end

    ready_writers.each do |io|
      monitor = @selectables[io]
      monitor.readiness = monitor.readiness == :r ? :rw : :w
      selected_monitors << monitor
    end
  end

  if block_given?
    selected_monitors.each { |m| yield m }
    selected_monitors.size
  else
    selected_monitors.to_a
  end
end