TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-03 15:00:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

阶段性提交

Browse Source
This commit is contained in:
GoEdgeLab
2020-07-30 22:41:49 +08:00
parent ed8be13072
commit f3156dd6b5
14 changed files with 1520 additions and 147 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
internal/db/models/node_dao.go
View File

@@ -1,6 +1,7 @@
package models
import (
"errors"
_ "github.com/go-sql-driver/mysql"
"github.com/iwind/TeaGo/Tea"
"github.com/iwind/TeaGo/dbs"
@@ -37,15 +38,16 @@ func (this *NodeDAO) EnableNode(id uint32) (rowsAffected int64, err error) {
}
// 禁用条目
func (this *NodeDAO) DisableNode(id uint32) (rowsAffected int64, err error) {
return this.Query().
func (this *NodeDAO) DisableNode(id int64) (err error) {
_, err = this.Query().
Pk(id).
Set("state", NodeStateDisabled).
Update()
return err
}
// 查找启用中的条目
func (this *NodeDAO) FindEnabledNode(id uint32) (*Node, error) {
func (this *NodeDAO) FindEnabledNode(id int64) (*Node, error) {
result, err := this.Query().
Pk(id).
Attr("state", NodeStateEnabled).
@@ -66,7 +68,7 @@ func (this *NodeDAO) FindNodeName(id uint32) (string, error) {
}
// 创建节点
func (this *NodeDAO) CreateNode(name string, clusterId int) (nodeId int, err error) {
func (this *NodeDAO) CreateNode(name string, clusterId int64) (nodeId int64, err error) {
op := NewNodeOperator()
op.Name = name
op.NodeId = rands.HexString(32)
@@ -79,11 +81,14 @@ func (this *NodeDAO) CreateNode(name string, clusterId int) (nodeId int, err err
return 0, err
}
return types.Int(op.Id), nil
return types.Int64(op.Id), nil
}
// 修改节点
func (this *NodeDAO) UpdateNode(nodeId int, name string, clusterId int) error {
func (this *NodeDAO) UpdateNode(nodeId int64, name string, clusterId int64) error {
if nodeId <= 0 {
return errors.New("invalid nodeId")
}
op := NewNodeOperator()
op.Id = nodeId
op.Name = name