orion/purescript-node-stream-pipes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: more yields

Browse Source
This commit is contained in:
orion 2024-05-13 14:42:23 -05:00
parent 9d8b500b8d
commit 820351f800
Signed by: orion
GPG Key ID: 6D4165AE4C928719
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/Pipes.Node.Stream.purs
View File

@ -94,17 +94,21 @@ fromTransform t =
liftEffect $ removeErrorListener liftEffect $ removeErrorListener
fromReadable t fromReadable t
pure $ Done unit pure $ Done unit
yieldFromReadableHalf = do yieldFromReadableHalf =
flip tailRecM unit $ const do
res <- liftEffect (O.read t) res <- liftEffect (O.read t)
case res of case res of
O.ReadJust a -> yield (Just a) *> yieldFromReadableHalf O.ReadJust a -> do
O.ReadWouldBlock -> pure unit yield $ Just a
O.ReadClosed -> yield Nothing *> pure unit pure $ Loop unit
O.ReadWouldBlock -> pure $ Done unit
O.ReadClosed -> yield Nothing $> Done unit
go { error, cancel } = do go { error, cancel } = do
liftAff $ delay $ wrap 0.0 liftAff $ delay $ wrap 0.0
err <- liftEffect $ liftST $ STRef.read error err <- liftEffect $ liftST $ STRef.read error
for_ err throwError for_ err throwError
yieldFromReadableHalf
ma <- await ma <- await
case ma of case ma of
Nothing -> cleanup cancel Nothing -> cleanup cancel