55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
package job
|
||
|
||
import (
|
||
"strconv"
|
||
"time"
|
||
|
||
"x-ui/web/service"
|
||
|
||
"github.com/shirou/gopsutil/v4/cpu"
|
||
)
|
||
|
||
// 连续超阈值告警实现
|
||
type CheckCpuJob struct {
|
||
tgbotService service.Tgbot
|
||
settingService service.SettingService
|
||
overThresholdCount int // 连续超阈值计数器
|
||
lastNotifyTime time.Time // 最近一次告警时间
|
||
}
|
||
|
||
func NewCheckCpuJob() *CheckCpuJob {
|
||
return &CheckCpuJob{}
|
||
}
|
||
|
||
// run 是 Job 接口方法
|
||
func (j *CheckCpuJob) Run() {
|
||
threshold, _ := j.settingService.GetTgCpu()
|
||
notifyInterval := 10 * time.Minute // 两次告警最小间隔，可做成配置项
|
||
|
||
percent, err := cpu.Percent(10*time.Second, false) // 10秒采样
|
||
if err != nil || len(percent) == 0 {
|
||
return
|
||
}
|
||
|
||
now := time.Now()
|
||
if percent[0] > float64(threshold) {
|
||
j.overThresholdCount++
|
||
} else {
|
||
j.overThresholdCount = 0
|
||
}
|
||
|
||
// 连续3次超阈值，且距离上次告警超过告警间隔
|
||
if j.overThresholdCount >= 3 && now.Sub(j.lastNotifyTime) > notifyInterval {
|
||
msg := j.tgbotService.I18nBot(
|
||
"tgbot.messages.cpuThreshold",
|
||
"Percent=="+strconv.FormatFloat(percent[0], 'f', 2, 64),
|
||
"Threshold=="+strconv.Itoa(threshold),
|
||
"SampleInterval==10s",
|
||
"NotifyPolicy==连续3次超阈值",
|
||
)
|
||
j.tgbotService.SendMsgToTgbotAdmins(msg)
|
||
j.lastNotifyTime = now
|
||
j.overThresholdCount = 0
|
||
}
|
||
}
|