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.
 
 

3162 lines
66 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 (
"github.com/pingcap/parser/mysql"
"github.com/pingcap/parser/terror"
"github.com/pingcap/tidb/types"
"github.com/pingcap/tidb/util/chunk"
)
func (b *builtinAddDatetimeAndDurationSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
if err := b.args[0].VecEvalTime(b.ctx, input, result); err != nil {
return err
}
buf0 := result
buf1, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalDuration(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf1)
arg0s := buf0.Times()
arg1s := buf1.GoDurations()
resultSlice := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
// get arg0 & arg1
arg0 := arg0s[i]
arg1 := arg1s[i]
// calculate
output, err := arg0.Add(b.ctx.GetSessionVars().StmtCtx, types.Duration{Duration: arg1, Fsp: -1})
if err != nil {
return err
}
// commit result
resultSlice[i] = output
}
return nil
}
func (b *builtinAddDatetimeAndDurationSig) vectorized() bool {
return true
}
func (b *builtinAddDatetimeAndStringSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
if err := b.args[0].VecEvalTime(b.ctx, input, result); err != nil {
return err
}
buf0 := result
buf1, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalString(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf1)
arg0s := buf0.Times()
resultSlice := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
// get arg0 & arg1
arg0 := arg0s[i]
arg1 := buf1.GetString(i)
// calculate
if !isDuration(arg1) {
result.SetNull(i, true) // fixed: true
continue
}
sc := b.ctx.GetSessionVars().StmtCtx
arg1Duration, err := types.ParseDuration(sc, arg1, types.GetFsp(arg1))
if err != nil {
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
sc.AppendWarning(err)
result.SetNull(i, true) // fixed: true
continue
}
return err
}
output, err := arg0.Add(sc, arg1Duration)
if err != nil {
return err
}
// commit result
resultSlice[i] = output
}
return nil
}
func (b *builtinAddDatetimeAndStringSig) vectorized() bool {
return true
}
func (b *builtinAddDurationAndDurationSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
if err := b.args[0].VecEvalDuration(b.ctx, input, result); err != nil {
return err
}
buf0 := result
buf1, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalDuration(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf1)
arg0s := buf0.GoDurations()
arg1s := buf1.GoDurations()
resultSlice := result.GoDurations()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
// get arg0 & arg1
arg0 := arg0s[i]
arg1 := arg1s[i]
// calculate
output, err := types.AddDuration(arg0, arg1)
if err != nil {
return err
}
// commit result
resultSlice[i] = output
}
return nil
}
func (b *builtinAddDurationAndDurationSig) vectorized() bool {
return true
}
func (b *builtinAddDurationAndStringSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
if err := b.args[0].VecEvalDuration(b.ctx, input, result); err != nil {
return err
}
buf0 := result
buf1, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalString(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf1)
arg0s := buf0.GoDurations()
resultSlice := result.GoDurations()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
// get arg0 & arg1
arg0 := arg0s[i]
arg1 := buf1.GetString(i)
// calculate
if !isDuration(arg1) {
result.SetNull(i, true) // fixed: true
continue
}
sc := b.ctx.GetSessionVars().StmtCtx
arg1Duration, err := types.ParseDuration(sc, arg1, types.GetFsp(arg1))
if err != nil {
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
sc.AppendWarning(err)
result.SetNull(i, true) // fixed: true
continue
}
return err
}
output, err := types.AddDuration(arg0, arg1Duration.Duration)
if err != nil {
return err
}
// commit result
resultSlice[i] = output
}
return nil
}
func (b *builtinAddDurationAndStringSig) vectorized() bool {
return true
}
func (b *builtinAddStringAndDurationSig) vecEvalString(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
buf0, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
if err := b.args[0].VecEvalString(b.ctx, input, buf0); err != nil {
return err
}
buf1, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalDuration(b.ctx, input, buf1); err != nil {
return err
}
result.ReserveString(n)
arg1s := buf1.GoDurations()
for i := 0; i < n; i++ {
if buf0.IsNull(i) || buf1.IsNull(i) {
result.AppendNull()
continue
}
// get arg0 & arg1
arg0 := buf0.GetString(i)
arg1 := arg1s[i]
// calculate
sc := b.ctx.GetSessionVars().StmtCtx
fsp1 := int8(b.args[1].GetType().Decimal)
arg1Duration := types.Duration{Duration: arg1, Fsp: fsp1}
var output string
if isDuration(arg0) {
output, err = strDurationAddDuration(sc, arg0, arg1Duration)
if err != nil {
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
sc.AppendWarning(err)
result.AppendNull() // fixed: false
continue
}
return err
}
} else {
output, err = strDatetimeAddDuration(sc, arg0, arg1Duration)
if err != nil {
return err
}
}
// commit result
result.AppendString(output)
}
return nil
}
func (b *builtinAddStringAndDurationSig) vectorized() bool {
return true
}
func (b *builtinAddStringAndStringSig) vecEvalString(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
buf0, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
if err := b.args[0].VecEvalString(b.ctx, input, buf0); err != nil {
return err
}
arg1Type := b.args[1].GetType()
if mysql.HasBinaryFlag(arg1Type.Flag) {
result.ReserveString(n)
for i := 0; i < n; i++ {
result.AppendNull()
}
return nil
}
buf1, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalString(b.ctx, input, buf1); err != nil {
return err
}
result.ReserveString(n)
for i := 0; i < n; i++ {
if buf0.IsNull(i) || buf1.IsNull(i) {
result.AppendNull()
continue
}
// get arg0 & arg1
arg0 := buf0.GetString(i)
arg1 := buf1.GetString(i)
// calculate
sc := b.ctx.GetSessionVars().StmtCtx
arg1Duration, err := types.ParseDuration(sc, arg1, getFsp4TimeAddSub(arg1))
if err != nil {
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
sc.AppendWarning(err)
result.AppendNull() // fixed: false
continue
}
return err
}
var output string
if isDuration(arg0) {
output, err = strDurationAddDuration(sc, arg0, arg1Duration)
if err != nil {
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
sc.AppendWarning(err)
result.AppendNull() // fixed: false
continue
}
return err
}
} else {
output, err = strDatetimeAddDuration(sc, arg0, arg1Duration)
if err != nil {
return err
}
}
// commit result
result.AppendString(output)
}
return nil
}
func (b *builtinAddStringAndStringSig) vectorized() bool {
return true
}
func (b *builtinAddDateAndDurationSig) vecEvalString(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
buf0, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
if err := b.args[0].VecEvalDuration(b.ctx, input, buf0); err != nil {
return err
}
buf1, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalDuration(b.ctx, input, buf1); err != nil {
return err
}
result.ReserveString(n)
arg0s := buf0.GoDurations()
arg1s := buf1.GoDurations()
for i := 0; i < n; i++ {
if buf0.IsNull(i) || buf1.IsNull(i) {
result.AppendNull()
continue
}
// get arg0 & arg1
arg0 := arg0s[i]
arg1 := arg1s[i]
// calculate
fsp0 := int8(b.args[0].GetType().Decimal)
fsp1 := int8(b.args[1].GetType().Decimal)
arg1Duration := types.Duration{Duration: arg1, Fsp: fsp1}
sum, err := types.Duration{Duration: arg0, Fsp: fsp0}.Add(arg1Duration)
if err != nil {
return err
}
output := sum.String()
// commit result
result.AppendString(output)
}
return nil
}
func (b *builtinAddDateAndDurationSig) vectorized() bool {
return true
}
func (b *builtinAddDateAndStringSig) vecEvalString(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
buf0, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
if err := b.args[0].VecEvalDuration(b.ctx, input, buf0); err != nil {
return err
}
buf1, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalString(b.ctx, input, buf1); err != nil {
return err
}
result.ReserveString(n)
arg0s := buf0.GoDurations()
for i := 0; i < n; i++ {
if buf0.IsNull(i) || buf1.IsNull(i) {
result.AppendNull()
continue
}
// get arg0 & arg1
arg0 := arg0s[i]
arg1 := buf1.GetString(i)
// calculate
if !isDuration(arg1) {
result.AppendNull() // fixed: false
continue
}
sc := b.ctx.GetSessionVars().StmtCtx
arg1Duration, err := types.ParseDuration(sc, arg1, getFsp4TimeAddSub(arg1))
if err != nil {
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
sc.AppendWarning(err)
result.AppendNull() // fixed: false
continue
}
return err
}
fsp0 := int8(b.args[0].GetType().Decimal)
sum, err := types.Duration{Duration: arg0, Fsp: fsp0}.Add(arg1Duration)
if err != nil {
return err
}
output := sum.String()
// commit result
result.AppendString(output)
}
return nil
}
func (b *builtinAddDateAndStringSig) vectorized() bool {
return true
}
func (b *builtinAddTimeDateTimeNullSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ResizeTime(n, true)
return nil
}
func (b *builtinAddTimeDateTimeNullSig) vectorized() bool {
return true
}
func (b *builtinAddTimeStringNullSig) vecEvalString(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ReserveString(n)
for i := 0; i < n; i++ {
result.AppendNull()
}
return nil
}
func (b *builtinAddTimeStringNullSig) vectorized() bool {
return true
}
func (b *builtinAddTimeDurationNullSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ResizeGoDuration(n, true)
return nil
}
func (b *builtinAddTimeDurationNullSig) vectorized() bool {
return true
}
func (b *builtinSubDatetimeAndDurationSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
if err := b.args[0].VecEvalTime(b.ctx, input, result); err != nil {
return err
}
buf0 := result
buf1, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalDuration(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf1)
arg0s := buf0.Times()
arg1s := buf1.GoDurations()
resultSlice := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
// get arg0 & arg1
arg0 := arg0s[i]
arg1 := arg1s[i]
// calculate
sc := b.ctx.GetSessionVars().StmtCtx
arg1Duration := types.Duration{Duration: arg1, Fsp: -1}
arg1time, err := arg1Duration.ConvertToTime(sc, mysql.TypeDatetime)
if err != nil {
return err
}
tmpDuration := arg0.Sub(sc, &arg1time)
output, err := tmpDuration.ConvertToTime(sc, arg0.Type())
if err != nil {
return err
}
// commit result
resultSlice[i] = output
}
return nil
}
func (b *builtinSubDatetimeAndDurationSig) vectorized() bool {
return true
}
func (b *builtinSubDatetimeAndStringSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
if err := b.args[0].VecEvalTime(b.ctx, input, result); err != nil {
return err
}
buf0 := result
buf1, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalString(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf1)
arg0s := buf0.Times()
resultSlice := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
// get arg0 & arg1
arg0 := arg0s[i]
arg1 := buf1.GetString(i)
// calculate
if !isDuration(arg1) {
result.SetNull(i, true) // fixed: true
continue
}
sc := b.ctx.GetSessionVars().StmtCtx
arg1Duration, err := types.ParseDuration(sc, arg1, types.GetFsp(arg1))
if err != nil {
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
sc.AppendWarning(err)
result.SetNull(i, true) // fixed: true
continue
}
return err
}
arg1time, err := arg1Duration.ConvertToTime(sc, mysql.TypeDatetime)
if err != nil {
return err
}
tmpDuration := arg0.Sub(sc, &arg1time)
output, err := tmpDuration.ConvertToTime(sc, mysql.TypeDatetime)
if err != nil {
return err
}
// commit result
resultSlice[i] = output
}
return nil
}
func (b *builtinSubDatetimeAndStringSig) vectorized() bool {
return true
}
func (b *builtinSubDurationAndDurationSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
if err := b.args[0].VecEvalDuration(b.ctx, input, result); err != nil {
return err
}
buf0 := result
buf1, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalDuration(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf1)
arg0s := buf0.GoDurations()
arg1s := buf1.GoDurations()
resultSlice := result.GoDurations()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
// get arg0 & arg1
arg0 := arg0s[i]
arg1 := arg1s[i]
// calculate
output, err := types.SubDuration(arg0, arg1)
if err != nil {
return err
}
// commit result
resultSlice[i] = output
}
return nil
}
func (b *builtinSubDurationAndDurationSig) vectorized() bool {
return true
}
func (b *builtinSubDurationAndStringSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
if err := b.args[0].VecEvalDuration(b.ctx, input, result); err != nil {
return err
}
buf0 := result
buf1, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalString(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf1)
arg0s := buf0.GoDurations()
resultSlice := result.GoDurations()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
// get arg0 & arg1
arg0 := arg0s[i]
arg1 := buf1.GetString(i)
// calculate
if !isDuration(arg1) {
result.SetNull(i, true) // fixed: true
continue
}
sc := b.ctx.GetSessionVars().StmtCtx
arg1Duration, err := types.ParseDuration(sc, arg1, types.GetFsp(arg1))
if err != nil {
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
sc.AppendWarning(err)
result.SetNull(i, true) // fixed: true
continue
}
return err
}
output, err := types.SubDuration(arg0, arg1Duration.Duration)
if err != nil {
return err
}
// commit result
resultSlice[i] = output
}
return nil
}
func (b *builtinSubDurationAndStringSig) vectorized() bool {
return true
}
func (b *builtinSubStringAndDurationSig) vecEvalString(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
buf0, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
if err := b.args[0].VecEvalString(b.ctx, input, buf0); err != nil {
return err
}
buf1, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalDuration(b.ctx, input, buf1); err != nil {
return err
}
result.ReserveString(n)
arg1s := buf1.GoDurations()
for i := 0; i < n; i++ {
if buf0.IsNull(i) || buf1.IsNull(i) {
result.AppendNull()
continue
}
// get arg0 & arg1
arg0 := buf0.GetString(i)
arg1 := arg1s[i]
// calculate
sc := b.ctx.GetSessionVars().StmtCtx
fsp1 := int8(b.args[1].GetType().Decimal)
arg1Duration := types.Duration{Duration: arg1, Fsp: fsp1}
var output string
if isDuration(arg0) {
output, err = strDurationSubDuration(sc, arg0, arg1Duration)
if err != nil {
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
sc.AppendWarning(err)
result.AppendNull() // fixed: false
continue
}
return err
}
} else {
output, err = strDatetimeSubDuration(sc, arg0, arg1Duration)
if err != nil {
return err
}
}
// commit result
result.AppendString(output)
}
return nil
}
func (b *builtinSubStringAndDurationSig) vectorized() bool {
return true
}
func (b *builtinSubStringAndStringSig) vecEvalString(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
buf0, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
if err := b.args[0].VecEvalString(b.ctx, input, buf0); err != nil {
return err
}
arg1Type := b.args[1].GetType()
if mysql.HasBinaryFlag(arg1Type.Flag) {
result.ReserveString(n)
for i := 0; i < n; i++ {
result.AppendNull()
}
return nil
}
buf1, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalString(b.ctx, input, buf1); err != nil {
return err
}
result.ReserveString(n)
for i := 0; i < n; i++ {
if buf0.IsNull(i) || buf1.IsNull(i) {
result.AppendNull()
continue
}
// get arg0 & arg1
arg0 := buf0.GetString(i)
arg1 := buf1.GetString(i)
// calculate
sc := b.ctx.GetSessionVars().StmtCtx
arg1Duration, err := types.ParseDuration(sc, arg1, getFsp4TimeAddSub(arg1))
if err != nil {
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
sc.AppendWarning(err)
result.AppendNull() // fixed: false
continue
}
return err
}
var output string
if isDuration(arg0) {
output, err = strDurationSubDuration(sc, arg0, arg1Duration)
if err != nil {
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
sc.AppendWarning(err)
result.AppendNull() // fixed: false
continue
}
return err
}
} else {
output, err = strDatetimeSubDuration(sc, arg0, arg1Duration)
if err != nil {
return err
}
}
// commit result
result.AppendString(output)
}
return nil
}
func (b *builtinSubStringAndStringSig) vectorized() bool {
return true
}
func (b *builtinSubDateAndDurationSig) vecEvalString(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
buf0, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
if err := b.args[0].VecEvalDuration(b.ctx, input, buf0); err != nil {
return err
}
buf1, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalDuration(b.ctx, input, buf1); err != nil {
return err
}
result.ReserveString(n)
arg0s := buf0.GoDurations()
arg1s := buf1.GoDurations()
for i := 0; i < n; i++ {
if buf0.IsNull(i) || buf1.IsNull(i) {
result.AppendNull()
continue
}
// get arg0 & arg1
arg0 := arg0s[i]
arg1 := arg1s[i]
// calculate
fsp0 := int8(b.args[0].GetType().Decimal)
fsp1 := int8(b.args[1].GetType().Decimal)
arg1Duration := types.Duration{Duration: arg1, Fsp: fsp1}
sum, err := types.Duration{Duration: arg0, Fsp: fsp0}.Sub(arg1Duration)
if err != nil {
return err
}
output := sum.String()
// commit result
result.AppendString(output)
}
return nil
}
func (b *builtinSubDateAndDurationSig) vectorized() bool {
return true
}
func (b *builtinSubDateAndStringSig) vecEvalString(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
buf0, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
if err := b.args[0].VecEvalDuration(b.ctx, input, buf0); err != nil {
return err
}
buf1, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[1].VecEvalString(b.ctx, input, buf1); err != nil {
return err
}
result.ReserveString(n)
arg0s := buf0.GoDurations()
for i := 0; i < n; i++ {
if buf0.IsNull(i) || buf1.IsNull(i) {
result.AppendNull()
continue
}
// get arg0 & arg1
arg0 := arg0s[i]
arg1 := buf1.GetString(i)
// calculate
if !isDuration(arg1) {
result.AppendNull() // fixed: false
continue
}
sc := b.ctx.GetSessionVars().StmtCtx
arg1Duration, err := types.ParseDuration(sc, arg1, getFsp4TimeAddSub(arg1))
if err != nil {
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
sc.AppendWarning(err)
result.AppendNull() // fixed: false
continue
}
return err
}
fsp0 := int8(b.args[0].GetType().Decimal)
sum, err := types.Duration{Duration: arg0, Fsp: fsp0}.Sub(arg1Duration)
if err != nil {
return err
}
output := sum.String()
// commit result
result.AppendString(output)
}
return nil
}
func (b *builtinSubDateAndStringSig) vectorized() bool {
return true
}
func (b *builtinSubTimeDateTimeNullSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ResizeTime(n, true)
return nil
}
func (b *builtinSubTimeDateTimeNullSig) vectorized() bool {
return true
}
func (b *builtinSubTimeStringNullSig) vecEvalString(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ReserveString(n)
for i := 0; i < n; i++ {
result.AppendNull()
}
return nil
}
func (b *builtinSubTimeStringNullSig) vectorized() bool {
return true
}
func (b *builtinSubTimeDurationNullSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ResizeGoDuration(n, true)
return nil
}
func (b *builtinSubTimeDurationNullSig) vectorized() bool {
return true
}
func (b *builtinNullTimeDiffSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ResizeGoDuration(n, true)
return nil
}
func (b *builtinNullTimeDiffSig) vectorized() bool {
return true
}
func (b *builtinTimeStringTimeDiffSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ResizeGoDuration(n, false)
r64s := result.GoDurations()
buf0, err := b.bufAllocator.get(types.ETDatetime, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
buf1, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[0].VecEvalTime(b.ctx, input, buf0); err != nil {
return err
}
if err := b.args[1].VecEvalString(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf0, buf1)
arg0 := buf0.Times()
stmtCtx := b.ctx.GetSessionVars().StmtCtx
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
lhsTime := arg0[i]
_, rhsTime, rhsIsDuration, err := convertStringToDuration(stmtCtx, buf1.GetString(i), int8(b.tp.Decimal))
if err != nil {
return err
}
if rhsIsDuration {
result.SetNull(i, true)
continue
}
d, isNull, err := calculateTimeDiff(stmtCtx, lhsTime, rhsTime)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
continue
}
r64s[i] = d.Duration
}
return nil
}
func (b *builtinTimeStringTimeDiffSig) vectorized() bool {
return true
}
func (b *builtinDurationStringTimeDiffSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ResizeGoDuration(n, false)
r64s := result.GoDurations()
buf0 := result
buf1, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[0].VecEvalDuration(b.ctx, input, buf0); err != nil {
return err
}
if err := b.args[1].VecEvalString(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf1)
arg0 := buf0.GoDurations()
var (
lhs types.Duration
rhs types.Duration
)
stmtCtx := b.ctx.GetSessionVars().StmtCtx
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
lhs.Duration = arg0[i]
rhsDur, _, rhsIsDuration, err := convertStringToDuration(stmtCtx, buf1.GetString(i), int8(b.tp.Decimal))
if err != nil {
return err
}
if !rhsIsDuration {
result.SetNull(i, true)
continue
}
rhs = rhsDur
d, isNull, err := calculateDurationTimeDiff(b.ctx, lhs, rhs)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
continue
}
r64s[i] = d.Duration
}
return nil
}
func (b *builtinDurationStringTimeDiffSig) vectorized() bool {
return true
}
func (b *builtinDurationDurationTimeDiffSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ResizeGoDuration(n, false)
r64s := result.GoDurations()
buf0 := result
buf1, err := b.bufAllocator.get(types.ETDuration, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[0].VecEvalDuration(b.ctx, input, buf0); err != nil {
return err
}
if err := b.args[1].VecEvalDuration(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf1)
arg0 := buf0.GoDurations()
arg1 := buf1.GoDurations()
var (
lhs types.Duration
rhs types.Duration
)
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
lhs.Duration = arg0[i]
rhs.Duration = arg1[i]
d, isNull, err := calculateDurationTimeDiff(b.ctx, lhs, rhs)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
continue
}
r64s[i] = d.Duration
}
return nil
}
func (b *builtinDurationDurationTimeDiffSig) vectorized() bool {
return true
}
func (b *builtinStringTimeTimeDiffSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ResizeGoDuration(n, false)
r64s := result.GoDurations()
buf0, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
buf1, err := b.bufAllocator.get(types.ETDatetime, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[0].VecEvalString(b.ctx, input, buf0); err != nil {
return err
}
if err := b.args[1].VecEvalTime(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf0, buf1)
arg1 := buf1.Times()
stmtCtx := b.ctx.GetSessionVars().StmtCtx
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
_, lhsTime, lhsIsDuration, err := convertStringToDuration(stmtCtx, buf0.GetString(i), int8(b.tp.Decimal))
if err != nil {
return err
}
if lhsIsDuration {
result.SetNull(i, true)
continue
}
rhsTime := arg1[i]
d, isNull, err := calculateTimeDiff(stmtCtx, lhsTime, rhsTime)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
continue
}
r64s[i] = d.Duration
}
return nil
}
func (b *builtinStringTimeTimeDiffSig) vectorized() bool {
return true
}
func (b *builtinStringDurationTimeDiffSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ResizeGoDuration(n, false)
r64s := result.GoDurations()
buf1 := result
buf0, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
if err := b.args[0].VecEvalString(b.ctx, input, buf0); err != nil {
return err
}
if err := b.args[1].VecEvalDuration(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf0)
arg1 := buf1.GoDurations()
var (
lhs types.Duration
rhs types.Duration
)
stmtCtx := b.ctx.GetSessionVars().StmtCtx
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
lhsDur, _, lhsIsDuration, err := convertStringToDuration(stmtCtx, buf0.GetString(i), int8(b.tp.Decimal))
if err != nil {
return err
}
if !lhsIsDuration {
result.SetNull(i, true)
continue
}
lhs = lhsDur
rhs.Duration = arg1[i]
d, isNull, err := calculateDurationTimeDiff(b.ctx, lhs, rhs)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
continue
}
r64s[i] = d.Duration
}
return nil
}
func (b *builtinStringDurationTimeDiffSig) vectorized() bool {
return true
}
func (b *builtinStringStringTimeDiffSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ResizeGoDuration(n, false)
r64s := result.GoDurations()
buf0, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
buf1, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[0].VecEvalString(b.ctx, input, buf0); err != nil {
return err
}
if err := b.args[1].VecEvalString(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf0, buf1)
stmtCtx := b.ctx.GetSessionVars().StmtCtx
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
lhsDur, lhsTime, lhsIsDuration, err := convertStringToDuration(stmtCtx, buf0.GetString(i), int8(b.tp.Decimal))
if err != nil {
return err
}
rhsDur, rhsTime, rhsIsDuration, err := convertStringToDuration(stmtCtx, buf1.GetString(i), int8(b.tp.Decimal))
if err != nil {
return err
}
if lhsIsDuration != rhsIsDuration {
result.SetNull(i, true)
continue
}
var (
d types.Duration
isNull bool
)
if lhsIsDuration {
d, isNull, err = calculateDurationTimeDiff(b.ctx, lhsDur, rhsDur)
} else {
d, isNull, err = calculateTimeDiff(stmtCtx, lhsTime, rhsTime)
}
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
continue
}
r64s[i] = d.Duration
}
return nil
}
func (b *builtinStringStringTimeDiffSig) vectorized() bool {
return true
}
func (b *builtinTimeTimeTimeDiffSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
result.ResizeGoDuration(n, false)
r64s := result.GoDurations()
buf0, err := b.bufAllocator.get(types.ETDatetime, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf0)
buf1, err := b.bufAllocator.get(types.ETDatetime, n)
if err != nil {
return err
}
defer b.bufAllocator.put(buf1)
if err := b.args[0].VecEvalTime(b.ctx, input, buf0); err != nil {
return err
}
if err := b.args[1].VecEvalTime(b.ctx, input, buf1); err != nil {
return err
}
result.MergeNulls(buf0, buf1)
arg0 := buf0.Times()
arg1 := buf1.Times()
stmtCtx := b.ctx.GetSessionVars().StmtCtx
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
lhsTime := arg0[i]
rhsTime := arg1[i]
d, isNull, err := calculateTimeDiff(stmtCtx, lhsTime, rhsTime)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
continue
}
r64s[i] = d.Duration
}
return nil
}
func (b *builtinTimeTimeTimeDiffSig) vectorized() bool {
return true
}
func (b *builtinAddDateStringStringSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromString(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromString(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinAddDateStringStringSig) vectorized() bool {
return true
}
func (b *builtinAddDateStringIntSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromInt(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromString(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinAddDateStringIntSig) vectorized() bool {
return true
}
func (b *builtinAddDateStringRealSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromReal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromString(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinAddDateStringRealSig) vectorized() bool {
return true
}
func (b *builtinAddDateStringDecimalSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromDecimal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromString(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinAddDateStringDecimalSig) vectorized() bool {
return true
}
func (b *builtinAddDateIntStringSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromString(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromInt(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinAddDateIntStringSig) vectorized() bool {
return true
}
func (b *builtinAddDateIntIntSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromInt(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromInt(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinAddDateIntIntSig) vectorized() bool {
return true
}
func (b *builtinAddDateIntRealSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromReal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromInt(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinAddDateIntRealSig) vectorized() bool {
return true
}
func (b *builtinAddDateIntDecimalSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromDecimal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromInt(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinAddDateIntDecimalSig) vectorized() bool {
return true
}
func (b *builtinAddDateDatetimeStringSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromString(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromDatetime(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinAddDateDatetimeStringSig) vectorized() bool {
return true
}
func (b *builtinAddDateDatetimeIntSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromInt(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromDatetime(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinAddDateDatetimeIntSig) vectorized() bool {
return true
}
func (b *builtinAddDateDatetimeRealSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromReal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromDatetime(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinAddDateDatetimeRealSig) vectorized() bool {
return true
}
func (b *builtinAddDateDatetimeDecimalSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromDecimal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromDatetime(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinAddDateDatetimeDecimalSig) vectorized() bool {
return true
}
func (b *builtinAddDateDurationStringSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeGoDuration(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromString(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
result.ResizeGoDuration(n, false)
if err := b.args[0].VecEvalDuration(b.ctx, input, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDurations := result.GoDurations()
iterDuration := types.Duration{Fsp: types.MaxFsp}
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
iterDuration.Duration = resDurations[i]
resDuration, isNull, err := b.addDuration(b.ctx, iterDuration, intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDurations[i] = resDuration.Duration
}
}
return nil
}
func (b *builtinAddDateDurationStringSig) vectorized() bool {
return true
}
func (b *builtinAddDateDurationIntSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeGoDuration(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromInt(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
result.ResizeGoDuration(n, false)
if err := b.args[0].VecEvalDuration(b.ctx, input, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDurations := result.GoDurations()
iterDuration := types.Duration{Fsp: types.MaxFsp}
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
iterDuration.Duration = resDurations[i]
resDuration, isNull, err := b.addDuration(b.ctx, iterDuration, intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDurations[i] = resDuration.Duration
}
}
return nil
}
func (b *builtinAddDateDurationIntSig) vectorized() bool {
return true
}
func (b *builtinAddDateDurationRealSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeGoDuration(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromReal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
result.ResizeGoDuration(n, false)
if err := b.args[0].VecEvalDuration(b.ctx, input, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDurations := result.GoDurations()
iterDuration := types.Duration{Fsp: types.MaxFsp}
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
iterDuration.Duration = resDurations[i]
resDuration, isNull, err := b.addDuration(b.ctx, iterDuration, intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDurations[i] = resDuration.Duration
}
}
return nil
}
func (b *builtinAddDateDurationRealSig) vectorized() bool {
return true
}
func (b *builtinAddDateDurationDecimalSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeGoDuration(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromDecimal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
result.ResizeGoDuration(n, false)
if err := b.args[0].VecEvalDuration(b.ctx, input, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDurations := result.GoDurations()
iterDuration := types.Duration{Fsp: types.MaxFsp}
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
iterDuration.Duration = resDurations[i]
resDuration, isNull, err := b.addDuration(b.ctx, iterDuration, intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDurations[i] = resDuration.Duration
}
}
return nil
}
func (b *builtinAddDateDurationDecimalSig) vectorized() bool {
return true
}
func (b *builtinSubDateStringStringSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromString(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromString(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinSubDateStringStringSig) vectorized() bool {
return true
}
func (b *builtinSubDateStringIntSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromInt(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromString(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinSubDateStringIntSig) vectorized() bool {
return true
}
func (b *builtinSubDateStringRealSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromReal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromString(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinSubDateStringRealSig) vectorized() bool {
return true
}
func (b *builtinSubDateStringDecimalSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromDecimal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromString(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinSubDateStringDecimalSig) vectorized() bool {
return true
}
func (b *builtinSubDateIntStringSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromString(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromInt(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinSubDateIntStringSig) vectorized() bool {
return true
}
func (b *builtinSubDateIntIntSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromInt(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromInt(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinSubDateIntIntSig) vectorized() bool {
return true
}
func (b *builtinSubDateIntRealSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromReal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromInt(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinSubDateIntRealSig) vectorized() bool {
return true
}
func (b *builtinSubDateIntDecimalSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromDecimal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromInt(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinSubDateIntDecimalSig) vectorized() bool {
return true
}
func (b *builtinSubDateDatetimeStringSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromString(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromDatetime(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinSubDateDatetimeStringSig) vectorized() bool {
return true
}
func (b *builtinSubDateDatetimeIntSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromInt(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromDatetime(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinSubDateDatetimeIntSig) vectorized() bool {
return true
}
func (b *builtinSubDateDatetimeRealSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromReal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromDatetime(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinSubDateDatetimeRealSig) vectorized() bool {
return true
}
func (b *builtinSubDateDatetimeDecimalSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeTime(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromDecimal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
if err := b.vecGetDateFromDatetime(&b.baseBuiltinFunc, input, unit, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDates := result.Times()
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDates[i] = resDate
}
}
return nil
}
func (b *builtinSubDateDatetimeDecimalSig) vectorized() bool {
return true
}
func (b *builtinSubDateDurationStringSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeGoDuration(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromString(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
result.ResizeGoDuration(n, false)
if err := b.args[0].VecEvalDuration(b.ctx, input, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDurations := result.GoDurations()
iterDuration := types.Duration{Fsp: types.MaxFsp}
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
iterDuration.Duration = resDurations[i]
resDuration, isNull, err := b.subDuration(b.ctx, iterDuration, intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDurations[i] = resDuration.Duration
}
}
return nil
}
func (b *builtinSubDateDurationStringSig) vectorized() bool {
return true
}
func (b *builtinSubDateDurationIntSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeGoDuration(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromInt(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
result.ResizeGoDuration(n, false)
if err := b.args[0].VecEvalDuration(b.ctx, input, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDurations := result.GoDurations()
iterDuration := types.Duration{Fsp: types.MaxFsp}
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
iterDuration.Duration = resDurations[i]
resDuration, isNull, err := b.subDuration(b.ctx, iterDuration, intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDurations[i] = resDuration.Duration
}
}
return nil
}
func (b *builtinSubDateDurationIntSig) vectorized() bool {
return true
}
func (b *builtinSubDateDurationRealSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeGoDuration(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromReal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
result.ResizeGoDuration(n, false)
if err := b.args[0].VecEvalDuration(b.ctx, input, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDurations := result.GoDurations()
iterDuration := types.Duration{Fsp: types.MaxFsp}
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
iterDuration.Duration = resDurations[i]
resDuration, isNull, err := b.subDuration(b.ctx, iterDuration, intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDurations[i] = resDuration.Duration
}
}
return nil
}
func (b *builtinSubDateDurationRealSig) vectorized() bool {
return true
}
func (b *builtinSubDateDurationDecimalSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
n := input.NumRows()
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
if err != nil {
return err
}
if isNull {
result.ResizeGoDuration(n, true)
return nil
}
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
if err != nil {
return err
}
defer b.bufAllocator.put(intervalBuf)
if err := b.vecGetIntervalFromDecimal(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
return err
}
result.ResizeGoDuration(n, false)
if err := b.args[0].VecEvalDuration(b.ctx, input, result); err != nil {
return err
}
result.MergeNulls(intervalBuf)
resDurations := result.GoDurations()
iterDuration := types.Duration{Fsp: types.MaxFsp}
for i := 0; i < n; i++ {
if result.IsNull(i) {
continue
}
iterDuration.Duration = resDurations[i]
resDuration, isNull, err := b.subDuration(b.ctx, iterDuration, intervalBuf.GetString(i), unit)
if err != nil {
return err
}
if isNull {
result.SetNull(i, true)
} else {
resDurations[i] = resDuration.Duration
}
}
return nil
}
func (b *builtinSubDateDurationDecimalSig) vectorized() bool {
return true
}