From fcd1ffc89370d7460479fbb95bd902860a0dd59a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E7=A5=A5=E8=B6=85?= Date: Wed, 1 Feb 2023 16:56:56 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=AE=89=E8=A3=85=E6=97=B6?= =?UTF-8?q?=E4=B8=8D=E5=86=8D=E8=87=AA=E5=8A=A8=E5=88=9D=E5=A7=8B=E5=8C=96?= =?UTF-8?q?admin=E8=B4=A6=E5=8F=B7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/setup/sql_executor.go | 32 +------------------------------- 1 file changed, 1 insertion(+), 31 deletions(-) diff --git a/internal/setup/sql_executor.go b/internal/setup/sql_executor.go index e7cd7a95..6bcf7953 100644 --- a/internal/setup/sql_executor.go +++ b/internal/setup/sql_executor.go @@ -12,7 +12,6 @@ import ( "github.com/iwind/TeaGo/maps" "github.com/iwind/TeaGo/rands" "github.com/iwind/TeaGo/types" - stringutil "github.com/iwind/TeaGo/utils/string" "gopkg.in/yaml.v3" "io" "os" @@ -82,14 +81,8 @@ func (this *SQLExecutor) Run(showLog bool) 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 { return err } @@ -139,29 +132,6 @@ func (this *SQLExecutor) checkData(db *dbs.DB) error { 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 { stmt, err := db.Prepare("SELECT COUNT(*) FROM edgeAPITokens WHERE role='admin'")