d5d4caf9b4
- Updated .pre-commit-config.yaml to change hooks to manual execution for performance. - Added UptimeHost and UptimeNotificationEvent models to support host-level monitoring. - Enhanced UptimeService to batch notifications for multiple service failures on the same host. - Implemented tests for notification batching and uptime host creation during sync. - Improved formatDuration function for better readability of durations.
55 lines
1.9 KiB
Go
55 lines
1.9 KiB
Go
package models
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type UptimeMonitor struct {
|
|
ID string `gorm:"primaryKey" json:"id"`
|
|
ProxyHostID *uint `json:"proxy_host_id"` // Optional link to proxy host
|
|
RemoteServerID *uint `json:"remote_server_id"` // Optional link to remote server
|
|
UptimeHostID *string `json:"uptime_host_id"` // Link to parent host for grouping
|
|
Name string `json:"name"`
|
|
Type string `json:"type"` // http, tcp, ping
|
|
URL string `json:"url"`
|
|
UpstreamHost string `json:"upstream_host"` // The actual backend host/IP (for grouping)
|
|
Interval int `json:"interval"` // seconds
|
|
Enabled bool `json:"enabled"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
|
|
// Current Status (Cached)
|
|
Status string `json:"status"` // up, down, maintenance, pending
|
|
LastCheck time.Time `json:"last_check"`
|
|
Latency int64 `json:"latency"` // ms
|
|
FailureCount int `json:"failure_count"`
|
|
LastStatusChange time.Time `json:"last_status_change"`
|
|
MaxRetries int `json:"max_retries" gorm:"default:3"`
|
|
|
|
// Notification tracking
|
|
LastNotifiedDown time.Time `json:"last_notified_down"` // Prevent duplicate notifications
|
|
NotifiedInBatch bool `json:"notified_in_batch"` // Was this included in a batch notification?
|
|
}
|
|
|
|
type UptimeHeartbeat struct {
|
|
ID uint `gorm:"primaryKey" json:"id"`
|
|
MonitorID string `json:"monitor_id" gorm:"index"`
|
|
Status string `json:"status"` // up, down
|
|
Latency int64 `json:"latency"`
|
|
Message string `json:"message"`
|
|
CreatedAt time.Time `json:"created_at" gorm:"index"`
|
|
}
|
|
|
|
func (m *UptimeMonitor) BeforeCreate(tx *gorm.DB) (err error) {
|
|
if m.ID == "" {
|
|
m.ID = uuid.New().String()
|
|
}
|
|
if m.Status == "" {
|
|
m.Status = "pending"
|
|
}
|
|
return
|
|
}
|