Fix comment

PHILO-HE · Nov 29, 2024 · ef14f24 · ef14f24
1 parent 61ffd2e
commit ef14f24
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/velox/functions/sparksql/ConcatWs.cpp b/velox/functions/sparksql/ConcatWs.cpp
@@ -38,7 +38,7 @@ class ConcatWs : public exec::VectorFunction {
       const std::vector<std::optional<std::string>>& constantStrings,
       const std::vector<exec::LocalDecodedVector>& decodedStringArgs,
       const exec::LocalDecodedVector& decodedSeparator) const {
-    size_t totalResultBytes = 0;
+    uint64_t totalResultBytes = 0;
     rows.applyToSelected([&](auto row) {
       // NULL separator produces NULL result.
       if (!isConstantSeparator() && decodedSeparator->isNullAt(row)) {
@@ -95,6 +95,7 @@ class ConcatWs : public exec::VectorFunction {
         totalResultBytes += (allElements - 1) * separatorSize;
       }
     });
+    VELOX_USER_CHECK_LE(totalResultBytes, UINT32_MAX);
     return totalResultBytes;
   }