Add support for #close_write.

socketry · Sep 10, 2024 · f2ffaca · f2ffaca
1 parent f5e1ea1
commit f2ffaca
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/lib/protocol/rack/body/streaming.rb b/lib/protocol/rack/body/streaming.rb
@@ -41,15 +41,23 @@ def write(chunk)
 						end
 					end
 
-					def close
+					def close(error = nil)
 						@fiber = nil
 
 						if from = @from
 							@from = nil
-							from.transfer(nil)
+							if error
+								from.raise(error)
+							else
+								from.transfer(nil)
+							end
 						end
 					end
 
+					def close_write(error = nil)
+						close(error)
+					end
+
 					def read
 						raise RuntimeError, "Stream is already being read!" if @from