// Copyright 2022 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.

package cache

import (
	"context"
	"encoding/binary"
	"encoding/hex"
	"fmt"
	"math"
	"time"

	"github.com/pingcap/errors"
	"github.com/pingcap/tidb/pkg/expression"
	"github.com/pingcap/tidb/pkg/expression/exprstatic"
	"github.com/pingcap/tidb/pkg/kv"
	"github.com/pingcap/tidb/pkg/meta/model"
	"github.com/pingcap/tidb/pkg/parser/ast"
	"github.com/pingcap/tidb/pkg/parser/charset"
	pmodel "github.com/pingcap/tidb/pkg/parser/model"
	"github.com/pingcap/tidb/pkg/parser/mysql"
	"github.com/pingcap/tidb/pkg/parser/terror"
	"github.com/pingcap/tidb/pkg/table/tables"
	"github.com/pingcap/tidb/pkg/tablecodec"
	"github.com/pingcap/tidb/pkg/ttl/session"
	"github.com/pingcap/tidb/pkg/types"
	"github.com/pingcap/tidb/pkg/util/chunk"
	"github.com/pingcap/tidb/pkg/util/codec"
	"github.com/pingcap/tidb/pkg/util/collate"
	"github.com/pingcap/tidb/pkg/util/intest"
	"github.com/pingcap/tidb/pkg/util/logutil"
	"github.com/tikv/client-go/v2/tikv"
	"go.uber.org/zap"
)

func getTableKeyColumns(tbl *model.TableInfo) ([]*model.ColumnInfo, []*types.FieldType, error) {
	if tbl.PKIsHandle {
		for i, col := range tbl.Columns {
			if mysql.HasPriKeyFlag(col.GetFlag()) {
				return []*model.ColumnInfo{tbl.Columns[i]}, []*types.FieldType{&tbl.Columns[i].FieldType}, nil
			}
		}
		return nil, nil, errors.Errorf("Cannot find primary key for table: %s", tbl.Name)
	}

	if tbl.IsCommonHandle {
		idxInfo := tables.FindPrimaryIndex(tbl)
		columns := make([]*model.ColumnInfo, len(idxInfo.Columns))
		fieldTypes := make([]*types.FieldType, len(idxInfo.Columns))
		for i, idxCol := range idxInfo.Columns {
			columns[i] = tbl.Columns[idxCol.Offset]
			fieldTypes[i] = &tbl.Columns[idxCol.Offset].FieldType
		}
		return columns, fieldTypes, nil
	}

	extraHandleColInfo := model.NewExtraHandleColInfo()
	return []*model.ColumnInfo{extraHandleColInfo}, []*types.FieldType{&extraHandleColInfo.FieldType}, nil
}

// ScanRange is the range to scan. The range is: [Start, End)
type ScanRange struct {
	Start []types.Datum
	End   []types.Datum
}

func newFullRange() ScanRange {
	return ScanRange{}
}

func newDatumRange(start types.Datum, end types.Datum) (r ScanRange) {
	if !start.IsNull() {
		r.Start = []types.Datum{start}
	}
	if !end.IsNull() {
		r.End = []types.Datum{end}
	}
	return r
}

func nullDatum() types.Datum {
	d := types.Datum{}
	d.SetNull()
	return d
}

// PhysicalTable is used to provide some information for a physical table in TTL job
type PhysicalTable struct {
	// ID is the physical ID of the table
	ID int64
	// Schema is the database name of the table
	Schema pmodel.CIStr
	*model.TableInfo
	// Partition is the partition name
	Partition pmodel.CIStr
	// PartitionDef is the partition definition
	PartitionDef *model.PartitionDefinition
	// KeyColumns is the cluster index key columns for the table
	KeyColumns []*model.ColumnInfo
	// KeyColumnTypes is the types of the key columns
	KeyColumnTypes []*types.FieldType
	// TimeColum is the time column used for TTL
	TimeColumn *model.ColumnInfo
}

// NewBasePhysicalTable create a new PhysicalTable with specific timeColumn.
func NewBasePhysicalTable(schema pmodel.CIStr,
	tbl *model.TableInfo,
	partition pmodel.CIStr,
	timeColumn *model.ColumnInfo,
) (*PhysicalTable, error) {
	if tbl.State != model.StatePublic {
		return nil, errors.Errorf("table '%s.%s' is not a public table", schema, tbl.Name)
	}

	keyColumns, keyColumTypes, err := getTableKeyColumns(tbl)
	if err != nil {
		return nil, err
	}

	var physicalID int64
	var partitionDef *model.PartitionDefinition
	if tbl.Partition == nil {
		if partition.L != "" {
			return nil, errors.Errorf("table '%s.%s' is not a partitioned table", schema, tbl.Name)
		}
		physicalID = tbl.ID
	} else {
		if partition.L == "" {
			return nil, errors.Errorf("partition name is required, table '%s.%s' is a partitioned table", schema, tbl.Name)
		}

		for i := range tbl.Partition.Definitions {
			def := &tbl.Partition.Definitions[i]
			if def.Name.L == partition.L {
				partitionDef = def
			}
		}

		if partitionDef == nil {
			return nil, errors.Errorf("partition '%s' is not found in ttl table '%s.%s'", partition.O, schema, tbl.Name)
		}

		physicalID = partitionDef.ID
	}

	return &PhysicalTable{
		ID:             physicalID,
		Schema:         schema,
		TableInfo:      tbl,
		Partition:      partition,
		PartitionDef:   partitionDef,
		KeyColumns:     keyColumns,
		KeyColumnTypes: keyColumTypes,
		TimeColumn:     timeColumn,
	}, nil
}

// NewPhysicalTable create a new PhysicalTable
func NewPhysicalTable(schema pmodel.CIStr, tbl *model.TableInfo, partition pmodel.CIStr) (*PhysicalTable, error) {
	ttlInfo := tbl.TTLInfo
	if ttlInfo == nil {
		return nil, errors.Errorf("table '%s.%s' is not a ttl table", schema, tbl.Name)
	}

	timeColumn := tbl.FindPublicColumnByName(ttlInfo.ColumnName.L)
	if timeColumn == nil {
		return nil, errors.Errorf("time column '%s' is not public in ttl table '%s.%s'", ttlInfo.ColumnName, schema, tbl.Name)
	}

	return NewBasePhysicalTable(schema, tbl, partition, timeColumn)
}

// ValidateKeyPrefix validates a key prefix
func (t *PhysicalTable) ValidateKeyPrefix(key []types.Datum) error {
	if len(key) > len(t.KeyColumns) {
		return errors.Errorf("invalid key length: %d, expected %d", len(key), len(t.KeyColumns))
	}
	return nil
}

type mockExpireTimeKey struct{}

// SetMockExpireTime can only used in test
func SetMockExpireTime(ctx context.Context, tm time.Time) context.Context {
	return context.WithValue(ctx, mockExpireTimeKey{}, tm)
}

// FullName returns the full name of the table
func (t *PhysicalTable) FullName() string {
	if t.Partition.L != "" {
		return fmt.Sprintf("%s.%s.%s", t.Schema.O, t.Name.O, t.Partition.O)
	}
	return fmt.Sprintf("%s.%s", t.Schema.O, t.Name.O)
}

// EvalExpireTime returns the expired time.
func EvalExpireTime(now time.Time, interval string, unit ast.TimeUnitType) (time.Time, error) {
	// Firstly, we should use the UTC time zone to compute the expired time to avoid time shift caused by DST.
	// The start time should be a time with the same datetime string as `now` but it is in the UTC timezone.
	// For example, if global timezone is `Asia/Shanghai` with a string format `2020-01-01 08:00:00 +0800`.
	// The startTime should be in timezone `UTC` and have a string format `2020-01-01 08:00:00 +0000` which is not the
	// same as the original one (`2020-01-01 00:00:00 +0000` in UTC actually).
	start := time.Date(
		now.Year(), now.Month(), now.Day(),
		now.Hour(), now.Minute(), now.Second(),
		now.Nanosecond(), time.UTC,
	)

	exprCtx := exprstatic.NewExprContext()
	// we need to set the location to UTC to make sure the time is in the same timezone as the start time.
	intest.Assert(exprCtx.GetEvalCtx().Location() == time.UTC)
	expr, err := expression.ParseSimpleExpr(
		exprCtx,
		fmt.Sprintf("FROM_UNIXTIME(0) + INTERVAL %d MICROSECOND - INTERVAL %s %s",
			start.UnixMicro(), interval, unit.String(),
		),
	)
	if err != nil {
		return time.Time{}, err
	}

	tm, _, err := expr.EvalTime(exprCtx.GetEvalCtx(), chunk.Row{})
	if err != nil {
		return time.Time{}, err
	}

	end, err := tm.GoTime(time.UTC)
	if err != nil {
		return time.Time{}, err
	}

	// Then we should add the duration between the time get from the previous SQL and the start time to the now time.
	expiredTime := now.
		Add(end.Sub(start)).
		// Truncate to second to make sure the precision is always the same with the one stored in a table to avoid some
		// comparing problems in testing.
		Truncate(time.Second)

	return expiredTime, nil
}

// EvalExpireTime returns the expired time for the current time.
// It uses the global timezone in session to evaluation the context
// and the return time is in the same timezone of now argument.
func (t *PhysicalTable) EvalExpireTime(ctx context.Context, se session.Session,
	now time.Time) (time.Time, error) {
	if intest.InTest {
		if tm, ok := ctx.Value(mockExpireTimeKey{}).(time.Time); ok {
			return tm, nil
		}
	}

	// Use the global time zone to compute expire time.
	// Different timezones may have different results event with the same "now" time and TTL expression.
	// Consider a TTL setting with the expiration `INTERVAL 1 MONTH`.
	// If the current timezone is `Asia/Shanghai` and now is `2021-03-01 00:00:00 +0800`
	// the expired time should be `2021-02-01 00:00:00 +0800`, corresponding to UTC time `2021-01-31 16:00:00 UTC`.
	// But if we use the `UTC` time zone, the current time is `2021-02-28 16:00:00 UTC`,
	// and the expired time should be `2021-01-28 16:00:00 UTC` that is not the same the previous one.
	globalTz, err := se.GlobalTimeZone(ctx)
	if err != nil {
		return time.Time{}, err
	}

	start := now.In(globalTz)
	expire, err := EvalExpireTime(start, t.TTLInfo.IntervalExprStr, ast.TimeUnitType(t.TTLInfo.IntervalTimeUnit))
	if err != nil {
		return time.Time{}, err
	}

	return expire.In(now.Location()), nil
}

// SplitScanRanges split ranges for TTL scan
func (t *PhysicalTable) SplitScanRanges(ctx context.Context, store kv.Storage, splitCnt int) ([]ScanRange, error) {
	if len(t.KeyColumns) < 1 || splitCnt <= 1 {
		return []ScanRange{newFullRange()}, nil
	}

	tikvStore, ok := store.(tikv.Storage)
	if !ok {
		return []ScanRange{newFullRange()}, nil
	}

	ft := t.KeyColumns[0].FieldType
	switch ft.GetType() {
	case mysql.TypeTiny, mysql.TypeShort, mysql.TypeLong, mysql.TypeLonglong, mysql.TypeInt24:
		if len(t.KeyColumns) > 1 {
			return t.splitCommonHandleRanges(ctx, tikvStore, splitCnt, true, mysql.HasUnsignedFlag(ft.GetFlag()), nil)
		}
		return t.splitIntRanges(ctx, tikvStore, splitCnt)
	case mysql.TypeBit:
		return t.splitCommonHandleRanges(ctx, tikvStore, splitCnt, false, false, nil)
	case mysql.TypeString, mysql.TypeVarString, mysql.TypeVarchar:
		var decode func([]byte) types.Datum
		if !mysql.HasBinaryFlag(ft.GetFlag()) {
			switch ft.GetCharset() {
			case charset.CharsetASCII, charset.CharsetLatin1:
				// ASCII and Latin1 are 8-bit charset, we can use GetASCIIPrefixDatumFromBytes to decode it.
				decode = GetASCIIPrefixDatumFromBytes
			case charset.CharsetUTF8, charset.CharsetUTF8MB4:
				switch ft.GetCollate() {
				case charset.CollationUTF8, charset.CollationUTF8MB4, "utf8mb4_0900_bin":
					// We can only use GetASCIIPrefixDatumFromBytes to decode UTF8 and UTF8MB4 when they are
					// "utf8_bin" or "utf8mb4_bin" collation.
					decode = GetASCIIPrefixDatumFromBytes
				}
			}
			if decode == nil {
				return []ScanRange{newFullRange()}, nil
			}
		}
		return t.splitCommonHandleRanges(ctx, tikvStore, splitCnt, false, false, decode)
	}
	return []ScanRange{newFullRange()}, nil
}

func unsignedEdge(d types.Datum) types.Datum {
	if d.IsNull() {
		return types.NewUintDatum(uint64(math.MaxInt64 + 1))
	}
	if d.GetInt64() == 0 {
		return nullDatum()
	}
	return types.NewUintDatum(uint64(d.GetInt64()))
}

func (t *PhysicalTable) splitIntRanges(ctx context.Context, store tikv.Storage, splitCnt int) ([]ScanRange, error) {
	recordPrefix := tablecodec.GenTableRecordPrefix(t.ID)
	startKey, endKey := tablecodec.GetTableHandleKeyRange(t.ID)
	keyRanges, err := t.splitRawKeyRanges(ctx, store, startKey, endKey, splitCnt)
	if err != nil {
		return nil, err
	}

	if len(keyRanges) <= 1 {
		return []ScanRange{newFullRange()}, nil
	}

	ft := t.KeyColumnTypes[0]
	unsigned := mysql.HasUnsignedFlag(ft.GetFlag())
	scanRanges := make([]ScanRange, 0, len(keyRanges)+1)
	curScanStart := nullDatum()
	for i, keyRange := range keyRanges {
		if i != 0 && curScanStart.IsNull() {
			break
		}

		curScanEnd := nullDatum()
		if i < len(keyRanges)-1 {
			if val := GetNextIntHandle(keyRange.EndKey, recordPrefix); val != nil {
				curScanEnd = types.NewIntDatum(val.IntValue())
			}
		}

		if !curScanStart.IsNull() && !curScanEnd.IsNull() && curScanStart.GetInt64() >= curScanEnd.GetInt64() {
			continue
		}

		if !unsigned {
			// primary key is signed or range
			scanRanges = append(scanRanges, newDatumRange(curScanStart, curScanEnd))
		} else if !curScanStart.IsNull() && curScanStart.GetInt64() >= 0 {
			// primary key is unsigned and range is in the right half side
			scanRanges = append(scanRanges, newDatumRange(unsignedEdge(curScanStart), unsignedEdge(curScanEnd)))
		} else if !curScanEnd.IsNull() && curScanEnd.GetInt64() <= 0 {
			// primary key is unsigned and range is in the left half side
			scanRanges = append(scanRanges, newDatumRange(unsignedEdge(curScanStart), unsignedEdge(curScanEnd)))
		} else {
			// primary key is unsigned and the start > math.MaxInt64 && end < math.MaxInt64
			// we must split it to two ranges
			scanRanges = append(scanRanges,
				newDatumRange(unsignedEdge(curScanStart), nullDatum()),
				newDatumRange(nullDatum(), unsignedEdge(curScanEnd)),
			)
		}
		curScanStart = curScanEnd
	}
	return scanRanges, nil
}

func (t *PhysicalTable) splitCommonHandleRanges(
	ctx context.Context, store tikv.Storage, splitCnt int, isInt bool, unsigned bool, decode func([]byte) types.Datum,
) ([]ScanRange, error) {
	recordPrefix := tablecodec.GenTableRecordPrefix(t.ID)
	startKey, endKey := recordPrefix, recordPrefix.PrefixNext()
	keyRanges, err := t.splitRawKeyRanges(ctx, store, startKey, endKey, splitCnt)
	if err != nil {
		return nil, err
	}

	if len(keyRanges) <= 1 {
		return []ScanRange{newFullRange()}, nil
	}

	scanRanges := make([]ScanRange, 0, len(keyRanges))
	curScanStart := nullDatum()
	for i, keyRange := range keyRanges {
		curScanEnd := nullDatum()
		if i != len(keyRanges)-1 {
			if isInt {
				curScanEnd = GetNextIntDatumFromCommonHandle(keyRange.EndKey, recordPrefix, unsigned)
			} else {
				curScanEnd = GetNextBytesHandleDatum(keyRange.EndKey, recordPrefix)
				if decode != nil {
					curScanEnd = decode(curScanEnd.GetBytes())
				}

				// "" is the smallest value for string/[]byte, skip to add it to ranges.
				if len(curScanEnd.GetBytes()) == 0 {
					continue
				}
			}
		}

		if !curScanStart.IsNull() && !curScanEnd.IsNull() {
			// Sometimes curScanStart >= curScanEnd because the edge datum is an approximate value.
			// At this time, we should skip this range to ensure the incremental of ranges.
			cmp, err := curScanStart.Compare(types.StrictContext, &curScanEnd, collate.GetBinaryCollator())
			intest.AssertNoError(err)
			if err != nil {
				return nil, err
			}

			if cmp >= 0 {
				continue
			}
		}

		scanRanges = append(scanRanges, newDatumRange(curScanStart, curScanEnd))
		if curScanEnd.IsNull() {
			break
		}
		curScanStart = curScanEnd
	}
	return scanRanges, nil
}

func (t *PhysicalTable) splitRawKeyRanges(ctx context.Context, store tikv.Storage,
	startKey, endKey kv.Key, splitCnt int) ([]kv.KeyRange, error) {
	maxSleep := 20000
	if intest.InTest {
		maxSleep = 500 // reduce the max sleep time in test
	}

	regionCache := store.GetRegionCache()
	regions, err := regionCache.LocateKeyRange(
		tikv.NewBackofferWithVars(ctx, maxSleep, nil), startKey, endKey)
	if err != nil {
		return nil, err
	}

	regionsCnt := len(regions)
	regionsPerRange := regionsCnt / splitCnt
	oversizeCnt := regionsCnt % splitCnt
	ranges := make([]kv.KeyRange, 0, min(regionsCnt, splitCnt))
	for len(regions) > 0 {
		startRegion := regions[0]

		endRegionIdx := regionsPerRange - 1
		if oversizeCnt > 0 {
			endRegionIdx++
		}

		endRegion := regions[endRegionIdx]

		rangeStartKey := kv.Key(startRegion.StartKey)
		if rangeStartKey.Cmp(startKey) < 0 {
			rangeStartKey = startKey
		}

		rangeEndKey := kv.Key(endRegion.EndKey)
		if rangeEndKey.Cmp(endKey) > 0 {
			rangeEndKey = endKey
		}

		ranges = append(ranges, kv.KeyRange{StartKey: rangeStartKey, EndKey: rangeEndKey})
		oversizeCnt--
		regions = regions[endRegionIdx+1:]
	}
	logutil.BgLogger().Info("TTL table raw key ranges split",
		zap.Int("regionsCnt", regionsCnt),
		zap.Int("shouldSplitCnt", splitCnt),
		zap.Int("actualSplitCnt", len(ranges)),
		zap.Int64("tableID", t.ID),
		zap.String("db", t.Schema.O),
		zap.String("table", t.Name.O),
		zap.String("partition", t.Partition.O),
	)
	return ranges, nil
}

var commonHandleBytesByte byte

var commonHandleIntByte byte

var commonHandleUintByte byte

func init() {
	key, err := codec.EncodeKey(time.UTC, nil, types.NewBytesDatum(nil))
	terror.MustNil(err)
	commonHandleBytesByte = key[0]

	key, err = codec.EncodeKey(time.UTC, nil, types.NewIntDatum(0))
	terror.MustNil(err)
	commonHandleIntByte = key[0]

	key, err = codec.EncodeKey(time.UTC, nil, types.NewUintDatum(0))
	terror.MustNil(err)
	commonHandleUintByte = key[0]
}

// GetNextIntHandle is used for int handle tables.
// It returns the min handle whose encoded key is or after argument `key`
// If it cannot find a valid value, a null datum will be returned.
func GetNextIntHandle(key kv.Key, recordPrefix []byte) kv.Handle {
	if key.Cmp(recordPrefix) > 0 && !key.HasPrefix(recordPrefix) {
		return nil
	}

	if key.Cmp(recordPrefix) <= 0 {
		return kv.IntHandle(math.MinInt64)
	}

	suffix := key[len(recordPrefix):]
	encodedVal := suffix
	if len(suffix) < 8 {
		encodedVal = make([]byte, 8)
		copy(encodedVal, suffix)
	}

	findNext := false
	if len(suffix) > 8 {
		findNext = true
		encodedVal = encodedVal[:8]
	}

	u := codec.DecodeCmpUintToInt(binary.BigEndian.Uint64(encodedVal))
	if !findNext {
		return kv.IntHandle(u)
	}

	if u == math.MaxInt64 {
		return nil
	}

	return kv.IntHandle(u + 1)
}

// GetNextIntDatumFromCommonHandle is used for common handle tables with int value.
// It returns the min handle whose encoded key is or after argument `key`
// If it cannot find a valid value, a null datum will be returned.
func GetNextIntDatumFromCommonHandle(key kv.Key, recordPrefix []byte, unsigned bool) (d types.Datum) {
	if key.Cmp(recordPrefix) > 0 && !key.HasPrefix(recordPrefix) {
		d.SetNull()
		return d
	}

	typeByte := commonHandleIntByte
	if unsigned {
		typeByte = commonHandleUintByte
	}

	var minDatum types.Datum
	if unsigned {
		minDatum.SetUint64(0)
	} else {
		minDatum.SetInt64(math.MinInt64)
	}

	if key.Cmp(recordPrefix) <= 0 {
		d = minDatum
		return d
	}

	encodedVal := key[len(recordPrefix):]
	if encodedVal[0] < typeByte {
		d = minDatum
		return d
	}

	if encodedVal[0] > typeByte {
		d.SetNull()
		return d
	}

	if len(encodedVal) < 9 {
		newVal := make([]byte, 9)
		copy(newVal, encodedVal)
		encodedVal = newVal
	}

	_, v, err := codec.DecodeOne(encodedVal)
	intest.AssertNoError(err)
	if err != nil {
		// should never happen
		terror.Log(errors.Annotatef(err, "TTL decode common handle failed, key: %s", hex.EncodeToString(key)))
		return nullDatum()
	}

	if len(encodedVal) > 9 {
		if (unsigned && v.GetUint64() == math.MaxUint64) || (!unsigned && v.GetInt64() == math.MaxInt64) {
			d.SetNull()
			return d
		}

		if unsigned {
			v.SetUint64(v.GetUint64() + 1)
		} else {
			v.SetInt64(v.GetInt64() + 1)
		}
	}

	return v
}

// GetNextBytesHandleDatum is used for a table with one binary or string column common handle.
// It returns the minValue whose encoded key is or after argument `key`
// If it cannot find a valid value, a null datum will be returned.
func GetNextBytesHandleDatum(key kv.Key, recordPrefix []byte) (d types.Datum) {
	if key.Cmp(recordPrefix) > 0 && !key.HasPrefix(recordPrefix) {
		d.SetNull()
		return d
	}

	if key.Cmp(recordPrefix) <= 0 {
		d.SetBytes([]byte{})
		return d
	}

	encodedVal := key[len(recordPrefix):]
	if encodedVal[0] < commonHandleBytesByte {
		d.SetBytes([]byte{})
		return d
	}

	if encodedVal[0] > commonHandleBytesByte {
		d.SetNull()
		return d
	}

	if remain, v, err := codec.DecodeOne(encodedVal); err == nil {
		if len(remain) > 0 {
			v.SetBytes(kv.Key(v.GetBytes()).Next())
		}
		return v
	}

	encodedVal = encodedVal[1:]
	brokenGroupEndIdx := len(encodedVal) - 1
	brokenGroupEmptyBytes := len(encodedVal) % 9
	for i := 7; i+1 < len(encodedVal); i += 9 {
		if emptyBytes := 255 - int(encodedVal[i+1]); emptyBytes != 0 || i+1 == len(encodedVal)-1 {
			brokenGroupEndIdx = i
			brokenGroupEmptyBytes = emptyBytes
			break
		}
	}

	for i := 0; i < brokenGroupEmptyBytes; i++ {
		if encodedVal[brokenGroupEndIdx] > 0 {
			break
		}
		brokenGroupEndIdx--
	}

	if brokenGroupEndIdx < 0 {
		d.SetBytes(nil)
		return d
	}

	val := make([]byte, 0, len(encodedVal))
	for i := 0; i <= brokenGroupEndIdx; i++ {
		if i%9 == 8 {
			continue
		}
		val = append(val, encodedVal[i])
	}
	d.SetBytes(val)
	return d
}

// GetASCIIPrefixDatumFromBytes is used to convert bytes to string datum which only contains ASCII prefix string.
// The ASCII prefix string only contains visible characters and `\t`, `\n`, `\r`.
// "abc" -> "abc"
// "\0abc" -> ""
// "ab\x01c" -> "ab"
// "ab\xffc" -> "ab"
// "ab\rc\xff" -> "ab\rc"
func GetASCIIPrefixDatumFromBytes(bs []byte) types.Datum {
	for i, c := range bs {
		if c >= 0x20 && c <= 0x7E {
			// visible characters from ` ` to `~`
			continue
		}

		if c == '\t' || c == '\n' || c == '\r' {
			continue
		}

		bs = bs[:i]
		break
	}
	return types.NewStringDatum(string(bs))
}