Function getKeysNeedCheckOneRow

pkg/executor/batch_checker.go:95–238

(ctx sessionctx.Context, t table.Table, row []types.Datum, nUnique int, handleCols []*table.Column,
	pkIdxInfo *model.IndexInfo, result []toBeCheckedRow)

}

func getKeysNeedCheckOneRow(ctx sessionctx.Context, t table.Table, row []types.Datum, nUnique int, handleCols []*table.Column,
	pkIdxInfo *model.IndexInfo, result []toBeCheckedRow) ([]toBeCheckedRow, error) {
	var err error
	if p, ok := t.(table.PartitionedTable); ok {
		t, err = p.GetPartitionByRow(ctx.GetExprCtx().GetEvalCtx(), row)
		if err != nil {
			if terr, ok := errors.Cause(err).(*terror.Error); ok && (terr.Code() == errno.ErrNoPartitionForGivenValue || terr.Code() == errno.ErrRowDoesNotMatchGivenPartitionSet) {
				ec := ctx.GetSessionVars().StmtCtx.ErrCtx()
				if err = ec.HandleError(terr); err != nil {
					return nil, err
				}
				result = append(result, toBeCheckedRow{ignored: true})
				return result, nil
			}
			return nil, err
		}
	}

	uniqueKeys := make([]*keyValueWithDupInfo, 0, nUnique)
	// Append record keys and errors.
	var handle kv.Handle
	if t.Meta().IsCommonHandle {
		var err error
		handle, err = buildHandleFromDatumRow(ctx.GetSessionVars().StmtCtx, row, handleCols, pkIdxInfo)
		if err != nil {
			return nil, err
		}
	} else if len(handleCols) > 0 {
		handle = kv.IntHandle(row[handleCols[0].Offset].GetInt64())
	}
	var handleKey *keyValueWithDupInfo
	if handle != nil {
		handleKey = &keyValueWithDupInfo{
			newKey: tablecodec.EncodeRecordKey(t.RecordPrefix(), handle),
		}

		var keyCols []string
		var err error
		if t.Meta().IsCommonHandle {
			data := make([]types.Datum, len(handleCols))
			for i, col := range handleCols {
				data[i] = row[col.Offset]
			}
			keyCols, err = dataToStrings(data)
		} else {
			var s string
			s, err = row[handleCols[0].Offset].ToString()
			keyCols = []string{s}
		}
		if err != nil {
			var handleData []types.Datum
			handleData, err = handle.Data()
			if err == nil {
				keyCols, err = dataToStrings(handleData)
			}
		}

		if err == nil {