git.iko.soy/streamly
on master help sign in
Provide a different impl for yieldLimit case
cecad7eHarendra Kumar
src/Streamly/Streams/Parallel.hs+24−1
compare with master
unifiedsplit
@@ -68,7 +68,30 @@ import qualified Streamly.Streams.StreamK as K
68runOne
68runOne
69 :: MonadIO m
69 :: MonadIO m
70 => State Stream m a -> Stream m a -> Maybe WorkerInfo -> m ()
70 => State Stream m a -> Stream m a -> Maybe WorkerInfo -> m ()
71−runOne st m0 winfo = go m0
71+runOne st m0 winfo =
72+ case getYieldLimit st of
73+ Nothing -> go m0
74+ Just _ -> runOneLimited st m0 winfo
75+
76+ where
77+
78+ go m = do
79+ liftIO $ decrementBufferLimit sv
80+ foldStreamShared st yieldk single stop m
81+
82+ sv = fromJust $ streamVar st
83+
84+ stop = liftIO $ do
85+ incrementBufferLimit sv
86+ sendStop sv winfo
87+ sendit a = liftIO $ void $ send sv (ChildYield a)
88+ single a = sendit a >> (liftIO $ sendStop sv winfo)
89+ yieldk a r = sendit a >> go r
90+
91+runOneLimited
92+ :: MonadIO m
93+ => State Stream m a -> Stream m a -> Maybe WorkerInfo -> m ()
94+runOneLimited st m0 winfo = go m0
72
95
73 where
96 where
74
97