Class: HTTP::Response::Body

Inherits:
Object
  • Object
show all
Extended by:
Forwardable
Includes:
Enumerable
Defined in:
lib/http/response/body.rb

Overview

A streamable response body, also easily converted into a string

Instance Method Summary collapse

Constructor Details

#initialize(client, encoding = Encoding::BINARY) ⇒ Body

Returns a new instance of Body.



12
13
14
15
16
17
 
# File 'lib/http/response/body.rb', line 12

def initialize(client, encoding = Encoding::BINARY)
  @client    = client
  @streaming = nil
  @contents  = nil
  @encoding  = encoding
end

Instance Method Details

#each ⇒ Object

Iterate over the body, allowing it to be enumerable



26
27
28
29
30
 
# File 'lib/http/response/body.rb', line 26

def each
  while (chunk = readpartial)
    yield chunk
  end
end

#inspect ⇒ Object

Easier to interpret string inspect



60
61
62
 
# File 'lib/http/response/body.rb', line 60

def inspect
  "#<#{self.class}:#{object_id.to_s(16)} @streaming=#{!!@streaming}>"
end

#readpartial(*args) ⇒ Object 



20
21
22
23
 
# File 'lib/http/response/body.rb', line 20

def readpartial(*args)
  stream!
  @client.readpartial(*args)
end

#stream! ⇒ Object

Assert that the body is actively being streamed



54
55
56
57
 
# File 'lib/http/response/body.rb', line 54

def stream!
  fail StateError, "body has already been consumed" if @streaming == false
  @streaming = true
end

#to_s ⇒ String Also known as: to_str

Returns eagerly consume the entire body as a string.

Returns:

  • (String) —

    eagerly consume the entire body as a string



33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'lib/http/response/body.rb', line 33

def to_s
  return @contents if @contents

  fail StateError, "body is being streamed" unless @streaming.nil?

  begin
    @streaming = false
    @contents = "".force_encoding(@encoding)
    while (chunk = @client.readpartial)
      @contents << chunk.force_encoding(@encoding)
    end
  rescue
    @contents = nil
    raise
  end

  @contents
end