Method WriteStringOrBinarySlot

be/src/exec/paimon/paimon-jni-row-reader.cc:286–314

/// memory later.
template <bool IS_BINARY>
Status PaimonJniRowReader::WriteStringOrBinarySlot(
    const arrow::Array* array, const int row_idx, void* slot, MemPool* tuple_data_pool) {
  std::string_view v;
  uint32_t jbuffer_size = 0;
  if constexpr (IS_BINARY) {
    const arrow::BinaryArray* binary_array =
        static_cast<const arrow::BinaryArray*>(array);
    v = binary_array->Value(row_idx);
  } else {
    const arrow::StringArray* string_array =
        static_cast<const arrow::StringArray*>(array);
    v = string_array->Value(row_idx);
  }

  jbuffer_size = v.size();
  // Allocate memory and copy the bytes from the JVM to the RowBatch.
  char* buffer =
      reinterpret_cast<char*>(tuple_data_pool->TryAllocateUnaligned(jbuffer_size));
  if (UNLIKELY(buffer == nullptr)) {
    string details = strings::Substitute("Failed to allocate $0 bytes for $1.",
        jbuffer_size, IS_BINARY ? "binary" : "string");
    return tuple_data_pool->mem_tracker()->MemLimitExceeded(
        nullptr, details, jbuffer_size);
  }

  memcpy(buffer, v.data(), jbuffer_size);
  reinterpret_cast<StringValue*>(slot)->Assign(buffer, jbuffer_size);
  return Status::OK();
}
} // namespace impala