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.
1399 lines
37 KiB
1399 lines
37 KiB
// Copyright 2016 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 (
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/pingcap/errors"
|
|
"github.com/pingcap/parser/ast"
|
|
"github.com/pingcap/parser/model"
|
|
"github.com/pingcap/parser/mysql"
|
|
"github.com/pingcap/tidb/sessionctx"
|
|
"github.com/pingcap/tidb/types"
|
|
"github.com/pingcap/tidb/types/json"
|
|
"github.com/pingcap/tidb/util/chunk"
|
|
"github.com/pingcap/tidb/util/collate"
|
|
"github.com/pingcap/tidb/util/set"
|
|
"github.com/pingcap/tidb/util/stringutil"
|
|
"github.com/pingcap/tipb/go-tipb"
|
|
)
|
|
|
|
var (
|
|
_ functionClass = &inFunctionClass{}
|
|
_ functionClass = &rowFunctionClass{}
|
|
_ functionClass = &setVarFunctionClass{}
|
|
_ functionClass = &getIntVarFunctionClass{}
|
|
_ functionClass = &getRealVarFunctionClass{}
|
|
_ functionClass = &getDecimalVarFunctionClass{}
|
|
_ functionClass = &getStringVarFunctionClass{}
|
|
_ functionClass = &lockFunctionClass{}
|
|
_ functionClass = &releaseLockFunctionClass{}
|
|
_ functionClass = &valuesFunctionClass{}
|
|
_ functionClass = &bitCountFunctionClass{}
|
|
_ functionClass = &getParamFunctionClass{}
|
|
)
|
|
|
|
var (
|
|
_ builtinFunc = &builtinSleepSig{}
|
|
_ builtinFunc = &builtinInIntSig{}
|
|
_ builtinFunc = &builtinInStringSig{}
|
|
_ builtinFunc = &builtinInDecimalSig{}
|
|
_ builtinFunc = &builtinInRealSig{}
|
|
_ builtinFunc = &builtinInTimeSig{}
|
|
_ builtinFunc = &builtinInDurationSig{}
|
|
_ builtinFunc = &builtinInJSONSig{}
|
|
_ builtinFunc = &builtinRowSig{}
|
|
_ builtinFunc = &builtinSetStringVarSig{}
|
|
_ builtinFunc = &builtinSetIntVarSig{}
|
|
_ builtinFunc = &builtinSetRealVarSig{}
|
|
_ builtinFunc = &builtinSetDecimalVarSig{}
|
|
_ builtinFunc = &builtinGetStringVarSig{}
|
|
_ builtinFunc = &builtinGetIntVarSig{}
|
|
_ builtinFunc = &builtinGetRealVarSig{}
|
|
_ builtinFunc = &builtinGetDecimalVarSig{}
|
|
_ builtinFunc = &builtinLockSig{}
|
|
_ builtinFunc = &builtinReleaseLockSig{}
|
|
_ builtinFunc = &builtinValuesIntSig{}
|
|
_ builtinFunc = &builtinValuesRealSig{}
|
|
_ builtinFunc = &builtinValuesDecimalSig{}
|
|
_ builtinFunc = &builtinValuesStringSig{}
|
|
_ builtinFunc = &builtinValuesTimeSig{}
|
|
_ builtinFunc = &builtinValuesDurationSig{}
|
|
_ builtinFunc = &builtinValuesJSONSig{}
|
|
_ builtinFunc = &builtinBitCountSig{}
|
|
_ builtinFunc = &builtinGetParamStringSig{}
|
|
)
|
|
|
|
type inFunctionClass struct {
|
|
baseFunctionClass
|
|
}
|
|
|
|
func (c *inFunctionClass) getFunction(ctx sessionctx.Context, args []Expression) (sig builtinFunc, err error) {
|
|
if err := c.verifyArgs(args); err != nil {
|
|
return nil, err
|
|
}
|
|
argTps := make([]types.EvalType, len(args))
|
|
for i := range args {
|
|
argTps[i] = args[0].GetType().EvalType()
|
|
}
|
|
bf, err := newBaseBuiltinFuncWithTp(ctx, c.funcName, args, types.ETInt, argTps...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bf.tp.Flen = 1
|
|
switch args[0].GetType().EvalType() {
|
|
case types.ETInt:
|
|
inInt := builtinInIntSig{baseInSig: baseInSig{baseBuiltinFunc: bf}}
|
|
err := inInt.buildHashMapForConstArgs(ctx)
|
|
if err != nil {
|
|
return &inInt, err
|
|
}
|
|
sig = &inInt
|
|
sig.setPbCode(tipb.ScalarFuncSig_InInt)
|
|
case types.ETString:
|
|
inStr := builtinInStringSig{baseInSig: baseInSig{baseBuiltinFunc: bf}}
|
|
err := inStr.buildHashMapForConstArgs(ctx)
|
|
if err != nil {
|
|
return &inStr, err
|
|
}
|
|
sig = &inStr
|
|
sig.setPbCode(tipb.ScalarFuncSig_InString)
|
|
case types.ETReal:
|
|
inReal := builtinInRealSig{baseInSig: baseInSig{baseBuiltinFunc: bf}}
|
|
err := inReal.buildHashMapForConstArgs(ctx)
|
|
if err != nil {
|
|
return &inReal, err
|
|
}
|
|
sig = &inReal
|
|
sig.setPbCode(tipb.ScalarFuncSig_InReal)
|
|
case types.ETDecimal:
|
|
inDecimal := builtinInDecimalSig{baseInSig: baseInSig{baseBuiltinFunc: bf}}
|
|
err := inDecimal.buildHashMapForConstArgs(ctx)
|
|
if err != nil {
|
|
return &inDecimal, err
|
|
}
|
|
sig = &inDecimal
|
|
sig.setPbCode(tipb.ScalarFuncSig_InDecimal)
|
|
case types.ETDatetime, types.ETTimestamp:
|
|
inTime := builtinInTimeSig{baseInSig: baseInSig{baseBuiltinFunc: bf}}
|
|
err := inTime.buildHashMapForConstArgs(ctx)
|
|
if err != nil {
|
|
return &inTime, err
|
|
}
|
|
sig = &inTime
|
|
sig.setPbCode(tipb.ScalarFuncSig_InTime)
|
|
case types.ETDuration:
|
|
inDuration := builtinInDurationSig{baseInSig: baseInSig{baseBuiltinFunc: bf}}
|
|
err := inDuration.buildHashMapForConstArgs(ctx)
|
|
if err != nil {
|
|
return &inDuration, err
|
|
}
|
|
sig = &inDuration
|
|
sig.setPbCode(tipb.ScalarFuncSig_InDuration)
|
|
case types.ETJson:
|
|
sig = &builtinInJSONSig{baseBuiltinFunc: bf}
|
|
sig.setPbCode(tipb.ScalarFuncSig_InJson)
|
|
}
|
|
return sig, nil
|
|
}
|
|
|
|
type baseInSig struct {
|
|
baseBuiltinFunc
|
|
nonConstArgs []Expression
|
|
hasNull bool
|
|
}
|
|
|
|
// builtinInIntSig see https://dev.mysql.com/doc/refman/5.7/en/comparison-operators.html#function_in
|
|
type builtinInIntSig struct {
|
|
baseInSig
|
|
// the bool value in the map is used to identify whether the constant stored in key is signed or unsigned
|
|
hashSet map[int64]bool
|
|
}
|
|
|
|
func (b *builtinInIntSig) buildHashMapForConstArgs(ctx sessionctx.Context) error {
|
|
b.nonConstArgs = []Expression{b.args[0]}
|
|
b.hashSet = make(map[int64]bool, len(b.args)-1)
|
|
for i := 1; i < len(b.args); i++ {
|
|
if b.args[i].ConstItem(b.ctx.GetSessionVars().StmtCtx) {
|
|
val, isNull, err := b.args[i].EvalInt(ctx, chunk.Row{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isNull {
|
|
b.hasNull = true
|
|
continue
|
|
}
|
|
b.hashSet[val] = mysql.HasUnsignedFlag(b.args[i].GetType().Flag)
|
|
} else {
|
|
b.nonConstArgs = append(b.nonConstArgs, b.args[i])
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (b *builtinInIntSig) Clone() builtinFunc {
|
|
newSig := &builtinInIntSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
newSig.nonConstArgs = make([]Expression, 0, len(b.nonConstArgs))
|
|
for _, arg := range b.nonConstArgs {
|
|
newSig.nonConstArgs = append(newSig.nonConstArgs, arg.Clone())
|
|
}
|
|
newSig.hashSet = b.hashSet
|
|
newSig.hasNull = b.hasNull
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinInIntSig) evalInt(row chunk.Row) (int64, bool, error) {
|
|
arg0, isNull0, err := b.args[0].EvalInt(b.ctx, row)
|
|
if isNull0 || err != nil {
|
|
return 0, isNull0, err
|
|
}
|
|
isUnsigned0 := mysql.HasUnsignedFlag(b.args[0].GetType().Flag)
|
|
|
|
args := b.args
|
|
if len(b.hashSet) != 0 {
|
|
args = b.nonConstArgs
|
|
if isUnsigned, ok := b.hashSet[arg0]; ok {
|
|
if (isUnsigned0 && isUnsigned) || (!isUnsigned0 && !isUnsigned) {
|
|
return 1, false, nil
|
|
}
|
|
if arg0 >= 0 {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
}
|
|
|
|
hasNull := b.hasNull
|
|
for _, arg := range args[1:] {
|
|
evaledArg, isNull, err := arg.EvalInt(b.ctx, row)
|
|
if err != nil {
|
|
return 0, true, err
|
|
}
|
|
if isNull {
|
|
hasNull = true
|
|
continue
|
|
}
|
|
isUnsigned := mysql.HasUnsignedFlag(arg.GetType().Flag)
|
|
if isUnsigned0 && isUnsigned {
|
|
if evaledArg == arg0 {
|
|
return 1, false, nil
|
|
}
|
|
} else if !isUnsigned0 && !isUnsigned {
|
|
if evaledArg == arg0 {
|
|
return 1, false, nil
|
|
}
|
|
} else if !isUnsigned0 && isUnsigned {
|
|
if arg0 >= 0 && evaledArg == arg0 {
|
|
return 1, false, nil
|
|
}
|
|
} else {
|
|
if evaledArg >= 0 && evaledArg == arg0 {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
}
|
|
return 0, hasNull, nil
|
|
}
|
|
|
|
// builtinInStringSig see https://dev.mysql.com/doc/refman/5.7/en/comparison-operators.html#function_in
|
|
type builtinInStringSig struct {
|
|
baseInSig
|
|
hashSet set.StringSet
|
|
}
|
|
|
|
func (b *builtinInStringSig) buildHashMapForConstArgs(ctx sessionctx.Context) error {
|
|
b.nonConstArgs = []Expression{b.args[0]}
|
|
b.hashSet = set.NewStringSet()
|
|
collator := collate.GetCollator(b.collation)
|
|
for i := 1; i < len(b.args); i++ {
|
|
if b.args[i].ConstItem(b.ctx.GetSessionVars().StmtCtx) {
|
|
val, isNull, err := b.args[i].EvalString(ctx, chunk.Row{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isNull {
|
|
b.hasNull = true
|
|
continue
|
|
}
|
|
b.hashSet.Insert(string(collator.Key(val))) // should do memory copy here
|
|
} else {
|
|
b.nonConstArgs = append(b.nonConstArgs, b.args[i])
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (b *builtinInStringSig) Clone() builtinFunc {
|
|
newSig := &builtinInStringSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
newSig.nonConstArgs = make([]Expression, 0, len(b.nonConstArgs))
|
|
for _, arg := range b.nonConstArgs {
|
|
newSig.nonConstArgs = append(newSig.nonConstArgs, arg.Clone())
|
|
}
|
|
newSig.hashSet = b.hashSet
|
|
newSig.hasNull = b.hasNull
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinInStringSig) evalInt(row chunk.Row) (int64, bool, error) {
|
|
arg0, isNull0, err := b.args[0].EvalString(b.ctx, row)
|
|
if isNull0 || err != nil {
|
|
return 0, isNull0, err
|
|
}
|
|
|
|
args := b.args
|
|
collator := collate.GetCollator(b.collation)
|
|
if len(b.hashSet) != 0 {
|
|
args = b.nonConstArgs
|
|
if b.hashSet.Exist(string(collator.Key(arg0))) {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
|
|
hasNull := b.hasNull
|
|
for _, arg := range args[1:] {
|
|
evaledArg, isNull, err := arg.EvalString(b.ctx, row)
|
|
if err != nil {
|
|
return 0, true, err
|
|
}
|
|
if isNull {
|
|
hasNull = true
|
|
continue
|
|
}
|
|
if types.CompareString(arg0, evaledArg, b.collation) == 0 {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
return 0, hasNull, nil
|
|
}
|
|
|
|
// builtinInRealSig see https://dev.mysql.com/doc/refman/5.7/en/comparison-operators.html#function_in
|
|
type builtinInRealSig struct {
|
|
baseInSig
|
|
hashSet set.Float64Set
|
|
}
|
|
|
|
func (b *builtinInRealSig) buildHashMapForConstArgs(ctx sessionctx.Context) error {
|
|
b.nonConstArgs = []Expression{b.args[0]}
|
|
b.hashSet = set.NewFloat64Set()
|
|
for i := 1; i < len(b.args); i++ {
|
|
if b.args[i].ConstItem(b.ctx.GetSessionVars().StmtCtx) {
|
|
val, isNull, err := b.args[i].EvalReal(ctx, chunk.Row{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isNull {
|
|
b.hasNull = true
|
|
continue
|
|
}
|
|
b.hashSet.Insert(val)
|
|
} else {
|
|
b.nonConstArgs = append(b.nonConstArgs, b.args[i])
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (b *builtinInRealSig) Clone() builtinFunc {
|
|
newSig := &builtinInRealSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
newSig.nonConstArgs = make([]Expression, 0, len(b.nonConstArgs))
|
|
for _, arg := range b.nonConstArgs {
|
|
newSig.nonConstArgs = append(newSig.nonConstArgs, arg.Clone())
|
|
}
|
|
newSig.hashSet = b.hashSet
|
|
newSig.hasNull = b.hasNull
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinInRealSig) evalInt(row chunk.Row) (int64, bool, error) {
|
|
arg0, isNull0, err := b.args[0].EvalReal(b.ctx, row)
|
|
if isNull0 || err != nil {
|
|
return 0, isNull0, err
|
|
}
|
|
args := b.args
|
|
if len(b.hashSet) != 0 {
|
|
args = b.nonConstArgs
|
|
if b.hashSet.Exist(arg0) {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
hasNull := b.hasNull
|
|
for _, arg := range args[1:] {
|
|
evaledArg, isNull, err := arg.EvalReal(b.ctx, row)
|
|
if err != nil {
|
|
return 0, true, err
|
|
}
|
|
if isNull {
|
|
hasNull = true
|
|
continue
|
|
}
|
|
if arg0 == evaledArg {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
return 0, hasNull, nil
|
|
}
|
|
|
|
// builtinInDecimalSig see https://dev.mysql.com/doc/refman/5.7/en/comparison-operators.html#function_in
|
|
type builtinInDecimalSig struct {
|
|
baseInSig
|
|
hashSet set.StringSet
|
|
}
|
|
|
|
func (b *builtinInDecimalSig) buildHashMapForConstArgs(ctx sessionctx.Context) error {
|
|
b.nonConstArgs = []Expression{b.args[0]}
|
|
b.hashSet = set.NewStringSet()
|
|
for i := 1; i < len(b.args); i++ {
|
|
if b.args[i].ConstItem(b.ctx.GetSessionVars().StmtCtx) {
|
|
val, isNull, err := b.args[i].EvalDecimal(ctx, chunk.Row{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isNull {
|
|
b.hasNull = true
|
|
continue
|
|
}
|
|
key, err := val.ToHashKey()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
b.hashSet.Insert(string(key))
|
|
} else {
|
|
b.nonConstArgs = append(b.nonConstArgs, b.args[i])
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (b *builtinInDecimalSig) Clone() builtinFunc {
|
|
newSig := &builtinInDecimalSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
newSig.nonConstArgs = make([]Expression, 0, len(b.nonConstArgs))
|
|
for _, arg := range b.nonConstArgs {
|
|
newSig.nonConstArgs = append(newSig.nonConstArgs, arg.Clone())
|
|
}
|
|
newSig.hashSet = b.hashSet
|
|
newSig.hasNull = b.hasNull
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinInDecimalSig) evalInt(row chunk.Row) (int64, bool, error) {
|
|
arg0, isNull0, err := b.args[0].EvalDecimal(b.ctx, row)
|
|
if isNull0 || err != nil {
|
|
return 0, isNull0, err
|
|
}
|
|
|
|
args := b.args
|
|
key, err := arg0.ToHashKey()
|
|
if err != nil {
|
|
return 0, true, err
|
|
}
|
|
if len(b.hashSet) != 0 {
|
|
args = b.nonConstArgs
|
|
if b.hashSet.Exist(string(key)) {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
|
|
hasNull := b.hasNull
|
|
for _, arg := range args[1:] {
|
|
evaledArg, isNull, err := arg.EvalDecimal(b.ctx, row)
|
|
if err != nil {
|
|
return 0, true, err
|
|
}
|
|
if isNull {
|
|
hasNull = true
|
|
continue
|
|
}
|
|
if arg0.Compare(evaledArg) == 0 {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
return 0, hasNull, nil
|
|
}
|
|
|
|
// builtinInTimeSig see https://dev.mysql.com/doc/refman/5.7/en/comparison-operators.html#function_in
|
|
type builtinInTimeSig struct {
|
|
baseInSig
|
|
hashSet map[types.CoreTime]struct{}
|
|
}
|
|
|
|
func (b *builtinInTimeSig) buildHashMapForConstArgs(ctx sessionctx.Context) error {
|
|
b.nonConstArgs = []Expression{b.args[0]}
|
|
b.hashSet = make(map[types.CoreTime]struct{}, len(b.args)-1)
|
|
for i := 1; i < len(b.args); i++ {
|
|
if b.args[i].ConstItem(b.ctx.GetSessionVars().StmtCtx) {
|
|
val, isNull, err := b.args[i].EvalTime(ctx, chunk.Row{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isNull {
|
|
b.hasNull = true
|
|
continue
|
|
}
|
|
b.hashSet[val.CoreTime()] = struct{}{}
|
|
} else {
|
|
b.nonConstArgs = append(b.nonConstArgs, b.args[i])
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (b *builtinInTimeSig) Clone() builtinFunc {
|
|
newSig := &builtinInTimeSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
newSig.nonConstArgs = make([]Expression, 0, len(b.nonConstArgs))
|
|
for _, arg := range b.nonConstArgs {
|
|
newSig.nonConstArgs = append(newSig.nonConstArgs, arg.Clone())
|
|
}
|
|
newSig.hashSet = b.hashSet
|
|
newSig.hasNull = b.hasNull
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinInTimeSig) evalInt(row chunk.Row) (int64, bool, error) {
|
|
arg0, isNull0, err := b.args[0].EvalTime(b.ctx, row)
|
|
if isNull0 || err != nil {
|
|
return 0, isNull0, err
|
|
}
|
|
args := b.args
|
|
if len(b.hashSet) != 0 {
|
|
args = b.nonConstArgs
|
|
if _, ok := b.hashSet[arg0.CoreTime()]; ok {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
hasNull := b.hasNull
|
|
for _, arg := range args[1:] {
|
|
evaledArg, isNull, err := arg.EvalTime(b.ctx, row)
|
|
if err != nil {
|
|
return 0, true, err
|
|
}
|
|
if isNull {
|
|
hasNull = true
|
|
continue
|
|
}
|
|
if arg0.Compare(evaledArg) == 0 {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
return 0, hasNull, nil
|
|
}
|
|
|
|
// builtinInDurationSig see https://dev.mysql.com/doc/refman/5.7/en/comparison-operators.html#function_in
|
|
type builtinInDurationSig struct {
|
|
baseInSig
|
|
hashSet map[time.Duration]struct{}
|
|
}
|
|
|
|
func (b *builtinInDurationSig) buildHashMapForConstArgs(ctx sessionctx.Context) error {
|
|
b.nonConstArgs = []Expression{b.args[0]}
|
|
b.hashSet = make(map[time.Duration]struct{}, len(b.args)-1)
|
|
for i := 1; i < len(b.args); i++ {
|
|
if b.args[i].ConstItem(b.ctx.GetSessionVars().StmtCtx) {
|
|
val, isNull, err := b.args[i].EvalDuration(ctx, chunk.Row{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isNull {
|
|
b.hasNull = true
|
|
continue
|
|
}
|
|
b.hashSet[val.Duration] = struct{}{}
|
|
} else {
|
|
b.nonConstArgs = append(b.nonConstArgs, b.args[i])
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (b *builtinInDurationSig) Clone() builtinFunc {
|
|
newSig := &builtinInDurationSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
newSig.nonConstArgs = make([]Expression, 0, len(b.nonConstArgs))
|
|
for _, arg := range b.nonConstArgs {
|
|
newSig.nonConstArgs = append(newSig.nonConstArgs, arg.Clone())
|
|
}
|
|
newSig.hashSet = b.hashSet
|
|
newSig.hasNull = b.hasNull
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinInDurationSig) evalInt(row chunk.Row) (int64, bool, error) {
|
|
arg0, isNull0, err := b.args[0].EvalDuration(b.ctx, row)
|
|
if isNull0 || err != nil {
|
|
return 0, isNull0, err
|
|
}
|
|
args := b.args
|
|
if len(b.hashSet) != 0 {
|
|
args = b.nonConstArgs
|
|
if _, ok := b.hashSet[arg0.Duration]; ok {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
hasNull := b.hasNull
|
|
for _, arg := range args[1:] {
|
|
evaledArg, isNull, err := arg.EvalDuration(b.ctx, row)
|
|
if err != nil {
|
|
return 0, true, err
|
|
}
|
|
if isNull {
|
|
hasNull = true
|
|
continue
|
|
}
|
|
if arg0.Compare(evaledArg) == 0 {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
return 0, hasNull, nil
|
|
}
|
|
|
|
// builtinInJSONSig see https://dev.mysql.com/doc/refman/5.7/en/comparison-operators.html#function_in
|
|
type builtinInJSONSig struct {
|
|
baseBuiltinFunc
|
|
}
|
|
|
|
func (b *builtinInJSONSig) Clone() builtinFunc {
|
|
newSig := &builtinInJSONSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinInJSONSig) evalInt(row chunk.Row) (int64, bool, error) {
|
|
arg0, isNull0, err := b.args[0].EvalJSON(b.ctx, row)
|
|
if isNull0 || err != nil {
|
|
return 0, isNull0, err
|
|
}
|
|
var hasNull bool
|
|
for _, arg := range b.args[1:] {
|
|
evaledArg, isNull, err := arg.EvalJSON(b.ctx, row)
|
|
if err != nil {
|
|
return 0, true, err
|
|
}
|
|
if isNull {
|
|
hasNull = true
|
|
continue
|
|
}
|
|
result := json.CompareBinary(evaledArg, arg0)
|
|
if result == 0 {
|
|
return 1, false, nil
|
|
}
|
|
}
|
|
return 0, hasNull, nil
|
|
}
|
|
|
|
type rowFunctionClass struct {
|
|
baseFunctionClass
|
|
}
|
|
|
|
func (c *rowFunctionClass) getFunction(ctx sessionctx.Context, args []Expression) (sig builtinFunc, err error) {
|
|
if err = c.verifyArgs(args); err != nil {
|
|
return nil, err
|
|
}
|
|
argTps := make([]types.EvalType, len(args))
|
|
for i := range argTps {
|
|
argTps[i] = args[i].GetType().EvalType()
|
|
}
|
|
bf, err := newBaseBuiltinFuncWithTp(ctx, c.funcName, args, types.ETString, argTps...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
sig = &builtinRowSig{bf}
|
|
return sig, nil
|
|
}
|
|
|
|
type builtinRowSig struct {
|
|
baseBuiltinFunc
|
|
}
|
|
|
|
func (b *builtinRowSig) Clone() builtinFunc {
|
|
newSig := &builtinRowSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
// evalString rowFunc should always be flattened in expression rewrite phrase.
|
|
func (b *builtinRowSig) evalString(row chunk.Row) (string, bool, error) {
|
|
panic("builtinRowSig.evalString() should never be called.")
|
|
}
|
|
|
|
type setVarFunctionClass struct {
|
|
baseFunctionClass
|
|
}
|
|
|
|
func (c *setVarFunctionClass) getFunction(ctx sessionctx.Context, args []Expression) (sig builtinFunc, err error) {
|
|
if err = c.verifyArgs(args); err != nil {
|
|
return nil, err
|
|
}
|
|
argTp := args[1].GetType().EvalType()
|
|
if argTp == types.ETTimestamp || argTp == types.ETDatetime || argTp == types.ETDuration || argTp == types.ETJson {
|
|
argTp = types.ETString
|
|
}
|
|
bf, err := newBaseBuiltinFuncWithTp(ctx, c.funcName, args, argTp, types.ETString, argTp)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bf.tp.Flen = args[1].GetType().Flen
|
|
switch argTp {
|
|
case types.ETString:
|
|
sig = &builtinSetStringVarSig{bf}
|
|
case types.ETReal:
|
|
sig = &builtinSetRealVarSig{bf}
|
|
case types.ETDecimal:
|
|
sig = &builtinSetDecimalVarSig{bf}
|
|
case types.ETInt:
|
|
sig = &builtinSetIntVarSig{bf}
|
|
default:
|
|
return nil, errors.Errorf("unexpected types.EvalType %v", argTp)
|
|
}
|
|
return sig, nil
|
|
}
|
|
|
|
type builtinSetStringVarSig struct {
|
|
baseBuiltinFunc
|
|
}
|
|
|
|
func (b *builtinSetStringVarSig) Clone() builtinFunc {
|
|
newSig := &builtinSetStringVarSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinSetStringVarSig) evalString(row chunk.Row) (res string, isNull bool, err error) {
|
|
var varName string
|
|
sessionVars := b.ctx.GetSessionVars()
|
|
varName, isNull, err = b.args[0].EvalString(b.ctx, row)
|
|
if isNull || err != nil {
|
|
return "", isNull, err
|
|
}
|
|
datum, err := b.args[1].Eval(row)
|
|
isNull = datum.IsNull()
|
|
if isNull || err != nil {
|
|
return "", isNull, err
|
|
}
|
|
res, err = datum.ToString()
|
|
if err != nil {
|
|
return "", isNull, err
|
|
}
|
|
varName = strings.ToLower(varName)
|
|
sessionVars.UsersLock.Lock()
|
|
sessionVars.SetUserVar(varName, stringutil.Copy(res), datum.Collation())
|
|
sessionVars.UsersLock.Unlock()
|
|
return res, false, nil
|
|
}
|
|
|
|
type builtinSetRealVarSig struct {
|
|
baseBuiltinFunc
|
|
}
|
|
|
|
func (b *builtinSetRealVarSig) Clone() builtinFunc {
|
|
newSig := &builtinSetRealVarSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinSetRealVarSig) evalReal(row chunk.Row) (res float64, isNull bool, err error) {
|
|
var varName string
|
|
sessionVars := b.ctx.GetSessionVars()
|
|
varName, isNull, err = b.args[0].EvalString(b.ctx, row)
|
|
if isNull || err != nil {
|
|
return 0, isNull, err
|
|
}
|
|
datum, err := b.args[1].Eval(row)
|
|
isNull = datum.IsNull()
|
|
if isNull || err != nil {
|
|
return 0, isNull, err
|
|
}
|
|
res = datum.GetFloat64()
|
|
varName = strings.ToLower(varName)
|
|
sessionVars.UsersLock.Lock()
|
|
sessionVars.Users[varName] = datum
|
|
sessionVars.UsersLock.Unlock()
|
|
return res, false, nil
|
|
}
|
|
|
|
type builtinSetDecimalVarSig struct {
|
|
baseBuiltinFunc
|
|
}
|
|
|
|
func (b *builtinSetDecimalVarSig) Clone() builtinFunc {
|
|
newSig := &builtinSetDecimalVarSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinSetDecimalVarSig) evalDecimal(row chunk.Row) (*types.MyDecimal, bool, error) {
|
|
sessionVars := b.ctx.GetSessionVars()
|
|
varName, isNull, err := b.args[0].EvalString(b.ctx, row)
|
|
if isNull || err != nil {
|
|
return nil, isNull, err
|
|
}
|
|
datum, err := b.args[1].Eval(row)
|
|
isNull = datum.IsNull()
|
|
if isNull || err != nil {
|
|
return nil, isNull, err
|
|
}
|
|
res := datum.GetMysqlDecimal()
|
|
varName = strings.ToLower(varName)
|
|
sessionVars.UsersLock.Lock()
|
|
sessionVars.Users[varName] = datum
|
|
sessionVars.UsersLock.Unlock()
|
|
return res, false, nil
|
|
}
|
|
|
|
type builtinSetIntVarSig struct {
|
|
baseBuiltinFunc
|
|
}
|
|
|
|
func (b *builtinSetIntVarSig) Clone() builtinFunc {
|
|
newSig := &builtinSetIntVarSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinSetIntVarSig) evalInt(row chunk.Row) (int64, bool, error) {
|
|
sessionVars := b.ctx.GetSessionVars()
|
|
varName, isNull, err := b.args[0].EvalString(b.ctx, row)
|
|
if isNull || err != nil {
|
|
return 0, isNull, err
|
|
}
|
|
datum, err := b.args[1].Eval(row)
|
|
isNull = datum.IsNull()
|
|
if isNull || err != nil {
|
|
return 0, isNull, err
|
|
}
|
|
res := datum.GetInt64()
|
|
varName = strings.ToLower(varName)
|
|
sessionVars.UsersLock.Lock()
|
|
sessionVars.Users[varName] = datum
|
|
sessionVars.UsersLock.Unlock()
|
|
return res, false, nil
|
|
}
|
|
|
|
// BuildGetVarFunction builds a GetVar ScalarFunction from the Expression.
|
|
func BuildGetVarFunction(ctx sessionctx.Context, expr Expression, retType *types.FieldType) (Expression, error) {
|
|
var fc functionClass
|
|
switch retType.EvalType() {
|
|
case types.ETInt:
|
|
fc = &getIntVarFunctionClass{getVarFunctionClass{baseFunctionClass{ast.GetVar, 1, 1}, retType}}
|
|
case types.ETDecimal:
|
|
fc = &getDecimalVarFunctionClass{getVarFunctionClass{baseFunctionClass{ast.GetVar, 1, 1}, retType}}
|
|
case types.ETReal:
|
|
fc = &getRealVarFunctionClass{getVarFunctionClass{baseFunctionClass{ast.GetVar, 1, 1}, retType}}
|
|
default:
|
|
fc = &getStringVarFunctionClass{getVarFunctionClass{baseFunctionClass{ast.GetVar, 1, 1}, retType}}
|
|
}
|
|
f, err := fc.getFunction(ctx, []Expression{expr})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if builtinRetTp := f.getRetTp(); builtinRetTp.Tp != mysql.TypeUnspecified || retType.Tp == mysql.TypeUnspecified {
|
|
retType = builtinRetTp
|
|
}
|
|
return &ScalarFunction{
|
|
FuncName: model.NewCIStr(ast.GetVar),
|
|
RetType: retType,
|
|
Function: f,
|
|
}, nil
|
|
}
|
|
|
|
type getVarFunctionClass struct {
|
|
baseFunctionClass
|
|
|
|
tp *types.FieldType
|
|
}
|
|
|
|
type getStringVarFunctionClass struct {
|
|
getVarFunctionClass
|
|
}
|
|
|
|
func (c *getStringVarFunctionClass) getFunction(ctx sessionctx.Context, args []Expression) (sig builtinFunc, err error) {
|
|
if err = c.verifyArgs(args); err != nil {
|
|
return nil, err
|
|
}
|
|
bf, err := newBaseBuiltinFuncWithTp(ctx, c.funcName, args, types.ETString, types.ETString)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bf.tp.Flen = c.tp.Flen
|
|
if len(c.tp.Charset) > 0 {
|
|
bf.tp.Charset = c.tp.Charset
|
|
bf.tp.Collate = c.tp.Collate
|
|
}
|
|
sig = &builtinGetStringVarSig{bf}
|
|
return sig, nil
|
|
}
|
|
|
|
type builtinGetStringVarSig struct {
|
|
baseBuiltinFunc
|
|
}
|
|
|
|
func (b *builtinGetStringVarSig) Clone() builtinFunc {
|
|
newSig := &builtinGetStringVarSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinGetStringVarSig) evalString(row chunk.Row) (string, bool, error) {
|
|
sessionVars := b.ctx.GetSessionVars()
|
|
varName, isNull, err := b.args[0].EvalString(b.ctx, row)
|
|
if isNull || err != nil {
|
|
return "", isNull, err
|
|
}
|
|
varName = strings.ToLower(varName)
|
|
sessionVars.UsersLock.RLock()
|
|
defer sessionVars.UsersLock.RUnlock()
|
|
if v, ok := sessionVars.Users[varName]; ok {
|
|
return v.GetString(), false, nil
|
|
}
|
|
return "", true, nil
|
|
}
|
|
|
|
type getIntVarFunctionClass struct {
|
|
getVarFunctionClass
|
|
}
|
|
|
|
func (c *getIntVarFunctionClass) getFunction(ctx sessionctx.Context, args []Expression) (sig builtinFunc, err error) {
|
|
if err = c.verifyArgs(args); err != nil {
|
|
return nil, err
|
|
}
|
|
bf, err := newBaseBuiltinFuncWithTp(ctx, c.funcName, args, types.ETInt, types.ETString)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bf.tp.Flen = c.tp.Flen
|
|
bf.tp.Flag = c.tp.Flag
|
|
sig = &builtinGetIntVarSig{bf}
|
|
return sig, nil
|
|
}
|
|
|
|
type builtinGetIntVarSig struct {
|
|
baseBuiltinFunc
|
|
}
|
|
|
|
func (b *builtinGetIntVarSig) Clone() builtinFunc {
|
|
newSig := &builtinGetIntVarSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinGetIntVarSig) evalInt(row chunk.Row) (int64, bool, error) {
|
|
sessionVars := b.ctx.GetSessionVars()
|
|
varName, isNull, err := b.args[0].EvalString(b.ctx, row)
|
|
if isNull || err != nil {
|
|
return 0, isNull, err
|
|
}
|
|
varName = strings.ToLower(varName)
|
|
sessionVars.UsersLock.RLock()
|
|
defer sessionVars.UsersLock.RUnlock()
|
|
if v, ok := sessionVars.Users[varName]; ok {
|
|
return v.GetInt64(), false, nil
|
|
}
|
|
return 0, true, nil
|
|
}
|
|
|
|
type getRealVarFunctionClass struct {
|
|
getVarFunctionClass
|
|
}
|
|
|
|
func (c *getRealVarFunctionClass) getFunction(ctx sessionctx.Context, args []Expression) (sig builtinFunc, err error) {
|
|
if err = c.verifyArgs(args); err != nil {
|
|
return nil, err
|
|
}
|
|
bf, err := newBaseBuiltinFuncWithTp(ctx, c.funcName, args, types.ETReal, types.ETString)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bf.tp.Flen = c.tp.Flen
|
|
sig = &builtinGetRealVarSig{bf}
|
|
return sig, nil
|
|
}
|
|
|
|
type builtinGetRealVarSig struct {
|
|
baseBuiltinFunc
|
|
}
|
|
|
|
func (b *builtinGetRealVarSig) Clone() builtinFunc {
|
|
newSig := &builtinGetRealVarSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinGetRealVarSig) evalReal(row chunk.Row) (float64, bool, error) {
|
|
sessionVars := b.ctx.GetSessionVars()
|
|
varName, isNull, err := b.args[0].EvalString(b.ctx, row)
|
|
if isNull || err != nil {
|
|
return 0, isNull, err
|
|
}
|
|
varName = strings.ToLower(varName)
|
|
sessionVars.UsersLock.RLock()
|
|
defer sessionVars.UsersLock.RUnlock()
|
|
if v, ok := sessionVars.Users[varName]; ok {
|
|
return v.GetFloat64(), false, nil
|
|
}
|
|
return 0, true, nil
|
|
}
|
|
|
|
type getDecimalVarFunctionClass struct {
|
|
getVarFunctionClass
|
|
}
|
|
|
|
func (c *getDecimalVarFunctionClass) getFunction(ctx sessionctx.Context, args []Expression) (sig builtinFunc, err error) {
|
|
if err = c.verifyArgs(args); err != nil {
|
|
return nil, err
|
|
}
|
|
bf, err := newBaseBuiltinFuncWithTp(ctx, c.funcName, args, types.ETDecimal, types.ETString)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bf.tp.Flen = c.tp.Flen
|
|
sig = &builtinGetDecimalVarSig{bf}
|
|
return sig, nil
|
|
}
|
|
|
|
type builtinGetDecimalVarSig struct {
|
|
baseBuiltinFunc
|
|
}
|
|
|
|
func (b *builtinGetDecimalVarSig) Clone() builtinFunc {
|
|
newSig := &builtinGetDecimalVarSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinGetDecimalVarSig) evalDecimal(row chunk.Row) (*types.MyDecimal, bool, error) {
|
|
sessionVars := b.ctx.GetSessionVars()
|
|
varName, isNull, err := b.args[0].EvalString(b.ctx, row)
|
|
if isNull || err != nil {
|
|
return nil, isNull, err
|
|
}
|
|
varName = strings.ToLower(varName)
|
|
sessionVars.UsersLock.RLock()
|
|
defer sessionVars.UsersLock.RUnlock()
|
|
if v, ok := sessionVars.Users[varName]; ok {
|
|
return v.GetMysqlDecimal(), false, nil
|
|
}
|
|
return nil, true, nil
|
|
}
|
|
|
|
type valuesFunctionClass struct {
|
|
baseFunctionClass
|
|
|
|
offset int
|
|
tp *types.FieldType
|
|
}
|
|
|
|
func (c *valuesFunctionClass) getFunction(ctx sessionctx.Context, args []Expression) (sig builtinFunc, err error) {
|
|
if err = c.verifyArgs(args); err != nil {
|
|
return nil, err
|
|
}
|
|
bf, err := newBaseBuiltinFunc(ctx, c.funcName, args)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bf.tp = c.tp
|
|
switch c.tp.EvalType() {
|
|
case types.ETInt:
|
|
sig = &builtinValuesIntSig{bf, c.offset}
|
|
case types.ETReal:
|
|
sig = &builtinValuesRealSig{bf, c.offset}
|
|
case types.ETDecimal:
|
|
sig = &builtinValuesDecimalSig{bf, c.offset}
|
|
case types.ETString:
|
|
sig = &builtinValuesStringSig{bf, c.offset}
|
|
case types.ETDatetime, types.ETTimestamp:
|
|
sig = &builtinValuesTimeSig{bf, c.offset}
|
|
case types.ETDuration:
|
|
sig = &builtinValuesDurationSig{bf, c.offset}
|
|
case types.ETJson:
|
|
sig = &builtinValuesJSONSig{bf, c.offset}
|
|
}
|
|
return sig, nil
|
|
}
|
|
|
|
type builtinValuesIntSig struct {
|
|
baseBuiltinFunc
|
|
|
|
offset int
|
|
}
|
|
|
|
func (b *builtinValuesIntSig) Clone() builtinFunc {
|
|
newSig := &builtinValuesIntSig{offset: b.offset}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
// evalInt evals a builtinValuesIntSig.
|
|
// See https://dev.mysql.com/doc/refman/5.7/en/miscellaneous-functions.html#function_values
|
|
func (b *builtinValuesIntSig) evalInt(_ chunk.Row) (int64, bool, error) {
|
|
if !b.ctx.GetSessionVars().StmtCtx.InInsertStmt {
|
|
return 0, true, nil
|
|
}
|
|
row := b.ctx.GetSessionVars().CurrInsertValues
|
|
if row.IsEmpty() {
|
|
return 0, true, errors.New("Session current insert values is nil")
|
|
}
|
|
if b.offset < row.Len() {
|
|
if row.IsNull(b.offset) {
|
|
return 0, true, nil
|
|
}
|
|
// For BinaryLiteral, see issue #15310
|
|
val := row.GetRaw(b.offset)
|
|
if len(val) > 8 {
|
|
return 0, true, errors.New("Session current insert values is too long")
|
|
}
|
|
if len(val) < 8 {
|
|
var binary types.BinaryLiteral = val
|
|
v, err := binary.ToInt(b.ctx.GetSessionVars().StmtCtx)
|
|
if err != nil {
|
|
return 0, true, errors.Trace(err)
|
|
}
|
|
return int64(v), false, nil
|
|
}
|
|
return row.GetInt64(b.offset), false, nil
|
|
}
|
|
return 0, true, errors.Errorf("Session current insert values len %d and column's offset %v don't match", row.Len(), b.offset)
|
|
}
|
|
|
|
type builtinValuesRealSig struct {
|
|
baseBuiltinFunc
|
|
|
|
offset int
|
|
}
|
|
|
|
func (b *builtinValuesRealSig) Clone() builtinFunc {
|
|
newSig := &builtinValuesRealSig{offset: b.offset}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
// evalReal evals a builtinValuesRealSig.
|
|
// See https://dev.mysql.com/doc/refman/5.7/en/miscellaneous-functions.html#function_values
|
|
func (b *builtinValuesRealSig) evalReal(_ chunk.Row) (float64, bool, error) {
|
|
if !b.ctx.GetSessionVars().StmtCtx.InInsertStmt {
|
|
return 0, true, nil
|
|
}
|
|
row := b.ctx.GetSessionVars().CurrInsertValues
|
|
if row.IsEmpty() {
|
|
return 0, true, errors.New("Session current insert values is nil")
|
|
}
|
|
if b.offset < row.Len() {
|
|
if row.IsNull(b.offset) {
|
|
return 0, true, nil
|
|
}
|
|
if b.getRetTp().Tp == mysql.TypeFloat {
|
|
return float64(row.GetFloat32(b.offset)), false, nil
|
|
}
|
|
return row.GetFloat64(b.offset), false, nil
|
|
}
|
|
return 0, true, errors.Errorf("Session current insert values len %d and column's offset %v don't match", row.Len(), b.offset)
|
|
}
|
|
|
|
type builtinValuesDecimalSig struct {
|
|
baseBuiltinFunc
|
|
|
|
offset int
|
|
}
|
|
|
|
func (b *builtinValuesDecimalSig) Clone() builtinFunc {
|
|
newSig := &builtinValuesDecimalSig{offset: b.offset}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
// evalDecimal evals a builtinValuesDecimalSig.
|
|
// See https://dev.mysql.com/doc/refman/5.7/en/miscellaneous-functions.html#function_values
|
|
func (b *builtinValuesDecimalSig) evalDecimal(_ chunk.Row) (*types.MyDecimal, bool, error) {
|
|
if !b.ctx.GetSessionVars().StmtCtx.InInsertStmt {
|
|
return nil, true, nil
|
|
}
|
|
row := b.ctx.GetSessionVars().CurrInsertValues
|
|
if row.IsEmpty() {
|
|
return nil, true, errors.New("Session current insert values is nil")
|
|
}
|
|
if b.offset < row.Len() {
|
|
if row.IsNull(b.offset) {
|
|
return nil, true, nil
|
|
}
|
|
return row.GetMyDecimal(b.offset), false, nil
|
|
}
|
|
return nil, true, errors.Errorf("Session current insert values len %d and column's offset %v don't match", row.Len(), b.offset)
|
|
}
|
|
|
|
type builtinValuesStringSig struct {
|
|
baseBuiltinFunc
|
|
|
|
offset int
|
|
}
|
|
|
|
func (b *builtinValuesStringSig) Clone() builtinFunc {
|
|
newSig := &builtinValuesStringSig{offset: b.offset}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
// evalString evals a builtinValuesStringSig.
|
|
// See https://dev.mysql.com/doc/refman/5.7/en/miscellaneous-functions.html#function_values
|
|
func (b *builtinValuesStringSig) evalString(_ chunk.Row) (string, bool, error) {
|
|
if !b.ctx.GetSessionVars().StmtCtx.InInsertStmt {
|
|
return "", true, nil
|
|
}
|
|
row := b.ctx.GetSessionVars().CurrInsertValues
|
|
if row.IsEmpty() {
|
|
return "", true, errors.New("Session current insert values is nil")
|
|
}
|
|
if b.offset >= row.Len() {
|
|
return "", true, errors.Errorf("Session current insert values len %d and column's offset %v don't match", row.Len(), b.offset)
|
|
}
|
|
|
|
if row.IsNull(b.offset) {
|
|
return "", true, nil
|
|
}
|
|
|
|
// Specially handle the ENUM/SET/BIT input value.
|
|
if retType := b.getRetTp(); retType.Hybrid() {
|
|
val := row.GetDatum(b.offset, retType)
|
|
res, err := val.ToString()
|
|
return res, err != nil, err
|
|
}
|
|
|
|
return row.GetString(b.offset), false, nil
|
|
}
|
|
|
|
type builtinValuesTimeSig struct {
|
|
baseBuiltinFunc
|
|
|
|
offset int
|
|
}
|
|
|
|
func (b *builtinValuesTimeSig) Clone() builtinFunc {
|
|
newSig := &builtinValuesTimeSig{offset: b.offset}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
// evalTime evals a builtinValuesTimeSig.
|
|
// See https://dev.mysql.com/doc/refman/5.7/en/miscellaneous-functions.html#function_values
|
|
func (b *builtinValuesTimeSig) evalTime(_ chunk.Row) (types.Time, bool, error) {
|
|
if !b.ctx.GetSessionVars().StmtCtx.InInsertStmt {
|
|
return types.ZeroTime, true, nil
|
|
}
|
|
row := b.ctx.GetSessionVars().CurrInsertValues
|
|
if row.IsEmpty() {
|
|
return types.ZeroTime, true, errors.New("Session current insert values is nil")
|
|
}
|
|
if b.offset < row.Len() {
|
|
if row.IsNull(b.offset) {
|
|
return types.ZeroTime, true, nil
|
|
}
|
|
return row.GetTime(b.offset), false, nil
|
|
}
|
|
return types.ZeroTime, true, errors.Errorf("Session current insert values len %d and column's offset %v don't match", row.Len(), b.offset)
|
|
}
|
|
|
|
type builtinValuesDurationSig struct {
|
|
baseBuiltinFunc
|
|
|
|
offset int
|
|
}
|
|
|
|
func (b *builtinValuesDurationSig) Clone() builtinFunc {
|
|
newSig := &builtinValuesDurationSig{offset: b.offset}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
// evalDuration evals a builtinValuesDurationSig.
|
|
// See https://dev.mysql.com/doc/refman/5.7/en/miscellaneous-functions.html#function_values
|
|
func (b *builtinValuesDurationSig) evalDuration(_ chunk.Row) (types.Duration, bool, error) {
|
|
if !b.ctx.GetSessionVars().StmtCtx.InInsertStmt {
|
|
return types.Duration{}, true, nil
|
|
}
|
|
row := b.ctx.GetSessionVars().CurrInsertValues
|
|
if row.IsEmpty() {
|
|
return types.Duration{}, true, errors.New("Session current insert values is nil")
|
|
}
|
|
if b.offset < row.Len() {
|
|
if row.IsNull(b.offset) {
|
|
return types.Duration{}, true, nil
|
|
}
|
|
duration := row.GetDuration(b.offset, b.getRetTp().Decimal)
|
|
return duration, false, nil
|
|
}
|
|
return types.Duration{}, true, errors.Errorf("Session current insert values len %d and column's offset %v don't match", row.Len(), b.offset)
|
|
}
|
|
|
|
type builtinValuesJSONSig struct {
|
|
baseBuiltinFunc
|
|
|
|
offset int
|
|
}
|
|
|
|
func (b *builtinValuesJSONSig) Clone() builtinFunc {
|
|
newSig := &builtinValuesJSONSig{offset: b.offset}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
// evalJSON evals a builtinValuesJSONSig.
|
|
// See https://dev.mysql.com/doc/refman/5.7/en/miscellaneous-functions.html#function_values
|
|
func (b *builtinValuesJSONSig) evalJSON(_ chunk.Row) (json.BinaryJSON, bool, error) {
|
|
if !b.ctx.GetSessionVars().StmtCtx.InInsertStmt {
|
|
return json.BinaryJSON{}, true, nil
|
|
}
|
|
row := b.ctx.GetSessionVars().CurrInsertValues
|
|
if row.IsEmpty() {
|
|
return json.BinaryJSON{}, true, errors.New("Session current insert values is nil")
|
|
}
|
|
if b.offset < row.Len() {
|
|
if row.IsNull(b.offset) {
|
|
return json.BinaryJSON{}, true, nil
|
|
}
|
|
return row.GetJSON(b.offset), false, nil
|
|
}
|
|
return json.BinaryJSON{}, true, errors.Errorf("Session current insert values len %d and column's offset %v don't match", row.Len(), b.offset)
|
|
}
|
|
|
|
type bitCountFunctionClass struct {
|
|
baseFunctionClass
|
|
}
|
|
|
|
func (c *bitCountFunctionClass) getFunction(ctx sessionctx.Context, args []Expression) (builtinFunc, error) {
|
|
if err := c.verifyArgs(args); err != nil {
|
|
return nil, err
|
|
}
|
|
bf, err := newBaseBuiltinFuncWithTp(ctx, c.funcName, args, types.ETInt, types.ETInt)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bf.tp.Flen = 2
|
|
sig := &builtinBitCountSig{bf}
|
|
return sig, nil
|
|
}
|
|
|
|
type builtinBitCountSig struct {
|
|
baseBuiltinFunc
|
|
}
|
|
|
|
func (b *builtinBitCountSig) Clone() builtinFunc {
|
|
newSig := &builtinBitCountSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
// evalInt evals BIT_COUNT(N).
|
|
// See https://dev.mysql.com/doc/refman/5.7/en/bit-functions.html#function_bit-count
|
|
func (b *builtinBitCountSig) evalInt(row chunk.Row) (int64, bool, error) {
|
|
n, isNull, err := b.args[0].EvalInt(b.ctx, row)
|
|
if err != nil || isNull {
|
|
if err != nil && types.ErrOverflow.Equal(err) {
|
|
return 64, false, nil
|
|
}
|
|
return 0, true, err
|
|
}
|
|
return bitCount(n), false, nil
|
|
}
|
|
|
|
// getParamFunctionClass for plan cache of prepared statements
|
|
type getParamFunctionClass struct {
|
|
baseFunctionClass
|
|
}
|
|
|
|
// getFunction gets function
|
|
// TODO: more typed functions will be added when typed parameters are supported.
|
|
func (c *getParamFunctionClass) getFunction(ctx sessionctx.Context, args []Expression) (builtinFunc, error) {
|
|
if err := c.verifyArgs(args); err != nil {
|
|
return nil, err
|
|
}
|
|
bf, err := newBaseBuiltinFuncWithTp(ctx, c.funcName, args, types.ETString, types.ETInt)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bf.tp.Flen = mysql.MaxFieldVarCharLength
|
|
sig := &builtinGetParamStringSig{bf}
|
|
return sig, nil
|
|
}
|
|
|
|
type builtinGetParamStringSig struct {
|
|
baseBuiltinFunc
|
|
}
|
|
|
|
func (b *builtinGetParamStringSig) Clone() builtinFunc {
|
|
newSig := &builtinGetParamStringSig{}
|
|
newSig.cloneFrom(&b.baseBuiltinFunc)
|
|
return newSig
|
|
}
|
|
|
|
func (b *builtinGetParamStringSig) evalString(row chunk.Row) (string, bool, error) {
|
|
sessionVars := b.ctx.GetSessionVars()
|
|
idx, isNull, err := b.args[0].EvalInt(b.ctx, row)
|
|
if isNull || err != nil {
|
|
return "", isNull, err
|
|
}
|
|
v := sessionVars.PreparedParams[idx]
|
|
|
|
str, err := v.ToString()
|
|
if err != nil {
|
|
return "", true, nil
|
|
}
|
|
return str, false, nil
|
|
}
|
|
|