1 files changed, 8 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/Internal.lhs b/src/Network/BitTorrent/Internal.lhs
index 7edcc977..5f6ad458 100644
--- a/src/Network/BitTorrent/Internal.lhs
+++ b/src/Network/BitTorrent/Internal.lhs
@@ -56,9 +56,8 @@
 >        , newSeeder
 >        , getClientBitfield
 >
->        , enterSwarm
->        , leaveSwarm
 >        , waitVacancy
+>        , forkThrottle
 >
 >        , pieceLength
 >
@@ -616,6 +615,13 @@ Peer sessions throttling
 >   bracket (enterSwarm se) (const (leaveSwarm se))
 >                   . const
 
+> forkThrottle :: SwarmSession -> IO () -> IO ThreadId
+> forkThrottle se action = do
+>   enterSwarm se
+>   (forkIO $ do
+>     action `finally` leaveSwarm se)
+>        `onException` leaveSwarm se
+
 Peer sessions
 ------------------------------------------------------------------------