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.
971 lines
35 KiB
971 lines
35 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.
|
|
|
|
//go:build ignore
|
|
// +build ignore
|
|
|
|
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"go/format"
|
|
"io/ioutil"
|
|
"log"
|
|
"path/filepath"
|
|
"text/template"
|
|
|
|
. "github.com/pingcap/tidb/expression/generator/helper"
|
|
)
|
|
|
|
var addOrSubTime = template.Must(template.New("").Parse(`
|
|
{{ if eq $.FuncName "AddTime" }}
|
|
// 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"
|
|
)
|
|
{{ end }}
|
|
{{ define "SetNull" }}{{if .Output.Fixed}}result.SetNull(i, true){{else}}result.AppendNull(){{end}} // fixed: {{.Output.Fixed }}{{ end }}
|
|
{{ define "ConvertStringToDuration" }}
|
|
{{ if and (ne .SigName "builtinAddStringAndStringSig") (ne .SigName "builtinSubStringAndStringSig") }}
|
|
if !isDuration(arg1) {
|
|
{{ template "SetNull" . }}
|
|
continue
|
|
}{{ end }}
|
|
sc := b.ctx.GetSessionVars().StmtCtx
|
|
arg1Duration, err := types.ParseDuration(sc, arg1, {{if eq .Output.TypeName "String"}}getFsp4TimeAddSub{{else}}types.GetFsp{{end}}(arg1))
|
|
if err != nil {
|
|
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
|
|
sc.AppendWarning(err)
|
|
{{ template "SetNull" . }}
|
|
continue
|
|
}
|
|
return err
|
|
}
|
|
{{ end }}
|
|
|
|
{{ range .Sigs }}
|
|
{{ if .AllNull}}
|
|
func (b *{{.SigName}}) vecEval{{ .Output.TypeName }}(input *chunk.Chunk, result *chunk.Column) error {
|
|
n := input.NumRows()
|
|
{{ if .Output.Fixed }}
|
|
result.Resize{{ .Output.TypeNameInColumn }}(n, true)
|
|
{{ else }}
|
|
result.Reserve{{ .Output.TypeNameInColumn }}(n)
|
|
for i := 0; i < n; i++ { result.AppendNull() }
|
|
{{ end }}
|
|
return nil
|
|
}
|
|
{{ else }}
|
|
func (b *{{.SigName}}) vecEval{{ .Output.TypeName }}(input *chunk.Chunk, result *chunk.Column) error {
|
|
n := input.NumRows()
|
|
{{ $reuse := (and (eq .TypeA.TypeName .Output.TypeName) .TypeA.Fixed) }}
|
|
{{ if $reuse }}
|
|
if err := b.args[0].VecEval{{ .TypeA.TypeName }}(b.ctx, input, result); err != nil {
|
|
return err
|
|
}
|
|
buf0 := result
|
|
{{ else }}
|
|
buf0, err := b.bufAllocator.get(types.ET{{.TypeA.ETName}}, n)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer b.bufAllocator.put(buf0)
|
|
if err := b.args[0].VecEval{{ .TypeA.TypeName }}(b.ctx, input, buf0); err != nil {
|
|
return err
|
|
}
|
|
{{ end }}
|
|
|
|
{{ if or (eq .SigName "builtinAddStringAndStringSig") (eq .SigName "builtinSubStringAndStringSig") }}
|
|
arg1Type := b.args[1].GetType()
|
|
if mysql.HasBinaryFlag(arg1Type.Flag) {
|
|
result.Reserve{{ .Output.TypeNameInColumn }}(n)
|
|
for i := 0; i < n; i++ {
|
|
result.AppendNull()
|
|
}
|
|
return nil
|
|
}
|
|
{{ end }}
|
|
|
|
buf1, err := b.bufAllocator.get(types.ET{{.TypeB.ETName}}, n)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer b.bufAllocator.put(buf1)
|
|
if err := b.args[1].VecEval{{ .TypeB.TypeName }}(b.ctx, input, buf1); err != nil {
|
|
return err
|
|
}
|
|
|
|
{{ if $reuse }}
|
|
result.MergeNulls(buf1)
|
|
{{ else if .Output.Fixed}}
|
|
result.Resize{{ .Output.TypeNameInColumn }}(n, false)
|
|
result.MergeNulls(buf0, buf1)
|
|
{{ else }}
|
|
result.Reserve{{ .Output.TypeNameInColumn}}(n)
|
|
{{ end }}
|
|
|
|
{{ if .TypeA.Fixed }}
|
|
arg0s := buf0.{{.TypeA.TypeNameInColumn}}s()
|
|
{{ end }}
|
|
{{ if .TypeB.Fixed }}
|
|
arg1s := buf1.{{.TypeB.TypeNameInColumn}}s()
|
|
{{ end }}
|
|
{{ if .Output.Fixed }}
|
|
resultSlice := result.{{.Output.TypeNameInColumn}}s()
|
|
{{ end }}
|
|
for i := 0; i < n; i++ {
|
|
{{ if .Output.Fixed }}
|
|
if result.IsNull(i) {
|
|
continue
|
|
}
|
|
{{ else }}
|
|
if buf0.IsNull(i) || buf1.IsNull(i) {
|
|
result.AppendNull()
|
|
continue
|
|
}
|
|
{{ end }}
|
|
|
|
// get arg0 & arg1
|
|
{{ if .TypeA.Fixed }}
|
|
arg0 := arg0s[i]
|
|
{{ else }}
|
|
arg0 := buf0.Get{{ .TypeA.TypeNameInColumn }}(i)
|
|
{{ end }}
|
|
{{ if .TypeB.Fixed }}
|
|
arg1 := arg1s[i]
|
|
{{ else }}
|
|
arg1 := buf1.Get{{ .TypeB.TypeNameInColumn }}(i)
|
|
{{ end }}
|
|
|
|
// calculate
|
|
{{ if or (eq .SigName "builtinAddDatetimeAndDurationSig") (eq .SigName "builtinSubDatetimeAndDurationSig") }}
|
|
{{ if eq $.FuncName "AddTime" }}
|
|
output, err := arg0.Add(b.ctx.GetSessionVars().StmtCtx, types.Duration{Duration: arg1, Fsp: -1})
|
|
{{ else }}
|
|
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())
|
|
{{ end }}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
{{ else if or (eq .SigName "builtinAddDatetimeAndStringSig") (eq .SigName "builtinSubDatetimeAndStringSig") }}
|
|
{{ if eq $.FuncName "AddTime" }}
|
|
{{ template "ConvertStringToDuration" . }}
|
|
output, err := arg0.Add(sc, arg1Duration)
|
|
{{ else }}
|
|
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)
|
|
{{ end }}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
{{ else if or (eq .SigName "builtinAddDurationAndDurationSig") (eq .SigName "builtinSubDurationAndDurationSig") }}
|
|
{{ if eq $.FuncName "AddTime" }}
|
|
output, err := types.AddDuration(arg0, arg1)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
{{ else }}
|
|
output, err := types.SubDuration(arg0, arg1)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
{{ end }}
|
|
{{ else if or (eq .SigName "builtinAddDurationAndStringSig") (eq .SigName "builtinSubDurationAndStringSig") }}
|
|
{{ template "ConvertStringToDuration" . }}
|
|
{{ if eq $.FuncName "AddTime" }}
|
|
output, err := types.AddDuration(arg0, arg1Duration.Duration)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
{{ else }}
|
|
output, err := types.SubDuration(arg0, arg1Duration.Duration)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
{{ end }}
|
|
{{ else if or (eq .SigName "builtinAddStringAndDurationSig") (eq .SigName "builtinSubStringAndDurationSig") }}
|
|
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) {
|
|
{{ if eq $.FuncName "AddTime" }}
|
|
output, err = strDurationAddDuration(sc, arg0, arg1Duration)
|
|
{{ else }}
|
|
output, err = strDurationSubDuration(sc, arg0, arg1Duration)
|
|
{{ end }}
|
|
if err != nil {
|
|
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
|
|
sc.AppendWarning(err)
|
|
{{ template "SetNull" . }}
|
|
continue
|
|
}
|
|
return err
|
|
}
|
|
} else {
|
|
{{ if eq $.FuncName "AddTime" }}
|
|
output, err = strDatetimeAddDuration(sc, arg0, arg1Duration)
|
|
{{ else }}
|
|
output, err = strDatetimeSubDuration(sc, arg0, arg1Duration)
|
|
{{ end }}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
{{ else if or (eq .SigName "builtinAddStringAndStringSig") (eq .SigName "builtinSubStringAndStringSig") }}
|
|
{{ template "ConvertStringToDuration" . }}
|
|
var output string
|
|
if isDuration(arg0) {
|
|
{{ if eq $.FuncName "AddTime" }}
|
|
output, err = strDurationAddDuration(sc, arg0, arg1Duration)
|
|
{{ else }}
|
|
output, err = strDurationSubDuration(sc, arg0, arg1Duration)
|
|
{{ end }}
|
|
if err != nil {
|
|
if terror.ErrorEqual(err, types.ErrTruncatedWrongVal) {
|
|
sc.AppendWarning(err)
|
|
{{ template "SetNull" . }}
|
|
continue
|
|
}
|
|
return err
|
|
}
|
|
} else {
|
|
{{ if eq $.FuncName "AddTime" }}
|
|
output, err = strDatetimeAddDuration(sc, arg0, arg1Duration)
|
|
{{ else }}
|
|
output, err = strDatetimeSubDuration(sc, arg0, arg1Duration)
|
|
{{ end }}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
{{ else if or (eq .SigName "builtinAddDateAndDurationSig") (eq .SigName "builtinSubDateAndDurationSig") }}
|
|
fsp0 := int8(b.args[0].GetType().Decimal)
|
|
fsp1 := int8(b.args[1].GetType().Decimal)
|
|
arg1Duration := types.Duration{Duration: arg1, Fsp: fsp1}
|
|
{{ if eq $.FuncName "AddTime" }}
|
|
sum, err := types.Duration{Duration: arg0, Fsp: fsp0}.Add(arg1Duration)
|
|
{{ else }}
|
|
sum, err := types.Duration{Duration: arg0, Fsp: fsp0}.Sub(arg1Duration)
|
|
{{ end }}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
output := sum.String()
|
|
{{ else if or (eq .SigName "builtinAddDateAndStringSig") (eq .SigName "builtinSubDateAndStringSig") }}
|
|
{{ template "ConvertStringToDuration" . }}
|
|
fsp0 := int8(b.args[0].GetType().Decimal)
|
|
{{ if eq $.FuncName "AddTime" }}
|
|
sum, err := types.Duration{Duration: arg0, Fsp: fsp0}.Add(arg1Duration)
|
|
{{ else }}
|
|
sum, err := types.Duration{Duration: arg0, Fsp: fsp0}.Sub(arg1Duration)
|
|
{{ end }}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
output := sum.String()
|
|
{{ end }}
|
|
|
|
// commit result
|
|
{{ if .Output.Fixed }}
|
|
resultSlice[i] = output
|
|
{{ else }}
|
|
result.Append{{ .Output.TypeNameInColumn }}(output)
|
|
{{ end }}
|
|
}
|
|
return nil
|
|
}
|
|
{{ end }}{{/* if .AllNull */}}
|
|
|
|
func (b *{{.SigName}}) vectorized() bool {
|
|
return true
|
|
}
|
|
{{ end }}{{/* range */}}
|
|
`))
|
|
|
|
var timeDiff = template.Must(template.New("").Parse(`
|
|
{{ define "BufAllocator0" }}
|
|
buf0, err := b.bufAllocator.get(types.ET{{.TypeA.ETName}}, n)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer b.bufAllocator.put(buf0)
|
|
{{ end }}
|
|
{{ define "BufAllocator1" }}
|
|
buf1, err := b.bufAllocator.get(types.ET{{.TypeB.ETName}}, n)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer b.bufAllocator.put(buf1)
|
|
{{ end }}
|
|
{{ define "ArgsVecEval" }}
|
|
if err := b.args[0].VecEval{{ .TypeA.TypeName }}(b.ctx, input, buf0); err != nil {
|
|
return err
|
|
}
|
|
if err := b.args[1].VecEval{{ .TypeB.TypeName }}(b.ctx, input, buf1); err != nil {
|
|
return err
|
|
}
|
|
{{ end }}
|
|
|
|
{{ range . }}
|
|
{{ $AIsString := (eq .TypeA.TypeName "String") }}
|
|
{{ $BIsString := (eq .TypeB.TypeName "String") }}
|
|
{{ $AIsTime := (eq .TypeA.TypeName "Time") }}
|
|
{{ $BIsTime := (eq .TypeB.TypeName "Time") }}
|
|
{{ $AIsDuration := (eq .TypeA.TypeName "Duration") }}
|
|
{{ $BIsDuration := (eq .TypeB.TypeName "Duration") }}
|
|
{{ $MaybeDuration := (or (or $AIsDuration $BIsDuration) (and $AIsString $AIsString)) }}
|
|
{{ $reuseA := (eq .TypeA.TypeName "Duration") }}
|
|
{{ $reuseB := (eq .TypeB.TypeName "Duration") }}
|
|
{{ $reuse := (or $reuseA $reuseB ) }}
|
|
{{ $noNull := (ne .SigName "builtinNullTimeDiffSig") }}
|
|
func (b *{{.SigName}}) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
|
|
n := input.NumRows()
|
|
{{- if not $noNull }}
|
|
result.ResizeGoDuration(n, true)
|
|
{{- else }}
|
|
result.ResizeGoDuration(n, false)
|
|
r64s := result.GoDurations()
|
|
{{- if $reuseA }}
|
|
buf0 := result
|
|
{{- template "BufAllocator1" . }}
|
|
{{- template "ArgsVecEval" . }}
|
|
result.MergeNulls(buf1)
|
|
{{- else if $reuseB }}
|
|
buf1 := result
|
|
{{- template "BufAllocator0" . }}
|
|
{{- template "ArgsVecEval" . }}
|
|
result.MergeNulls(buf0)
|
|
{{- else }}
|
|
{{- template "BufAllocator0" . }}
|
|
{{- template "BufAllocator1" . }}
|
|
{{- template "ArgsVecEval" . }}
|
|
result.MergeNulls(buf0, buf1)
|
|
{{- end }}
|
|
{{- if .TypeA.Fixed }}
|
|
arg0 := buf0.{{.TypeA.TypeNameInColumn}}s()
|
|
{{- end }}
|
|
{{- if .TypeB.Fixed }}
|
|
arg1 := buf1.{{.TypeB.TypeNameInColumn}}s()
|
|
{{- end }}
|
|
|
|
{{- if (or $AIsDuration $BIsDuration) }}
|
|
var (
|
|
lhs types.Duration
|
|
rhs types.Duration
|
|
)
|
|
{{- end }}
|
|
{{- if or (or $AIsString $BIsString) (and $AIsTime $BIsTime) }}
|
|
stmtCtx := b.ctx.GetSessionVars().StmtCtx
|
|
{{- end }}
|
|
for i:=0; i<n ; i++{
|
|
if result.IsNull(i) {
|
|
continue
|
|
}
|
|
|
|
{{- if $AIsString }}
|
|
{{ if $BIsDuration }} lhsDur, _, lhsIsDuration,
|
|
{{- else if $BIsTime }} _, lhsTime, lhsIsDuration,
|
|
{{- else if $BIsString }} lhsDur, lhsTime, lhsIsDuration,
|
|
{{- end }} err := convertStringToDuration(stmtCtx, buf0.GetString(i), int8(b.tp.Decimal))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
{{- if $BIsDuration }}
|
|
if !lhsIsDuration {
|
|
result.SetNull(i, true)
|
|
continue
|
|
}
|
|
lhs = lhsDur
|
|
{{- else if $BIsTime }}
|
|
if lhsIsDuration {
|
|
result.SetNull(i, true)
|
|
continue
|
|
}
|
|
{{- end }}
|
|
{{- else if $AIsTime }}
|
|
lhsTime := arg0[i]
|
|
{{- else }}
|
|
lhs.Duration = arg0[i]
|
|
{{- end }}
|
|
|
|
{{- if $BIsString }}
|
|
{{ if $AIsDuration }} rhsDur, _, rhsIsDuration,
|
|
{{- else if $AIsTime }}_, rhsTime, rhsIsDuration,
|
|
{{- else if $AIsString }} rhsDur, rhsTime, rhsIsDuration,
|
|
{{- end}} err := convertStringToDuration(stmtCtx, buf1.GetString(i), int8(b.tp.Decimal))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
{{- if $AIsDuration }}
|
|
if !rhsIsDuration {
|
|
result.SetNull(i, true)
|
|
continue
|
|
}
|
|
rhs = rhsDur
|
|
{{- else if $AIsTime }}
|
|
if rhsIsDuration {
|
|
result.SetNull(i, true)
|
|
continue
|
|
}
|
|
{{- end }}
|
|
{{- else if $BIsTime }}
|
|
rhsTime := arg1[i]
|
|
{{- else }}
|
|
rhs.Duration = arg1[i]
|
|
{{- end }}
|
|
|
|
{{- if and $AIsString $BIsString }}
|
|
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)
|
|
}
|
|
{{- else if or $AIsDuration $BIsDuration }}
|
|
d, isNull, err := calculateDurationTimeDiff(b.ctx, lhs, rhs)
|
|
{{- else if or $AIsTime $BIsTime }}
|
|
d, isNull, err := calculateTimeDiff(stmtCtx, lhsTime, rhsTime)
|
|
{{- end }}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isNull {
|
|
result.SetNull(i, true)
|
|
continue
|
|
}
|
|
r64s[i] = d.Duration
|
|
}
|
|
{{- end }} {{/* if $noNull */}}
|
|
return nil
|
|
}
|
|
|
|
func (b *{{.SigName}}) vectorized() bool {
|
|
return true
|
|
}
|
|
{{ end }}{{/* range */}}
|
|
`))
|
|
|
|
var addOrSubDate = template.Must(template.New("").Parse(`
|
|
{{ range .Sigs }}
|
|
{{- if eq .TypeA.TypeName "Duration" }}
|
|
func (b *{{.SigName}}) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
|
|
{{- else }}
|
|
func (b *{{.SigName}}) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
|
|
{{- end }}
|
|
n := input.NumRows()
|
|
unit, isNull, err := b.args[2].EvalString(b.ctx, chunk.Row{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isNull {
|
|
{{- if eq .TypeA.TypeName "Duration" }}
|
|
result.ResizeGoDuration(n, true)
|
|
{{- else }}
|
|
result.ResizeTime(n, true)
|
|
{{- end }}
|
|
return nil
|
|
}
|
|
|
|
intervalBuf, err := b.bufAllocator.get(types.ETString, n)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer b.bufAllocator.put(intervalBuf)
|
|
if err := b.vecGetIntervalFrom{{.TypeB.ETName}}(&b.baseBuiltinFunc, input, unit, intervalBuf); err != nil {
|
|
return err
|
|
}
|
|
|
|
{{ if eq .TypeA.TypeName "Duration" }}
|
|
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}
|
|
{{ else }}
|
|
if err := b.vecGetDateFrom{{.TypeA.ETName}}(&b.baseBuiltinFunc, input, unit, result); err != nil {
|
|
return err
|
|
}
|
|
|
|
result.MergeNulls(intervalBuf)
|
|
resDates := result.Times()
|
|
{{ end -}}
|
|
|
|
for i := 0; i < n; i++ {
|
|
if result.IsNull(i) {
|
|
continue
|
|
}
|
|
|
|
{{- if eq .TypeA.TypeName "Duration" }}
|
|
iterDuration.Duration = resDurations[i]
|
|
{{- if eq $.FuncName "AddDate" }}
|
|
resDuration, isNull, err := b.addDuration(b.ctx, iterDuration, intervalBuf.GetString(i), unit)
|
|
{{- else }}
|
|
resDuration, isNull, err := b.subDuration(b.ctx, iterDuration, intervalBuf.GetString(i), unit)
|
|
{{- end }}
|
|
{{- else }}
|
|
{{- if eq $.FuncName "AddDate" }}
|
|
resDate, isNull, err := b.add(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
|
|
{{- else }}
|
|
resDate, isNull, err := b.sub(b.ctx, resDates[i], intervalBuf.GetString(i), unit)
|
|
{{- end }}
|
|
{{- end }}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isNull {
|
|
result.SetNull(i, true)
|
|
} else {
|
|
{{- if eq .TypeA.TypeName "Duration" }}
|
|
resDurations[i] = resDuration.Duration
|
|
{{- else }}
|
|
resDates[i] = resDate
|
|
{{- end }}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (b *{{.SigName}}) vectorized() bool {
|
|
return true
|
|
}
|
|
{{ end }}{{/* range */}}
|
|
`))
|
|
|
|
var testFileFuncs = template.FuncMap{
|
|
"getIntervalUnitList": func() []string {
|
|
return []string{
|
|
"MICROSECOND",
|
|
"SECOND",
|
|
"MINUTE",
|
|
"HOUR",
|
|
"DAY",
|
|
"WEEK",
|
|
"MONTH",
|
|
"QUARTER",
|
|
"YEAR",
|
|
"SECOND_MICROSECOND",
|
|
"MINUTE_MICROSECOND",
|
|
"MINUTE_SECOND",
|
|
"HOUR_MICROSECOND",
|
|
"HOUR_SECOND",
|
|
"HOUR_MINUTE",
|
|
"DAY_MICROSECOND",
|
|
"DAY_SECOND",
|
|
"DAY_MINUTE",
|
|
"DAY_HOUR",
|
|
"YEAR_MONTH",
|
|
}
|
|
},
|
|
}
|
|
|
|
var testFile = template.Must(template.New("").Funcs(testFileFuncs).Parse(`
|
|
// 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 (
|
|
"math"
|
|
"testing"
|
|
|
|
. "github.com/pingcap/check"
|
|
"github.com/pingcap/parser/ast"
|
|
"github.com/pingcap/parser/mysql"
|
|
"github.com/pingcap/tidb/types"
|
|
)
|
|
|
|
type gener struct {
|
|
defaultGener
|
|
}
|
|
|
|
func (g gener) gen() interface{} {
|
|
result := g.defaultGener.gen()
|
|
if _, ok := result.(string); ok {
|
|
dg := newDefaultGener(0, types.ETDuration)
|
|
d := dg.gen().(types.Duration)
|
|
if int8(d.Duration)%2 == 0 {
|
|
d.Fsp = 0
|
|
} else {
|
|
d.Fsp = 1
|
|
}
|
|
result = d.String()
|
|
}
|
|
return result
|
|
}
|
|
|
|
{{ define "addOrSubDateCases" }}
|
|
{{- $unitList := getIntervalUnitList -}}
|
|
{{- range $sig := .Sigs }}
|
|
// {{ $sig.SigName }}
|
|
{{- range $unit := $unitList }}
|
|
{
|
|
retEvalType: types.ET{{ $sig.Output.ETName }},
|
|
childrenTypes: []types.EvalType{types.ET{{ $sig.TypeA.ETName }}, types.ET{{ $sig.TypeB.ETName }}, types.ETString},
|
|
geners: []dataGenerator{
|
|
{{- if eq $sig.TypeA.ETName "String"}}
|
|
&dateStrGener{NullRation: 0.2, randGen: newDefaultRandGen()},
|
|
{{- else if eq $sig.TypeA.ETName "Int"}}
|
|
&dateTimeIntGener{dateTimeGener: dateTimeGener{randGen: newDefaultRandGen()}, nullRation: 0.2},
|
|
{{- else }}
|
|
newDefaultGener(0.2, types.ET{{$sig.TypeA.ETName}}),
|
|
{{- end }}
|
|
|
|
{{- if eq $sig.TypeB.ETName "String" }}
|
|
&numStrGener{rangeInt64Gener{math.MinInt32 + 1, math.MaxInt32, newDefaultRandGen()}},
|
|
{{- else }}
|
|
newDefaultGener(0.2, types.ET{{$sig.TypeB.ETName}}),
|
|
{{- end }}
|
|
},
|
|
constants: []*Constant{nil, nil, {Value: types.NewStringDatum("{{$unit}}"), RetType: types.NewFieldType(mysql.TypeString)}},
|
|
chunkSize: 128,
|
|
},
|
|
{{- end }}
|
|
{{- end }}
|
|
{{ end }}
|
|
|
|
{{ define "addOrSubTimeCases" }}
|
|
{{- range $sig := .Sigs }}
|
|
// {{ $sig.SigName }}
|
|
{
|
|
retEvalType: types.ET{{ .Output.ETName }},
|
|
{{- if eq .TestTypeA "" }}
|
|
childrenTypes: []types.EvalType{types.ET{{ .TypeA.ETName }}, types.ET{{ .TypeB.ETName }}},
|
|
{{- else }}
|
|
childrenTypes: []types.EvalType{types.ET{{ .TestTypeA }}, types.ET{{ .TestTypeB }}},
|
|
{{- end }}
|
|
{{- if ne .FieldTypeA "" }}
|
|
childrenFieldTypes: []*types.FieldType{types.NewFieldType(mysql.Type{{.FieldTypeA}}), types.NewFieldType(mysql.Type{{.FieldTypeB}})},
|
|
{{- end }}
|
|
geners: []dataGenerator{
|
|
{{- if eq .TestTypeA "" }}
|
|
gener{*newDefaultGener(0.2, types.ET{{.TypeA.ETName}})},
|
|
gener{*newDefaultGener(0.2, types.ET{{.TypeB.ETName}})},
|
|
{{- else }}
|
|
gener{*newDefaultGener(0.2, types.ET{{ .TestTypeA }})},
|
|
gener{*newDefaultGener(0.2, types.ET{{ .TestTypeB }})},
|
|
{{- end }}
|
|
},
|
|
},
|
|
{{- end }}
|
|
{{ end }}
|
|
|
|
{{/* Add more test cases here if we have more functions in this file */}}
|
|
var vecBuiltin{{.Category}}GeneratedCases = map[string][]vecExprBenchCase{
|
|
{{- range .Functions }}
|
|
{{- if eq .FuncName "AddTime" }}
|
|
ast.AddTime: {
|
|
{{- template "addOrSubTimeCases" . -}}
|
|
},
|
|
{{ end }}
|
|
{{- if eq .FuncName "SubTime" }}
|
|
ast.SubTime: {
|
|
{{- template "addOrSubTimeCases" . -}}
|
|
},
|
|
{{ end }}
|
|
{{- if eq .FuncName "TimeDiff" }}
|
|
ast.TimeDiff: {
|
|
// builtinNullTimeDiffSig
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETDuration, types.ETDatetime}},
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETDuration, types.ETTimestamp}},
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETDatetime, types.ETDuration}},
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETTimestamp, types.ETDuration}},
|
|
// builtinDurationDurationTimeDiffSig
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETDuration, types.ETDuration}},
|
|
// builtinDurationStringTimeDiffSig
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETDuration, types.ETString}, geners: []dataGenerator{nil, &dateTimeStrGener{Year: 2019, Month: 11, randGen: newDefaultRandGen()}}},
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETDuration, types.ETString}, geners: []dataGenerator{nil, &dateTimeStrGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}}},
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETDuration, types.ETString}, geners: []dataGenerator{nil, &dateTimeStrGener{Year: 2019, Month: 10, Fsp: 4, randGen: newDefaultRandGen()}}},
|
|
// builtinTimeTimeTimeDiffSig
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETDatetime, types.ETDatetime}, geners: []dataGenerator{&dateTimeGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}, &dateTimeGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}}},
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETDatetime, types.ETTimestamp}, geners: []dataGenerator{&dateTimeGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}, &dateTimeGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}}},
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETTimestamp, types.ETTimestamp}, geners: []dataGenerator{&dateTimeGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}, &dateTimeGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}}},
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETTimestamp, types.ETDatetime}, geners: []dataGenerator{&dateTimeGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}, &dateTimeGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}}},
|
|
// builtinTimeStringTimeDiffSig
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETDatetime, types.ETString}, geners: []dataGenerator{&dateTimeGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}, &dateTimeStrGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}}},
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETTimestamp, types.ETString}, geners: []dataGenerator{&dateTimeGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}, &dateTimeStrGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}}},
|
|
// builtinStringDurationTimeDiffSig
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETString, types.ETDuration}, geners: []dataGenerator{&dateTimeStrGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}, nil}},
|
|
// builtinStringTimeTimeDiffSig
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETString, types.ETDatetime}, geners: []dataGenerator{&dateTimeStrGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}, &dateTimeGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}}},
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETString, types.ETTimestamp}, geners: []dataGenerator{&dateTimeStrGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}, &dateTimeGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}}},
|
|
// builtinStringStringTimeDiffSig
|
|
{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETString, types.ETString}, geners: []dataGenerator{&dateTimeStrGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}, &dateTimeStrGener{Year: 2019, Month: 10, randGen: newDefaultRandGen()}}},
|
|
},
|
|
{{ end }}
|
|
{{- if eq .FuncName "AddDate" }}
|
|
ast.AddDate: {
|
|
{{- template "addOrSubDateCases" . -}}
|
|
},
|
|
{{ end }}
|
|
{{- if eq .FuncName "SubDate" }}
|
|
ast.SubDate: {
|
|
{{- template "addOrSubDateCases" . -}}
|
|
},
|
|
{{ end }}
|
|
{{ end }}
|
|
}
|
|
|
|
func (s *testVectorizeSuite1) TestVectorizedBuiltin{{.Category}}EvalOneVecGenerated(c *C) {
|
|
testVectorizedEvalOneVec(c, vecBuiltin{{.Category}}GeneratedCases)
|
|
}
|
|
|
|
func (s *testVectorizeSuite1) TestVectorizedBuiltin{{.Category}}FuncGenerated(c *C) {
|
|
testVectorizedBuiltinFunc(c, vecBuiltin{{.Category}}GeneratedCases)
|
|
}
|
|
|
|
func BenchmarkVectorizedBuiltin{{.Category}}EvalOneVecGenerated(b *testing.B) {
|
|
benchmarkVectorizedEvalOneVec(b, vecBuiltin{{.Category}}GeneratedCases)
|
|
}
|
|
|
|
func BenchmarkVectorizedBuiltin{{.Category}}FuncGenerated(b *testing.B) {
|
|
benchmarkVectorizedBuiltinFunc(b, vecBuiltin{{.Category}}GeneratedCases)
|
|
}
|
|
`))
|
|
|
|
var addTimeSigsTmpl = []sig{
|
|
{SigName: "builtinAddDatetimeAndDurationSig", TypeA: TypeDatetime, TypeB: TypeDuration, Output: TypeDatetime},
|
|
{SigName: "builtinAddDatetimeAndStringSig", TypeA: TypeDatetime, TypeB: TypeString, Output: TypeDatetime},
|
|
{SigName: "builtinAddDurationAndDurationSig", TypeA: TypeDuration, TypeB: TypeDuration, Output: TypeDuration},
|
|
{SigName: "builtinAddDurationAndStringSig", TypeA: TypeDuration, TypeB: TypeString, Output: TypeDuration},
|
|
{SigName: "builtinAddStringAndDurationSig", TypeA: TypeString, TypeB: TypeDuration, Output: TypeString},
|
|
{SigName: "builtinAddStringAndStringSig", TypeA: TypeString, TypeB: TypeString, Output: TypeString},
|
|
{SigName: "builtinAddDateAndDurationSig", TypeA: TypeDuration, TypeB: TypeDuration, Output: TypeString, FieldTypeA: "Date", FieldTypeB: "Duration", TestTypeA: "Datetime", TestTypeB: "Duration"},
|
|
{SigName: "builtinAddDateAndStringSig", TypeA: TypeDuration, TypeB: TypeString, Output: TypeString, FieldTypeA: "Date", FieldTypeB: "String", TestTypeA: "Datetime", TestTypeB: "String"},
|
|
|
|
{SigName: "builtinAddTimeDateTimeNullSig", TypeA: TypeDatetime, TypeB: TypeDatetime, Output: TypeDatetime, AllNull: true},
|
|
{SigName: "builtinAddTimeStringNullSig", TypeA: TypeDatetime, TypeB: TypeDatetime, Output: TypeString, AllNull: true, FieldTypeA: "Date", FieldTypeB: "Datetime"},
|
|
{SigName: "builtinAddTimeDurationNullSig", TypeA: TypeDuration, TypeB: TypeDatetime, Output: TypeDuration, AllNull: true},
|
|
}
|
|
|
|
var subTimeSigsTmpl = []sig{
|
|
{SigName: "builtinSubDatetimeAndDurationSig", TypeA: TypeDatetime, TypeB: TypeDuration, Output: TypeDatetime},
|
|
{SigName: "builtinSubDatetimeAndStringSig", TypeA: TypeDatetime, TypeB: TypeString, Output: TypeDatetime},
|
|
{SigName: "builtinSubDurationAndDurationSig", TypeA: TypeDuration, TypeB: TypeDuration, Output: TypeDuration},
|
|
{SigName: "builtinSubDurationAndStringSig", TypeA: TypeDuration, TypeB: TypeString, Output: TypeDuration},
|
|
{SigName: "builtinSubStringAndDurationSig", TypeA: TypeString, TypeB: TypeDuration, Output: TypeString},
|
|
{SigName: "builtinSubStringAndStringSig", TypeA: TypeString, TypeB: TypeString, Output: TypeString},
|
|
{SigName: "builtinSubDateAndDurationSig", TypeA: TypeDuration, TypeB: TypeDuration, Output: TypeString, FieldTypeA: "Date", FieldTypeB: "Duration", TestTypeA: "Datetime", TestTypeB: "Duration"},
|
|
{SigName: "builtinSubDateAndStringSig", TypeA: TypeDuration, TypeB: TypeString, Output: TypeString, FieldTypeA: "Date", FieldTypeB: "String", TestTypeA: "Datetime", TestTypeB: "String"},
|
|
|
|
{SigName: "builtinSubTimeDateTimeNullSig", TypeA: TypeDatetime, TypeB: TypeDatetime, Output: TypeDatetime, AllNull: true},
|
|
{SigName: "builtinSubTimeStringNullSig", TypeA: TypeDatetime, TypeB: TypeDatetime, Output: TypeString, AllNull: true, FieldTypeA: "Date", FieldTypeB: "Datetime"},
|
|
{SigName: "builtinSubTimeDurationNullSig", TypeA: TypeDuration, TypeB: TypeDatetime, Output: TypeDuration, AllNull: true},
|
|
}
|
|
|
|
var timeDiffSigsTmpl = []sig{
|
|
{SigName: "builtinNullTimeDiffSig", Output: TypeDuration},
|
|
{SigName: "builtinTimeStringTimeDiffSig", TypeA: TypeDatetime, TypeB: TypeString, Output: TypeDuration},
|
|
{SigName: "builtinDurationStringTimeDiffSig", TypeA: TypeDuration, TypeB: TypeString, Output: TypeDuration},
|
|
{SigName: "builtinDurationDurationTimeDiffSig", TypeA: TypeDuration, TypeB: TypeDuration, Output: TypeDuration},
|
|
{SigName: "builtinStringTimeTimeDiffSig", TypeA: TypeString, TypeB: TypeDatetime, Output: TypeDuration},
|
|
{SigName: "builtinStringDurationTimeDiffSig", TypeA: TypeString, TypeB: TypeDuration, Output: TypeDuration},
|
|
{SigName: "builtinStringStringTimeDiffSig", TypeA: TypeString, TypeB: TypeString, Output: TypeDuration},
|
|
{SigName: "builtinTimeTimeTimeDiffSig", TypeA: TypeDatetime, TypeB: TypeDatetime, Output: TypeDuration},
|
|
}
|
|
|
|
var addDataSigsTmpl = []sig{
|
|
{SigName: "builtinAddDateStringStringSig", TypeA: TypeString, TypeB: TypeString, Output: TypeDatetime},
|
|
{SigName: "builtinAddDateStringIntSig", TypeA: TypeString, TypeB: TypeInt, Output: TypeDatetime},
|
|
{SigName: "builtinAddDateStringRealSig", TypeA: TypeString, TypeB: TypeReal, Output: TypeDatetime},
|
|
{SigName: "builtinAddDateStringDecimalSig", TypeA: TypeString, TypeB: TypeDecimal, Output: TypeDatetime},
|
|
{SigName: "builtinAddDateIntStringSig", TypeA: TypeInt, TypeB: TypeString, Output: TypeDatetime},
|
|
{SigName: "builtinAddDateIntIntSig", TypeA: TypeInt, TypeB: TypeInt, Output: TypeDatetime},
|
|
{SigName: "builtinAddDateIntRealSig", TypeA: TypeInt, TypeB: TypeReal, Output: TypeDatetime},
|
|
{SigName: "builtinAddDateIntDecimalSig", TypeA: TypeInt, TypeB: TypeDecimal, Output: TypeDatetime},
|
|
{SigName: "builtinAddDateDatetimeStringSig", TypeA: TypeDatetime, TypeB: TypeString, Output: TypeDatetime},
|
|
{SigName: "builtinAddDateDatetimeIntSig", TypeA: TypeDatetime, TypeB: TypeInt, Output: TypeDatetime},
|
|
{SigName: "builtinAddDateDatetimeRealSig", TypeA: TypeDatetime, TypeB: TypeReal, Output: TypeDatetime},
|
|
{SigName: "builtinAddDateDatetimeDecimalSig", TypeA: TypeDatetime, TypeB: TypeDecimal, Output: TypeDatetime},
|
|
{SigName: "builtinAddDateDurationStringSig", TypeA: TypeDuration, TypeB: TypeString, Output: TypeDuration},
|
|
{SigName: "builtinAddDateDurationIntSig", TypeA: TypeDuration, TypeB: TypeInt, Output: TypeDuration},
|
|
{SigName: "builtinAddDateDurationRealSig", TypeA: TypeDuration, TypeB: TypeReal, Output: TypeDuration},
|
|
{SigName: "builtinAddDateDurationDecimalSig", TypeA: TypeDuration, TypeB: TypeDecimal, Output: TypeDuration},
|
|
}
|
|
|
|
var subDataSigsTmpl = []sig{
|
|
{SigName: "builtinSubDateStringStringSig", TypeA: TypeString, TypeB: TypeString, Output: TypeDatetime},
|
|
{SigName: "builtinSubDateStringIntSig", TypeA: TypeString, TypeB: TypeInt, Output: TypeDatetime},
|
|
{SigName: "builtinSubDateStringRealSig", TypeA: TypeString, TypeB: TypeReal, Output: TypeDatetime},
|
|
{SigName: "builtinSubDateStringDecimalSig", TypeA: TypeString, TypeB: TypeDecimal, Output: TypeDatetime},
|
|
{SigName: "builtinSubDateIntStringSig", TypeA: TypeInt, TypeB: TypeString, Output: TypeDatetime},
|
|
{SigName: "builtinSubDateIntIntSig", TypeA: TypeInt, TypeB: TypeInt, Output: TypeDatetime},
|
|
{SigName: "builtinSubDateIntRealSig", TypeA: TypeInt, TypeB: TypeReal, Output: TypeDatetime},
|
|
{SigName: "builtinSubDateIntDecimalSig", TypeA: TypeInt, TypeB: TypeDecimal, Output: TypeDatetime},
|
|
{SigName: "builtinSubDateDatetimeStringSig", TypeA: TypeDatetime, TypeB: TypeString, Output: TypeDatetime},
|
|
{SigName: "builtinSubDateDatetimeIntSig", TypeA: TypeDatetime, TypeB: TypeInt, Output: TypeDatetime},
|
|
{SigName: "builtinSubDateDatetimeRealSig", TypeA: TypeDatetime, TypeB: TypeReal, Output: TypeDatetime},
|
|
{SigName: "builtinSubDateDatetimeDecimalSig", TypeA: TypeDatetime, TypeB: TypeDecimal, Output: TypeDatetime},
|
|
{SigName: "builtinSubDateDurationStringSig", TypeA: TypeDuration, TypeB: TypeString, Output: TypeDuration},
|
|
{SigName: "builtinSubDateDurationIntSig", TypeA: TypeDuration, TypeB: TypeInt, Output: TypeDuration},
|
|
{SigName: "builtinSubDateDurationRealSig", TypeA: TypeDuration, TypeB: TypeReal, Output: TypeDuration},
|
|
{SigName: "builtinSubDateDurationDecimalSig", TypeA: TypeDuration, TypeB: TypeDecimal, Output: TypeDuration},
|
|
}
|
|
|
|
type sig struct {
|
|
SigName string
|
|
TypeA, TypeB, Output TypeContext
|
|
FieldTypeA, FieldTypeB string // Optional
|
|
TestTypeA, TestTypeB string // Optional, specific Type for test in builtinAddDateAndDurationSig & builtinAddDateAndStringSig
|
|
AllNull bool
|
|
}
|
|
|
|
type function struct {
|
|
FuncName string
|
|
Sigs []sig
|
|
}
|
|
|
|
var tmplVal = struct {
|
|
Category string
|
|
Functions []function
|
|
}{
|
|
Category: "Time",
|
|
Functions: []function{
|
|
{FuncName: "AddTime", Sigs: addTimeSigsTmpl},
|
|
{FuncName: "SubTime", Sigs: subTimeSigsTmpl},
|
|
{FuncName: "TimeDiff", Sigs: timeDiffSigsTmpl},
|
|
{FuncName: "AddDate", Sigs: addDataSigsTmpl},
|
|
{FuncName: "SubDate", Sigs: subDataSigsTmpl},
|
|
},
|
|
}
|
|
|
|
func generateDotGo(fileName string) error {
|
|
w := new(bytes.Buffer)
|
|
err := addOrSubTime.Execute(w, function{FuncName: "AddTime", Sigs: addTimeSigsTmpl})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = addOrSubTime.Execute(w, function{FuncName: "SubTime", Sigs: subTimeSigsTmpl})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = timeDiff.Execute(w, timeDiffSigsTmpl)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = addOrSubDate.Execute(w, function{FuncName: "AddDate", Sigs: addDataSigsTmpl})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = addOrSubDate.Execute(w, function{FuncName: "SubDate", Sigs: subDataSigsTmpl})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
data, err := format.Source(w.Bytes())
|
|
if err != nil {
|
|
log.Println("[Warn]", fileName+": gofmt failed", err)
|
|
data = w.Bytes() // write original data for debugging
|
|
}
|
|
return ioutil.WriteFile(fileName, data, 0644)
|
|
}
|
|
|
|
func generateTestDotGo(fileName string) error {
|
|
w := new(bytes.Buffer)
|
|
err := testFile.Execute(w, tmplVal)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
data, err := format.Source(w.Bytes())
|
|
if err != nil {
|
|
log.Println("[Warn]", fileName+": gofmt failed", err)
|
|
data = w.Bytes() // write original data for debugging
|
|
}
|
|
return ioutil.WriteFile(fileName, data, 0644)
|
|
}
|
|
|
|
// generateOneFile generate one xxx.go file and the associated xxx_test.go file.
|
|
func generateOneFile(fileNamePrefix string) (err error) {
|
|
|
|
err = generateDotGo(fileNamePrefix + ".go")
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = generateTestDotGo(fileNamePrefix + "_test.go")
|
|
return
|
|
}
|
|
|
|
func main() {
|
|
var err error
|
|
outputDir := "."
|
|
err = generateOneFile(filepath.Join(outputDir, "builtin_time_vec_generated"))
|
|
if err != nil {
|
|
log.Fatalln("generateOneFile", err)
|
|
}
|
|
}
|
|
|