expression: go generate vectorized addtime functions

expression/builtin.go

@@ -16,6 +16,7 @@
 // limitations under the License.
 
 //go:generate go run generator/control_vec.go
+//go:generate go run generator/time_vec.go
 
 package expression
 

expression/builtin_cast_vec.go

@@ -143,3 +143,38 @@ func (b *builtinCastRealAsRealSig) vecEvalReal(input *chunk.Chunk, result *chunk
 func (b *builtinCastRealAsRealSig) vectorized() bool {
 	return true
 }
+
+func (b *builtinCastTimeAsDurationSig) vecEvalDuration(input *chunk.Chunk, result *chunk.Column) error {
+	n := input.NumRows()
+	arg0, err := b.bufAllocator.get(types.ETDatetime, n)
+	if err != nil {
+		return err
+	}
+	defer b.bufAllocator.put(arg0)
+	if err := b.args[0].VecEvalTime(b.ctx, input, arg0); err != nil {
+		return err
+	}
+	arg0s := arg0.Times()
+	result.ResizeGoDuration(n, false)
+	result.MergeNulls(arg0)
+	ds := result.GoDurations()
+	for i, t := range arg0s {
+		if result.IsNull(i) {
+			continue
+		}
+		d, err := t.ConvertToDuration()
+		if err != nil {
+			return err
+		}
+		d, err = d.RoundFrac(int8(b.tp.Decimal))
+		if err != nil {
+			return err
+		}
+		ds[i] = d.Duration
+	}
+	return nil
+}
+
+func (b *builtinCastTimeAsDurationSig) vectorized() bool {
+	return true
+}

expression/builtin_cast_vec_test.go

@@ -27,9 +27,29 @@ var vecBuiltinCastCases = map[string][]vecExprBenchCase{
 		{retEvalType: types.ETReal, childrenTypes: []types.EvalType{types.ETInt}},
 		{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETInt}, geners: []dataGenerator{new(randDurInt)}},
 		{retEvalType: types.ETReal, childrenTypes: []types.EvalType{types.ETReal}},
+		{retEvalType: types.ETDuration, childrenTypes: []types.EvalType{types.ETDatetime},
+			geners: []dataGenerator{&dateTimeGenerWithFsp{
+				defaultGener: defaultGener{nullRation: 0.2, eType: types.ETDatetime},
+				fsp:          1,
+			}},
+		},
 	},
 }
 
+type dateTimeGenerWithFsp struct {
+	defaultGener
+	fsp int8
+}
+
+func (g *dateTimeGenerWithFsp) gen() interface{} {
+	result := g.defaultGener.gen()
+	if t, ok := result.(types.Time); ok {
+		t.Fsp = g.fsp
+		return t
+	}
+	return result
+}
+
 func (s *testEvaluatorSuite) TestVectorizedBuiltinCastEvalOneVec(c *C) {
 	testVectorizedEvalOneVec(c, vecBuiltinCastCases)
 }