move sendStopToProducer inside fromProducer

Author: harendra-kumar

src/Streamly/Internal/Data/Stream/Parallel.hs

@@ -335,17 +335,7 @@ newFoldSVar stt f = do
     -- Add the producer thread-id to the SVar.
     liftIO myThreadId >>= modifyThread sv
 
-    -- XXX move this in fromProducer itself.
-    --
-    -- A wrapper to send a Stop event back to the producer when the fold
-    -- receives a stop from the producer.
-    let pull m = mkStream $ \st yld _ stp -> do
-            let stop = sendStopToProducer sv >> stp
-                single a = yld a (K.nilM (sendStopToProducer sv))
-                yieldk a r = yld a (pull r)
-             in foldStreamShared st yieldk single stop m
-
-    void $ doFork (void $ f $ fromStream $ pull $ fromProducer sv)
+    void $ doFork (void $ f $ fromStream $ fromProducer sv)
                   (svarMrun sv)
                   (handleFoldException sv)
     return sv

src/Streamly/Internal/Data/Stream/SVar.hs

@@ -133,6 +133,7 @@ fromProducer sv = mkStream $ \st yld sng stp -> do
             t <- liftIO $ getTime Monotonic
             liftIO $ writeIORef (svarStopTime (svarStats sv)) (Just t)
             liftIO $ printSVar sv "SVar Done"
+        sendStopToProducer sv
         stp
 
     {-# INLINE processEvents #-}