expression: implement vectorized evaluation for `builtinCurren… (ping…

…cap#12611)

Makefile

@@ -41,6 +41,11 @@ CHECK_LDFLAGS += $(LDFLAGS) ${TEST_LDFLAGS}
 
 TARGET = ""
 
+# VB = Vector Benchmark
+VB_FILE =
+VB_FUNC =
+
+
 .PHONY: all build update clean todo test gotest interpreter server dev benchkv benchraw check checklist parser tidy ddltest
 
 default: server buildsucc
@@ -266,3 +271,13 @@ tools/bin/misspell:tools/check/go.mod
 tools/bin/ineffassign:tools/check/go.mod
 	cd tools/check; \
 	$(GO) build -o ../bin/ineffassign github.com/gordonklaus/ineffassign
+
+# Usage:
+#
+# 	$ make vectorized-bench VB_FILE=Time VB_FUNC=builtinCurrentDateSig
+vectorized-bench:
+	cd ./expression && \
+		go test -v -benchmem \
+			-bench=BenchmarkVectorizedBuiltin$(VB_FILE)Func \
+			-run=BenchmarkVectorizedBuiltin$(VB_FILE)Func \
+			-args "$(VB_FUNC)"

expression/bench_test.go

@@ -701,11 +701,25 @@ func getColumnLen(col *chunk.Column, eType types.EvalType) int {
 	return chk.NumRows()
 }
 
+// removeTestOptions removes all not needed options like '-test.timeout=' from argument list
+func removeTestOptions(args []string) []string {
+	argList := args[:0]
+
+	// args contains '-test.timeout=' option for example
+	// excluding it to be able to run all tests
+	for _, arg := range args {
+		if strings.HasPrefix(arg, "builtin") {
+			argList = append(argList, arg)
+		}
+	}
+	return argList
+}
+
 // testVectorizedBuiltinFunc is used to verify that the vectorized
 // expression is evaluated correctly
 func testVectorizedBuiltinFunc(c *C, vecExprCases vecExprBenchCases) {
 	testFunc := make(map[string]bool)
-	argList := flag.Args()
+	argList := removeTestOptions(flag.Args())
 	testAll := len(argList) == 0
 	for _, arg := range argList {
 		testFunc[arg] = true
@@ -862,7 +876,7 @@ func testVectorizedBuiltinFunc(c *C, vecExprCases vecExprBenchCases) {
 func benchmarkVectorizedBuiltinFunc(b *testing.B, vecExprCases vecExprBenchCases) {
 	ctx := mock.NewContext()
 	testFunc := make(map[string]bool)
-	argList := flag.Args()
+	argList := removeTestOptions(flag.Args())
 	testAll := len(argList) == 0
 	for _, arg := range argList {
 		testFunc[arg] = true

expression/builtin_time_vec.go

@@ -790,11 +790,29 @@ func (b *builtinDateDiffSig) vecEvalInt(input *chunk.Chunk, result *chunk.Column
 }
 
 func (b *builtinCurrentDateSig) vectorized() bool {
-	return false
+	return true
 }
 
 func (b *builtinCurrentDateSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
-	return errors.Errorf("not implemented")
+	nowTs, err := getStmtTimestamp(b.ctx)
+	if err != nil {
+		return err
+	}
+
+	tz := b.ctx.GetSessionVars().Location()
+	year, month, day := nowTs.In(tz).Date()
+	timeValue := types.Time{
+		Time: types.FromDate(year, int(month), day, 0, 0, 0, 0),
+		Type: mysql.TypeDate,
+		Fsp:  0}
+
+	n := input.NumRows()
+	result.ResizeTime(n, false)
+	times := result.Times()
+	for i := 0; i < n; i++ {
+		times[i] = timeValue
+	}
+	return nil
 }
 
 func (b *builtinAddDateStringStringSig) vectorized() bool {

expression/builtin_time_vec_test.go

@@ -38,12 +38,14 @@ var vecBuiltinTimeCases = map[string][]vecExprBenchCase{
 	ast.DayOfYear:   {},
 	ast.Day:         {},
 	ast.CurrentTime: {},
-	ast.CurrentDate: {},
-	ast.MakeDate:    {},
-	ast.MakeTime:    {},
-	ast.PeriodAdd:   {},
-	ast.PeriodDiff:  {},
-	ast.Quarter:     {},
+	ast.CurrentDate: {
+		{retEvalType: types.ETDatetime},
+	},
+	ast.MakeDate:   {},
+	ast.MakeTime:   {},
+	ast.PeriodAdd:  {},
+	ast.PeriodDiff: {},
+	ast.Quarter:    {},
 	ast.TimeFormat: {
 		{retEvalType: types.ETString, childrenTypes: []types.EvalType{types.ETDuration, types.ETString}, geners: []dataGenerator{&rangeDurationGener{0.5}, &timeFormatGener{0.5}}},
 	},