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