+ Add session exception.

1 files changed, 11 insertions, 2 deletions

diff --git a/exsamples/Main.hs b/exsamples/Main.hs
index a5b865b0..88796642 100644
--- a/exsamples/Main.hs
+++ b/exsamples/Main.hs
@@ -6,7 +6,7 @@ import Data.Bitfield
 import Network.BitTorrent
 import System.Environment
 import Control.Monad.Reader
-
+import Data.IORef
 
 main :: IO ()
 main = do
@@ -16,6 +16,8 @@ main = do
   client  <- newClient []
   swarm   <- newLeacher  client torrent
 
+  ref <- newIORef 0
+
   discover swarm $ do
     addr <- asks connectedPeerAddr
     liftIO $ print $ "connected to" ++ show addr
@@ -27,7 +29,14 @@ main = do
           | Just m <- findMin bf -> yieldEvent (Want (BlockIx m 0 10))
           |     otherwise        -> return ()
         Want     bix -> liftIO $ print bix
-        Fragment blk -> liftIO $ print (ppBlock blk)
+        Fragment blk -> do
+
+          liftIO $ do
+            readIORef ref >>= print
+            modifyIORef ref succ
+            print (ppBlock blk)
+
+          yieldEvent (Want (BlockIx 0 0 (16 * 1024)))
 
 
   print "Bye-bye! =_="
\ No newline at end of file