lib/grpc_kit/session/stream.rb in grpc_kit-0.1.7 vs lib/grpc_kit/session/stream.rb in grpc_kit-0.1.8

@@ -1,11 +1,12 @@
 # frozen_string_literal: false
 
 require 'forwardable'
-require 'grpc_kit/session/buffer'
 require 'grpc_kit/session/headers'
 require 'grpc_kit/session/stream_status'
+require 'grpc_kit/session/recv_buffer'
+require 'grpc_kit/session/send_buffer'
 
 module GrpcKit
   module Session
     class Stream
       extend Forwardable
@@ -19,11 +20,11 @@
 
       def initialize(stream_id:, send_data: nil, recv_data: nil)
         @stream_id = stream_id
         @end_read_stream = false
         @headers = GrpcKit::Session::Headers.new
-        @pending_send_data = send_data || GrpcKit::Session::Buffer.new
-        @pending_recv_data = recv_data || GrpcKit::Session::Buffer.new
+        @pending_send_data = send_data || GrpcKit::Session::SendBuffer.new
+        @pending_recv_data = recv_data || GrpcKit::Session::RecvBuffer.new
 
         @inflight = false
         @trailer_data = {}
         @status = GrpcKit::Session::StreamStatus.new
         @draining = false