mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-08 19:40:24 +08:00
修复安装时不再自动初始化admin账号
This commit is contained in:
@@ -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'")
|
||||||
|
|||||||
Reference in New Issue
Block a user