using yield

haskell · Jun 10, 2024 · b3c898e · b3c898e
1 parent 507ab0e
commit b3c898e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Network/Socket/Shutdown.hs b/Network/Socket/Shutdown.hs
@@ -11,7 +11,7 @@ module Network.Socket.Shutdown (
 import qualified Control.Exception as E
 import Foreign.Marshal.Alloc (mallocBytes, free)
 
-import Control.Concurrent (threadDelay)
+import Control.Concurrent (threadDelay, yield)
 
 import Network.Socket.Buffer
 import Network.Socket.Imports
@@ -55,6 +55,9 @@ gracefulClose s tmout0 = sendRecvFIN `E.finally` close s
         case ex of
           Left (E.SomeException _) -> return ()
           Right () -> do
+              -- Giving CPU time to other threads hoping that
+              -- FIN arrives meanwhile.
+              yield
               -- Waiting TCP FIN.
               E.bracket (mallocBytes bufSize) free recvEOFloop
     recvEOFloop buf = loop 1 0