fix bug

Jrohy · Jul 31, 2020 · 2af17a0 · 2af17a0
1 parent 989671d
commit 2af17a0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/trojan/client.go b/trojan/client.go
@@ -15,9 +15,9 @@ func GenClientJson() {
 	var user core.User
 	domain, port := GetDomainAndPort()
 	mysql := core.GetMysql()
-	userList := mysql.GetData()
-	if userList == nil {
-		fmt.Println("连接mysql失败!")
+	userList, err := mysql.GetData()
+	if err != nil {
+		fmt.Println(err.Error())
 		return
 	}
 	if len(userList) == 1 {

diff --git a/trojan/install.go b/trojan/install.go
@@ -190,7 +190,7 @@ func InstallMysql() {
 	}
 	mysql.CreateTable()
 	core.WriteMysql(&mysql)
-	if len(mysql.GetData()) == 0 {
+	if userList, _ := mysql.GetData(); len(userList) == 0 {
 		AddUser()
 	}
 	Restart()