sinatra-stream-chunked-response.rb

# -- coding: utf-8

require "pp"
require "rubygems"
require 'sinatra'
require 'sinatra/base'

module Sinatra
  module Helpers
    class Stream
      def each(&front)
        @front = front
        callback do
          @front.call("0\r\n\r\n")
        end

        @scheduler.defer do
          begin
            @back.call(self)
          rescue Exception => e
            @scheduler.schedule { raise e }
          end
          close unless @keep_open
        end
      end

      def <<(data)
        @scheduler.schedule do
          size = data.to_s.bytesize
          @front.call([size.to_s(16), "\r\n", data.to_s, "\r\n"].join)
        end
        self
      end
    end
  end
end

get "/" do
  headers "Transfer-Encoding" => "chunked"
  stream do |out|
    out << "foo"
    sleep 0.3
    out << "bar"*10
    sleep 0.5
    out << "jkjllkjk"
  end
end