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

初始化时修改默认生成的用户名，并将用户自动关联到默认集群

Browse Source
This commit is contained in:
GoEdgeLab
2023-06-21 11:51:07 +08:00
parent b92e3c6fb0
commit 99e7875ea3
1 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
internal/setup/sql_executor.go
View File

@@ -87,14 +87,8 @@ func (this *SQLExecutor) Run(showLog bool) error {
// 检查数据 // 检查数据
func (this *SQLExecutor) checkData(db *dbs.DB) error { func (this *SQLExecutor) checkData(db *dbs.DB) error {
// 检查初始化用户
err := this.checkUser(db)
if err != nil {
return err
}
// 检查管理员平台节点 // 检查管理员平台节点
err = this.checkAdminNode(db) err := this.checkAdminNode(db)
if err != nil { if err != nil {
return err return err
} }
@@ -111,6 +105,13 @@ func (this *SQLExecutor) checkData(db *dbs.DB) error {
return err return err
} }
// 检查初始化用户
// 需要放在检查集群后面
err = this.checkUser(db)
if err != nil {
return err
}
// 检查IP名单 // 检查IP名单
err = this.checkIPList(db) err = this.checkIPList(db)
if err != nil { if err != nil {
@@ -154,7 +155,14 @@ func (this *SQLExecutor) checkUser(db *dbs.DB) error {
return nil return nil
} }
_, err = db.Exec("INSERT INTO edgeUsers (`username`, `password`, `fullname`, `isOn`, `state`, `createdAt`) VALUES (?, ?, ?, ?, ?, ?)", "USER-"+rands.HexString(10), stringutil.Md5(rands.HexString(32)), "默认用户", 1, 1, time.Now().Unix()) // 读取默认集群ID
// Read default cluster id
clusterId, err := db.FindCol(0, "SELECT id FROM edgeNodeClusters WHERE state=1 ORDER BY id ASC LIMIT 1")
if err != nil {
return err
}
_, err = db.Exec("INSERT INTO edgeUsers (`username`, `password`, `fullname`, `isOn`, `state`, `createdAt`, `clusterId`) VALUES (?, ?, ?, ?, ?, ?, ?)", "USER_"+rands.HexString(10), stringutil.Md5(rands.HexString(32)), "默认用户", 1, 1, time.Now().Unix(), clusterId)
return err return err
} }