dbf5b46 · streamly · git.iko.soy

git.iko.soy/streamly

on master help sign in

Add parMergeBy and parMergeByM

dbf5b46Harendra Kumar4 years ago

src/Streamly/Internal/Data/Stream/Concurrent.hs+32

compare with master

unifiedsplit

@@ -65,6 +65,8 @@ module Streamly.Internal.Data.Stream.Concurrent

65 , parTwo

66 , parZipWithM

67 , parZipWith

68+ , parMergeByM

69+ , parMergeBy

69 -- ** List of streams

71 -- ** List of streams

70 -- | Shares a single channel across many streams.

72 -- | Shares a single channel across many streams.

@@ -582,6 +584,36 @@ parZipWith :: MonadAsync m

582 -> Stream m c

584 -> Stream m c

583parZipWith cfg f = parZipWithM cfg (\a b -> return $ f a b)

585parZipWith cfg f = parZipWithM cfg (\a b -> return $ f a b)

584

586

587+-- | Like 'mergeByM' but evaluates both the streams concurrently.

588+--

589+-- Definition:

590+--

591+-- >>> parMergeByM cfg f m1 m2 = Stream.mergeByM f (Stream.parEval cfg m1) (Stream.parEval cfg m2)

592+--

593+{-# INLINE parMergeByM #-}

594+parMergeByM :: MonadAsync m

595+ => (Config -> Config)

596+ -> (a -> a -> m Ordering)

597+ -> Stream m a

598+ -> Stream m a

599+ -> Stream m a

600+parMergeByM cfg f m1 m2 = Stream.mergeByM f (parEval cfg m1) (parEval cfg m2)

601+

602+-- | Like 'mergeBy' but evaluates both the streams concurrently.

603+--

604+-- Definition:

605+--

606+-- >>> parMergeBy cfg f = Stream.parMergeByM cfg (\a b -> return $ f a b)

607+--

608+{-# INLINE parMergeBy #-}

609+parMergeBy :: MonadAsync m

610+ => (Config -> Config)

611+ -> (a -> a -> Ordering)

612+ -> Stream m a

613+ -> Stream m a

614+ -> Stream m a

615+parMergeBy cfg f = parMergeByM cfg (\a b -> return $ f a b)

616+

585-------------------------------------------------------------------------------

617-------------------------------------------------------------------------------

586-- concatIterate

618-- concatIterate

587-------------------------------------------------------------------------------

619-------------------------------------------------------------------------------