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()
}