planner: add hash64 equals for tableDual/topN/unionScan/Window. (ping…

…cap#57412)

ref pingcap#51664

pkg/meta/model/index.go

@@ -18,6 +18,7 @@ import (
 	"github.com/pingcap/tidb/pkg/parser/ast"
 	"github.com/pingcap/tidb/pkg/parser/model"
 	"github.com/pingcap/tidb/pkg/parser/types"
+	"github.com/pingcap/tidb/pkg/planner/cascades/base"
 )
 
 // DistanceMetric is the distance metric used by the vector index.
@@ -78,6 +79,27 @@ type IndexInfo struct {
 	VectorInfo    *VectorIndexInfo `json:"vector_index"` // VectorInfo is the vector index information.
 }
 
+// Hash64 implement HashEquals interface.
+func (index *IndexInfo) Hash64(h base.Hasher) {
+	h.HashInt64(index.ID)
+}
+
+// Equals implements HashEquals interface.
+func (index *IndexInfo) Equals(other any) bool {
+	// any(nil) can still be converted as (*IndexInfo)(nil)
+	index2, ok := other.(*IndexInfo)
+	if !ok {
+		return false
+	}
+	if index == nil {
+		return index2 == nil
+	}
+	if index2 == nil {
+		return false
+	}
+	return index.ID == index2.ID
+}
+
 // Clone clones IndexInfo.
 func (index *IndexInfo) Clone() *IndexInfo {
 	if index == nil {

pkg/planner/core/generator/hash64_equals/hash64_equals_generator.go

@@ -40,6 +40,7 @@ func GenHash64Equals4LogicalOps() ([]byte, error) {
 		logicalop.LogicalExpand{}, logicalop.LogicalLimit{}, logicalop.LogicalMaxOneRow{}, logicalop.DataSource{},
 		logicalop.LogicalMemTable{}, logicalop.LogicalUnionAll{}, logicalop.LogicalPartitionUnionAll{}, logicalop.LogicalProjection{},
 		logicalop.LogicalSelection{}, logicalop.LogicalShow{}, logicalop.LogicalShowDDLJobs{}, logicalop.LogicalSort{},
+		logicalop.LogicalTableDual{}, logicalop.LogicalTopN{}, logicalop.LogicalUnionScan{}, logicalop.LogicalWindow{},
 	}
 	c := new(cc)
 	c.write(codeGenHash64EqualsPrefix)
@@ -144,6 +145,14 @@ func logicalOpName2PlanCodecString(name string) string {
 		return "plancodec.TypeShowDDLJobs"
 	case "LogicalSort":
 		return "plancodec.TypeSort"
+	case "LogicalTableDual":
+		return "plancodec.TypeDual"
+	case "LogicalTopN":
+		return "plancodec.TypeTopN"
+	case "LogicalUnionScan":
+		return "plancodec.TypeUnionScan"
+	case "LogicalWindow":
+		return "plancodec.TypeWindow"
 	default:
 		return ""
 	}

pkg/planner/core/operator/logicalop/logical_table_dual.go

@@ -33,10 +33,10 @@ import (
 // outputting 0/1 row with zero column. This semantic may be different from your expectation sometimes but should not
 // cause any actual problems now.
 type LogicalTableDual struct {
-	LogicalSchemaProducer
+	LogicalSchemaProducer `hash64-equals:"true"`
 
 	// RowCount could only be 0 or 1.
-	RowCount int
+	RowCount int `hash64-equals:"true"`
 }
 
 // Init initializes LogicalTableDual.

pkg/planner/core/operator/logicalop/logical_top_n.go

@@ -32,12 +32,12 @@ import (
 type LogicalTopN struct {
 	BaseLogicalPlan
 
-	ByItems []*util.ByItems
+	ByItems []*util.ByItems `hash64-equals:"true"`
 	// PartitionBy is used for extended TopN to consider K heaps. Used by rule_derive_topn_from_window
-	PartitionBy      []property.SortItem // This is used for enhanced topN optimization
-	Offset           uint64
-	Count            uint64
-	PreferLimitToCop bool
+	PartitionBy      []property.SortItem `hash64-equals:"true"` // This is used for enhanced topN optimization
+	Offset           uint64              `hash64-equals:"true"`
+	Count            uint64              `hash64-equals:"true"`
+	PreferLimitToCop bool                `hash64-equals:"true"`
 }
 
 // Init initializes LogicalTopN.

pkg/planner/core/operator/logicalop/logical_union_scan.go

@@ -32,9 +32,9 @@ import (
 type LogicalUnionScan struct {
 	BaseLogicalPlan
 
-	Conditions []expression.Expression
+	Conditions []expression.Expression `hash64-equals:"true"`
 
-	HandleCols util.HandleCols
+	HandleCols util.HandleCols `hash64-equals:"true"`
 }
 
 // Init initializes LogicalUnionScan.