diff --git a/expression/builtin_math_vec.go b/expression/builtin_math_vec.go index fe08b43cc1..66ee1cb65a 100644 --- a/expression/builtin_math_vec.go +++ b/expression/builtin_math_vec.go @@ -697,11 +697,38 @@ func (b *builtinCeilIntToDecSig) vecEvalDecimal(input *chunk.Chunk, result *chun } func (b *builtinTruncateIntSig) vectorized() bool { - return false + return true } func (b *builtinTruncateIntSig) vecEvalInt(input *chunk.Chunk, result *chunk.Column) error { - return errors.Errorf("not implemented") + if err := b.args[0].VecEvalInt(b.ctx, input, result); err != nil { + return err + } + + n := input.NumRows() + buf, err := b.bufAllocator.get(types.ETInt, n) + if err != nil { + return err + } + defer b.bufAllocator.put(buf) + + if err := b.args[1].VecEvalInt(b.ctx, input, buf); err != nil { + return err + } + result.MergeNulls(buf) + i64s := result.Int64s() + buf64s := buf.Int64s() + + for i := 0; i < len(i64s); i++ { + if result.IsNull(i) { + continue + } + if buf64s[i] < 0 { + shift := int64(math.Pow10(int(-buf64s[i]))) + i64s[i] = i64s[i] / shift * shift + } + } + return nil } func (b *builtinTruncateUintSig) vectorized() bool { diff --git a/expression/builtin_math_vec_test.go b/expression/builtin_math_vec_test.go index 4f93984862..3b64d8c073 100644 --- a/expression/builtin_math_vec_test.go +++ b/expression/builtin_math_vec_test.go @@ -97,6 +97,7 @@ var vecBuiltinMathCases = map[string][]vecExprBenchCase{ }, ast.Truncate: { {retEvalType: types.ETReal, childrenTypes: []types.EvalType{types.ETReal, types.ETInt}, geners: []dataGenerator{nil, &rangeInt64Gener{-10, 10}}}, + {retEvalType: types.ETInt, childrenTypes: []types.EvalType{types.ETInt, types.ETInt}, geners: []dataGenerator{nil, &rangeInt64Gener{-10, 10}}}, }, }