codex-storage/nim-libp2p
1
0
Fork 0
mirror of https://github.com/codex-storage/nim-libp2p.git synced 2025-01-28 03:35:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: add basic metrics for rendezvous (#905)

Browse Source
This commit is contained in:
Vaclav Pavlin 2023-06-07 15:45:06 +02:00 committed by GitHub
parent 3640b4dd89
commit 3725f6a95b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
libp2p/protocols/rendezvous.nim
View File

@ -10,6 +10,7 @@
{.push raises: [].} {.push raises: [].}
import tables, sequtils, sugar, sets, options import tables, sequtils, sugar, sets, options
import metrics except collect
import chronos, import chronos,
chronicles, chronicles,
bearssl/rand, bearssl/rand,
@ -27,6 +28,11 @@ export chronicles
logScope: logScope:
topics = "libp2p discovery rendezvous" topics = "libp2p discovery rendezvous"
declareCounter(libp2p_rendezvous_register, "number of advertise requests")
declareCounter(libp2p_rendezvous_discover, "number of discovery requests")
declareGauge(libp2p_rendezvous_registered, "number of registered peers")
declareGauge(libp2p_rendezvous_namespaces, "number of registered namespaces")
const const
RendezVousCodec* = "/rendezvous/1.0.0" RendezVousCodec* = "/rendezvous/1.0.0"
MinimumDuration* = 2.hours MinimumDuration* = 2.hours
@ -375,6 +381,7 @@ proc save(rdv: RendezVous,
proc register(rdv: RendezVous, conn: Connection, r: Register): Future[void] = proc register(rdv: RendezVous, conn: Connection, r: Register): Future[void] =
trace "Received Register", peerId = conn.peerId, ns = r.ns trace "Received Register", peerId = conn.peerId, ns = r.ns
libp2p_rendezvous_register.inc()
if r.ns.len notin 1..255: if r.ns.len notin 1..255:
return conn.sendRegisterResponseError(InvalidNamespace) return conn.sendRegisterResponseError(InvalidNamespace)
let ttl = r.ttl.get(MinimumTTL) let ttl = r.ttl.get(MinimumTTL)
@ -386,6 +393,8 @@ proc register(rdv: RendezVous, conn: Connection, r: Register): Future[void] =
if rdv.countRegister(conn.peerId) >= RegistrationLimitPerPeer: if rdv.countRegister(conn.peerId) >= RegistrationLimitPerPeer:
return conn.sendRegisterResponseError(NotAuthorized, "Registration limit reached") return conn.sendRegisterResponseError(NotAuthorized, "Registration limit reached")
rdv.save(r.ns, conn.peerId, r) rdv.save(r.ns, conn.peerId, r)
libp2p_rendezvous_registered.inc()
libp2p_rendezvous_namespaces.set(int64(rdv.namespaces.len))
conn.sendRegisterResponse(ttl) conn.sendRegisterResponse(ttl)
proc unregister(rdv: RendezVous, conn: Connection, u: Unregister) = proc unregister(rdv: RendezVous, conn: Connection, u: Unregister) =
@ -395,11 +404,13 @@ proc unregister(rdv: RendezVous, conn: Connection, u: Unregister) =
for index in rdv.namespaces[nsSalted]: for index in rdv.namespaces[nsSalted]:
if rdv.registered[index].peerId == conn.peerId: if rdv.registered[index].peerId == conn.peerId:
rdv.registered[index].expiration = rdv.defaultDT rdv.registered[index].expiration = rdv.defaultDT
libp2p_rendezvous_registered.dec()
except KeyError: except KeyError:
return return
proc discover(rdv: RendezVous, conn: Connection, d: Discover) {.async.} = proc discover(rdv: RendezVous, conn: Connection, d: Discover) {.async.} =
trace "Received Discover", peerId = conn.peerId, ns = d.ns trace "Received Discover", peerId = conn.peerId, ns = d.ns
libp2p_rendezvous_discover.inc()
if d.ns.len notin 0..255: if d.ns.len notin 0..255:
await conn.sendDiscoverResponseError(InvalidNamespace) await conn.sendDiscoverResponseError(InvalidNamespace)
return return
@ -654,9 +665,13 @@ proc new*(T: typedesc[RendezVous],
proc deletesRegister(rdv: RendezVous) {.async.} = proc deletesRegister(rdv: RendezVous) {.async.} =
heartbeat "Register timeout", 1.minutes: heartbeat "Register timeout", 1.minutes:
let n = Moment.now() let n = Moment.now()
var total = 0
rdv.registered.flushIfIt(it.expiration < n) rdv.registered.flushIfIt(it.expiration < n)
for data in rdv.namespaces.mvalues(): for data in rdv.namespaces.mvalues():
data.keepItIf(it >= rdv.registered.offset) data.keepItIf(it >= rdv.registered.offset)
total += data.len
libp2p_rendezvous_registered.set(int64(total))
libp2p_rendezvous_namespaces.set(int64(rdv.namespaces.len))
method start*(rdv: RendezVous) {.async.} = method start*(rdv: RendezVous) {.async.} =
if not rdv.registerDeletionLoop.isNil: if not rdv.registerDeletionLoop.isNil: