Skip to content

Commit

Permalink
[fix](round) fix round decimal128 overflow (apache#37733)
Browse files Browse the repository at this point in the history
  • Loading branch information
cambyzju committed Jul 17, 2024
1 parent 5f5ec1c commit b5b773f
Show file tree
Hide file tree
Showing 3 changed files with 41 additions and 4 deletions.
10 changes: 6 additions & 4 deletions be/src/vec/functions/round.h
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@

#include "vec/columns/column_const.h"
#include "vec/columns/columns_number.h"
#include "vec/exec/format/format_common.h"
#include "vec/functions/function.h"
#if defined(__SSE4_1__) || defined(__aarch64__)
#include "util/sse_util.hpp"
Expand Down Expand Up @@ -116,7 +117,7 @@ struct IntegerRoundingComputation {
__builtin_unreachable();
}

static ALWAYS_INLINE T compute(T x, T scale, size_t target_scale) {
static ALWAYS_INLINE T compute(T x, T scale, T target_scale) {
switch (scale_mode) {
case ScaleMode::Zero:
case ScaleMode::Positive:
Expand Down Expand Up @@ -153,21 +154,22 @@ class DecimalRoundingImpl {
Int16 out_scale) {
Int16 scale_arg = in_scale - out_scale;
if (scale_arg > 0) {
size_t scale = int_exp10(scale_arg);
auto scale = DecimalScaleParams::get_scale_factor<NativeType>(scale_arg);

const NativeType* __restrict p_in = reinterpret_cast<const NativeType*>(in.data());
const NativeType* end_in = reinterpret_cast<const NativeType*>(in.data()) + in.size();
NativeType* __restrict p_out = reinterpret_cast<NativeType*>(out.data());

if (out_scale < 0) {
auto negative_scale = DecimalScaleParams::get_scale_factor<NativeType>(-out_scale);
while (p_in < end_in) {
Op::compute(p_in, scale, p_out, int_exp10(-out_scale));
*p_out = Op::compute(*p_in, scale, negative_scale);
++p_in;
++p_out;
}
} else {
while (p_in < end_in) {
Op::compute(p_in, scale, p_out, 1);
*p_out = Op::compute(*p_in, scale, 1);
++p_in;
++p_out;
}
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
-- This file is automatically generated. You should know what you did if you want to edit this
-- !select1 --
186

-- !select2 --
0

-- !select3 --
20000000000000000000000

Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you 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, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.

suite("test_round_overflow") {
sql "SET enable_nereids_planner=true"
sql "SET enable_fallback_to_original_planner=false"

qt_select1 "select round(coalesce(186,-33280029.8473323000000000000000000));"
qt_select2 "select round(coalesce(186,-33280029.8473323000000000000000000), -20);"
qt_select3 "select round(coalesce(18618500001234567890123.0,-33280029.847332300000000),-22);"
}

0 comments on commit b5b773f

Please sign in to comment.