82 lines
1.2 KiB
Go
82 lines
1.2 KiB
Go
package aggregate
|
|
|
|
import "time"
|
|
|
|
type Window struct {
|
|
start time.Time
|
|
|
|
cpuSum float64
|
|
cpuCount uint64
|
|
|
|
memSum float64
|
|
memCount uint64
|
|
|
|
diskSum float64
|
|
diskCount uint64
|
|
|
|
uptime uint64
|
|
}
|
|
|
|
type Snapshot struct {
|
|
Timestamp time.Time
|
|
|
|
CPUAvg float64
|
|
MemAvg float64
|
|
DiskAvg float64
|
|
|
|
Uptime uint64
|
|
}
|
|
|
|
func NewWindow() *Window {
|
|
return &Window{
|
|
start: time.Now(),
|
|
}
|
|
}
|
|
|
|
func (w *Window) AddCPU(value float64) {
|
|
w.cpuSum += value
|
|
w.cpuCount++
|
|
}
|
|
|
|
func (w *Window) AddMemory(value float64) {
|
|
w.memSum += value
|
|
w.memCount++
|
|
}
|
|
|
|
func (w *Window) SetUptime(value uint64) {
|
|
w.uptime = value
|
|
}
|
|
|
|
func (w *Window) Ready(windowDuration time.Duration) bool {
|
|
return time.Since(w.start) >= windowDuration
|
|
}
|
|
|
|
func (w *Window) Snapshot() Snapshot {
|
|
var cpuAvg float64
|
|
if w.cpuCount > 0 {
|
|
cpuAvg = w.cpuSum / float64(w.cpuCount)
|
|
}
|
|
|
|
var memAvg float64
|
|
if w.memCount > 0 {
|
|
memAvg = w.memSum / float64(w.memCount)
|
|
}
|
|
|
|
var diskAvg float64
|
|
if w.diskCount > 0 {
|
|
diskAvg = w.diskSum / float64(w.diskCount)
|
|
}
|
|
|
|
return Snapshot{
|
|
Timestamp: time.Now(),
|
|
CPUAvg: cpuAvg,
|
|
MemAvg: memAvg,
|
|
DiskAvg: diskAvg,
|
|
Uptime: w.uptime,
|
|
}
|
|
}
|
|
|
|
func (w *Window) AddDisk(value float64) {
|
|
w.diskSum += value
|
|
w.diskCount++
|
|
} |