You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
535 lines
26 KiB
535 lines
26 KiB
// Copyright 2019 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,
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package expression
|
|
|
|
import (
|
|
"math/rand"
|
|
"testing"
|
|
|
|
. "github.com/pingcap/check"
|
|
"github.com/pingcap/parser/ast"
|
|
"github.com/pingcap/parser/charset"
|
|
"github.com/pingcap/parser/mysql"
|
|
"github.com/pingcap/tidb/types"
|
|
)
|
|
|
|
type randSpaceStrGener struct {
|
|
lenBegin int
|
|
lenEnd int
|
|
}
|
|
|
|
func (g *randSpaceStrGener) gen() interface{} {
|
|
n := rand.Intn(g.lenEnd-g.lenBegin) + g.lenBegin
|
|
buf := make([]byte, n)
|
|
for i := range buf {
|
|
x := rand.Intn(150)
|
|
if x < 10 {
|
|
buf[i] = byte('0' + x)
|
|
} else if x-10 < 26 {
|
|
buf[i] = byte('a' + x - 10)
|
|
} else if x < 62 {
|
|
buf[i] = byte('A' + x - 10 - 26)
|
|
} else {
|
|
buf[i] = byte(' ')
|
|
}
|
|
}
|
|
return string(buf)
|
|
}
|
|
|
|
var vecBuiltinStringCases = map[string][]vecExprBenchCase{
|
|
ast.Length: {
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{newDefaultGener(0.2, types.ETString)}},
|
|
},
|
|
ast.ASCII: {
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{newDefaultGener(0.2, types.ETString)}},
|
|
},
|
|
ast.Concat: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETString}},
|
|
},
|
|
ast.ConcatWS: {
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETString, types.ETString},
|
|
geners: []dataGenerator{&constStrGener{","}},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETString, types.ETString},
|
|
geners: []dataGenerator{newDefaultGener(1, types.ETString)},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETString},
|
|
geners: []dataGenerator{
|
|
&constStrGener{"<------------------>"},
|
|
&constStrGener{"1413006"},
|
|
&constStrGener{"idlfmv"},
|
|
},
|
|
},
|
|
},
|
|
ast.Convert: {
|
|
{
|
|
retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
constants: []*Constant{nil, {Value: types.NewDatum("utf8"), RetType: types.NewFieldType(mysql.TypeString)}},
|
|
},
|
|
{
|
|
retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
constants: []*Constant{nil, {Value: types.NewDatum("binary"), RetType: types.NewFieldType(mysql.TypeString)}},
|
|
},
|
|
{
|
|
retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
constants: []*Constant{nil, {Value: types.NewDatum("utf8mb4"), RetType: types.NewFieldType(mysql.TypeString)}},
|
|
},
|
|
{
|
|
retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
constants: []*Constant{nil, {Value: types.NewDatum("ascii"), RetType: types.NewFieldType(mysql.TypeString)}},
|
|
},
|
|
{
|
|
retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
constants: []*Constant{nil, {Value: types.NewDatum("latin1"), RetType: types.NewFieldType(mysql.TypeString)}},
|
|
},
|
|
},
|
|
ast.Substring: {
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETInt},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 20), newRangeInt64Gener(-25, 25)},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETInt, types.ETInt},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 20), newRangeInt64Gener(-25, 25), newRangeInt64Gener(-25, 25)},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETInt, types.ETInt},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin},
|
|
{Tp: mysql.TypeLonglong}, {Tp: mysql.TypeLonglong}},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 20), newRangeInt64Gener(-25, 25), newRangeInt64Gener(-25, 25)},
|
|
},
|
|
},
|
|
ast.SubstringIndex: {
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETInt},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 20), newRandLenStrGener(0, 2), newRangeInt64Gener(-4, 4)},
|
|
},
|
|
},
|
|
ast.Locate: {
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 10), newRandLenStrGener(0, 20)},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
geners: []dataGenerator{newRandLenStrGener(1, 2), newRandLenStrGener(0, 20)},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
geners: []dataGenerator{newSelectStringGener([]string{"01", "10", "001", "110", "0001", "1110"}), newSelectStringGener([]string{"010010001000010", "101101110111101"})},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETInt},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 10), newRandLenStrGener(0, 20), newRangeInt64Gener(-10, 20)},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETInt},
|
|
geners: []dataGenerator{newRandLenStrGener(1, 2), newRandLenStrGener(0, 10), newRangeInt64Gener(0, 8)},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
geners: []dataGenerator{newSelectStringGener([]string{"01", "10", "001", "110", "0001", "1110"}), newSelectStringGener([]string{"010010001000010", "101101110111101"})},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{nil, {Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 10), newRandLenStrGener(0, 20)},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{nil, {Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}},
|
|
geners: []dataGenerator{newRandLenStrGener(1, 2), newRandLenStrGener(0, 20)},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{nil, {Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}},
|
|
geners: []dataGenerator{newSelectStringGener([]string{"01", "10", "001", "110", "0001", "1110"}), newSelectStringGener([]string{"010010001000010", "101101110111101"})},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}, nil},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 10), newRandLenStrGener(0, 20)},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}, nil},
|
|
geners: []dataGenerator{newRandLenStrGener(1, 2), newRandLenStrGener(0, 20)},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}, nil},
|
|
geners: []dataGenerator{newSelectStringGener([]string{"01", "10", "001", "110", "0001", "1110"}), newSelectStringGener([]string{"010010001000010", "101101110111101"})},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}, {Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 10), newRandLenStrGener(0, 20)},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}, {Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}},
|
|
geners: []dataGenerator{newRandLenStrGener(1, 2), newRandLenStrGener(0, 20)},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}, {Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}},
|
|
geners: []dataGenerator{newSelectStringGener([]string{"01", "10", "001", "110", "0001", "1110"}), newSelectStringGener([]string{"010010001000010", "101101110111101"})},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETInt},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}, {Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}, {Tp: mysql.TypeInt24}},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 10), newRandLenStrGener(0, 20), newRangeInt64Gener(-10, 20)},
|
|
},
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETInt},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}, {Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}, {Tp: mysql.TypeInt24}},
|
|
geners: []dataGenerator{newSelectStringGener([]string{"01", "10", "001", "110", "0001", "1110"}), newSelectStringGener([]string{"010010001000010", "101101110111101"}), newRangeInt64Gener(-10, 20)},
|
|
},
|
|
},
|
|
ast.Hex: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{newRandHexStrGener(10, 100)}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETInt}},
|
|
},
|
|
ast.Unhex: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{newRandHexStrGener(10, 100)}},
|
|
},
|
|
ast.Trim: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{&randSpaceStrGener{10, 100}}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETString}, geners: []dataGenerator{newRandLenStrGener(10, 20), newRandLenStrGener(5, 25)}},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETInt},
|
|
geners: []dataGenerator{newRandLenStrGener(10, 20), newRandLenStrGener(5, 25), nil},
|
|
constants: []*Constant{nil, nil, {Value: types.NewDatum(ast.TrimBoth), RetType: types.NewFieldType(mysql.TypeLonglong)}},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETInt},
|
|
geners: []dataGenerator{newRandLenStrGener(10, 20), newRandLenStrGener(5, 25), nil},
|
|
constants: []*Constant{nil, nil, {Value: types.NewDatum(ast.TrimLeading), RetType: types.NewFieldType(mysql.TypeLonglong)}},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETInt},
|
|
geners: []dataGenerator{newRandLenStrGener(10, 20), newRandLenStrGener(5, 25), nil},
|
|
constants: []*Constant{nil, nil, {Value: types.NewDatum(ast.TrimTrailing), RetType: types.NewFieldType(mysql.TypeLonglong)}},
|
|
},
|
|
},
|
|
ast.LTrim: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{&randSpaceStrGener{10, 100}}},
|
|
},
|
|
ast.RTrim: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{&randSpaceStrGener{10, 100}}},
|
|
},
|
|
ast.Lpad: {
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETInt, types.ETString},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 20), newRangeInt64Gener(168435456, 368435456), newRandLenStrGener(0, 10)},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETInt, types.ETString},
|
|
geners: []dataGenerator{newDefaultGener(0.2, types.ETString), newDefaultGener(0.2, types.ETInt), newDefaultGener(0.2, types.ETString)},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETInt, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 20), newRangeInt64Gener(168435456, 368435456), newRandLenStrGener(0, 10)},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETInt, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}},
|
|
geners: []dataGenerator{newDefaultGener(0.2, types.ETString), newDefaultGener(0.2, types.ETInt), newDefaultGener(0.2, types.ETString)},
|
|
},
|
|
},
|
|
ast.Rpad: {
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETInt, types.ETString},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 20), newRangeInt64Gener(168435456, 368435456), newRandLenStrGener(0, 10)},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETInt, types.ETString},
|
|
geners: []dataGenerator{newDefaultGener(0.2, types.ETString), newDefaultGener(0.2, types.ETInt), newDefaultGener(0.2, types.ETString)},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETInt, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}},
|
|
geners: []dataGenerator{newRandLenStrGener(0, 20), newRangeInt64Gener(168435456, 368435456), newRandLenStrGener(0, 10)},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETString, types.ETInt, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}},
|
|
geners: []dataGenerator{newDefaultGener(0.2, types.ETString), newDefaultGener(0.2, types.ETInt), newDefaultGener(0.2, types.ETString)},
|
|
},
|
|
},
|
|
ast.CharLength: {
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString}},
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}},
|
|
},
|
|
},
|
|
ast.BitLength: {
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString}},
|
|
},
|
|
ast.CharFunc: {
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETInt, types.ETInt, types.ETInt, types.ETString},
|
|
geners: []dataGenerator{&charInt64Gener{}, &charInt64Gener{}, &charInt64Gener{}, nil},
|
|
constants: []*Constant{nil, nil, nil, {Value: types.NewDatum("ascii"), RetType: types.NewFieldType(mysql.TypeString)}},
|
|
},
|
|
},
|
|
ast.FindInSet: {
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString, types.ETString}},
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString, types.ETString}, geners: []dataGenerator{&constStrGener{"case"}, &constStrGener{"test,case"}}},
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString, types.ETString}, geners: []dataGenerator{&constStrGener{""}, &constStrGener{"test,case"}}},
|
|
},
|
|
ast.MakeSet: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETInt, types.ETString, types.ETString, types.ETString, types.ETString, types.ETString, types.ETString, types.ETString, types.ETString}},
|
|
},
|
|
ast.Oct: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETInt}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{&numStrGener{*newRangeInt64Gener(-10, 10)}}},
|
|
},
|
|
ast.Quote: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}},
|
|
},
|
|
ast.Ord: {
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString}},
|
|
},
|
|
ast.Bin: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETInt}},
|
|
},
|
|
ast.ToBase64: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{newRandLenStrGener(0, 10)}},
|
|
},
|
|
ast.FromBase64: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{newRandLenStrGener(10, 100)}},
|
|
},
|
|
ast.ExportSet: {
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETInt, types.ETString, types.ETString},
|
|
geners: []dataGenerator{newRangeInt64Gener(10, 100), &constStrGener{"Y"}, &constStrGener{"N"}},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETInt, types.ETString, types.ETString, types.ETString},
|
|
geners: []dataGenerator{newRangeInt64Gener(10, 100), &constStrGener{"Y"}, &constStrGener{"N"}, &constStrGener{","}},
|
|
},
|
|
{
|
|
retEvalType: types.ETString,
|
|
childrenTypes: []types.EvalType{types.ETInt, types.ETString, types.ETString, types.ETString, types.ETInt},
|
|
geners: []dataGenerator{newRangeInt64Gener(10, 100), &constStrGener{"Y"}, &constStrGener{"N"}, &constStrGener{","}, newRangeInt64Gener(-10, 70)},
|
|
},
|
|
},
|
|
ast.Repeat: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETInt}, geners: []dataGenerator{newRandLenStrGener(10, 20), newRangeInt64Gener(-10, 10)}},
|
|
},
|
|
ast.Lower: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{newSelectStringGener([]string{"one week’s time TEST", "one week's time TEST", "ABC测试DEF", "ABCテストABC"})}},
|
|
},
|
|
ast.IsNull: {
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{newRandLenStrGener(10, 20)}},
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{newDefaultGener(0.2, types.ETString)}},
|
|
},
|
|
ast.Upper: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{newSelectStringGener([]string{"one week’s time TEST", "one week's time TEST", "abc测试DeF", "AbCテストAbC"})}},
|
|
},
|
|
ast.Right: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETInt}},
|
|
// need to add BinaryFlag for the Binary func
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETInt},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin},
|
|
{Tp: mysql.TypeLonglong}},
|
|
geners: []dataGenerator{
|
|
newRandLenStrGener(10, 20),
|
|
newRangeInt64Gener(-10, 20),
|
|
},
|
|
},
|
|
},
|
|
ast.Left: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETInt}},
|
|
// need to add BinaryFlag for the Binary func
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETInt},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin},
|
|
{Tp: mysql.TypeLonglong}},
|
|
geners: []dataGenerator{
|
|
newRandLenStrGener(10, 20),
|
|
newRangeInt64Gener(-10, 20),
|
|
},
|
|
},
|
|
},
|
|
ast.Space: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETInt}, geners: []dataGenerator{newRangeInt64Gener(-10, 2000)}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETInt}, geners: []dataGenerator{newRangeInt64Gener(5, 10)}},
|
|
},
|
|
ast.Reverse: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{newRandLenStrGener(10, 20)}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString}, geners: []dataGenerator{newDefaultGener(0.2, types.ETString)}},
|
|
// need to add BinaryFlag for the Binary func
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin}},
|
|
},
|
|
},
|
|
ast.Instr: {
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString, types.ETString}},
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString, types.ETString}, geners: []dataGenerator{&constStrGener{"test,case"}, &constStrGener{"case"}}},
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString, types.ETString}, geners: []dataGenerator{&constStrGener{"test,case"}, &constStrGener{"testcase"}}},
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{
|
|
{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin},
|
|
{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin},
|
|
},
|
|
},
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{
|
|
{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin},
|
|
{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin},
|
|
},
|
|
geners: []dataGenerator{&constStrGener{"test,case"}, &constStrGener{"case"}},
|
|
},
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{
|
|
{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin},
|
|
{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin},
|
|
},
|
|
geners: []dataGenerator{&constStrGener{"test,case"}, &constStrGener{""}},
|
|
},
|
|
},
|
|
ast.Replace: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETString, types.ETString}, geners: []dataGenerator{newRandLenStrGener(10, 20), newRandLenStrGener(0, 10), newRandLenStrGener(0, 10)}},
|
|
},
|
|
ast.InsertFunc: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETInt, types.ETInt, types.ETString}, geners: []dataGenerator{newRandLenStrGener(10, 20), newRangeInt64Gener(-10, 20), newRangeInt64Gener(0, 100), newRandLenStrGener(0, 10)}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETInt, types.ETInt, types.ETString},
|
|
childrenFieldTypes: []*types.FieldType{
|
|
{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin},
|
|
{Tp: mysql.TypeLonglong},
|
|
{Tp: mysql.TypeLonglong},
|
|
{Tp: mysql.TypeString, Flag: mysql.BinaryFlag, Collate: charset.CollationBin},
|
|
},
|
|
},
|
|
},
|
|
ast.Elt: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETInt, types.ETString, types.ETString, types.ETString}, geners: []dataGenerator{newRangeInt64Gener(-1, 5)}},
|
|
},
|
|
ast.FromUnixTime: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETDecimal, types.ETString},
|
|
geners: []dataGenerator{
|
|
gener{*newDefaultGener(0.9, types.ETDecimal)},
|
|
&constStrGener{"%y-%m-%d"},
|
|
},
|
|
},
|
|
},
|
|
ast.Strcmp: {
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString, types.ETString}},
|
|
{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString, types.ETString}, geners: []dataGenerator{
|
|
newSelectStringGener(
|
|
[]string{
|
|
"test",
|
|
},
|
|
),
|
|
newSelectStringGener(
|
|
[]string{
|
|
"test",
|
|
},
|
|
),
|
|
}},
|
|
},
|
|
ast.Format: {
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETDecimal, types.ETInt}, geners: []dataGenerator{
|
|
newRangeDecimalGener(-10000, 10000, 0),
|
|
newRangeInt64Gener(-10, 40),
|
|
}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETReal, types.ETInt}, geners: []dataGenerator{
|
|
newRangeRealGener(-10000, 10000, 0),
|
|
newRangeInt64Gener(-10, 40),
|
|
}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETDecimal, types.ETInt}, geners: []dataGenerator{
|
|
newRangeDecimalGener(-10000, 10000, 1),
|
|
newRangeInt64Gener(-10, 40),
|
|
}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETReal, types.ETInt}, geners: []dataGenerator{
|
|
newRangeRealGener(-10000, 10000, 1),
|
|
newRangeInt64Gener(-10, 40),
|
|
}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETString, types.ETString}, geners: []dataGenerator{
|
|
newRealStringGener(),
|
|
&numStrGener{*newRangeInt64Gener(-10, 40)},
|
|
}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETDecimal, types.ETInt, types.ETString}, geners: []dataGenerator{
|
|
newRangeDecimalGener(-10000, 10000, 0.5),
|
|
newRangeInt64Gener(-10, 40),
|
|
newNullWrappedGener(0.1, &constStrGener{"en_US"}),
|
|
}},
|
|
{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETReal, types.ETInt, types.ETString}, geners: []dataGenerator{
|
|
newRangeRealGener(-10000, 10000, 0.5),
|
|
newRangeInt64Gener(-10, 40),
|
|
newNullWrappedGener(0.1, &constStrGener{"en_US"}),
|
|
}},
|
|
},
|
|
}
|
|
|
|
func (s *testVectorizeSuite1) TestVectorizedBuiltinStringEvalOneVec(c *C) {
|
|
testVectorizedEvalOneVec(c, vecBuiltinStringCases)
|
|
}
|
|
|
|
func (s *testVectorizeSuite1) TestVectorizedBuiltinStringFunc(c *C) {
|
|
testVectorizedBuiltinFunc(c, vecBuiltinStringCases)
|
|
}
|
|
|
|
func BenchmarkVectorizedBuiltinStringEvalOneVec(b *testing.B) {
|
|
benchmarkVectorizedEvalOneVec(b, vecBuiltinStringCases)
|
|
}
|
|
|
|
func BenchmarkVectorizedBuiltinStringFunc(b *testing.B) {
|
|
benchmarkVectorizedBuiltinFunc(b, vecBuiltinStringCases)
|
|
}
|
|
|