T3MP3ST public launch build

2026-07-03 10:57:20 -04:00
1 parent 972247b881
commit 184c7d4e59
@@ -0,0 +1,141 @@
+package model
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/goccy/go-json"
+	"github.com/robfig/cron/v3"
+	"gorm.io/gorm"
+
+	pb "github.com/nezhahq/nezha/proto"
+)
+
+const (
+	_ = iota
+	TaskTypeHTTPGet
+	TaskTypeICMPPing
+	TaskTypeTCPPing
+	TaskTypeCommand
+	TaskTypeTerminal
+	TaskTypeUpgrade
+	TaskTypeKeepalive
+	TaskTypeTerminalGRPC
+	TaskTypeNAT
+	TaskTypeReportHostInfoDeprecated
+	TaskTypeFM
+	TaskTypeReportConfig
+	TaskTypeApplyConfig
+)
+
+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"`
+	DisplayIndex        int    `json:"display_index"` // 展示排序，越大越靠前
+	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 := json.Marshal(m.SkipServers); err != nil {
+		return err
+	} else {
+		m.SkipServersRaw = string(data)
+	}
+	if data, err := json.Marshal(m.FailTriggerTasks); err != nil {
+		return err
+	} else {
+		m.FailTriggerTasksRaw = string(data)
+	}
+	if data, err := 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 := json.Unmarshal([]byte(m.SkipServersRaw), &m.SkipServers); err != nil {
+		log.Println("NEZHA>> Service.AfterFind:", err)
+		return nil
+	}
+
+	// 加载触发任务列表
+	if err := json.Unmarshal([]byte(m.FailTriggerTasksRaw), &m.FailTriggerTasks); err != nil {
+		return err
+	}
+	if err := json.Unmarshal([]byte(m.RecoverTriggerTasksRaw), &m.RecoverTriggerTasks); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// IsServiceSentinelNeeded 判断该任务类型是否需要进行服务监控 需要则返回true
+func IsServiceSentinelNeeded(t uint64) bool {
+	switch t {
+	case TaskTypeCommand, TaskTypeTerminalGRPC, TaskTypeUpgrade,
+		TaskTypeKeepalive, TaskTypeNAT, TaskTypeFM,
+		TaskTypeReportConfig, TaskTypeApplyConfig:
+		return false
+	default:
+		return true
+	}
+}