Sha256: aac5767bbbf707c90c0127aba52c0cd60b0a9753c61944f82d5369dcdf072a29
Contents?: true
Size: 1.57 KB
Versions: 7
Compression:
Stored size: 1.57 KB
Contents
require "forwardable"
require "http/client"
module HTTP
class Response
# A streamable response body, also easily converted into a string
class Body
extend Forwardable
include Enumerable
def_delegator :to_s, :empty?
def initialize(client, encoding = Encoding::BINARY)
@client = client
@streaming = nil
@contents = nil
@encoding = encoding
end
# (see HTTP::Client#readpartial)
def readpartial(*args)
stream!
@client.readpartial(*args)
end
# Iterate over the body, allowing it to be enumerable
def each
while (chunk = readpartial)
yield chunk
end
end
# @return [String] eagerly consume the entire body as a string
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
alias_method :to_str, :to_s
# Assert that the body is actively being streamed
def stream!
fail StateError, "body has already been consumed" if @streaming == false
@streaming = true
end
# Easier to interpret string inspect
def inspect
"#<#{self.class}:#{object_id.to_s(16)} @streaming=#{!!@streaming}>"
end
end
end
end
Version data entries
7 entries across 7 versions & 1 rubygems