Graceful shutdown?

ioquatix · ioquatix · commit f23f323f2738 · 2024-09-03T21:30:09.000+12:00
diff --git a/lib/async/websocket/adapters/http.rb b/lib/async/websocket/adapters/http.rb
@@ -40,7 +40,12 @@ def self.open(request, headers: [], protocols: [], handler: Connection, extensio
 							
 							yield connection
 							
-							connection.close unless connection.closed?
+							connection.send_close
+							until connection.closed?
+								connection.read_frame
+							end
+						ensure
+							stream.close
 						end
 						
 						# Once we get to this point, we no longer need to hold on to the request:
diff --git a/lib/async/websocket/client.rb b/lib/async/websocket/client.rb
@@ -62,6 +62,11 @@ def self.connect(endpoint, *arguments, **options, &block)
 					
 					begin
 						yield connection
+						
+						connection.send_close
+						until connection.closed?
+							connection.read_frame
+						end
 					ensure
 						connection.close
 						client.close
