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



67
68
69
 
# File 'lib/http/response/body.rb', line 67

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



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

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
51
52
53
54
55
56
57
 
# File 'lib/http/response/body.rb', line 33

def to_s
  return @contents if @contents

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

  # see issue 312
  begin
    encoding = Encoding.find @encoding
  rescue ArgumentError
    encoding = Encoding::BINARY
  end

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

  @contents
end