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.
299 lines
8.3 KiB
299 lines
8.3 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.
|
|
|
|
// Code generated by go generate in expression/generator; DO NOT EDIT.
|
|
|
|
package expression
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"testing"
|
|
"time"
|
|
|
|
. "github.com/pingcap/check"
|
|
"github.com/pingcap/parser/ast"
|
|
"github.com/pingcap/parser/mysql"
|
|
"github.com/pingcap/tidb/types"
|
|
"github.com/pingcap/tidb/types/json"
|
|
)
|
|
|
|
type inGener struct {
|
|
defaultGener
|
|
}
|
|
|
|
func (g inGener) gen() interface{} {
|
|
if rand.Float64() < g.nullRation {
|
|
return nil
|
|
}
|
|
randNum := rand.Int63n(10)
|
|
switch g.eType {
|
|
case types.ETInt:
|
|
if rand.Float64() < 0.5 {
|
|
return -randNum
|
|
}
|
|
return randNum
|
|
case types.ETReal:
|
|
if rand.Float64() < 0.5 {
|
|
return -float64(randNum)
|
|
}
|
|
return float64(randNum)
|
|
case types.ETDecimal:
|
|
d := new(types.MyDecimal)
|
|
f := float64(randNum * 100000)
|
|
if err := d.FromFloat64(f); err != nil {
|
|
panic(err)
|
|
}
|
|
return d
|
|
case types.ETDatetime, types.ETTimestamp:
|
|
gt := types.FromDate(2019, 11, 2, 22, 00, int(randNum), rand.Intn(1000000))
|
|
t := types.NewTime(gt, convertETType(g.eType), 0)
|
|
return t
|
|
case types.ETDuration:
|
|
return types.Duration{Duration: time.Duration(randNum)}
|
|
case types.ETJson:
|
|
j := new(json.BinaryJSON)
|
|
jsonStr := fmt.Sprintf("{\"key\":%v}", randNum)
|
|
if err := j.UnmarshalJSON([]byte(jsonStr)); err != nil {
|
|
panic(err)
|
|
}
|
|
return *j
|
|
case types.ETString:
|
|
return fmt.Sprint(randNum)
|
|
}
|
|
return randNum
|
|
}
|
|
|
|
var vecBuiltinOtherGeneratedCases = map[string][]vecExprBenchCase{
|
|
ast.In: {
|
|
// builtinInIntSig
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETInt,
|
|
types.ETInt,
|
|
types.ETInt,
|
|
types.ETInt,
|
|
},
|
|
geners: []dataGenerator{
|
|
inGener{*newDefaultGener(0.2, types.ETInt)},
|
|
inGener{*newDefaultGener(0.2, types.ETInt)},
|
|
inGener{*newDefaultGener(0.2, types.ETInt)},
|
|
inGener{*newDefaultGener(0.2, types.ETInt)},
|
|
},
|
|
},
|
|
// builtinInStringSig
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETString,
|
|
types.ETString,
|
|
types.ETString,
|
|
types.ETString,
|
|
},
|
|
geners: []dataGenerator{
|
|
inGener{*newDefaultGener(0.2, types.ETString)},
|
|
inGener{*newDefaultGener(0.2, types.ETString)},
|
|
inGener{*newDefaultGener(0.2, types.ETString)},
|
|
inGener{*newDefaultGener(0.2, types.ETString)},
|
|
},
|
|
},
|
|
// builtinInDecimalSig
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETDecimal,
|
|
types.ETDecimal,
|
|
types.ETDecimal,
|
|
types.ETDecimal,
|
|
},
|
|
geners: []dataGenerator{
|
|
inGener{*newDefaultGener(0.2, types.ETDecimal)},
|
|
inGener{*newDefaultGener(0.2, types.ETDecimal)},
|
|
inGener{*newDefaultGener(0.2, types.ETDecimal)},
|
|
inGener{*newDefaultGener(0.2, types.ETDecimal)},
|
|
},
|
|
},
|
|
// builtinInRealSig
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETReal,
|
|
types.ETReal,
|
|
types.ETReal,
|
|
types.ETReal,
|
|
},
|
|
geners: []dataGenerator{
|
|
inGener{*newDefaultGener(0.2, types.ETReal)},
|
|
inGener{*newDefaultGener(0.2, types.ETReal)},
|
|
inGener{*newDefaultGener(0.2, types.ETReal)},
|
|
inGener{*newDefaultGener(0.2, types.ETReal)},
|
|
},
|
|
},
|
|
// builtinInTimeSig
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETDatetime,
|
|
types.ETDatetime,
|
|
types.ETDatetime,
|
|
types.ETDatetime,
|
|
},
|
|
geners: []dataGenerator{
|
|
inGener{*newDefaultGener(0.2, types.ETDatetime)},
|
|
inGener{*newDefaultGener(0.2, types.ETDatetime)},
|
|
inGener{*newDefaultGener(0.2, types.ETDatetime)},
|
|
inGener{*newDefaultGener(0.2, types.ETDatetime)},
|
|
},
|
|
},
|
|
// builtinInDurationSig
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETDuration,
|
|
types.ETDuration,
|
|
types.ETDuration,
|
|
types.ETDuration,
|
|
},
|
|
geners: []dataGenerator{
|
|
inGener{*newDefaultGener(0.2, types.ETDuration)},
|
|
inGener{*newDefaultGener(0.2, types.ETDuration)},
|
|
inGener{*newDefaultGener(0.2, types.ETDuration)},
|
|
inGener{*newDefaultGener(0.2, types.ETDuration)},
|
|
},
|
|
},
|
|
// builtinInJSONSig
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETJson,
|
|
types.ETJson,
|
|
types.ETJson,
|
|
types.ETJson,
|
|
},
|
|
geners: []dataGenerator{
|
|
inGener{*newDefaultGener(0.2, types.ETJson)},
|
|
inGener{*newDefaultGener(0.2, types.ETJson)},
|
|
inGener{*newDefaultGener(0.2, types.ETJson)},
|
|
inGener{*newDefaultGener(0.2, types.ETJson)},
|
|
},
|
|
},
|
|
// builtinInIntSig with const arguments
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETInt,
|
|
types.ETInt, types.ETInt,
|
|
},
|
|
constants: []*Constant{
|
|
nil,
|
|
{Value: types.NewDatum(1), RetType: types.NewFieldType(mysql.TypeInt24)},
|
|
{Value: types.NewDatum(2), RetType: types.NewFieldType(mysql.TypeInt24)},
|
|
},
|
|
},
|
|
// builtinInStringSig with const arguments
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETString,
|
|
types.ETString, types.ETString,
|
|
},
|
|
constants: []*Constant{
|
|
nil,
|
|
{Value: types.NewStringDatum("aaaa"), RetType: types.NewFieldType(mysql.TypeString)},
|
|
{Value: types.NewStringDatum("bbbb"), RetType: types.NewFieldType(mysql.TypeString)},
|
|
},
|
|
},
|
|
// builtinInDecimalSig with const arguments
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETDecimal,
|
|
types.ETDecimal, types.ETDecimal,
|
|
},
|
|
constants: []*Constant{
|
|
nil,
|
|
{Value: types.NewDecimalDatum(types.NewDecFromInt(10)), RetType: types.NewFieldType(mysql.TypeDecimal)},
|
|
{Value: types.NewDecimalDatum(types.NewDecFromInt(20)), RetType: types.NewFieldType(mysql.TypeDecimal)},
|
|
},
|
|
},
|
|
// builtinInRealSig with const arguments
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETReal,
|
|
types.ETReal, types.ETReal,
|
|
},
|
|
constants: []*Constant{
|
|
nil,
|
|
{Value: types.NewFloat64Datum(0.1), RetType: types.NewFieldType(mysql.TypeFloat)},
|
|
{Value: types.NewFloat64Datum(0.2), RetType: types.NewFieldType(mysql.TypeFloat)},
|
|
},
|
|
},
|
|
// builtinInTimeSig with const arguments
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETDatetime,
|
|
types.ETDatetime, types.ETDatetime,
|
|
},
|
|
constants: []*Constant{
|
|
nil,
|
|
{Value: types.NewTimeDatum(dateTimeFromString("2019-01-01")), RetType: types.NewFieldType(mysql.TypeDatetime)},
|
|
{Value: types.NewTimeDatum(dateTimeFromString("2019-01-01")), RetType: types.NewFieldType(mysql.TypeDatetime)},
|
|
},
|
|
},
|
|
// builtinInDurationSig with const arguments
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETDuration,
|
|
types.ETDuration, types.ETDuration,
|
|
},
|
|
constants: []*Constant{
|
|
nil,
|
|
{Value: types.NewDurationDatum(types.Duration{Duration: time.Duration(1000)}), RetType: types.NewFieldType(mysql.TypeDuration)},
|
|
{Value: types.NewDurationDatum(types.Duration{Duration: time.Duration(2000)}), RetType: types.NewFieldType(mysql.TypeDuration)},
|
|
},
|
|
},
|
|
// builtinInJSONSig with const arguments
|
|
{
|
|
retEvalType: types.ETInt,
|
|
childrenTypes: []types.EvalType{
|
|
types.ETJson,
|
|
types.ETJson, types.ETJson,
|
|
},
|
|
constants: []*Constant{
|
|
nil,
|
|
{Value: types.NewJSONDatum(json.CreateBinary("aaaa")), RetType: types.NewFieldType(mysql.TypeJSON)},
|
|
{Value: types.NewJSONDatum(json.CreateBinary("bbbb")), RetType: types.NewFieldType(mysql.TypeJSON)},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
func (s *testEvaluatorSuite) TestVectorizedBuiltinOtherEvalOneVecGenerated(c *C) {
|
|
testVectorizedEvalOneVec(c, vecBuiltinOtherGeneratedCases)
|
|
}
|
|
|
|
func (s *testEvaluatorSuite) TestVectorizedBuiltinOtherFuncGenerated(c *C) {
|
|
testVectorizedBuiltinFunc(c, vecBuiltinOtherGeneratedCases)
|
|
}
|
|
|
|
func BenchmarkVectorizedBuiltinOtherEvalOneVecGenerated(b *testing.B) {
|
|
benchmarkVectorizedEvalOneVec(b, vecBuiltinOtherGeneratedCases)
|
|
}
|
|
|
|
func BenchmarkVectorizedBuiltinOtherFuncGenerated(b *testing.B) {
|
|
benchmarkVectorizedBuiltinFunc(b, vecBuiltinOtherGeneratedCases)
|
|
}
|
|
|