executor: verify the existence of the resource group before switching (…

…#56145) ref #54434
pingcap · Sep 22, 2024 · 1acbbec · 1acbbec
1 parent 2761fe0
commit 1acbbec
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/executor/adapter.go b/pkg/executor/adapter.go
@@ -555,7 +555,12 @@ func (a *ExecStmt) Exec(ctx context.Context) (_ sqlexec.RecordSet, err error) {
 			return nil, err
 		}
 		if len(switchGroupName) > 0 {
-			stmtCtx.ResourceGroupName = switchGroupName
+			group, err := rm.ResourceGroupCtl.GetResourceGroup(switchGroupName)
+			if err != nil || group == nil {
+				logutil.BgLogger().Debug("invalid switch resource group", zap.String("switch-group-name", switchGroupName), zap.Error(err))
+			} else {
+				stmtCtx.ResourceGroupName = switchGroupName
+			}
 		}
 	}
 	ctx = a.observeStmtBeginForTopSQL(ctx)