优化数据库升级

Jrohy · Feb 24, 2021 · a8d17a9 · a8d17a9
1 parent 400527e
commit a8d17a9
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/core/tools.go b/core/tools.go
@@ -55,10 +55,11 @@ ADD COLUMN expiryDate char(10) DEFAULT '';
 			return err
 		}
 	}
-	result := db.QueryRow(fmt.Sprintf(
+	var tableName string
+	error = db.QueryRow(fmt.Sprintf(
 		"SELECT * FROM information_schema.TABLES WHERE TABLE_NAME = 'users' AND TABLE_SCHEMA = '%s' ",
-		mysql.Database) + " AND TABLE_COLLATION LIKE 'utf8%';")
-	if result.Err() == sql.ErrNoRows {
+		mysql.Database) + " AND TABLE_COLLATION LIKE 'utf8%';").Scan(&tableName)
+	if error == sql.ErrNoRows {
 		tempFile := "temp.sql"
 		mysql.DumpSql(tempFile)
 		mysql.ExecSql(tempFile)