statistics: updating stats cache can trigger evict (#53979)

close #53742
pingcap · Jun 13, 2024 · 2ee6d22 · 2ee6d22
1 parent fab0940
commit 2ee6d22
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/pkg/statistics/handle/cache/internal/lfu/BUILD.bazel b/pkg/statistics/handle/cache/internal/lfu/BUILD.bazel
@@ -30,7 +30,7 @@ go_test(
     embed = [":lfu"],
     flaky = True,
     race = "on",
-    shard_count = 9,
+    shard_count = 10,
     deps = [
         "//pkg/statistics",
         "//pkg/statistics/handle/cache/internal/testutil",

diff --git a/pkg/statistics/handle/cache/internal/lfu/lfu_cache.go b/pkg/statistics/handle/cache/internal/lfu/lfu_cache.go
@@ -207,6 +207,7 @@ func (s *LFU) onExit(val any) {
 	if s.closed.Load() {
 		return
 	}
+	s.triggerEvict()
 	// Subtract the memory usage of the table from the total memory usage.
 	s.addCost(-val.(*statistics.Table).MemoryUsage().TotalTrackingMemUsage())
 }

diff --git a/pkg/statistics/handle/cache/internal/lfu/lfu_cache_test.go b/pkg/statistics/handle/cache/internal/lfu/lfu_cache_test.go
@@ -295,3 +295,15 @@ func TestMemoryControl(t *testing.T) {
 	lfu.wait()
 	require.Equal(t, int64(10)*(2*mockCMSMemoryUsage+mockCMSMemoryUsage), lfu.Cost())
 }
+
+func TestMemoryControlWithUpdate(t *testing.T) {
+	capacity := int64(100)
+	lfu, err := NewLFU(capacity)
+	require.NoError(t, err)
+	for i := 0; i < 100; i++ {
+		t1 := testutil.NewMockStatisticsTable(i, 1, true, false, false)
+		lfu.Put(1, t1)
+	}
+	time.Sleep(1 * time.Second)
+	require.Equal(t, int64(0), lfu.Cost())
+}