Files
tidb/pkg/planner/core/plan_clone_generated.go
2024-11-08 08:12:25 +00:00

540 lines
18 KiB
Go

// Copyright 2024 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Code generated by plan_clone_generator; DO NOT EDIT IT DIRECTLY.
package core
import (
"github.com/pingcap/tidb/pkg/expression"
"github.com/pingcap/tidb/pkg/planner/core/base"
"github.com/pingcap/tidb/pkg/planner/util"
)
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalTableScan) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalTableScan)
*cloned = *op
basePlan, baseOK := op.physicalSchemaProducer.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.physicalSchemaProducer = *basePlan
cloned.AccessCondition = cloneExpressionsForPlanCache(op.AccessCondition, nil)
cloned.filterCondition = cloneExpressionsForPlanCache(op.filterCondition, nil)
cloned.LateMaterializationFilterCondition = cloneExpressionsForPlanCache(op.LateMaterializationFilterCondition, nil)
cloned.HandleIdx = make([]int, len(op.HandleIdx))
copy(cloned.HandleIdx, op.HandleIdx)
if op.HandleCols != nil {
cloned.HandleCols = op.HandleCols.Clone(newCtx.GetSessionVars().StmtCtx)
}
cloned.ByItems = util.CloneByItemss(op.ByItems)
cloned.PlanPartInfo = op.PlanPartInfo.cloneForPlanCache()
if op.SampleInfo != nil {
return nil, false
}
cloned.constColsByCond = make([]bool, len(op.constColsByCond))
copy(cloned.constColsByCond, op.constColsByCond)
if op.runtimeFilterList != nil {
return nil, false
}
if op.AnnIndexExtra != nil {
return nil, false
}
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalIndexScan) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalIndexScan)
*cloned = *op
basePlan, baseOK := op.physicalSchemaProducer.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.physicalSchemaProducer = *basePlan
cloned.AccessCondition = cloneExpressionsForPlanCache(op.AccessCondition, nil)
cloned.IdxCols = cloneColumnsForPlanCache(op.IdxCols, nil)
cloned.IdxColLens = make([]int, len(op.IdxColLens))
copy(cloned.IdxColLens, op.IdxColLens)
if op.GenExprs != nil {
return nil, false
}
cloned.ByItems = util.CloneByItemss(op.ByItems)
if op.pkIsHandleCol != nil {
if op.pkIsHandleCol.SafeToShareAcrossSession() {
cloned.pkIsHandleCol = op.pkIsHandleCol
} else {
cloned.pkIsHandleCol = op.pkIsHandleCol.Clone().(*expression.Column)
}
}
cloned.constColsByCond = make([]bool, len(op.constColsByCond))
copy(cloned.constColsByCond, op.constColsByCond)
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalSelection) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalSelection)
*cloned = *op
basePlan, baseOK := op.BasePhysicalPlan.CloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.BasePhysicalPlan = *basePlan
cloned.Conditions = cloneExpressionsForPlanCache(op.Conditions, nil)
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalProjection) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalProjection)
*cloned = *op
basePlan, baseOK := op.physicalSchemaProducer.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.physicalSchemaProducer = *basePlan
cloned.Exprs = cloneExpressionsForPlanCache(op.Exprs, nil)
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalSort) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalSort)
*cloned = *op
basePlan, baseOK := op.BasePhysicalPlan.CloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.BasePhysicalPlan = *basePlan
cloned.ByItems = util.CloneByItemss(op.ByItems)
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalTopN) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalTopN)
*cloned = *op
basePlan, baseOK := op.BasePhysicalPlan.CloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.BasePhysicalPlan = *basePlan
cloned.ByItems = util.CloneByItemss(op.ByItems)
cloned.PartitionBy = util.CloneSortItems(op.PartitionBy)
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalStreamAgg) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalStreamAgg)
*cloned = *op
basePlan, baseOK := op.basePhysicalAgg.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.basePhysicalAgg = *basePlan
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalHashAgg) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalHashAgg)
*cloned = *op
basePlan, baseOK := op.basePhysicalAgg.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.basePhysicalAgg = *basePlan
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalHashJoin) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalHashJoin)
*cloned = *op
basePlan, baseOK := op.basePhysicalJoin.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.basePhysicalJoin = *basePlan
cloned.EqualConditions = cloneScalarFunctionsForPlanCache(op.EqualConditions, nil)
cloned.NAEqualConditions = cloneScalarFunctionsForPlanCache(op.NAEqualConditions, nil)
if op.runtimeFilterList != nil {
return nil, false
}
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalMergeJoin) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalMergeJoin)
*cloned = *op
basePlan, baseOK := op.basePhysicalJoin.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.basePhysicalJoin = *basePlan
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalTableReader) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalTableReader)
*cloned = *op
basePlan, baseOK := op.physicalSchemaProducer.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.physicalSchemaProducer = *basePlan
if op.tablePlan != nil {
tablePlan, ok := op.tablePlan.CloneForPlanCache(newCtx)
if !ok {
return nil, false
}
cloned.tablePlan = tablePlan.(base.PhysicalPlan)
}
cloned.TablePlans = flattenPushDownPlan(cloned.tablePlan)
cloned.PlanPartInfo = op.PlanPartInfo.cloneForPlanCache()
if op.TableScanAndPartitionInfos != nil {
return nil, false
}
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalIndexReader) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalIndexReader)
*cloned = *op
basePlan, baseOK := op.physicalSchemaProducer.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.physicalSchemaProducer = *basePlan
if op.indexPlan != nil {
indexPlan, ok := op.indexPlan.CloneForPlanCache(newCtx)
if !ok {
return nil, false
}
cloned.indexPlan = indexPlan.(base.PhysicalPlan)
}
cloned.IndexPlans = flattenPushDownPlan(cloned.indexPlan)
cloned.OutputColumns = cloneColumnsForPlanCache(op.OutputColumns, nil)
cloned.PlanPartInfo = op.PlanPartInfo.cloneForPlanCache()
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PointGetPlan) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PointGetPlan)
*cloned = *op
cloned.Plan = *op.Plan.CloneWithNewCtx(newCtx)
if op.PartitionIdx != nil {
cloned.PartitionIdx = new(int)
*cloned.PartitionIdx = *op.PartitionIdx
}
if op.Handle != nil {
cloned.Handle = op.Handle.Copy()
}
if op.HandleConstant != nil {
if op.HandleConstant.SafeToShareAcrossSession() {
cloned.HandleConstant = op.HandleConstant
} else {
cloned.HandleConstant = op.HandleConstant.Clone().(*expression.Constant)
}
}
cloned.IndexValues = util.CloneDatums(op.IndexValues)
cloned.IndexConstants = cloneConstantsForPlanCache(op.IndexConstants, nil)
cloned.IdxCols = cloneColumnsForPlanCache(op.IdxCols, nil)
cloned.IdxColLens = make([]int, len(op.IdxColLens))
copy(cloned.IdxColLens, op.IdxColLens)
cloned.AccessConditions = cloneExpressionsForPlanCache(op.AccessConditions, nil)
cloned.accessCols = cloneColumnsForPlanCache(op.accessCols, nil)
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *BatchPointGetPlan) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(BatchPointGetPlan)
*cloned = *op
cloned.baseSchemaProducer = *op.baseSchemaProducer.cloneForPlanCache(newCtx)
probeParents, ok := clonePhysicalPlansForPlanCache(newCtx, op.probeParents)
if !ok {
return nil, false
}
cloned.probeParents = probeParents
cloned.ctx = newCtx
cloned.Handles = util.CloneHandles(op.Handles)
cloned.HandleParams = cloneConstantsForPlanCache(op.HandleParams, nil)
cloned.IndexValues = util.CloneDatum2D(op.IndexValues)
cloned.IndexValueParams = cloneConstant2DForPlanCache(op.IndexValueParams)
cloned.AccessConditions = cloneExpressionsForPlanCache(op.AccessConditions, nil)
cloned.IdxCols = cloneColumnsForPlanCache(op.IdxCols, nil)
cloned.IdxColLens = make([]int, len(op.IdxColLens))
copy(cloned.IdxColLens, op.IdxColLens)
cloned.PartitionIdxs = make([]int, len(op.PartitionIdxs))
copy(cloned.PartitionIdxs, op.PartitionIdxs)
cloned.accessCols = cloneColumnsForPlanCache(op.accessCols, nil)
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalLimit) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalLimit)
*cloned = *op
basePlan, baseOK := op.physicalSchemaProducer.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.physicalSchemaProducer = *basePlan
cloned.PartitionBy = util.CloneSortItems(op.PartitionBy)
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalIndexJoin) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalIndexJoin)
*cloned = *op
basePlan, baseOK := op.basePhysicalJoin.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.basePhysicalJoin = *basePlan
if op.innerPlan != nil {
innerPlan, ok := op.innerPlan.CloneForPlanCache(newCtx)
if !ok {
return nil, false
}
cloned.innerPlan = innerPlan.(base.PhysicalPlan)
}
cloned.Ranges = op.Ranges.CloneForPlanCache()
cloned.KeyOff2IdxOff = make([]int, len(op.KeyOff2IdxOff))
copy(cloned.KeyOff2IdxOff, op.KeyOff2IdxOff)
cloned.IdxColLens = make([]int, len(op.IdxColLens))
copy(cloned.IdxColLens, op.IdxColLens)
cloned.CompareFilters = op.CompareFilters.cloneForPlanCache()
cloned.OuterHashKeys = cloneColumnsForPlanCache(op.OuterHashKeys, nil)
cloned.InnerHashKeys = cloneColumnsForPlanCache(op.InnerHashKeys, nil)
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalIndexHashJoin) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalIndexHashJoin)
*cloned = *op
inlj, ok := op.PhysicalIndexJoin.CloneForPlanCache(newCtx)
if !ok {
return nil, false
}
cloned.PhysicalIndexJoin = *inlj.(*PhysicalIndexJoin)
cloned.Self = cloned
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalIndexLookUpReader) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalIndexLookUpReader)
*cloned = *op
basePlan, baseOK := op.physicalSchemaProducer.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.physicalSchemaProducer = *basePlan
if op.indexPlan != nil {
indexPlan, ok := op.indexPlan.CloneForPlanCache(newCtx)
if !ok {
return nil, false
}
cloned.indexPlan = indexPlan.(base.PhysicalPlan)
}
if op.tablePlan != nil {
tablePlan, ok := op.tablePlan.CloneForPlanCache(newCtx)
if !ok {
return nil, false
}
cloned.tablePlan = tablePlan.(base.PhysicalPlan)
}
cloned.IndexPlans = flattenPushDownPlan(cloned.indexPlan)
cloned.TablePlans = flattenPushDownPlan(cloned.tablePlan)
if op.ExtraHandleCol != nil {
if op.ExtraHandleCol.SafeToShareAcrossSession() {
cloned.ExtraHandleCol = op.ExtraHandleCol
} else {
cloned.ExtraHandleCol = op.ExtraHandleCol.Clone().(*expression.Column)
}
}
cloned.PushedLimit = op.PushedLimit.Clone()
cloned.CommonHandleCols = cloneColumnsForPlanCache(op.CommonHandleCols, nil)
cloned.PlanPartInfo = op.PlanPartInfo.cloneForPlanCache()
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalIndexMergeReader) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalIndexMergeReader)
*cloned = *op
basePlan, baseOK := op.physicalSchemaProducer.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.physicalSchemaProducer = *basePlan
cloned.PushedLimit = op.PushedLimit.Clone()
cloned.ByItems = util.CloneByItemss(op.ByItems)
partialPlans, ok := clonePhysicalPlansForPlanCache(newCtx, op.partialPlans)
if !ok {
return nil, false
}
cloned.partialPlans = partialPlans
if op.tablePlan != nil {
tablePlan, ok := op.tablePlan.CloneForPlanCache(newCtx)
if !ok {
return nil, false
}
cloned.tablePlan = tablePlan.(base.PhysicalPlan)
}
cloned.PartialPlans = make([][]base.PhysicalPlan, len(op.PartialPlans))
for i, plan := range cloned.partialPlans {
cloned.PartialPlans[i] = flattenPushDownPlan(plan)
}
cloned.TablePlans = flattenPushDownPlan(cloned.tablePlan)
cloned.PlanPartInfo = op.PlanPartInfo.cloneForPlanCache()
if op.HandleCols != nil {
cloned.HandleCols = op.HandleCols.Clone(newCtx.GetSessionVars().StmtCtx)
}
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *Update) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(Update)
*cloned = *op
cloned.baseSchemaProducer = *op.baseSchemaProducer.cloneForPlanCache(newCtx)
cloned.OrderedList = util.CloneAssignments(op.OrderedList)
if op.SelectPlan != nil {
SelectPlan, ok := op.SelectPlan.CloneForPlanCache(newCtx)
if !ok {
return nil, false
}
cloned.SelectPlan = SelectPlan.(base.PhysicalPlan)
}
if op.PartitionedTable != nil {
return nil, false
}
if op.FKChecks != nil {
return nil, false
}
if op.FKCascades != nil {
return nil, false
}
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *Delete) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(Delete)
*cloned = *op
cloned.baseSchemaProducer = *op.baseSchemaProducer.cloneForPlanCache(newCtx)
if op.SelectPlan != nil {
SelectPlan, ok := op.SelectPlan.CloneForPlanCache(newCtx)
if !ok {
return nil, false
}
cloned.SelectPlan = SelectPlan.(base.PhysicalPlan)
}
if op.FKChecks != nil {
return nil, false
}
if op.FKCascades != nil {
return nil, false
}
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *Insert) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(Insert)
*cloned = *op
cloned.baseSchemaProducer = *op.baseSchemaProducer.cloneForPlanCache(newCtx)
cloned.Lists = cloneExpression2DForPlanCache(op.Lists)
cloned.OnDuplicate = util.CloneAssignments(op.OnDuplicate)
cloned.GenCols = op.GenCols.cloneForPlanCache()
if op.SelectPlan != nil {
SelectPlan, ok := op.SelectPlan.CloneForPlanCache(newCtx)
if !ok {
return nil, false
}
cloned.SelectPlan = SelectPlan.(base.PhysicalPlan)
}
if op.FKChecks != nil {
return nil, false
}
if op.FKCascades != nil {
return nil, false
}
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalLock) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalLock)
*cloned = *op
basePlan, baseOK := op.BasePhysicalPlan.CloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.BasePhysicalPlan = *basePlan
if op.TblID2Handle != nil {
cloned.TblID2Handle = make(map[int64][]util.HandleCols, len(op.TblID2Handle))
for k, v := range op.TblID2Handle {
cloned.TblID2Handle[k] = util.CloneHandleCols(newCtx.GetSessionVars().StmtCtx, v)
}
}
if op.TblID2PhysTblIDCol != nil {
cloned.TblID2PhysTblIDCol = make(map[int64]*expression.Column, len(op.TblID2PhysTblIDCol))
for k, v := range op.TblID2PhysTblIDCol {
cloned.TblID2PhysTblIDCol[k] = v.Clone().(*expression.Column)
}
}
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalUnionScan) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalUnionScan)
*cloned = *op
basePlan, baseOK := op.BasePhysicalPlan.CloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.BasePhysicalPlan = *basePlan
cloned.Conditions = cloneExpressionsForPlanCache(op.Conditions, nil)
if op.HandleCols != nil {
cloned.HandleCols = op.HandleCols.Clone(newCtx.GetSessionVars().StmtCtx)
}
return cloned, true
}
// CloneForPlanCache implements the base.Plan interface.
func (op *PhysicalUnionAll) CloneForPlanCache(newCtx base.PlanContext) (base.Plan, bool) {
cloned := new(PhysicalUnionAll)
*cloned = *op
basePlan, baseOK := op.physicalSchemaProducer.cloneForPlanCacheWithSelf(newCtx, cloned)
if !baseOK {
return nil, false
}
cloned.physicalSchemaProducer = *basePlan
return cloned, true
}