lib/grpc_kit/session/stream.rb in grpc_kit-0.1.3 vs lib/grpc_kit/session/stream.rb in grpc_kit-0.1.4

@@ -25,9 +25,14 @@
         @pending_recv_data = recv_data || GrpcKit::Session::Buffer.new
 
         @inflight = false
         @trailer_data = {}
         @status = GrpcKit::Session::StreamStatus.new
+        @draining = false
+      end
+
+      def drain
+        @draining = true
       end
 
       def write_trailers_data(tariler)
         @trailer_data = tariler
       end