@@ -207,16 +207,14 @@ take n m = fromStream $ go n (toStream m)
208 else (runStream m1) ctx stp yield
208 else (runStream m1) ctx stp yield
210+-- XXX This is not as efficient as it could be. We need a short circuiting at
211+-- a lower level. Compare with simple-conduit, filtering there cuts down time
212+-- due to short circuting whereas the time spent remains the same here.
210-- | Include only those elements that pass a predicate.
214-- | Include only those elements that pass a predicate.
215+{-# INLINE filter #-}
211−filter :: Streaming t => (a -> Bool) -> t m a -> t m a
216+filter :: (Streaming t, Monad (t m)) => (a -> Bool) -> t m a -> t m a
212−filter p m = fromStream $ go (toStream m)
217+filter p m = m >>= \x -> if p x then return x else nil
214− go m1 = Stream $ \ctx stp yld -> do
215− let yield a Nothing | p a = yld a Nothing
217− yield a (Just x) | p a = yld a (Just (go x))
218− | otherwise = (runStream (go x)) ctx stp yield
219− in (runStream m1) ctx stp yield
221-- | End the stream as soon as the predicate fails on an element.
219-- | End the stream as soon as the predicate fails on an element.
222takeWhile :: Streaming t => (a -> Bool) -> t m a -> t m a
220takeWhile :: Streaming t => (a -> Bool) -> t m a -> t m a