Use K parameter in DHT session

1 files changed, 6 insertions, 3 deletions

diff --git a/src/Network/BitTorrent/DHT/Session.hs b/src/Network/BitTorrent/DHT/Session.hs
index debe9694..ac285bdc 100644
--- a/src/Network/BitTorrent/DHT/Session.hs
+++ b/src/Network/BitTorrent/DHT/Session.hs
@@ -254,12 +254,15 @@ getNodeId :: DHT ip NodeId
 getNodeId = thisId <$> getTable
 
 getClosest :: Eq ip => NodeId -> DHT ip [NodeInfo ip]
-getClosest nid = kclosest 8 nid <$> getTable
+getClosest nid = do
+  k <- asks (optK . options)
+  kclosest k nid <$> getTable
 
 getClosestHash :: Eq ip => InfoHash -> DHT ip [NodeInfo ip]
-getClosestHash ih = kclosestHash 8 ih <$> getTable
+getClosestHash ih = do
+  k <- asks (optK . options)
+  kclosestHash k ih <$> getTable
 
--- FIXME some nodes can be ommited
 insertNode :: Address ip => NodeInfo ip -> DHT ip ThreadId
 insertNode info = fork $ do
   var <- asks routingTable