fix round(5) crash, check type error (#5967)

fix round(5) crash, add check type
vesoft-inc · Nov 11, 2024 · 6c59b79 · 6c59b79
1 parent b4f83eb
commit 6c59b79
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/common/function/FunctionManager.cpp b/src/common/function/FunctionManager.cpp
@@ -659,7 +659,12 @@ FunctionManager::FunctionManager() {
               return Value::kNullBadType;
             }
           }
-          return std::round(args[0].get().getFloat());
+          if (args[0].get().type() == Value::Type::INT) {
+            return std::round(args[0].get().getInt());
+          } else if (args[0].get().type() == Value::Type::FLOAT) {
+            return std::round(args[0].get().getFloat());
+          }
+          return Value::kNullBadType;
         }
         default: {
           return Value::kNullBadType;