Sha256: caca9943dc2e50400beee664ed94b19e5d23a579ac3188048d7e3e45a2d086a7
Contents?: true
Size: 1.19 KB
Versions: 7
Compression:
Stored size: 1.19 KB
Contents
module Reel
# Represents the bodies of Requests
class RequestBody
include Enumerable
def initialize(request)
@request = request
@streaming = nil
@contents = nil
end
# Read exactly the given amount of data
def read(length)
stream!
@request.read(length)
end
# Read up to length bytes, but return any data that's available
def readpartial(length = nil)
stream!
@request.readpartial(length)
end
# Iterate over the body, allowing it to be enumerable
def each
while chunk = readpartial
yield chunk
end
end
# Eagerly consume the entire body as a string
def to_s
return @contents if @contents
raise StateError, "body is being streamed" unless @streaming.nil?
begin
@streaming = false
@contents = ""
while chunk = @request.readpartial
@contents << chunk
end
rescue
@contents = nil
raise
end
@contents
end
# Assert that the body is actively being streamed
def stream!
raise StateError, "body has already been consumed" if @streaming == false
@streaming = true
end
end
end
Version data entries
7 entries across 7 versions & 1 rubygems