luciano/amazon-kinesis-client-go
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #56 from Clever/INFRANG-3918

Browse source 
clear count stats after logging them
This commit is contained in:
Gavi Hirsch 2020-09-11 15:02:06 -07:00 committed by GitHub
commit 611f0c0a60
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
batchconsumer/stats/stats.go
View file

@ -30,16 +30,17 @@ type datum struct {
var queue = make(chan datum, 1000)
func init() {
data := map[string]int{}
countData := map[string]int{}
gaugeData := map[string]int{}
tick := time.Tick(time.Minute)
go func() {
for {
select {
case d := <-queue:
if d.category == "counter" {
data[d.name] = data[d.name] + d.value
countData[d.name] = countData[d.name] + d.value
} else if d.category == "gauge" {
data[d.name] = d.value
gaugeData[d.name] = d.value
} else {
log.ErrorD("unknown-stat-category", logger.M{"category": d.category})
}
@ -48,10 +49,14 @@ func init() {
for _, k := range DefaultCounters {
tmp[k] = 0
}
for k, v := range data {
for k, v := range countData {
tmp[k] = v
}
for k, v := range gaugeData {
tmp[k] = v
}
log.InfoD("stats", tmp)
countData = map[string]int{}
}
}
}()