[Fix]fix_partition_column_orders_issue.

apache · May 24, 2024 · de0dffc · de0dffc
1 parent b2edc80
commit de0dffc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/be/src/vec/sink/writer/vhive_partition_writer.cpp b/be/src/vec/sink/writer/vhive_partition_writer.cpp
@@ -189,7 +189,7 @@ Status VHivePartitionWriter::_projection_and_filter_block(doris::vectorized::Blo
     }
 
     std::vector<uint32_t> columns_to_filter;
-    int column_to_keep = input_block.columns();
+    int column_to_keep = output_block.columns();
     columns_to_filter.resize(column_to_keep);
     for (uint32_t i = 0; i < column_to_keep; ++i) {
         columns_to_filter[i] = i;

diff --git a/be/src/vec/sink/writer/vhive_table_writer.cpp b/be/src/vec/sink/writer/vhive_table_writer.cpp
@@ -323,8 +323,12 @@ std::vector<std::string> VHiveTableWriter::_create_partition_values(vectorized::
     std::vector<std::string> partition_values;
     for (int i = 0; i < _partition_columns_input_index.size(); ++i) {
         int partition_column_idx = _partition_columns_input_index[i];
+        DCHECK(_vec_output_expr_ctxs[partition_column_idx]->root()->is_slot_ref());
+        int partition_column_pos_in_block =
+                ((vectorized::VSlotRef*)_vec_output_expr_ctxs[partition_column_idx]->root().get())
+                        ->column_id();
         vectorized::ColumnWithTypeAndName partition_column =
-                block.get_by_position(partition_column_idx);
+                block.get_by_position(partition_column_pos_in_block);
         std::string value =
                 _to_partition_value(_vec_output_expr_ctxs[partition_column_idx]->root()->type(),
                                     partition_column, position);