forked from pachyderm/pachyderm
/
metrics.go
60 lines (49 loc) · 1.25 KB
/
metrics.go
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
package metrics
import (
"sync/atomic"
"time"
"github.com/pachyderm/pachyderm/src/client/pkg/uuid"
"github.com/pachyderm/pachyderm/src/client/version"
"go.pedge.io/lion/proto"
kube "k8s.io/kubernetes/pkg/client/unversioned"
)
var metrics = &Metrics{}
var modified int64
func AddRepos(num int64) {
atomic.AddInt64(&metrics.Repos, num)
atomic.SwapInt64(&modified, 1)
}
func AddCommits(num int64) {
atomic.AddInt64(&metrics.Commits, num)
atomic.SwapInt64(&modified, 1)
}
func AddFiles(num int64) {
atomic.AddInt64(&metrics.Files, num)
atomic.SwapInt64(&modified, 1)
}
func AddBytes(num int64) {
atomic.AddInt64(&metrics.Bytes, num)
atomic.SwapInt64(&modified, 1)
}
func AddJobs(num int64) {
atomic.AddInt64(&metrics.Jobs, num)
atomic.SwapInt64(&modified, 1)
}
func AddPipelines(num int64) {
atomic.AddInt64(&metrics.Pipelines, num)
atomic.SwapInt64(&modified, 1)
}
func ReportMetrics(clusterID string, kubeClient *kube.Client) {
metrics.ID = clusterID
metrics.PodID = uuid.NewWithoutDashes()
metrics.Version = version.PrettyPrintVersion(version.Version)
for {
write := atomic.SwapInt64(&modified, 0)
if write == 1 {
externalMetrics(kubeClient, metrics)
protolion.Info(metrics)
reportSegment(metrics)
}
<-time.After(15 * time.Second)
}
}