extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2024-11-07 16:54:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

metrics aggregation in frontend component (#74)

Browse Source
This commit is contained in:
Michael Quigley 2022-10-10 17:11:18 -04:00
parent 87122024da
commit ce941012b8
No known key found for this signature in database
GPG Key ID: 9B60314A9DD20A62
1 changed files with 25 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
endpoints/frontend/metrics.go
View File

@ -1,7 +1,8 @@
package frontend
import (
"github.com/sirupsen/logrus"
"fmt"
"github.com/openziti-test-kitchen/zrok/util"
"time"
)
@ -33,7 +34,29 @@ func (ma *metricsAgent) run() {
for {
select {
case update := <-ma.updates:
logrus.Infof("update: [%v] read: %d, written: %d", update.id, update.bytesRead, update.bytesWritten)
if sm, found := ma.metrics[update.id]; found {
sm.bytesRead += update.bytesRead
sm.bytesWritten += update.bytesWritten
sm.lastUpdate = time.Now()
ma.metrics[update.id] = sm
} else {
sm := sessionMetrics{
bytesRead: update.bytesRead,
bytesWritten: update.bytesWritten,
lastUpdate: time.Now(),
}
ma.metrics[update.id] = sm
}
case <-time.After(5 * time.Second):
now := time.Now()
out := "metrics = {\n"
for k, v := range ma.metrics {
age := now.Sub(v.lastUpdate)
out += fmt.Sprintf("\t[%v]: %s/%s (%s)\n", k, util.BytesToSize(v.bytesRead), util.BytesToSize(v.bytesWritten), age.String())
}
out += "}\n"
//fmt.Println(out)
}
}
}