Improve connection&.close handling.

Author: ioquatix

lib/async/websocket/adapters/http.rb

@@ -39,8 +39,9 @@ def self.open(request, headers: [], protocols: [], handler: Connection, extensio
 							connection = handler.call(framer, protocol, extensions)
 
 							yield connection
-							
-							connection.close unless connection.closed?
+						ensure
+							connection&.close
+							stream.close
 						end
 
 						# Once we get to this point, we no longer need to hold on to the request:

lib/async/websocket/client.rb

@@ -62,9 +62,10 @@ def self.connect(endpoint, *arguments, **options, &block)
 
 					begin
 						yield connection
+						
 					ensure
-						connection.close
-						client.close
+						connection&.close
+						client&.close
 					end
 				end
 			end