find statistics table in table wrappers during replace count star

angelamayxie · angelamayxie · commit e9bfab1400eb · 2025-07-02T17:58:52.000-07:00
diff --git a/sql/analyzer/catalog.go b/sql/analyzer/catalog.go
@@ -467,6 +467,8 @@ func getStatisticsTable(table sql.Table, prevTable sql.Table) (sql.StatisticsTab
 		return t, true
 	case sql.TableNode:
 		return getStatisticsTable(t.UnderlyingTable(), table)
+	case sql.TableWrapper:
+		return getStatisticsTable(t.Underlying(), table)
 	default:
 		return nil, false
 	}
diff --git a/sql/analyzer/replace_count_star.go b/sql/analyzer/replace_count_star.go
@@ -95,7 +95,7 @@ func replaceCountStar(ctx *sql.Context, a *Analyzer, n sql.Node, _ *plan.Scope,
 					return n, transform.SameTree, nil
 				}
 
-				if statsTable, ok := rt.Table.(sql.StatisticsTable); ok {
+				if statsTable, ok := getStatisticsTable(rt.Table, nil); ok {
 					rowCnt, exact, err := statsTable.RowCount(ctx)
 					if err == nil && exact {
 						return plan.NewProject(

Original file line number	Diff line number	Diff line change
`@@ -467,6 +467,8 @@ func getStatisticsTable(table sql.Table, prevTable sql.Table) (sql.StatisticsTab`
`467`	`467`	`return t, true`
`468`	`468`	`case sql.TableNode:`
`469`	`469`	`return getStatisticsTable(t.UnderlyingTable(), table)`
	`470`	`+ case sql.TableWrapper:`
	`471`	`+ return getStatisticsTable(t.Underlying(), table)`
`470`	`472`	`default:`
`471`	`473`	`return nil, false`
`472`	`474`	`}`
Original file line number	Diff line number	Diff line change
`@@ -95,7 +95,7 @@ func replaceCountStar(ctx sql.Context, a Analyzer, n sql.Node, _ *plan.Scope,`
`95`	`95`	`return n, transform.SameTree, nil`
`96`	`96`	`}`
`97`	`97`
`98`		`- if statsTable, ok := rt.Table.(sql.StatisticsTable); ok {`
	`98`	`+ if statsTable, ok := getStatisticsTable(rt.Table, nil); ok {`
`99`	`99`	`rowCnt, exact, err := statsTable.RowCount(ctx)`
`100`	`100`	`if err == nil && exact {`
`101`	`101`	`return plan.NewProject(`