mirror of
https://github.com/Awuqing/BackupX.git
synced 2026-05-27 19:19:35 +08:00
0f30e7bf52
抽取备份/恢复/验证/复制四服务的复制粘贴逻辑到 execution_helpers.go（净减约 250 行）；节点状态改为按 LastSeen 实时推导，消除过期 online 误判；Agent systemd 单元补齐 LimitNOFILE 与单机端一致。go build/test 全绿。
100 lines
3.9 KiB
Go
100 lines
3.9 KiB
Go
package model
|
||
|
||
import (
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
const (
|
||
NodeStatusOnline = "online"
|
||
NodeStatusOffline = "offline"
|
||
)
|
||
|
||
// OfflineGracePeriod 节点心跳超时判定阈值：超过该时长未心跳的远程节点视为离线。
|
||
// Agent 默认 15s 心跳一次，预留 3 次重试空间。
|
||
const OfflineGracePeriod = 45 * time.Second
|
||
|
||
// EffectiveStatus 返回节点的「实时」在线状态。
|
||
//
|
||
// 存储字段 Status 由心跳置 online、由后台离线监控置 offline，二者之间存在最长一个
|
||
// 监控周期的滞后窗口；期间 List/Get/调度器可能读到过期的 "online"，进而把任务下发
|
||
// 给一台刚刚失联的节点。本方法直接以 LastSeen 推导：远程节点若超过 OfflineGracePeriod
|
||
// 未心跳即视为 offline，消除该滞后导致的误判。本机节点恒以存储状态为准（它就是 Master
|
||
// 自身，不依赖心跳）。
|
||
func (n *Node) EffectiveStatus(now time.Time) string {
|
||
if n == nil {
|
||
return NodeStatusOffline
|
||
}
|
||
if n.IsLocal {
|
||
return n.Status
|
||
}
|
||
if now.Sub(n.LastSeen) > OfflineGracePeriod {
|
||
return NodeStatusOffline
|
||
}
|
||
return n.Status
|
||
}
|
||
|
||
// Node represents a managed server node in the cluster.
|
||
// The default "local" node is auto-created for single-machine backward compatibility.
|
||
type Node struct {
|
||
ID uint `gorm:"primaryKey" json:"id"`
|
||
Name string `gorm:"size:128;uniqueIndex;not null" json:"name"`
|
||
Hostname string `gorm:"size:255" json:"hostname"`
|
||
IPAddress string `gorm:"column:ip_address;size:64" json:"ipAddress"`
|
||
Token string `gorm:"size:128;uniqueIndex;not null" json:"-"`
|
||
Status string `gorm:"size:20;not null;default:'offline'" json:"status"`
|
||
IsLocal bool `gorm:"not null;default:false" json:"isLocal"`
|
||
OS string `gorm:"size:64" json:"os"`
|
||
Arch string `gorm:"size:32" json:"arch"`
|
||
AgentVer string `gorm:"column:agent_version;size:32" json:"agentVersion"`
|
||
LastSeen time.Time `gorm:"column:last_seen" json:"lastSeen"`
|
||
PrevToken string `gorm:"size:128;index" json:"-"`
|
||
PrevTokenExpires *time.Time `gorm:"column:prev_token_expires" json:"-"`
|
||
// MaxConcurrent 该节点允许的最大并发任务数（0=不限制，沿用全局 cfg.Backup.MaxConcurrent）。
|
||
// 用于大集群中限制单节点资源占用：例如小内存 Agent 节点可配 1，避免多个大备份同时跑挤爆。
|
||
MaxConcurrent int `gorm:"column:max_concurrent;not null;default:0" json:"maxConcurrent"`
|
||
// BandwidthLimit 该节点上传带宽上限（rclone 可识别格式：10M / 1G / 0=不限）。
|
||
// 对集群感知的上传场景有效（Master 本地与 Agent 运行时均会应用）。
|
||
BandwidthLimit string `gorm:"column:bandwidth_limit;size:32" json:"bandwidthLimit"`
|
||
// Labels 节点标签（CSV，如 "prod,db-host,high-mem"）。
|
||
// 用于任务调度的节点池选择：任务配置 NodePoolTag 时，调度器会从 Labels 包含该 tag 的
|
||
// 在线节点中自动挑选一台执行（按当前运行中任务数升序）。单节点可属多个池。
|
||
Labels string `gorm:"column:labels;size:500" json:"labels"`
|
||
CreatedAt time.Time `json:"createdAt"`
|
||
UpdatedAt time.Time `json:"updatedAt"`
|
||
}
|
||
|
||
// LabelSet 把 CSV Labels 解析为 set，便于做成员判定。
|
||
// 空白与空 token 自动忽略。
|
||
func (n *Node) LabelSet() map[string]struct{} {
|
||
if n == nil {
|
||
return nil
|
||
}
|
||
out := make(map[string]struct{})
|
||
for _, raw := range strings.Split(n.Labels, ",") {
|
||
label := strings.TrimSpace(raw)
|
||
if label != "" {
|
||
out[label] = struct{}{}
|
||
}
|
||
}
|
||
return out
|
||
}
|
||
|
||
// HasLabel 判断节点是否属于指定池。nil/空 tag 返回 false。
|
||
func (n *Node) HasLabel(tag string) bool {
|
||
tag = strings.TrimSpace(tag)
|
||
if n == nil || tag == "" {
|
||
return false
|
||
}
|
||
for _, raw := range strings.Split(n.Labels, ",") {
|
||
if strings.TrimSpace(raw) == tag {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
func (Node) TableName() string {
|
||
return "nodes"
|
||
}
|