oomtest: fix data race in the oomtest (#43398)

close #43396
pingcap · Apr 25, 2023 · 451bda4 · 451bda4
1 parent 355958a
commit 451bda4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/executor/oomtest/oom_test.go b/executor/oomtest/oom_test.go
@@ -262,14 +262,13 @@ func (h *oomCapture) Write(entry zapcore.Entry, fields []zapcore.Field) error {
 		h.mu.Unlock()
 		return nil
 	}
+	h.mu.Lock()
+	defer h.mu.Unlock()
 	// They are just common background task and not related to the oom.
 	if !h.messageFilter.Empty() && !h.messageFilter.Exist(entry.Message) {
 		return nil
 	}
-
-	h.mu.Lock()
 	h.tracker = entry.Message
-	h.mu.Unlock()
 	return nil
 }