Bug-fix: Enabled storing multiple peers in a swarm.

1 files changed, 4 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/DHT/ContactInfo.hs b/src/Network/BitTorrent/DHT/ContactInfo.hs
index b7d7f2a2..9c336f93 100644
--- a/src/Network/BitTorrent/DHT/ContactInfo.hs
+++ b/src/Network/BitTorrent/DHT/ContactInfo.hs
@@ -174,10 +174,12 @@ swarmSingleton a = SwarmData
     , name = Nothing }
 
 swarmInsert :: SwarmData -> SwarmData -> SwarmData
-swarmInsert old new = SwarmData
-    { peers = L.foldl' (\q (a :-> t) -> PSQ.insertWith (\p _ -> p) a t q) (peers old) (PSQ.toList $ peers new)
+swarmInsert new old = SwarmData
+    { peers = L.foldl' (\q (a :-> t) -> PSQ.insertWith newerTimeStamp a t q) (peers old) (PSQ.toList $ peers new)
     , name = name new <|> name old -- TODO: decodeUtf8' check
     }
+ where
+    newerTimeStamp newtime oldtime = if newtime > oldtime then newtime else oldtime
 
 isSwarmOccupied SwarmData{..} = not $ PSQ.null peers