1 files changed, 8 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/DHT/Query.hs b/src/Network/BitTorrent/DHT/Query.hs
index c5fcccb4..7f20ad6d 100644
--- a/src/Network/BitTorrent/DHT/Query.hs
+++ b/src/Network/BitTorrent/DHT/Query.hs
@@ -41,6 +41,7 @@ module Network.BitTorrent.DHT.Query
        , publish
 
        -- ** Routing table
+       , restoreTable
        , insertNode
        , refreshNodes
 
@@ -266,3 +267,10 @@ queryNode addr q = do
 q <@> addr = snd <$> queryNode addr q
 {-# INLINE (<@>) #-}
 
+restoreTable :: Address ip => Table ip -> DHT ip ()
+restoreTable tbl = do
+  tblvar <- asks routingTable
+  tbl0 <- liftIO $ takeMVar tblvar
+  mb <- routing $ merge tbl tbl0
+  maybe (return ()) (liftIO . putMVar tblvar) mb
+