ArrowResultSet.cpp File Reference

#include "QueryEngine/ArrowResultSet.h"
#include <arrow/api.h>
#include <arrow/io/memory.h>
#include <arrow/ipc/api.h>
#include "QueryEngine/Descriptors/RelAlgExecutionDescriptor.h"
#include "Shared/ArrowUtil.h"

Include dependency graph for ArrowResultSet.cpp:

Go to the source code of this file.

Namespaces
	anonymous_namespace{ArrowResultSet.cpp}

Functions
SQLTypeInfo	anonymous_namespace{ArrowResultSet.cpp}::type_from_arrow_field (const arrow::Field &field)
std::unique_ptr< ArrowResultSet >	result_set_arrow_loopback (const ExecutionResult &results)
std::unique_ptr< ArrowResultSet >	result_set_arrow_loopback (const ExecutionResult *results, const std::shared_ptr< ResultSet > &rows, const ExecutorDeviceType device_type)
std::unique_ptr< ArrowResultSet >	result_set_arrow_loopback (const ExecutionResult *results, const std::shared_ptr< ResultSet > &rows, const ExecutorDeviceType device_type, const size_t min_result_size_for_bulk_dictionary_fetch, const double max_dictionary_to_result_size_ratio_for_bulk_dictionary_fetch)

Function Documentation

std::unique_ptr<ArrowResultSet> result_set_arrow_loopback

(

const ExecutionResult &

results

)

Definition at line 328 of file ArrowResultSet.cpp.

References ExecutionResult::getRows(), and ExecutionResult::getTargetsMeta().

                                    {
  // NOTE(wesm): About memory ownership

  // After calling ReadRecordBatch, the buffers inside arrow::RecordBatch now
  // share ownership of the memory in serialized_arrow_output.records (zero
  // copy). Not necessary to retain these buffers. Same is true of any
  // dictionaries contained in serialized_arrow_output.schema; the arrays
  // reference that memory (zero copy).
  return std::make_unique<ArrowResultSet>(results.getRows(), results.getTargetsMeta());
}

Here is the call graph for this function:

std::unique_ptr<ArrowResultSet> result_set_arrow_loopback	(	const ExecutionResult *	results,
		const std::shared_ptr< ResultSet > &	rows,
		const ExecutorDeviceType	device_type
	)

Definition at line 340 of file ArrowResultSet.cpp.

References ExecutionResult::getTargetsMeta().

                                          {
  return results ? std::make_unique<ArrowResultSet>(
                       rows, results->getTargetsMeta(), device_type)
                 : std::make_unique<ArrowResultSet>(rows, device_type);
}

Here is the call graph for this function:

std::unique_ptr<ArrowResultSet> result_set_arrow_loopback	(	const ExecutionResult *	results,
		const std::shared_ptr< ResultSet > &	rows,
		const ExecutorDeviceType	device_type,
		const size_t	min_result_size_for_bulk_dictionary_fetch,
		const double	max_dictionary_to_result_size_ratio_for_bulk_dictionary_fetch
	)

Definition at line 349 of file ArrowResultSet.cpp.

References ExecutionResult::getTargetsMeta().

                                                                                {
  std::vector<TargetMetaInfo> dummy_targets_meta;
  return results ? std::make_unique<ArrowResultSet>(
                       rows,
                       results->getTargetsMeta(),
                       device_type,
                       min_result_size_for_bulk_dictionary_fetch,
                       max_dictionary_to_result_size_ratio_for_bulk_dictionary_fetch)
                 : std::make_unique<ArrowResultSet>(
                       rows,
                       dummy_targets_meta,
                       device_type,
                       min_result_size_for_bulk_dictionary_fetch,
                       max_dictionary_to_result_size_ratio_for_bulk_dictionary_fetch);
}

Here is the call graph for this function: