阶段性提交

internal/db/models/node_login_dao.go

@@ -1,9 +1,11 @@
 package models
 
 import (
+	"errors"
 	_ "github.com/go-sql-driver/mysql"
 	"github.com/iwind/TeaGo/Tea"
 	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/types"
 )
 
 const (
@@ -62,3 +64,53 @@ func (this *NodeLoginDAO) FindNodeLoginName(id uint32) (string, error) {
 		FindCol("")
 	return name.(string), err
 }
+
+// 创建认证
+func (this *NodeLoginDAO) CreateNodeLogin(nodeId int64, name string, loginType string, paramsJSON []byte) (loginId int64, err error) {
+	login := NewNodeLoginOperator()
+	login.NodeId = nodeId
+	login.Name = name
+	login.Type = loginType
+	login.Params = string(paramsJSON)
+	login.State = NodeLoginStateEnabled
+	_, err = this.Save(login)
+	return types.Int64(login.Id), err
+}
+
+// 修改认证
+func (this *NodeLoginDAO) UpdateNodeLogin(loginId int64, name string, loginType string, paramsJSON []byte) error {
+	if loginId <= 0 {
+		return errors.New("invalid loginId")
+	}
+	login := NewNodeLoginOperator()
+	login.Id = loginId
+	login.Name = name
+	login.Type = loginType
+	login.Params = string(paramsJSON)
+	_, err := this.Save(login)
+	return err
+}
+
+// 查找认证
+func (this *NodeLoginDAO) FindEnabledNodeLoginWithNodeId(nodeId int64) (*NodeLogin, error) {
+	one, err := this.Query().
+		Attr("nodeId", nodeId).
+		State(NodeLoginStateEnabled).
+		Find()
+	if err != nil {
+		return nil, err
+	}
+	if one == nil {
+		return nil, nil
+	}
+	return one.(*NodeLogin), nil
+}
+
+// 禁用某个节点的认证
+func (this *NodeLoginDAO) DisableNodeLogins(nodeId int64) error {
+	_, err := this.Query().
+		Attr("nodeId", nodeId).
+		Set("state", NodeLoginStateDisabled).
+		Update()
+	return err
+}