lib/grpc_kit/session/stream.rb in grpc_kit-0.3.9 vs lib/grpc_kit/session/stream.rb in grpc_kit-0.4.0

@@ -11,11 +11,10 @@
     class Stream
       extend Forwardable
 
       delegate %i[end_write?] => :@pending_send_data
       delegate %i[end_read?] => :@pending_recv_data
-      delegate %i[close close_remote close_local close? close_remote? close_local?] => :@status
 
       attr_reader :headers, :pending_send_data, :pending_recv_data, :trailer_data, :status
       attr_accessor :inflight, :stream_id
 
       # @param stream_id [Integer]
@@ -57,19 +56,32 @@
       def write_send_data(data, last: false)
         @pending_send_data.write(data, last: last)
       end
 
       # @param last [Boolean]
+      # @param blocking [Boolean]
       # @return [void]
-      def read_recv_data(last: false)
-        @pending_recv_data.read(last: last)
+      def read_recv_data(last: false, blocking:)
+        @pending_recv_data.read(last: last, blocking: blocking)
       end
 
       # @param name [String]
       # @param value [String]
       # @return [void]
       def add_header(name, value)
         @headers.add(name, value)
+      end
+
+      delegate %i[close_local close? close_remote? close_local?] => :@status
+
+      def close
+        status.close
+        pending_recv_data.close
+      end
+
+      def close_remote
+        status.close_remote
+        pending_recv_data.close
       end
     end
   end
 end