Method CodegenWriteCollectionToSlot

be/src/runtime/descriptors.cc:1353–1392

} /* anonymous namespace */

void SlotDescriptor::CodegenWriteCollectionToSlot(
    const CodegenAnyValReadWriteInfo& read_write_info,
    llvm::Value* slot_ptr, llvm::Value* pool_val, const SlotDescriptor* slot_desc,
    const NonWritableBasicBlock& insert_before) {
  LlvmCodeGen* codegen = read_write_info.codegen();
  LlvmBuilder* builder = read_write_info.builder();
  const ColumnType& type = read_write_info.type();
  DCHECK(type.IsCollectionType());

  // Convert to 'CollectionValue'.
  llvm::Type* raw_type = codegen->GetSlotType(type);
  llvm::Value* coll_value = llvm::Constant::getNullValue(raw_type);
  coll_value = CodegenCollValueSetLen(codegen, builder, coll_value,
      read_write_info.GetPtrAndLen().len);
  if (pool_val != nullptr) {
    llvm::Value* num_tuples = read_write_info.GetPtrAndLen().len;
    DCHECK(slot_desc != nullptr) << "SlotDescriptor needed to calculate the size of "
        << "the collection for copying.";
    // For a 'CollectionValue', 'len' is not the byte size of the whole data but the
    // number of items, so we have to multiply it with the byte size of the item tuple
    // to get the data size.
    int item_tuple_byte_size = slot_desc->children_tuple_descriptor()->byte_size();
    llvm::Value* byte_len = builder->CreateMul(num_tuples,
        codegen->GetI32Constant(item_tuple_byte_size), "coll_tuple_byte_len");

    // Allocate a 'new_ptr' from 'pool_val' and copy the data from 'read_write_info->ptr'.
    llvm::Value* new_ptr = codegen->CodegenMemPoolAllocate(
        builder, pool_val, byte_len, "new_coll_val_ptr");
    codegen->CodegenMemcpy(builder, new_ptr, read_write_info.GetPtrAndLen().ptr,
        byte_len);
    coll_value = CodegenCollValueSetPtr(codegen, builder, coll_value, new_ptr);

    slot_desc->CodegenWriteCollectionItemsToSlot(codegen, builder, new_ptr,
        read_write_info.GetPtrAndLen().len, pool_val, insert_before);
  } else {
    coll_value = CodegenCollValueSetPtr(codegen, builder, coll_value,
        read_write_info.GetPtrAndLen().ptr);
  }
  builder->CreateStore(coll_value, slot_ptr);
}

void SlotDescriptor::CodegenWriteStringToSlot(
    const CodegenAnyValReadWriteInfo& read_write_info,