executor: push down LIKE patterns for some infoschema readers

pkg/executor/infoschema_reader.go

@@ -1199,7 +1199,7 @@ func (e *memtableRetriever) setDataFromPartitions(ctx context.Context, sctx sess
 				avgRowLength = dataLength / rowCount
 			}
 			// If there are any condition on the `PARTITION_NAME` in the extractor, this record should be ignored
-			if len(ex.ColPredicates["partition_name"]) > 0 {
+			if ex.HasPartitionPred() {
 				continue
 			}
 			record := types.MakeDatums(
@@ -3773,50 +3773,47 @@ func (e *memtableRetriever) setDataFromIndexUsage(ctx context.Context, sctx sess
 	dom := domain.GetDomain(sctx)
 	rows := make([][]types.Datum, 0, 100)
 	checker := privilege.GetPrivilegeManager(sctx)
-	extractor, ok := e.extractor.(*plannercore.InfoSchemaTiDBIndexUsageExtractor)
+	ex, ok := e.extractor.(*plannercore.InfoSchemaTiDBIndexUsageExtractor)
 	if !ok {
 		return errors.Errorf("wrong extractor type: %T, expected InfoSchemaIndexUsageExtractor", e.extractor)
 	}
-	if extractor.SkipRequest {
+	if ex.SkipRequest {
 		return nil
 	}
 
-	schemas := extractor.ListSchemas(e.is)
-	for _, schema := range schemas {
-		tbls, err := extractor.ListTables(ctx, schema, e.is)
-		if err != nil {
-			return errors.Trace(err)
+	schemas, tbls, err := ex.ListSchemasAndTables(ctx, e.is)
+	if err != nil {
+		return errors.Trace(err)
+	}
+	for i, tbl := range tbls {
+		schema := schemas[i]
+		if checker != nil && !checker.RequestVerification(
+			sctx.GetSessionVars().ActiveRoles,
+			schema.L, tbl.Name.L, "", mysql.AllPrivMask) {
+			continue
 		}
 
-		for _, tbl := range tbls {
-			if checker != nil && !checker.RequestVerification(
-				sctx.GetSessionVars().ActiveRoles,
-				schema.L, tbl.Name.L, "", mysql.AllPrivMask) {
-				continue
-			}
-
-			idxs := extractor.ListIndexes(tbl)
-			for _, idx := range idxs {
-				row := make([]types.Datum, 0, 14)
-				usage := dom.StatsHandle().GetIndexUsage(tbl.ID, idx.ID)
-				row = append(row, types.NewStringDatum(schema.O))
-				row = append(row, types.NewStringDatum(tbl.Name.O))
-				row = append(row, types.NewStringDatum(idx.Name.O))
-				row = append(row, types.NewIntDatum(int64(usage.QueryTotal)))
-				row = append(row, types.NewIntDatum(int64(usage.KvReqTotal)))
-				row = append(row, types.NewIntDatum(int64(usage.RowAccessTotal)))
-				for _, percentage := range usage.PercentageAccess {
-					row = append(row, types.NewIntDatum(int64(percentage)))
-				}
-				lastUsedAt := types.Datum{}
-				lastUsedAt.SetNull()
-				if !usage.LastUsedAt.IsZero() {
-					t := types.NewTime(types.FromGoTime(usage.LastUsedAt), mysql.TypeTimestamp, 0)
-					lastUsedAt = types.NewTimeDatum(t)
-				}
-				row = append(row, lastUsedAt)
-				rows = append(rows, row)
-			}
+		idxs := ex.ListIndexes(tbl)
+		for _, idx := range idxs {
+			row := make([]types.Datum, 0, 14)
+			usage := dom.StatsHandle().GetIndexUsage(tbl.ID, idx.ID)
+			row = append(row, types.NewStringDatum(schema.O))
+			row = append(row, types.NewStringDatum(tbl.Name.O))
+			row = append(row, types.NewStringDatum(idx.Name.O))
+			row = append(row, types.NewIntDatum(int64(usage.QueryTotal)))
+			row = append(row, types.NewIntDatum(int64(usage.KvReqTotal)))
+			row = append(row, types.NewIntDatum(int64(usage.RowAccessTotal)))
+			for _, percentage := range usage.PercentageAccess {
+				row = append(row, types.NewIntDatum(int64(percentage)))
+			}
+			lastUsedAt := types.Datum{}
+			lastUsedAt.SetNull()
+			if !usage.LastUsedAt.IsZero() {
+				t := types.NewTime(types.FromGoTime(usage.LastUsedAt), mysql.TypeTimestamp, 0)
+				lastUsedAt = types.NewTimeDatum(t)
+			}
+			row = append(row, lastUsedAt)
+			rows = append(rows, row)
 		}
 	}