sink to run with the await oneshot rule

oneShot.hs

{-#LANGUAGE BangPatterns #-}
import Pipes
import qualified Pipes.Prelude as P
import Control.Monad.Trans.State.Strict
import GHC.Magic
import qualified Pipes.Internal as I
import Data.IORef
import Control.Monad

emit :: Int -> Int -> IORef Int -> Producer Int IO ()
emit !large !m ref = do 
  n <- liftIO (readIORef ref)
  when (n < large) $ yield n >> emit large (m+1) ref

receive :: IORef Int -> Consumer Int IO ()
receive ref = do
  n <- await
  liftIO $ writeIORef ref $! (n+1::Int)
  receive ref 

main = do 
  ref <- newIORef 0
  let b = receive ref
      large = 10000000 :: Int
      larger = 2*large
  runEffect $ emit large 0 ref >-> b
  readIORef ref >>= print
  runEffect $ emit larger 0 ref >-> b
  readIORef ref >>= print