LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

keep same node id between runs

Browse source
This commit is contained in:
Victor Shyba 2021-12-06 11:07:55 -03:00
parent 7ed5fe8f66
commit 371df6e6c2
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
scripts/dht_node.py
View file

@ -2,6 +2,7 @@ import asyncio
import argparse import argparse
import logging import logging
import csv import csv
import os.path
from io import StringIO from io import StringIO
from typing import Optional from typing import Optional
from aiohttp import web from aiohttp import web
@ -96,6 +97,16 @@ class SimpleMetrics:
async def main(host: str, port: int, db_file_path: str, bootstrap_node: Optional[str], prometheus_port: int): async def main(host: str, port: int, db_file_path: str, bootstrap_node: Optional[str], prometheus_port: int):
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
conf = Config() conf = Config()
if not db_file_path.startswith(':memory:'):
node_id_file_path = db_file_path + 'node_id'
if os.path.exists(node_id_file_path):
with open(node_id_file_path, 'rb') as node_id_file:
node_id = node_id_file.read()
else:
with open(node_id_file_path, 'wb') as node_id_file:
node_id = generate_id()
node_id_file.write(node_id)
storage = SQLiteStorage(conf, db_file_path, loop, loop.time) storage = SQLiteStorage(conf, db_file_path, loop, loop.time)
if bootstrap_node: if bootstrap_node:
nodes = bootstrap_node.split(':') nodes = bootstrap_node.split(':')
@ -104,13 +115,14 @@ async def main(host: str, port: int, db_file_path: str, bootstrap_node: Optional
nodes = conf.known_dht_nodes nodes = conf.known_dht_nodes
await storage.open() await storage.open()
node = Node( node = Node(
loop, PeerManager(loop), generate_id(), port, port, 3333, None, loop, PeerManager(loop), node_id, port, port, 3333, None,
storage=storage storage=storage
) )
if prometheus_port > 0: if prometheus_port > 0:
metrics = SimpleMetrics(prometheus_port, node) metrics = SimpleMetrics(prometheus_port, node)
await metrics.start() await metrics.start()
node.start(host, nodes) node.start(host, nodes)
log.info("Peer with id %s started", node_id.hex())
while True: while True:
await asyncio.sleep(10) await asyncio.sleep(10)
PEERS.labels('main').set(len(node.protocol.routing_table.get_peers())) PEERS.labels('main').set(len(node.protocol.routing_table.get_peers()))