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)

Function Documentation

result_set_arrow_loopback() [1/2]

std::unique_ptr<ArrowResultSet> result_set_arrow_loopback

(

const ExecutionResult &

results

)

Definition at line 260 of file ArrowResultSet.cpp.

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

Referenced by ArrowResultSetRowIterator::operator*().

                                    {
  // 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:

Here is the caller graph for this function:

result_set_arrow_loopback() [2/2]

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

Definition at line 272 of file ArrowResultSet.cpp.

References ExecutionResult::getTargetsMeta().

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

Here is the call graph for this function: