Sha256: b6215e9c06aa4b30675658a23f10d98653084a799b480ee948251a0ed96053a7
Contents?: true
Size: 1.74 KB
Versions: 13
Compression:
Stored size: 1.74 KB
Contents
module HTTP
module Timeout
class PerOperation < Null
CONNECT_TIMEOUT = 0.25
WRITE_TIMEOUT = 0.25
READ_TIMEOUT = 0.25
attr_reader :read_timeout, :write_timeout, :connect_timeout
def initialize(*args)
super
@read_timeout = options.fetch(:read_timeout, READ_TIMEOUT)
@write_timeout = options.fetch(:write_timeout, WRITE_TIMEOUT)
@connect_timeout = options.fetch(:connect_timeout, CONNECT_TIMEOUT)
end
def connect(socket_class, host, port)
::Timeout.timeout(connect_timeout, TimeoutError) do
@socket = socket_class.open(host, port)
end
end
def connect_ssl
socket.connect_nonblock
rescue IO::WaitReadable
if IO.select([socket], nil, nil, connect_timeout)
retry
else
raise TimeoutError, "Connection timed out after #{connect_timeout} seconds"
end
rescue IO::WaitWritable
if IO.select(nil, [socket], nil, connect_timeout)
retry
else
raise TimeoutError, "Connection timed out after #{connect_timeout} seconds"
end
end
# Read data from the socket
def readpartial(size)
socket.read_nonblock(size)
rescue IO::WaitReadable
if IO.select([socket], nil, nil, read_timeout)
retry
else
raise TimeoutError, "Read timed out after #{read_timeout} seconds"
end
end
# Write data to the socket
def write(data)
socket.write_nonblock(data)
rescue IO::WaitWritable
if IO.select(nil, [socket], nil, write_timeout)
retry
else
raise TimeoutError, "Read timed out after #{write_timeout} seconds"
end
end
end
end
end
Version data entries
13 entries across 13 versions & 1 rubygems