-
Notifications
You must be signed in to change notification settings - Fork 443
/
tracked-map.ts
65 lines (51 loc) · 1.18 KB
/
tracked-map.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
import type { Metric, Metrics } from '@libp2p/interface'
export interface TrackedMapInit {
name: string
metrics: Metrics
}
class TrackedMap<K, V> extends Map<K, V> {
private readonly metric: Metric
constructor (init: TrackedMapInit) {
super()
const { name, metrics } = init
this.metric = metrics.registerMetric(name)
this.updateComponentMetric()
}
set (key: K, value: V): this {
super.set(key, value)
this.updateComponentMetric()
return this
}
delete (key: K): boolean {
const deleted = super.delete(key)
this.updateComponentMetric()
return deleted
}
clear (): void {
super.clear()
this.updateComponentMetric()
}
private updateComponentMetric (): void {
this.metric.update(this.size)
}
}
export interface CreateTrackedMapInit {
/**
* The metric name to use
*/
name: string
/**
* A metrics implementation
*/
metrics?: Metrics
}
export function trackedMap <K, V> (config: CreateTrackedMapInit): Map<K, V> {
const { name, metrics } = config
let map: Map<K, V>
if (metrics != null) {
map = new TrackedMap<K, V>({ name, metrics })
} else {
map = new Map<K, V>()
}
return map
}