network-mux: forkJob return Async

IntersectMBO · May 4, 2021 · 73d48c0 · 73d48c0
1 parent 4fd6231
commit 73d48c0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/network-mux/src/Control/Concurrent/JobPool.hs b/network-mux/src/Control/Concurrent/JobPool.hs
@@ -58,7 +58,7 @@ forkJob :: forall m a.
            (MonadAsync m, MonadMask m)
         => JobPool m a
         -> Job     m a
-        -> m ()
+        -> m (Async m ())
 forkJob JobPool{jobsVar, completionQueue} (Job action handler label) =
     mask $ \restore -> do
       jobAsync <- async $ do
@@ -72,6 +72,7 @@ forkJob JobPool{jobsVar, completionQueue} (Job action handler label) =
 
       let !tid = asyncThreadId (Proxy :: Proxy m) jobAsync
       atomically $ modifyTVar' jobsVar (Map.insert tid jobAsync)
+      return jobAsync
   where
     notAsyncExceptions :: SomeException -> Maybe SomeException
     notAsyncExceptions e