Invoke close_write(error) if possible.

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