[用户节点]可以管理用户节点

This commit is contained in:
GoEdgeLab
2020-12-14 21:25:11 +08:00
parent 641df4ccc5
commit c05d060aa1
16 changed files with 749 additions and 7 deletions

View File

@@ -77,6 +77,12 @@ func (this *SQLExecutor) checkData(db *dbs.DB) error {
return err
}
// 检查用户平台节点
err = this.checkUserNode(db)
if err != nil {
return err
}
// 检查集群配置
err = this.checkCluster(db)
if err != nil {
@@ -143,6 +149,34 @@ func (this *SQLExecutor) checkAdminNode(db *dbs.DB) error {
return nil
}
// 检查用户平台节点
func (this *SQLExecutor) checkUserNode(db *dbs.DB) error {
stmt, err := db.Prepare("SELECT COUNT(*) FROM edgeAPITokens WHERE role='user'")
if err != nil {
return err
}
defer func() {
_ = stmt.Close()
}()
col, err := stmt.FindCol(0)
if err != nil {
return err
}
count := types.Int(col)
if count > 0 {
return nil
}
nodeId := rands.HexString(32)
secret := rands.String(32)
_, err = db.Exec("INSERT INTO edgeAPITokens (nodeId, secret, role) VALUES (?, ?, ?)", nodeId, secret, "user")
if err != nil {
return err
}
return nil
}
// 检查集群配置
func (this *SQLExecutor) checkCluster(db *dbs.DB) error {
/// 检查是否有集群数字