mirror of
https://github.com/nezhahq/nezha.git
synced 2025-01-22 12:48:14 -05:00
132 lines
3.3 KiB
Go
132 lines
3.3 KiB
Go
package model
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/robfig/cron/v3"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/nezhahq/nezha/pkg/utils"
|
|
pb "github.com/nezhahq/nezha/proto"
|
|
)
|
|
|
|
const (
|
|
_ = iota
|
|
TaskTypeHTTPGet
|
|
TaskTypeICMPPing
|
|
TaskTypeTCPPing
|
|
TaskTypeCommand
|
|
TaskTypeTerminal
|
|
TaskTypeUpgrade
|
|
TaskTypeKeepalive
|
|
TaskTypeTerminalGRPC
|
|
TaskTypeNAT
|
|
TaskTypeReportHostInfo
|
|
TaskTypeFM
|
|
)
|
|
|
|
type TerminalTask struct {
|
|
StreamID string
|
|
}
|
|
|
|
type TaskNAT struct {
|
|
StreamID string
|
|
Host string
|
|
}
|
|
|
|
type TaskFM struct {
|
|
StreamID string
|
|
}
|
|
|
|
const (
|
|
ServiceCoverAll = iota
|
|
ServiceCoverIgnoreAll
|
|
)
|
|
|
|
type Service struct {
|
|
Common
|
|
Name string `json:"name"`
|
|
Type uint8 `json:"type"`
|
|
Target string `json:"target"`
|
|
SkipServersRaw string `json:"-"`
|
|
Duration uint64 `json:"duration"`
|
|
Notify bool `json:"notify,omitempty"`
|
|
NotificationGroupID uint64 `json:"notification_group_id"` // 当前服务监控所属的通知组 ID
|
|
Cover uint8 `json:"cover"`
|
|
|
|
EnableTriggerTask bool `gorm:"default: false" json:"enable_trigger_task,omitempty"`
|
|
EnableShowInService bool `gorm:"default: false" json:"enable_show_in_service,omitempty"`
|
|
FailTriggerTasksRaw string `gorm:"default:'[]'" json:"-"`
|
|
RecoverTriggerTasksRaw string `gorm:"default:'[]'" json:"-"`
|
|
|
|
FailTriggerTasks []uint64 `gorm:"-" json:"fail_trigger_tasks"` // 失败时执行的触发任务id
|
|
RecoverTriggerTasks []uint64 `gorm:"-" json:"recover_trigger_tasks"` // 恢复时执行的触发任务id
|
|
|
|
MinLatency float32 `json:"min_latency"`
|
|
MaxLatency float32 `json:"max_latency"`
|
|
LatencyNotify bool `json:"latency_notify,omitempty"`
|
|
|
|
SkipServers map[uint64]bool `gorm:"-" json:"skip_servers"`
|
|
CronJobID cron.EntryID `gorm:"-" json:"-"`
|
|
}
|
|
|
|
func (m *Service) PB() *pb.Task {
|
|
return &pb.Task{
|
|
Id: m.ID,
|
|
Type: uint64(m.Type),
|
|
Data: m.Target,
|
|
}
|
|
}
|
|
|
|
// CronSpec 返回服务监控请求间隔对应的 cron 表达式
|
|
func (m *Service) CronSpec() string {
|
|
if m.Duration == 0 {
|
|
// 默认间隔 30 秒
|
|
m.Duration = 30
|
|
}
|
|
return fmt.Sprintf("@every %ds", m.Duration)
|
|
}
|
|
|
|
func (m *Service) BeforeSave(tx *gorm.DB) error {
|
|
if data, err := utils.Json.Marshal(m.SkipServers); err != nil {
|
|
return err
|
|
} else {
|
|
m.SkipServersRaw = string(data)
|
|
}
|
|
if data, err := utils.Json.Marshal(m.FailTriggerTasks); err != nil {
|
|
return err
|
|
} else {
|
|
m.FailTriggerTasksRaw = string(data)
|
|
}
|
|
if data, err := utils.Json.Marshal(m.RecoverTriggerTasks); err != nil {
|
|
return err
|
|
} else {
|
|
m.RecoverTriggerTasksRaw = string(data)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *Service) AfterFind(tx *gorm.DB) error {
|
|
m.SkipServers = make(map[uint64]bool)
|
|
if err := utils.Json.Unmarshal([]byte(m.SkipServersRaw), &m.SkipServers); err != nil {
|
|
log.Println("NEZHA>> Service.AfterFind:", err)
|
|
return nil
|
|
}
|
|
|
|
// 加载触发任务列表
|
|
if err := utils.Json.Unmarshal([]byte(m.FailTriggerTasksRaw), &m.FailTriggerTasks); err != nil {
|
|
return err
|
|
}
|
|
if err := utils.Json.Unmarshal([]byte(m.RecoverTriggerTasksRaw), &m.RecoverTriggerTasks); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// IsServiceSentinelNeeded 判断该任务类型是否需要进行服务监控 需要则返回true
|
|
func IsServiceSentinelNeeded(t uint64) bool {
|
|
return t != TaskTypeCommand && t != TaskTypeTerminalGRPC && t != TaskTypeUpgrade && t != TaskTypeKeepalive
|
|
}
|