Module: Redis::Connection::SocketMixin

Included in:
SSLSocket, TCPSocket, UNIXSocket
Defined in:
lib/redis/connection/ruby.rb

Constant Summary collapse

CRLF =
"\r\n"

Instance Method Summary collapse

Instance Method Details

#_read_from_socket(nbytes, buffer = nil) ⇒ Object



53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# File 'lib/redis/connection/ruby.rb', line 53

def _read_from_socket(nbytes, buffer = nil)
  loop do
    case chunk = read_nonblock(nbytes, buffer, exception: false)
    when :wait_readable
      unless wait_readable(@timeout)
        raise Redis::TimeoutError
      end
    when :wait_writable
      unless wait_writable(@timeout)
        raise Redis::TimeoutError
      end
    when nil
      raise Errno::ECONNRESET
    when String
      return chunk
    end
  end
end

#getsObject



45
46
47
48
49
50
51
# File 'lib/redis/connection/ruby.rb', line 45

def gets
  while (crlf = @buffer.index(CRLF)).nil?
    @buffer << _read_from_socket(16_384)
  end

  @buffer.slice!(0, crlf + CRLF.bytesize)
end

#initialize(*args) ⇒ Object



21
22
23
24
25
26
# File 'lib/redis/connection/ruby.rb', line 21

def initialize(*args)
  super(*args)

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

#read(nbytes) ⇒ Object



36
37
38
39
40
41
42
43
# File 'lib/redis/connection/ruby.rb', line 36

def read(nbytes)
  result = @buffer.slice!(0, nbytes)

  buffer = String.new(capacity: nbytes, encoding: Encoding::ASCII_8BIT)
  result << _read_from_socket(nbytes - result.bytesize, buffer) while result.bytesize < nbytes

  result
end

#timeout=(timeout) ⇒ Object



28
29
30
# File 'lib/redis/connection/ruby.rb', line 28

def timeout=(timeout)
  @timeout = (timeout if timeout && timeout > 0)
end

#write(buffer) ⇒ Object



72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# File 'lib/redis/connection/ruby.rb', line 72

def write(buffer)
  return super(buffer) unless @write_timeout

  bytes_to_write = buffer.bytesize
  total_bytes_written = 0
  loop do
    case bytes_written = write_nonblock(buffer, exception: false)
    when :wait_readable
      unless wait_readable(@write_timeout)
        raise Redis::TimeoutError
      end
    when :wait_writable
      unless wait_writable(@write_timeout)
        raise Redis::TimeoutError
      end
    when nil
      raise Errno::ECONNRESET
    when Integer
      total_bytes_written += bytes_written

      if total_bytes_written >= bytes_to_write
        return total_bytes_written
      end

      buffer = buffer.byteslice(bytes_written..-1)
    end
  end
end

#write_timeout=(timeout) ⇒ Object



32
33
34
# File 'lib/redis/connection/ruby.rb', line 32

def write_timeout=(timeout)
  @write_timeout = (timeout if timeout && timeout > 0)
end