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

first attempt at crawling

Browse source
This commit is contained in:
Victor Shyba 2021-11-28 21:58:15 -03:00
parent a4dce8cf9f
commit 9cdcff0e1e
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
scripts/dht_node.py
View file

@ -56,11 +56,20 @@ class SimpleMetrics:
writer.writerow({"blob_hash": blob.hex()}) writer.writerow({"blob_hash": blob.hex()})
return web.Response(text=out.getvalue(), content_type='text/csv') return web.Response(text=out.getvalue(), content_type='text/csv')
async def estimate_peers(self, request: web.Request):
amount = 2000
peers = await self.dht_node.peer_search(self.dht_node.protocol.node_id, count=amount, max_results=amount)
close_ids = [peer for peer in peers if peer.node_id[0] == self.dht_node.protocol.node_id[0]]
print(self.dht_node.protocol.node_id.hex())
print([cid.node_id.hex() for cid in close_ids])
return web.json_response({"total": len(peers), "close": len(close_ids)})
async def start(self): async def start(self):
prom_app = web.Application() prom_app = web.Application()
prom_app.router.add_get('/metrics', self.handle_metrics_get_request) prom_app.router.add_get('/metrics', self.handle_metrics_get_request)
prom_app.router.add_get('/peers.csv', self.handle_peers_csv) prom_app.router.add_get('/peers.csv', self.handle_peers_csv)
prom_app.router.add_get('/blobs.csv', self.handle_blobs_csv) prom_app.router.add_get('/blobs.csv', self.handle_blobs_csv)
prom_app.router.add_get('/estimate', self.estimate_peers)
metrics_runner = web.AppRunner(prom_app) metrics_runner = web.AppRunner(prom_app)
await metrics_runner.setup() await metrics_runner.setup()
prom_site = web.TCPSite(metrics_runner, "0.0.0.0", self.prometheus_port) prom_site = web.TCPSite(metrics_runner, "0.0.0.0", self.prometheus_port)