Sha256: 3a6c0fbcef0bf27a0b47361817ea04c9a46b3fe0b6a501d04ae9cbaf54111bb2

Contents?: true

Size: 1.67 KB

Versions: 12

Compression:

Stored size: 1.67 KB

Contents

# -*- encoding: binary -*-
# :enddoc:
# used to wrap a BasicSocket to use with +q+ for all writes
# this is compatible with IO.select
class Rainbows::WriterThreadPool::Client < Struct.new(:to_io, :q)
  include Rainbows::SocketProxy
  include Rainbows::ProcessClient

  module Methods
    def write_body_each(body)
      return if @hp.hijacked?
      q << [ to_io, :write_body_each, body ]
    end

    def write_response_close(status, headers, body, alive)
      to_io.instance_variable_set(:@hp, @hp) # XXX ugh
      return if @hp.hijacked?
      Rainbows::SyncClose.new(body) { |sync_body|
        q << [ to_io, :write_response, status, headers, sync_body, alive ]
      }
    end

    if Rainbows::Response::COPY_STREAM || IO.method_defined?(:trysendfile)
      def write_response(status, headers, body, alive)
        if body.respond_to?(:close)
          write_response_close(status, headers, body, alive)
        elsif body.respond_to?(:to_path)
          write_response_path(status, headers, body, alive)
        else
          super
        end
      end

      def write_body_file(body, range)
        q << [ to_io, :write_body_file, body, range ]
      end

      def write_body_stream(body)
        q << [ to_io, :write_body_stream, body ]
      end
    else # each-only body response
      def write_response(status, headers, body, alive)
        if body.respond_to?(:close)
          write_response_close(status, headers, body, alive)
        else
          super
        end
      end
    end # each-only body response
  end # module Methods
  include Methods

  def write(buf)
    q << [ to_io, buf ]
  end

  def close
    q << [ to_io, :close ]
  end

  def closed?
    to_io.closed?
  end
end

Version data entries

12 entries across 12 versions & 1 rubygems

Version Path
rainbows-5.2.1 lib/rainbows/writer_thread_pool/client.rb
rainbows-5.2.0 lib/rainbows/writer_thread_pool/client.rb
rainbows-5.1.1 lib/rainbows/writer_thread_pool/client.rb
rainbows-5.1.0 lib/rainbows/writer_thread_pool/client.rb
rainbows-5.0.0.5.ge717 lib/rainbows/writer_thread_pool/client.rb
rainbows-5.0.0 lib/rainbows/writer_thread_pool/client.rb
rainbows-4.7.0 lib/rainbows/writer_thread_pool/client.rb
rainbows-4.6.2 lib/rainbows/writer_thread_pool/client.rb
rainbows-4.6.1 lib/rainbows/writer_thread_pool/client.rb
rainbows-4.6.0.4.g4108 lib/rainbows/writer_thread_pool/client.rb
rainbows-4.6.0 lib/rainbows/writer_thread_pool/client.rb
rainbows-4.5.0 lib/rainbows/writer_thread_pool/client.rb