find statistics table in table wrappers during replace count star

Author: angelamayxie

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
 	}

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(