github.com/whtcorpsinc/milevadb-prod@v0.0.0-20211104133533-f57f4be3b597/causetstore/petri/acyclic/causet/implementation/datasource.go

// Copyright 2020 WHTCORPS INC, 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,
// See the License for the specific language governing permissions and
// limitations under the License.

package implementation

import (
	"math"

	causetembedded "github.com/whtcorpsinc/milevadb/causet/embedded"
	"github.com/whtcorpsinc/milevadb/causet/memo"
	"github.com/whtcorpsinc/milevadb/ekv"
	"github.com/whtcorpsinc/milevadb/memex"
	"github.com/whtcorpsinc/milevadb/statistics"
)

// BlockDualImpl implementation of PhysicalBlockDual.
type BlockDualImpl struct {
	baseImpl
}

// NewBlockDualImpl creates a new causet dual Implementation.
func NewBlockDualImpl(dual *causetembedded.PhysicalBlockDual) *BlockDualImpl {
	return &BlockDualImpl{baseImpl{plan: dual}}
}

// CalcCost calculates the cost of the causet dual Implementation.
func (impl *BlockDualImpl) CalcCost(outCount float64, children ...memo.Implementation) float64 {
	return 0
}

// MemBlockScanImpl implementation of PhysicalBlockDual.
type MemBlockScanImpl struct {
	baseImpl
}

// NewMemBlockScanImpl creates a new causet dual Implementation.
func NewMemBlockScanImpl(dual *causetembedded.PhysicalMemBlock) *MemBlockScanImpl {
	return &MemBlockScanImpl{baseImpl{plan: dual}}
}

// CalcCost calculates the cost of the causet dual Implementation.
func (impl *MemBlockScanImpl) CalcCost(outCount float64, children ...memo.Implementation) float64 {
	return 0
}

// BlockReaderImpl implementation of PhysicalBlockReader.
type BlockReaderImpl struct {
	baseImpl
	tblDefCausHists *statistics.HistDefCausl
}

// NewBlockReaderImpl creates a new causet reader Implementation.
func NewBlockReaderImpl(reader *causetembedded.PhysicalBlockReader, hists *statistics.HistDefCausl) *BlockReaderImpl {
	base := baseImpl{plan: reader}
	impl := &BlockReaderImpl{
		baseImpl:        base,
		tblDefCausHists: hists,
	}
	return impl
}

// CalcCost calculates the cost of the causet reader Implementation.
func (impl *BlockReaderImpl) CalcCost(outCount float64, children ...memo.Implementation) float64 {
	reader := impl.plan.(*causetembedded.PhysicalBlockReader)
	width := impl.tblDefCausHists.GetAvgRowSize(impl.plan.SCtx(), reader.Schema().DeferredCausets, false, false)
	sessVars := reader.SCtx().GetStochastikVars()
	networkCost := outCount * sessVars.NetworkFactor * width
	// CausetTasks are run in parallel, to make the estimated cost closer to execution time, we amortize
	// the cost to cop iterator workers. According to `CopClient::Send`, the concurrency
	// is Min(DistALLEGROSQLScanConcurrency, numRegionsInvolvedInScan), since we cannot infer
	// the number of regions involved, we simply use DistALLEGROSQLScanConcurrency.
	copIterWorkers := float64(sessVars.DistALLEGROSQLScanConcurrency())
	impl.cost = (networkCost + children[0].GetCost()) / copIterWorkers
	return impl.cost
}

// GetCostLimit implements Implementation interface.
func (impl *BlockReaderImpl) GetCostLimit(costLimit float64, children ...memo.Implementation) float64 {
	reader := impl.plan.(*causetembedded.PhysicalBlockReader)
	sessVars := reader.SCtx().GetStochastikVars()
	copIterWorkers := float64(sessVars.DistALLEGROSQLScanConcurrency())
	if math.MaxFloat64/copIterWorkers < costLimit {
		return math.MaxFloat64
	}
	return costLimit * copIterWorkers
}

// BlockScanImpl implementation of PhysicalBlockScan.
type BlockScanImpl struct {
	baseImpl
	tblDefCausHists *statistics.HistDefCausl
	tblDefCauss     []*memex.DeferredCauset
}

// NewBlockScanImpl creates a new causet scan Implementation.
func NewBlockScanImpl(ts *causetembedded.PhysicalBlockScan, defcaus []*memex.DeferredCauset, hists *statistics.HistDefCausl) *BlockScanImpl {
	base := baseImpl{plan: ts}
	impl := &BlockScanImpl{
		baseImpl:        base,
		tblDefCausHists: hists,
		tblDefCauss:     defcaus,
	}
	return impl
}

// CalcCost calculates the cost of the causet scan Implementation.
func (impl *BlockScanImpl) CalcCost(outCount float64, children ...memo.Implementation) float64 {
	ts := impl.plan.(*causetembedded.PhysicalBlockScan)
	width := impl.tblDefCausHists.GetBlockAvgRowSize(impl.plan.SCtx(), impl.tblDefCauss, ekv.EinsteinDB, true)
	sessVars := ts.SCtx().GetStochastikVars()
	impl.cost = outCount * sessVars.ScanFactor * width
	if ts.Desc {
		impl.cost = outCount * sessVars.DescScanFactor * width
	}
	return impl.cost
}

// IndexReaderImpl is the implementation of PhysicalIndexReader.
type IndexReaderImpl struct {
	baseImpl
	tblDefCausHists *statistics.HistDefCausl
}

// GetCostLimit implements Implementation interface.
func (impl *IndexReaderImpl) GetCostLimit(costLimit float64, children ...memo.Implementation) float64 {
	reader := impl.plan.(*causetembedded.PhysicalIndexReader)
	sessVars := reader.SCtx().GetStochastikVars()
	copIterWorkers := float64(sessVars.DistALLEGROSQLScanConcurrency())
	if math.MaxFloat64/copIterWorkers < costLimit {
		return math.MaxFloat64
	}
	return costLimit * copIterWorkers
}

// CalcCost implements Implementation interface.
func (impl *IndexReaderImpl) CalcCost(outCount float64, children ...memo.Implementation) float64 {
	reader := impl.plan.(*causetembedded.PhysicalIndexReader)
	sessVars := reader.SCtx().GetStochastikVars()
	networkCost := outCount * sessVars.NetworkFactor * impl.tblDefCausHists.GetAvgRowSize(reader.SCtx(), children[0].GetCauset().Schema().DeferredCausets, true, false)
	copIterWorkers := float64(sessVars.DistALLEGROSQLScanConcurrency())
	impl.cost = (networkCost + children[0].GetCost()) / copIterWorkers
	return impl.cost
}

// NewIndexReaderImpl creates a new IndexReader Implementation.
func NewIndexReaderImpl(reader *causetembedded.PhysicalIndexReader, tblDefCausHists *statistics.HistDefCausl) *IndexReaderImpl {
	return &IndexReaderImpl{
		baseImpl:        baseImpl{plan: reader},
		tblDefCausHists: tblDefCausHists,
	}
}

// IndexScanImpl is the Implementation of PhysicalIndexScan.
type IndexScanImpl struct {
	baseImpl
	tblDefCausHists *statistics.HistDefCausl
}

// CalcCost implements Implementation interface.
func (impl *IndexScanImpl) CalcCost(outCount float64, children ...memo.Implementation) float64 {
	is := impl.plan.(*causetembedded.PhysicalIndexScan)
	sessVars := is.SCtx().GetStochastikVars()
	rowSize := impl.tblDefCausHists.GetIndexAvgRowSize(is.SCtx(), is.Schema().DeferredCausets, is.Index.Unique)
	cost := outCount * rowSize * sessVars.ScanFactor
	if is.Desc {
		cost = outCount * rowSize * sessVars.DescScanFactor
	}
	cost += float64(len(is.Ranges)) * sessVars.SeekFactor
	impl.cost = cost
	return impl.cost
}

// NewIndexScanImpl creates a new IndexScan Implementation.
func NewIndexScanImpl(scan *causetembedded.PhysicalIndexScan, tblDefCausHists *statistics.HistDefCausl) *IndexScanImpl {
	return &IndexScanImpl{
		baseImpl:        baseImpl{plan: scan},
		tblDefCausHists: tblDefCausHists,
	}
}