fankes/beszel
2
0
Fork 0
mirror of https://github.com/fankes/beszel.git synced 2025-10-20 02:09:28 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

temperature alerts

Browse Source
This commit is contained in:
Henry Dollman
2024-10-12 14:57:46 -04:00
parent 6cf6661f2e
commit 10d348c052
3 changed files with 40 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
beszel/internal/alerts/alerts.go
View File

@@ -39,7 +39,7 @@ func NewAlertManager(app *pocketbase.PocketBase) *AlertManager {
}
}
func (am *AlertManager) HandleSystemInfoAlerts(systemRecord *models.Record, systemInfo system.Info) {
func (am *AlertManager) HandleSystemAlerts(systemRecord *models.Record, systemInfo system.Info, temperatures map[string]float64) {
alertRecords, err := am.app.Dao().FindRecordsByExpr("alerts",
dbx.NewExp("system={:system}", dbx.Params{"system": systemRecord.GetId()}),
)
@@ -51,19 +51,28 @@ func (am *AlertManager) HandleSystemInfoAlerts(systemRecord *models.Record, syst
for _, alertRecord := range alertRecords {
name := alertRecord.GetString("name")
switch name {
case "CPU", "Memory", "Disk":
if name == "CPU" {
am.handleSlidingValueAlert(systemRecord, alertRecord, name, systemInfo.Cpu)
} else if name == "Memory" {
am.handleSlidingValueAlert(systemRecord, alertRecord, name, systemInfo.MemPct)
} else if name == "Disk" {
am.handleSlidingValueAlert(systemRecord, alertRecord, name, systemInfo.DiskPct)
case "CPU":
am.handleSlidingValueAlert(systemRecord, alertRecord, name, "%", systemInfo.Cpu)
case "Memory":
am.handleSlidingValueAlert(systemRecord, alertRecord, name, "%", systemInfo.MemPct)
case "Disk":
am.handleSlidingValueAlert(systemRecord, alertRecord, name+" usage", "%", systemInfo.DiskPct)
case "Temperature":
if temperatures == nil {
continue
}
highTemp := 0.0
for _, temp := range temperatures {
if temp > highTemp {
highTemp = temp
}
}
am.handleSlidingValueAlert(systemRecord, alertRecord, name, "°C", highTemp)
}
}
}
func (am *AlertManager) handleSlidingValueAlert(systemRecord *models.Record, alertRecord *models.Record, name string, curValue float64) {
func (am *AlertManager) handleSlidingValueAlert(systemRecord *models.Record, alertRecord *models.Record, name, unit string, curValue float64) {
triggered := alertRecord.GetBool("triggered")
threshold := alertRecord.GetFloat("value")
// fmt.Println(name, curValue, "threshold", threshold, "triggered", triggered)
@@ -73,13 +82,13 @@ func (am *AlertManager) handleSlidingValueAlert(systemRecord *models.Record, ale
if !triggered && curValue > threshold {
alertRecord.Set("triggered", true)
systemName = systemRecord.GetString("name")
subject = fmt.Sprintf("%s usage above threshold on %s", name, systemName)
body = fmt.Sprintf("%s usage on %s is %.1f%%.", name, systemName, curValue)
subject = fmt.Sprintf("%s above threshold on %s", name, systemName)
body = fmt.Sprintf("%s on %s is %v%s.", name, systemName, curValue, unit)
} else if triggered && curValue <= threshold {
alertRecord.Set("triggered", false)
systemName = systemRecord.GetString("name")
subject = fmt.Sprintf("%s usage below threshold on %s", name, systemName)
body = fmt.Sprintf("%s usage on %s is below threshold at %.1f%%.", name, systemName, curValue)
subject = fmt.Sprintf("%s below threshold on %s", name, systemName)
body = fmt.Sprintf("%s on %s is below threshold at %v%s.", name, systemName, curValue, unit)
} else {
// fmt.Println(name, "not triggered")
return

4
beszel/internal/hub/hub.go
View File

@@ -306,8 +306,8 @@ func (h *Hub) updateSystem(record *models.Record) {
h.app.Logger().Error("Failed to save record: ", "err", err.Error())
}
}
// system info alerts (todo: temp alerts, extra fs alerts)
h.am.HandleSystemInfoAlerts(record, systemData.Info)
// system info alerts (todo: extra fs alerts)
h.am.HandleSystemAlerts(record, systemData.Info, systemData.Stats.Temperatures)
}
// set system to specified status and save record