[GLUTEN-5009][CH] Fix TPCDS q9 failed with debug build (#5015)

apache · Mar 20, 2024 · 8dbca6c · 8dbca6c
1 parent 01cce51
commit 8dbca6c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/cpp-ch/local-engine/Parser/CHColumnToSparkRow.cpp b/cpp-ch/local-engine/Parser/CHColumnToSparkRow.cpp
@@ -696,6 +696,12 @@ int64_t VariableLengthDataWriter::writeArray(size_t row_idx, const DB::Array & a
                     writer.unsafeWrite(
                         reinterpret_cast<const char *>(&v), buffer_address + offset + start + 8 + len_null_bitmap + i * elem_size);
                 }
+                else if (writer.getWhichDataType().isFloat64())
+                {
+                    // Fix 'Invalid Field get from type Float64 to type Int64' in debug build.
+                    auto v = elem.get<Float64>();
+                    writer.unsafeWrite(reinterpret_cast<const char *>(&v), buffer_address + offset + start + 8 + len_null_bitmap + i * elem_size);
+                }
                 else
                     writer.unsafeWrite(
                         reinterpret_cast<const char *>(&elem.get<char>()),
@@ -808,6 +814,12 @@ int64_t VariableLengthDataWriter::writeStruct(size_t row_idx, const DB::Tuple &
                 auto v = static_cast<Float32>(field_value.get<Float32>());
                 writer.unsafeWrite(reinterpret_cast<const char *>(&v), buffer_address + offset + start + len_null_bitmap + i * 8);
             }
+            else if (writer.getWhichDataType().isFloat64())
+            {
+                // Fix 'Invalid Field get from type Float64 to type Int64' in debug build.
+                auto v = field_value.get<Float64>();
+                writer.unsafeWrite(reinterpret_cast<const char *>(&v), buffer_address + offset + start + len_null_bitmap + i * 8);
+            }
             else
                 writer.unsafeWrite(
                     reinterpret_cast<const char *>(&field_value.get<char>()), buffer_address + offset + start + len_null_bitmap + i * 8);