forked from pachyderm/pachyderm
/
metrics.go
68 lines (57 loc) · 1.71 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
61
62
63
64
65
66
67
68
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
// AddRepos atomically adds a number of repos to metrics.
func AddRepos(num int64) {
atomic.AddInt64(&metrics.Repos, num)
atomic.SwapInt64(&modified, 1)
}
// AddCommits atomically adds a number of commits to metrics.
func AddCommits(num int64) {
atomic.AddInt64(&metrics.Commits, num)
atomic.SwapInt64(&modified, 1)
}
// AddFiles atomically adds a number of files to metrics.
func AddFiles(num int64) {
atomic.AddInt64(&metrics.Files, num)
atomic.SwapInt64(&modified, 1)
}
// AddBytes atomically adds a number of bytes to metrics.
func AddBytes(num int64) {
atomic.AddInt64(&metrics.Bytes, num)
atomic.SwapInt64(&modified, 1)
}
// AddJobs atomically adds a number of jobs to metrics.
func AddJobs(num int64) {
atomic.AddInt64(&metrics.Jobs, num)
atomic.SwapInt64(&modified, 1)
}
// AddPipelines atomically adds a number of pipelines to metrics.
func AddPipelines(num int64) {
atomic.AddInt64(&metrics.Pipelines, num)
atomic.SwapInt64(&modified, 1)
}
// ReportMetrics blocks and reports metrics, if modified, to the
// given kubernetes client every 15 seconds.
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)
}
}