97 lines
2.4 KiB
Go
97 lines
2.4 KiB
Go
package models
|
||
|
||
import (
|
||
"errors"
|
||
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
type Node struct {
|
||
gorm.Model
|
||
Name string `json:"name" gorm:"not null"`
|
||
DaemonTaskNum uint `json:"daemonTaskNum"`
|
||
Disabled bool `json:"disabled"` // 通信失败时Disabled会被设置为true
|
||
CrontabTaskNum uint `json:"crontabTaskNum"`
|
||
GroupID uint `json:"groupID" gorm:"not null;uniqueIndex:uni_group_addr" `
|
||
CrontabJobAuditNum uint `json:"crontabJobAuditNum"`
|
||
DaemonJobAuditNum uint `json:"daemonJobAuditNum"`
|
||
CrontabJobFailNum uint `json:"crontabJobFailNum"`
|
||
Addr string `json:"addr" gorm:"not null;uniqueIndex:uni_group_addr;size:100"`
|
||
Group Group `json:"group"`
|
||
}
|
||
|
||
func (n *Node) VerifyUserGroup(userID, groupID uint, addr string) bool {
|
||
var user User
|
||
|
||
if groupID == SuperGroup.ID {
|
||
return true
|
||
}
|
||
|
||
if DB().Take(&user, "id=? and group_id=?", userID, groupID).Error != nil {
|
||
return false
|
||
}
|
||
|
||
return n.Exists(groupID, addr)
|
||
}
|
||
|
||
func (n *Node) Delete(groupID uint, addr string) error {
|
||
var ret *gorm.DB
|
||
DB().Take(n, "group_id=? and addr=?", groupID, addr)
|
||
if groupID == SuperGroup.ID {
|
||
// 超级管理员分组采用软删除
|
||
ret = DB().Delete(n, "group_id=? and addr=?", groupID, addr)
|
||
} else {
|
||
ret = DB().Unscoped().Delete(n, "group_id=? and addr=?", groupID, addr)
|
||
}
|
||
|
||
if ret.Error != nil {
|
||
return ret.Error
|
||
}
|
||
|
||
if ret.RowsAffected == 0 {
|
||
return errors.New("Delete failed")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (n *Node) Rename(groupID uint, addr string) error {
|
||
return DB().Model(n).Where("group_id=? and addr=?", groupID, addr).Updates(n).Error
|
||
}
|
||
|
||
// GroupNode 为节点分组，复制groupID=1分组中node至目标分组
|
||
func (n *Node) GroupNode(addr string, targetGroupID uint, targetNodeName, targetGroupName string) error {
|
||
|
||
// 新建分组
|
||
if targetGroupID == 0 {
|
||
group := &Group{
|
||
Name: targetGroupName,
|
||
}
|
||
if err := DB().Save(group).Error; err != nil {
|
||
return err
|
||
}
|
||
targetGroupID = group.ID
|
||
}
|
||
|
||
err := DB().Preload("Group").Where("group_id=? and addr=?", SuperGroup.ID, addr).Take(n).Error
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
if targetNodeName == "" {
|
||
targetNodeName = n.Name
|
||
}
|
||
|
||
return DB().Save(&Node{
|
||
Addr: addr,
|
||
GroupID: targetGroupID,
|
||
Name: targetNodeName,
|
||
}).Error
|
||
}
|
||
|
||
func (n *Node) Exists(groupID uint, addr string) bool {
|
||
if DB().Take(n, "group_id=? and addr=?", groupID, addr).Error != nil {
|
||
return false
|
||
}
|
||
return true
|
||
}
|