Max known nodes configurable

Bitmessage · Jun 24, 2017 · b9d60f8 · b9d60f8
1 parent 26eb54a
commit b9d60f8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/bmconfigparser.py b/src/bmconfigparser.py
@@ -23,6 +23,9 @@
         "storage": "sqlite",
         "acceptmismatch": False,
     },
+    "knownnodes": {
+        "maxnodes": 20000,
+    },
     "zlib": {
         'maxsize': 1048576
     }

diff --git a/src/knownnodes.py b/src/knownnodes.py
@@ -1,12 +1,12 @@
 import pickle
 import threading
 
+from bmconfigparser import BMConfigParser
 import state
 
 knownNodesLock = threading.Lock()
 knownNodes = {}
 
-knownNodesMax = 20000
 knownNodesTrimAmount = 2000
 
 def saveKnownNodes(dirName = None):
@@ -17,7 +17,7 @@ def saveKnownNodes(dirName = None):
             pickle.dump(knownNodes, output)
 
 def trimKnownNodes(recAddrStream = 1):
-    if len(knownNodes[recAddrStream]) < knownNodesMax:
+    if len(knownNodes[recAddrStream]) < BMConfigParser().get("knownnodes", "maxnodes"):
         return
     with knownNodesLock:
         oldestList = sorted(knownNodes[recAddrStream], key=knownNodes[recAddrStream].get)[:knownNodesTrimAmount]