close socket on failed handshake

plausible · Jan 27, 2024 · 9e27b6e · 9e27b6e
1 parent b64082b
commit 9e27b6e
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/lib/ch/connection.ex b/lib/ch/connection.ex
@@ -34,14 +34,17 @@ defmodule Ch.Connection do
               {:ok, conn}
 
             result ->
-              {:error,
-               Error.exception("unexpected result for '#{handshake}': " <> inspect(result))}
+              {:ok, _conn} = HTTP.close(conn)
+              reason = Error.exception("unexpected result for '#{handshake}': #{inspect(result)}")
+              {:error, reason}
           end
 
-        {:error, reason, _conn} ->
+        {:error, reason, conn} ->
+          {:ok, _conn} = HTTP.close(conn)
           {:error, reason}
 
-        {:disconnect, reason, _conn} ->
+        {:disconnect, reason, conn} ->
+          {:ok, _conn} = HTTP.close(conn)
           {:error, reason}
       end
     end