package system

import "time"

type Monitor struct {
	Dur      time.Duration `json:"dur" yaml:"dur"`
	Diff     int64         `json:"diff" yaml:"diff"`
	Count    int           `json:"count" yaml:"count"`
	Cpu      int           `json:"cpu" yaml:"cpu"`
	Mem      int           `json:"mem" yaml:"mem"`
	Disk     int           `json:"disk" yaml:"disk"`
	DiskPath string        `json:"diskPath" yaml:"diskPath"`
}

func (mon *Monitor) RunMonitor(call func(d []DiskPath, m *MemInfo, cpuPercent float64, flag int)) {
	if call == nil {
		return
	}
	if mon.Dur < time.Second {
		mon.Dur = time.Second
	}
	for {
		select {
		case <-time.After(mon.Dur):
			flag := 0
			c := CpuPercent()
			if int(c) >= mon.Cpu {
				flag = 1
			}
			var disk []DiskPath
			if mon.DiskPath == "" { //所有磁盘
				d := DiskAll()
				for i := range d {
					if int(d[i].UsedPercent) >= mon.Disk {
						flag |= 2
						disk = append(disk, d[i])
					}
				}
			} else { //当前运行磁盘
				d := DiskWithPath(mon.DiskPath)
				if int(d.UsedPercent) >= mon.Disk {
					flag |= 2
					disk = append(disk, d)
				}
			}
			m := Mem()
			if int(m.UsedPercent) >= mon.Mem {
				flag |= 4
			}
			if flag > 0 {
				call(disk, &m, c, flag)
			}
		}
	}
}