Method: HTTPX::Selector#select

Defined in:
lib/httpx/selector.rb

#select(interval) ⇒ Object

waits for read/write events for interval. Yields for monitors of selected IO objects.



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
138
 
# File 'lib/httpx/selector.rb', line 98

def select(interval)
  begin
    r = nil
    w = nil
    @lock.synchronize do
      r = @readers.keys
      w = @writers.keys
    end
    r.unshift(@__r__)

    readers, writers = IO.select(r, w, nil, interval)

    raise HTTPX::TimeoutError, "timed out while waiting on select" if readers.nil? && writers.nil?
  rescue IOError, SystemCallError
    @lock.synchronize do
      @readers.reject! { |io, _| io.closed? }
      @writers.reject! { |io, _| io.closed? }
    end
    retry
  end

  readers.each do |io|
    if io == @__r__
      # clean up wakeups
      @__r__.read(@__r__.stat.size)
    else
      monitor = io.closed? ? @readers.delete(io) : @readers[io]
      next unless monitor
      monitor.readiness = writers.delete(io) ? :rw : :r
      yield monitor
    end
  end if readers

  writers.each do |io|
    monitor = io.closed? ? @writers.delete(io) : @writers[io]
    next unless monitor
    # don't double run this, the last iteration might have run this task already
    monitor.readiness = :w
    yield monitor
  end if writers
end