Invoke close_write(error) if possible.

socketry · Oct 29, 2024 · 3d0cf2e · 3d0cf2e
1 parent 968ae4e
commit 3d0cf2e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/xrb/template.rb b/lib/xrb/template.rb
@@ -119,6 +119,14 @@ def to_buffer(scope)
 		def to_proc(scope = @binding.dup)
 			proc do |output|
 				to_string(scope, output)
+			rescue => error
+				raise
+			ensure
+				if output.method(:close_write).arity == 1
+					output.close_write(error)
+				else
+					output.close_write
+				end
 			end
 		end