-
Notifications
You must be signed in to change notification settings - Fork 443
/
tracked-set.ts
79 lines (65 loc) · 1.55 KB
/
tracked-set.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
import { PeerSet } from './set.js'
import type { Metric, Metrics, PeerId } from '@libp2p/interface'
export interface TrackedPeerSetInit {
name: string
metrics: Metrics
}
class TrackedPeerSet extends PeerSet {
private readonly metric: Metric
constructor (init: TrackedPeerSetInit) {
super()
const { name, metrics } = init
this.metric = metrics.registerMetric(name)
this.updateComponentMetric()
}
add (peer: PeerId): void {
super.add(peer)
this.updateComponentMetric()
}
delete (peer: PeerId): void {
super.delete(peer)
this.updateComponentMetric()
}
clear (): void {
super.clear()
this.updateComponentMetric()
}
private updateComponentMetric (): void {
this.metric.update(this.size)
}
}
export interface CreateTrackedPeerSetInit {
/**
* The metric name to use
*/
name: string
/**
* A metrics implementation
*/
metrics?: Metrics
}
/**
* Creates a PeerSet that reports it's size to the libp2p Metrics service
*
* @example Tracked peer sets
*
* * ```Typescript
* import { trackedPeerSet } from '@libp2p/peer-collections'
* import { createLibp2p } from 'libp2p'
*
* const libp2p = await createLibp2p()
*
* const list = trackedPeerSet({ name: 'my_metric_name', metrics: libp2p.metrics })
* map.add(peerId)
* ```
*/
export function trackedPeerSet (config: CreateTrackedPeerSetInit): PeerSet {
const { name, metrics } = config
let map: PeerSet
if (metrics != null) {
map = new TrackedPeerSet({ name, metrics })
} else {
map = new PeerSet()
}
return map
}