clustvirt/lib/host/stats.go

46 lines
834 B
Go
Raw Permalink Normal View History

2024-04-21 17:50:26 +00:00
package host
import (
"context"
"libvirt.org/go/libvirtxml"
)
// Stats for hosts with channels to subscribe to
// HostStats works as a handler that can be subscribed to.
// It runs a go routine that sends Stats via a channel
// when subscribed to
type HostStats struct {
h *Host
}
type Stats struct {
Memory NodeMemoryInfo
VMs []*libvirtxml.Domain
StoragePools []*libvirtxml.StoragePool
}
func (hs *HostStats) Subscribe(parent context.Context, pubto <-chan Stats) error {
ctx, cancel := context.WithCancel(parent)
go func() {
defer cancel()
for {
select {
case <-ctx.Done():
break
default:
hs.update()
}
}
}()
return nil
}
// update runs as a goroutine, and updates host stats
func (hs *HostStats) update() {
hs.h.memoryInfo()
hs.h.getStoragePools()
hs.h.getDomainInfo()
}