More strictness.

1 files changed, 7 insertions, 6 deletions

diff --git a/src/Network/BitTorrent/DHT/ContactInfo.hs b/src/Network/BitTorrent/DHT/ContactInfo.hs
index 26bdeae9..f9dae567 100644
--- a/src/Network/BitTorrent/DHT/ContactInfo.hs
+++ b/src/Network/BitTorrent/DHT/ContactInfo.hs
@@ -1,3 +1,4 @@
+{-# LANGUAGE BangPatterns #-}
 module Network.BitTorrent.DHT.ContactInfo
        ( PeerStore
        , Network.BitTorrent.DHT.ContactInfo.lookup
@@ -120,8 +121,8 @@ newtype PeerStore ip = PeerStore (HashMap InfoHash (SwarmData ip))
 type Timestamp = POSIXTime
 
 data SwarmData ip = SwarmData
-    { peers :: PSQ (PeerAddr ip) Timestamp
-    , name :: Maybe ByteString
+    { peers :: !(PSQ (PeerAddr ip) Timestamp)
+    , name :: !(Maybe ByteString)
     }
 
 
@@ -210,16 +211,16 @@ freshPeers ih tm (PeerStore m) = fromMaybe ([],PeerStore m) $ do
                     _         -> peers swarm
         ps = L.map (key . fst) ps0
         m' = HM.insert ih swarm { peers = L.foldl' (\q p -> PSQ.insert p tm q) peers' ps } m
-    return (ps,PeerStore m')
+    return $! (ps,PeerStore m')
 
 incomp :: (x -> Maybe (r,x)) -> x -> Maybe ((r,x),x)
-incomp f x = do
+incomp !f !x = do
    (result,x') <- f x
-   pure $ ( (result,x'), x' )
+   pure $! ( (result,x'), x' )
 
 -- | Used in 'announce_peer' DHT queries.
 insertPeer :: Ord a => InfoHash -> Maybe ByteString -> PeerAddr a -> PeerStore a -> PeerStore a
-insertPeer ih name a (PeerStore m) = PeerStore (HM.insertWith swarmInsert ih a' m)
+insertPeer !ih !name !a !(PeerStore m) = seq a' $ PeerStore (HM.insertWith swarmInsert ih a' m)
  where
     a' = SwarmData { peers = PSQ.singleton a 0
                    , name = name }