Method Prepare

be/src/exec/hdfs-table-sink.cc:103–132

}

Status HdfsTableSink::Prepare(RuntimeState* state, MemTracker* parent_mem_tracker) {
  SCOPED_TIMER(profile()->total_time_counter());
  RETURN_IF_ERROR(TableSinkBase::Prepare(state, parent_mem_tracker));
  unique_id_str_ = PrintId(state->fragment_instance_id(), "-");

  // Resolve table id and set input tuple descriptor.
  table_desc_ = static_cast<const HdfsTableDescriptor*>(
      state->desc_tbl().GetTableDescriptor(table_id_));

  if (table_desc_ == nullptr) {
    stringstream error_msg;
    error_msg << "Failed to get table descriptor for table id: " << table_id_;
    return Status(error_msg.str());
  }

  staging_dir_ = Substitute("$0/_impala_insert_staging/$1", table_desc_->hdfs_base_dir(),
      PrintId(state->query_id(), "_"));

  // Sanity check.
  if (!IsIceberg()) {
    DCHECK_LE(partition_key_expr_evals_.size(), table_desc_->num_cols())
        << DebugString();
    DCHECK_EQ(partition_key_expr_evals_.size(), table_desc_->num_clustering_cols())
        << DebugString();
    DCHECK_GE(output_expr_evals_.size(),
      table_desc_->num_cols() - table_desc_->num_clustering_cols()) << DebugString();
  }

  return Status::OK();
}

void HdfsTableSink::BuildPartitionDescMap() {
  for (const HdfsTableDescriptor::PartitionIdToDescriptorMap::value_type& id_to_desc: