cloudreve/models/node.go

package model

import (
	"encoding/json"
	"github.com/jinzhu/gorm"
)

// Node 从机节点信息模型
type Node struct {
	gorm.Model
	Status       NodeStatus // 节点状态
	Name         string     // 节点别名
	Type         ModelType  // 节点状态
	Server       string     // 服务器地址
	SlaveKey     string     `gorm:"type:text"` // 主->从 通信密钥
	MasterKey    string     `gorm:"type:text"` // 从->主 通信密钥
	Aria2Enabled bool       // 是否支持用作离线下载节点
	Aria2Options string     `gorm:"type:text"` // 离线下载配置
	Rank         int        // 负载均衡权重

	// 数据库忽略字段
	Aria2OptionsSerialized Aria2Option `gorm:"-"`
}

// Aria2Option 非公有的Aria2配置属性
type Aria2Option struct {
	// RPC 服务器地址
	Server string `json:"server,omitempty"`
	// RPC 密钥
	Token string `json:"token,omitempty"`
	// 临时下载目录
	TempPath string `json:"temp_path,omitempty"`
	// 附加下载配置
	Options string `json:"options,omitempty"`
	// 下载监控间隔
	Interval int `json:"interval,omitempty"`
	// RPC API 请求超时
	Timeout int `json:"timeout,omitempty"`
}

type NodeStatus int
type ModelType int

const (
	NodeActive NodeStatus = iota
	NodeSuspend
)

const (
	SlaveNodeType ModelType = iota
	MasterNodeType
)

// GetNodeByID 用ID获取节点
func GetNodeByID(ID interface{}) (Node, error) {
	var node Node
	result := DB.First(&node, ID)
	return node, result.Error
}

// GetNodesByStatus 根据给定状态获取节点
func GetNodesByStatus(status ...NodeStatus) ([]Node, error) {
	var nodes []Node
	result := DB.Where("status in (?)", status).Find(&nodes)
	return nodes, result.Error
}

// AfterFind 找到节点后的钩子
func (node *Node) AfterFind() (err error) {
	// 解析离线下载设置到 Aria2OptionsSerialized
	if node.Aria2Options != "" {
		err = json.Unmarshal([]byte(node.Aria2Options), &node.Aria2OptionsSerialized)
	}

	return err
}

// BeforeSave Save策略前的钩子
func (node *Node) BeforeSave() (err error) {
	optionsValue, err := json.Marshal(&node.Aria2OptionsSerialized)
	node.Aria2Options = string(optionsValue)
	return err
}

// SetStatus 设置节点启用状态
func (node *Node) SetStatus(status NodeStatus) error {
	node.Status = status
	return DB.Model(node).Updates(map[string]interface{}{
		"status": status,
	}).Error
}