mirror of
https://github.com/Awuqing/BackupX.git
synced 2026-05-11 18:10:23 +08:00
757b0fa5ed
基础修复： - 新增节点离线检测：每 15s 扫描，超 45s 未心跳的远程节点自动置离线 - 节点删除前检查关联任务，避免孤立备份任务 - BackupTaskRepository 新增 CountByNodeID/ListByNodeID Master 端 Agent 协议： - 新增 AgentCommand 模型与命令队列仓储（pending/dispatched/succeeded/failed/timeout） - 新增 AgentService：任务下发、命令轮询、结果回收、超时扫描 - 新增专用 Agent HTTP API（X-Agent-Token 认证）： /api/agent/heartbeat /api/agent/commands/poll /api/agent/commands/:id/result /api/agent/tasks/:id /api/agent/records/:id - BackupExecutionService 支持 node 路由：task.NodeID 指向远程节点时自动入队派发 Agent CLI（backupx agent 子命令）： - 配置：YAML 文件 / 环境变量 / CLI 参数，优先级 CLI > 文件 > 环境 - 心跳循环 + 命令轮询循环 + 优雅退出 - 本地复用 BackupRunner 与 storage registry 执行备份并直接上传 - 支持 run_task 和 list_dir 两种命令 远程目录浏览： - NodeService 支持通过 Agent RPC 列出远程节点目录（15s 超时） 前端： - NodesPage 添加节点后展示 Agent 启动命令和环境变量配置 文档： - README 中英文重写"多节点集群"章节，含架构图、步骤、限制、CLI 参考
96 lines
2.8 KiB
Go
96 lines
2.8 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"backupx/server/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type NodeRepository interface {
|
|
List(context.Context) ([]model.Node, error)
|
|
FindByID(context.Context, uint) (*model.Node, error)
|
|
FindByToken(context.Context, string) (*model.Node, error)
|
|
FindLocal(context.Context) (*model.Node, error)
|
|
Create(context.Context, *model.Node) error
|
|
Update(context.Context, *model.Node) error
|
|
Delete(context.Context, uint) error
|
|
MarkStaleOffline(ctx context.Context, threshold time.Time) (int64, error)
|
|
}
|
|
|
|
type GormNodeRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewNodeRepository(db *gorm.DB) *GormNodeRepository {
|
|
return &GormNodeRepository{db: db}
|
|
}
|
|
|
|
func (r *GormNodeRepository) List(ctx context.Context) ([]model.Node, error) {
|
|
var items []model.Node
|
|
if err := r.db.WithContext(ctx).Order("is_local desc, updated_at desc").Find(&items).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return items, nil
|
|
}
|
|
|
|
func (r *GormNodeRepository) FindByID(ctx context.Context, id uint) (*model.Node, error) {
|
|
var item model.Node
|
|
if err := r.db.WithContext(ctx).First(&item, id).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
return &item, nil
|
|
}
|
|
|
|
func (r *GormNodeRepository) FindByToken(ctx context.Context, token string) (*model.Node, error) {
|
|
var item model.Node
|
|
if err := r.db.WithContext(ctx).Where("token = ?", token).First(&item).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
return &item, nil
|
|
}
|
|
|
|
func (r *GormNodeRepository) FindLocal(ctx context.Context) (*model.Node, error) {
|
|
var item model.Node
|
|
if err := r.db.WithContext(ctx).Where("is_local = ?", true).First(&item).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
return &item, nil
|
|
}
|
|
|
|
func (r *GormNodeRepository) Create(ctx context.Context, item *model.Node) error {
|
|
return r.db.WithContext(ctx).Create(item).Error
|
|
}
|
|
|
|
func (r *GormNodeRepository) Update(ctx context.Context, item *model.Node) error {
|
|
return r.db.WithContext(ctx).Save(item).Error
|
|
}
|
|
|
|
func (r *GormNodeRepository) Delete(ctx context.Context, id uint) error {
|
|
return r.db.WithContext(ctx).Delete(&model.Node{}, id).Error
|
|
}
|
|
|
|
// MarkStaleOffline 把最近心跳早于 threshold 的在线远程节点标记为离线。
|
|
// 本机节点 (is_local=true) 不受影响，由主程序自己维护 online 状态。
|
|
// 返回受影响行数。
|
|
func (r *GormNodeRepository) MarkStaleOffline(ctx context.Context, threshold time.Time) (int64, error) {
|
|
result := r.db.WithContext(ctx).Model(&model.Node{}).
|
|
Where("is_local = ? AND status = ? AND last_seen < ?", false, model.NodeStatusOnline, threshold).
|
|
Update("status", model.NodeStatusOffline)
|
|
if result.Error != nil {
|
|
return 0, result.Error
|
|
}
|
|
return result.RowsAffected, nil
|
|
}
|