ilyakooo0/streamly
1
0
Fork 0
mirror of https://github.com/ilyakooo0/streamly.git synced 2024-09-11 08:25:40 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix O(n) space issue in evalStateT/runStateT

Browse Source 
Added strictness annotation. Space consumption now is O(1).
This commit is contained in:
Harendra Kumar 2021-06-02 23:04:32 +05:30
parent 28e17ee82e
commit 8072636a12
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/Streamly/Internal/Data/Stream/StreamD/Lift.hs
View File

@ -92,7 +92,7 @@ evalStateT initial (Stream step state) = Stream step' (state, initial)
{-# INLINE_LATE step' #-}
step' gst (st, action) = do
sv <- action
(r, sv') <- State.runStateT (step (adaptState gst) st) sv
(r, !sv') <- State.runStateT (step (adaptState gst) st) sv
return $ case r of
Yield x s -> Yield x (s, return sv')
Skip s -> Skip (s, return sv')
@ -105,7 +105,7 @@ runStateT initial (Stream step state) = Stream step' (state, initial)
{-# INLINE_LATE step' #-}
step' gst (st, action) = do
sv <- action
(r, sv') <- State.runStateT (step (adaptState gst) st) sv
(r, !sv') <- State.runStateT (step (adaptState gst) st) sv
return $ case r of
Yield x s -> Yield (sv', x) (s, return sv')
Skip s -> Skip (s, return sv')