module Redis::Connection::SocketMixin

Constants

CRLF

Public Class Methods

new(*args) click to toggle source
Calls superclass method
# File lib/redis/connection/ruby.rb, line 20
def initialize(*args)
  super(*args)

  @timeout = @write_timeout = nil
  @buffer = "".dup
end

Public Instance Methods

_read_from_socket(nbytes) click to toggle source
# File lib/redis/connection/ruby.rb, line 51
def _read_from_socket(nbytes)
  begin
    read_nonblock(nbytes)
  rescue IO::WaitReadable
    if IO.select([self], nil, nil, @timeout)
      retry
    else
      raise Redis::TimeoutError
    end
  rescue IO::WaitWritable
    if IO.select(nil, [self], nil, @timeout)
      retry
    else
      raise Redis::TimeoutError
    end
  end
rescue EOFError
  raise Errno::ECONNRESET
end
_write_to_socket(data) click to toggle source
# File lib/redis/connection/ruby.rb, line 71
def _write_to_socket(data)
  begin
    write_nonblock(data)
  rescue IO::WaitWritable
    if IO.select(nil, [self], nil, @write_timeout)
      retry
    else
      raise Redis::TimeoutError
    end
  rescue IO::WaitReadable
    if IO.select([self], nil, nil, @write_timeout)
      retry
    else
      raise Redis::TimeoutError
    end
  end
rescue EOFError
  raise Errno::ECONNRESET
end
gets() click to toggle source
# File lib/redis/connection/ruby.rb, line 43
def gets
  while (crlf = @buffer.index(CRLF)).nil?
    @buffer << _read_from_socket(16_384)
  end

  @buffer.slice!(0, crlf + CRLF.bytesize)
end
read(nbytes) click to toggle source
# File lib/redis/connection/ruby.rb, line 35
def read(nbytes)
  result = @buffer.slice!(0, nbytes)

  result << _read_from_socket(nbytes - result.bytesize) while result.bytesize < nbytes

  result
end
timeout=(timeout) click to toggle source
# File lib/redis/connection/ruby.rb, line 27
def timeout=(timeout)
  @timeout = (timeout if timeout && timeout > 0)
end
write(data) click to toggle source
Calls superclass method
# File lib/redis/connection/ruby.rb, line 91
def write(data)
  return super(data) unless @write_timeout

  length = data.bytesize
  total_count = 0
  loop do
    count = _write_to_socket(data)

    total_count += count
    return total_count if total_count >= length

    data = data.byteslice(count..-1)
  end
end
write_timeout=(timeout) click to toggle source
# File lib/redis/connection/ruby.rb, line 31
def write_timeout=(timeout)
  @write_timeout = (timeout if timeout && timeout > 0)
end