address comment

Signed-off-by: Cabinfever_B <cabinfeveroier@gmail.com> address comment Signed-off-by: Cabinfever_B <cabinfeveroier@gmail.com>
pingcap · Feb 20, 2024 · ae03f2e · ae03f2e
1 parent fd5aed0
commit ae03f2e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/domain/resourcegroup/runaway.go b/pkg/domain/resourcegroup/runaway.go
@@ -540,7 +540,11 @@ func (r *RunawayChecker) CheckCopRespError(err error) error {
 			if r.deadline.Before(now) && r.marked.CompareAndSwap(false, true) {
 				r.markRunaway(RunawayMatchTypeIdentify, r.setting.Action, &now)
 				r.markQuarantine(&now)
+				return exeerrors.ErrResourceGroupQueryRunawayInterrupted
 			}
+		}
+		// Due to concurrency, check again.
+		if r.marked.Load() {
 			return exeerrors.ErrResourceGroupQueryRunawayInterrupted
 		}
 	}