Add support for defer_stop.

socketry · Mar 27, 2024 · cf288fe · cf288fe
1 parent f681cd4
commit cf288fe
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/lib/async/http/protocol/http1/server.rb b/lib/async/http/protocol/http1/server.rb
@@ -55,7 +55,7 @@ def each(task: Task.current)
 								return
 							end
 
-							begin
+							task.defer_stop do
 								# If a response was generated, send it:
 								if response
 									trailer = response.headers.trailer!

diff --git a/lib/async/http/protocol/http2/connection.rb b/lib/async/http/protocol/http2/connection.rb
@@ -89,7 +89,7 @@ def read_in_background(parent: Task.current)
 
 							task.annotate("#{version} reading data for #{self.class}.")
 
-							begin
+							task.defer_stop do
 								while !self.closed?
 									self.consume_window
 									self.read_frame

diff --git a/lib/async/http/protocol/http2/server.rb b/lib/async/http/protocol/http2/server.rb
@@ -53,7 +53,7 @@ def each(task: Task.current)
 
 							@count += 1
 
-							begin
+							task.defer_stop do
 								response = yield(request)
 							rescue
 								# We need to close the stream if the user code blows up while generating a response: