infoschema: add cluster_tidb_statements_stats table

pingcap · Dec 3, 2024 · 8900c16 · 8900c16
1 parent 26734f7
commit 8900c16
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 3 deletions.
diff --git a/pkg/executor/builder.go b/pkg/executor/builder.go
@@ -2328,7 +2328,8 @@ func (b *executorBuilder) buildMemTable(v *plannercore.PhysicalMemTable) exec.Ex
 			strings.ToLower(infoschema.TableTiDBStatementsStats),
 			strings.ToLower(infoschema.ClusterTableStatementsSummary),
 			strings.ToLower(infoschema.ClusterTableStatementsSummaryHistory),
-			strings.ToLower(infoschema.ClusterTableStatementsSummaryEvicted):
+			strings.ToLower(infoschema.ClusterTableStatementsSummaryEvicted),
+			strings.ToLower(infoschema.ClusterTableTiDBStatementsStats):
 			var extractor *plannercore.StatementsSummaryExtractor
 			if v.Extractor != nil {
 				extractor = v.Extractor.(*plannercore.StatementsSummaryExtractor)

diff --git a/pkg/executor/stmtsummary.go b/pkg/executor/stmtsummary.go
@@ -346,15 +346,22 @@ func isClusterTable(originalTableName string) bool {
 	switch originalTableName {
 	case infoschema.ClusterTableStatementsSummary,
 		infoschema.ClusterTableStatementsSummaryHistory,
-		infoschema.ClusterTableStatementsSummaryEvicted:
+		infoschema.ClusterTableStatementsSummaryEvicted,
+		infoschema.ClusterTableTiDBStatementsStats:
 		return true
 	}
 
 	return false
 }
 
 func isCumulativeTable(originalTableName string) bool {
-	return originalTableName == infoschema.TableTiDBStatementsStats
+	switch originalTableName {
+	case infoschema.TableTiDBStatementsStats,
+		infoschema.ClusterTableTiDBStatementsStats:
+		return true
+	}
+
+	return false
 }
 
 func isCurrentTable(originalTableName string) bool {

diff --git a/pkg/infoschema/cluster.go b/pkg/infoschema/cluster.go
@@ -42,6 +42,8 @@ const (
 	ClusterTableStatementsSummaryHistory = "CLUSTER_STATEMENTS_SUMMARY_HISTORY"
 	// ClusterTableStatementsSummaryEvicted is the string constant of cluster statement summary evict table.
 	ClusterTableStatementsSummaryEvicted = "CLUSTER_STATEMENTS_SUMMARY_EVICTED"
+	// ClusterTableTiDBStatementsStats is the string constant of the cluster statement stats table.
+	ClusterTableTiDBStatementsStats = "CLUSTER_TIDB_STATEMENTS_STATS"
 	// ClusterTableTiDBTrx is the string constant of cluster transaction running table.
 	ClusterTableTiDBTrx = "CLUSTER_TIDB_TRX"
 	// ClusterTableDeadlocks is the string constant of cluster dead lock table.
@@ -65,6 +67,7 @@ var memTableToAllTiDBClusterTables = map[string]string{
 	TableStatementsSummary:        ClusterTableStatementsSummary,
 	TableStatementsSummaryHistory: ClusterTableStatementsSummaryHistory,
 	TableStatementsSummaryEvicted: ClusterTableStatementsSummaryEvicted,
+	TableTiDBStatementsStats:      ClusterTableTiDBStatementsStats,
 	TableTiDBTrx:                  ClusterTableTiDBTrx,
 	TableDeadlocks:                ClusterTableDeadlocks,
 	TableTrxSummary:               ClusterTableTrxSummary,

diff --git a/pkg/infoschema/tables.go b/pkg/infoschema/tables.go
@@ -345,6 +345,7 @@ var tableIDMap = map[string]int64{
 	TableTiDBPlanCache:                   autoid.InformationSchemaDBID + 96,
 	ClusterTableTiDBPlanCache:            autoid.InformationSchemaDBID + 97,
 	TableTiDBStatementsStats:             autoid.InformationSchemaDBID + 98,
+	ClusterTableTiDBStatementsStats:      autoid.InformationSchemaDBID + 99,
 }
 
 // columnInfo represents the basic column information of all kinds of INFORMATION_SCHEMA tables

diff --git a/pkg/planner/core/operator/logicalop/logical_mem_table.go b/pkg/planner/core/operator/logicalop/logical_mem_table.go
@@ -84,6 +84,7 @@ func (p *LogicalMemTable) PruneColumns(parentUsedCols []*expression.Column, opt
 		infoschema.TableSlowQuery,
 		infoschema.ClusterTableStatementsSummary,
 		infoschema.ClusterTableStatementsSummaryHistory,
+		infoschema.ClusterTableTiDBStatementsStats,
 		infoschema.ClusterTableSlowLog,
 		infoschema.TableTiDBTrx,
 		infoschema.ClusterTableTiDBTrx,