Contents

# frozen_string_literal: true

module GrpcKit
  module Session
    class SendBuffer
      def initialize
        @buffer = ''.b
        @end_write = false
        @deferred_read = false
      end

      def write(data, last: false)
        end_write if last
        @buffer << data
      end

      def need_resume?
        @deferred_read
      end

      def end_write
        @end_write = true
      end

      def end_write?
        @end_write
      end

      def read(size = nil)
        if @buffer.empty?
          if end_write?
            @deferred_read = false
            return nil # EOF
          end

          @deferred_read = true
          return DS9::ERR_DEFERRED
        end

        if size.nil? || @buffer.bytesize < size
          buf = @buffer
          @buffer = ''.b
          buf
        else
          @buffer.freeze
          rbuf = @buffer.byteslice(0, size)
          @buffer = @buffer.byteslice(size, @buffer.bytesize)
          rbuf
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
grpc_kit-0.1.8	lib/grpc_kit/session/send_buffer.rb