安装时自动检查服务器上安装的MySQL

internal/web/actions/default/setup/detectDB.go

@@ -0,0 +1,61 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+
+package setup
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
+	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/maps"
+	"net"
+	"strings"
+	"time"
+)
+
+// DetectDBAction 尝试从本地服务器中发现MySQL
+type DetectDBAction struct {
+	actionutils.ParentAction
+}
+
+func (this *DetectDBAction) RunPost(params struct{}) {
+	var localHost = ""
+	var localPort = ""
+	var localUsername = ""
+	var localPassword = ""
+
+	// 本地的3306端口是否可以连接
+	conn, err := net.DialTimeout("tcp", "127.0.0.1:3306", 3*time.Second)
+	if err == nil {
+		_ = conn.Close()
+		localHost = "127.0.0.1"
+		localPort = "3306"
+
+		var username = "root"
+		for _, pass := range []string{"", "123456", "654321", "Aa_123456"} {
+			db, err := dbs.NewInstanceFromConfig(&dbs.DBConfig{
+				Driver: "mysql",
+				Dsn:    username + ":" + pass + "@tcp(" + configutils.QuoteIP(localHost) + ":" + localPort + ")/edges11111",
+				Prefix: "",
+			})
+			if err == nil {
+				err = db.Raw().Ping()
+				_ = db.Close()
+
+				if err == nil || strings.Contains(err.Error(), "Error 1049") {
+					localUsername = username
+					localPassword = pass
+					break
+				}
+			}
+		}
+	}
+
+	this.Data["localDB"] = maps.Map{
+		"host":     localHost,
+		"port":     localPort,
+		"username": localUsername,
+		"password": localPassword,
+	}
+
+	this.Success()
+}

internal/web/actions/default/setup/init.go

@@ -13,6 +13,7 @@ func init() {
 			Post("/validateAdmin", new(ValidateAdminAction)).
 			Post("/install", new(InstallAction)).
 			Post("/status", new(StatusAction)).
+			Post("/detectDB", new(DetectDBAction)).
 			EndAll()
 	})
 }