Class: SSE::HTTPResponseReader

Inherits:
Object
  • Object
show all
Defined in:
lib/sse_client/streaming_http.rb

Overview

Used internally to read the HTTP response, either all at once or as a stream of text lines. Incoming data is fed into an instance of HTTPTools::Parser, which gives us the header and chunks of the body via callbacks.

Constant Summary collapse

DEFAULT_CHUNK_SIZE = 
10000

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(socket, read_timeout) ⇒ HTTPResponseReader

Returns a new instance of HTTPResponseReader.



125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
 
# File 'lib/sse_client/streaming_http.rb', line 125

def initialize(socket, read_timeout)
  @socket = socket
  @read_timeout = read_timeout
  @parser = HTTPTools::Parser.new
  @buffer = ""
  @done = false
  @lock = Mutex.new

  # Provide callbacks for the Parser to give us the headers and body. This has to be done
  # before we start piping any data into the parser.
  have_headers = false
  @parser.on(:header) do
    have_headers = true
  end
  @parser.on(:stream) do |data|
    @lock.synchronize { @buffer << data }  # synchronize because we're called from another thread in Socketry
  end
  @parser.on(:finish) do
    @lock.synchronize { @done = true }
  end

  # Block until the status code and headers have been successfully read.
  while !have_headers
    raise EOFError if !read_chunk_into_buffer
  end
  @headers = Hash[@parser.header.map { |k,v| [k.downcase, v] }]
  @status = @parser.status_code
end

Instance Attribute Details

#headers ⇒ Object (readonly)

Returns the value of attribute headers.



123
124
125
 
# File 'lib/sse_client/streaming_http.rb', line 123

def headers
  @headers
end

#status ⇒ Object (readonly)

Returns the value of attribute status.



123
124
125
 
# File 'lib/sse_client/streaming_http.rb', line 123

def status
  @status
end

Instance Method Details

#read_all ⇒ Object 



164
165
166
167
168
 
# File 'lib/sse_client/streaming_http.rb', line 164

def read_all
  while read_chunk_into_buffer
  end
  @buffer
end

#read_lines ⇒ Object 



154
155
156
157
158
159
160
161
162
 
# File 'lib/sse_client/streaming_http.rb', line 154

def read_lines
  Enumerator.new do |gen|
    loop do
      line = read_line
      break if line.nil?
      gen.yield line
    end
  end
end