From 53f31fd94c542d465d7bacb6d32ca67fbf6653d2 Mon Sep 17 00:00:00 2001
From: syeopite <syeopite@syeopite.dev>
Date: Wed, 9 Jun 2021 15:52:46 -0700
Subject: [PATCH] Revert "Update recieve function to read full data from socket
 at once"

This reverts commit a3d2458c282b0b2c1a8907dca81851618efb2b1b.
---
 src/lsquic/client.cr | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/lsquic/client.cr b/src/lsquic/client.cr
index 63a7a24..b99f3db 100644
--- a/src/lsquic/client.cr
+++ b/src/lsquic/client.cr
@@ -225,10 +225,11 @@ module QUIC
           LibLsquic.engine_packet_in(engine, buffer[0, bytes_read], bytes_read, socket.local_address, socket.remote_address, Box.box(socket), 0) if bytes_read != 0
           LibLsquic.engine_process_conns(engine)
         end
-        @socket.try &.close
-        @socket = nil
       rescue IO::Error
         # may have already been closed
+      ensure
+        @socket.try &.close
+        @socket = nil
       end
     end