expression:implement vectorized evaluation for builtinExtractDurationSig

expression/bench_test.go

@@ -643,6 +643,8 @@ type vecExprBenchCase struct {
 	// geners[gen1, gen2] will be regarded as geners[gen1, gen2, nil].
 	// This field is optional.
 	geners []dataGenerator
+	// constants are used to generate constant data for children[i].
+	constants []*Constant
 }
 
 type vecExprBenchCases map[string][]vecExprBenchCase

expression/builtin_time.go

@@ -2531,13 +2531,16 @@ func (c *extractFunctionClass) getFunction(ctx sessionctx.Context, args []Expres
 		"YEAR_MONTH":      {},
 	}
 	isDatetimeUnit := true
+	fmt.Println("args0--", args[0].ConstItem())
 	args[0] = WrapWithCastAsString(ctx, args[0])
+	fmt.Println("args0--", args[0].ConstItem())
 	if _, isCon := args[0].(*Constant); isCon {
 		unit, _, err1 := args[0].EvalString(ctx, chunk.Row{})
 		if err1 != nil {
 			return nil, err1
 		}
 		_, isDatetimeUnit = datetimeUnits[unit]
+		fmt.Println("unit---", unit)
 	}
 	var bf baseBuiltinFunc
 	if isDatetimeUnit {

expression/builtin_time_vec.go

@@ -935,11 +935,45 @@ func (b *builtinWeekWithModeSig) vecEvalInt(input *chunk.Chunk, result *chunk.Co
 }
 
 func (b *builtinExtractDurationSig) vectorized() bool {
-	return false
+	return true
 }
 
 func (b *builtinExtractDurationSig) vecEvalInt(input *chunk.Chunk, result *chunk.Column) error {
-	return errors.Errorf("not implemented")
+	n := input.NumRows()
+	unit, err := b.bufAllocator.get(types.ETString, n)
+	if err != nil {
+		return err
+	}
+	defer b.bufAllocator.put(unit)
+	if err := b.args[0].VecEvalString(b.ctx, input, unit); err != nil {
+		return err
+	}
+	dur, err := b.bufAllocator.get(types.ETDuration, n)
+	if err != nil {
+		return err
+	}
+	defer b.bufAllocator.put(dur)
+	if err = b.args[1].VecEvalDuration(b.ctx, input, dur); err != nil {
+		return err
+	}
+	result.ResizeInt64(n, false)
+	result.MergeNulls(unit, dur)
+	i64s := result.Int64s()
+	durIs := dur.GoDurations()
+	var duration types.Duration
+	for i := 0; i < n; i++ {
+		if result.IsNull(i) {
+			continue
+		}
+		unitI := unit.GetString(i)
+		duration.Duration = durIs[i]
+		duration.Fsp = int8(types.UnspecifiedFsp)
+		i64s[i], err = types.ExtractDurationNum(&duration, unitI)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
 }
 
 func (b *builtinStrToDateDurationSig) vectorized() bool {

expression/builtin_time_vec_test.go

@@ -98,6 +98,11 @@ var vecBuiltinTimeCases = map[string][]vecExprBenchCase{
 		},
 	},
 	ast.MakeTime: {},
+	ast.Extract: {
+		{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETString, types.ETDuration},
+			constants: []*Constant{&Constant{Value: types.NewStringDatum("test"), RetType: types.NewFieldType(mysql.TypeString)}, nil},
+		},
+	},
 	ast.PeriodAdd: {
 		{retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETInt, types.ETInt}, geners: []dataGenerator{new(periodGener), new(periodGener)}},
 	},