fix dump panic(#1.2-dev) (#16903)

fix dump panic Approved by: @daviszhen, @sukki37
matrixorigin · Jun 14, 2024 · 79577c2 · 79577c2
1 parent 6071110
commit 79577c2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pkg/frontend/session.go b/pkg/frontend/session.go
@@ -1223,7 +1223,7 @@ func (ses *Session) AuthenticateUser(ctx context.Context, userInput string, dbNa
 	// If the login information contains the database name, verify if the database exists
 	if dbName != "" {
 		ses.timestampMap[TSCheckDbNameStart] = time.Now()
-		_, err = executeSQLInBackgroundSession(tenantCtx, ses, "use "+dbName)
+		_, err = executeSQLInBackgroundSession(tenantCtx, ses, "use `"+dbName+"`")
 		if err != nil {
 			return nil, err
 		}
@@ -1523,7 +1523,7 @@ func (d *dbMigration) Migrate(ctx context.Context, ses *Session) error {
 		skipRespClient: true,
 		ses:            ses,
 	}
-	return doComQuery(ses, tempExecCtx, &UserInput{sql: "use " + d.db})
+	return doComQuery(ses, tempExecCtx, &UserInput{sql: "use `" + d.db + "`"})
 }
 
 type prepareStmtMigration struct {

diff --git a/pkg/frontend/snapshot.go b/pkg/frontend/snapshot.go
@@ -717,7 +717,7 @@ func restoreViews(
 		if tblInfo, ok := viewMap[key]; ok {
 			getLogger().Info(fmt.Sprintf("[%s] start to restore view: %v", snapshotName, tblInfo.tblName))
 
-			if err = bh.Exec(toCtx, "use "+tblInfo.dbName); err != nil {
+			if err = bh.Exec(toCtx, "use `"+tblInfo.dbName+"`"); err != nil {
 				return err
 			}
 
@@ -747,7 +747,7 @@ func recreateTable(
 
 	ctx = defines.AttachAccountId(ctx, toAccountId)
 
-	if err = bh.Exec(ctx, fmt.Sprintf("use %s", tblInfo.dbName)); err != nil {
+	if err = bh.Exec(ctx, fmt.Sprintf("use `%s`", tblInfo.dbName)); err != nil {
 		return
 	}