summaryrefslogtreecommitdiffstats
path: root/src/go/parser/testdata/metrics.go2
blob: ef1c66b241cbb0bb60a3ad7e4177c7716e108faa (plain)
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
package metrics

import "sync"

type Metric1[T comparable] struct {
	mu sync.Mutex
	m  map[T]int
}

func (m *Metric1[T]) Add(v T) {
	m.mu.Lock()
	defer m.mu.Unlock()
	if m.m == nil {
		m.m = make(map[T]int)
	}
	m[v]++
}

type key2[T1, T2 comparable] struct {
	f1 T1
	f2 T2
}

type Metric2[T1, T2 cmp2] struct {
	mu sync.Mutex
	m  map[key2[T1, T2]]int
}

func (m *Metric2[T1, T2]) Add(v1 T1, v2 T2) {
	m.mu.Lock()
	defer m.mu.Unlock()
	if m.m == nil {
		m.m = make(map[key2[T1, T2]]int)
	}
	m[key[T1, T2]{v1, v2}]++
}

type key3[T1, T2, T3 comparable] struct {
	f1 T1
	f2 T2
	f3 T3
}

type Metric3[T1, T2, T3 comparable] struct {
	mu sync.Mutex
	m  map[key3[T1, T2, T3]]int
}

func (m *Metric3[T1, T2, T3]) Add(v1 T1, v2 T2, v3 T3) {
	m.mu.Lock()
	defer m.mu.Unlock()
	if m.m == nil {
		m.m = make(map[key3]int)
	}
	m[key[T1, T2, T3]{v1, v2, v3}]++
}

// Repeat for the maximum number of permitted arguments.