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

修复安装时不再自动初始化admin账号

Browse Source
This commit is contained in:
刘祥超
2023-02-01 16:56:56 +08:00
parent c201f3e07c
commit fcd1ffc893
1 changed files with 1 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
internal/setup/sql_executor.go
View File

@@ -12,7 +12,6 @@ import (
"github.com/iwind/TeaGo/maps" "github.com/iwind/TeaGo/maps"
"github.com/iwind/TeaGo/rands" "github.com/iwind/TeaGo/rands"
"github.com/iwind/TeaGo/types" "github.com/iwind/TeaGo/types"
stringutil "github.com/iwind/TeaGo/utils/string"
"gopkg.in/yaml.v3" "gopkg.in/yaml.v3"
"io" "io"
"os" "os"
@@ -82,14 +81,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.checkAdmin(db)
if err != nil {
return err
}
// 检查管理员平台节点 // 检查管理员平台节点
err = this.checkAdminNode(db) err := this.checkAdminNode(db)
if err != nil { if err != nil {
return err return err
} }
@@ -139,29 +132,6 @@ func (this *SQLExecutor) checkData(db *dbs.DB) error {
return nil return nil
} }
// 检查管理员
func (this *SQLExecutor) checkAdmin(db *dbs.DB) error {
stmt, err := db.Prepare("SELECT COUNT(*) FROM edgeAdmins")
if err != nil {
return errors.New("check admin failed: " + err.Error())
}
defer func() {
_ = stmt.Close()
}()
col, err := stmt.FindCol(0)
if err != nil {
return errors.New("check admin failed: " + err.Error())
}
count := types.Int(col)
if count == 0 {
_, err = db.Exec("INSERT INTO edgeAdmins (username, password, fullname, isSuper, createdAt, state) VALUES (?, ?, ?, ?, ?, ?)", "admin", stringutil.Md5("123456"), "管理员", 1, time.Now().Unix(), 1)
if err != nil {
return errors.New("create admin failed: " + err.Error())
}
}
return nil
}
// 检查管理员平台节点 // 检查管理员平台节点
func (this *SQLExecutor) checkAdminNode(db *dbs.DB) error { func (this *SQLExecutor) checkAdminNode(db *dbs.DB) error {
stmt, err := db.Prepare("SELECT COUNT(*) FROM edgeAPITokens WHERE role='admin'") stmt, err := db.Prepare("SELECT COUNT(*) FROM edgeAPITokens WHERE role='admin'")