ArrowResultSet.h File Reference

#include "CompilationOptions.h"
#include "DataMgr/DataMgr.h"
#include "Descriptors/RelAlgExecutionDescriptor.h"
#include "ResultSet.h"
#include "TargetMetaInfo.h"
#include "TargetValue.h"
#include <type_traits>
#include "arrow/api.h"
#include "arrow/ipc/api.h"

Include dependency graph for ArrowResultSet.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	ArrowResultSetRowIterator
struct	ArrowResult
class	ArrowResultSet
class	ArrowResultSetConverter
struct	ArrowResultSetConverter::SerializedArrowOutput
struct	ArrowResultSetConverter::ColumnBuilder

Typedefs
using	ValueArray = boost::variant< std::vector< bool >, std::vector< int8_t >, std::vector< int16_t >, std::vector< int32_t >, std::vector< int64_t >, std::vector< float >, std::vector< double >, std::vector< std::string > >

Functions
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)
template<typename T >
constexpr auto	scale_epoch_values ()

Typedef Documentation

ValueArray

using ValueArray = boost::variant<std::vector<bool>, std::vector<int8_t>, std::vector<int16_t>, std::vector<int32_t>, std::vector<int64_t>, std::vector<float>, std::vector<double>, std::vector<std::string> >

Definition at line 39 of file ArrowResultSet.h.

Function Documentation

result_set_arrow_loopback() [1/2]

std::unique_ptr<ArrowResultSet> result_set_arrow_loopback

(

const ExecutionResult &

results

)

Definition at line 234 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 246 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:

scale_epoch_values()

template<typename T >

constexpr auto scale_epoch_values

(

)

Definition at line 242 of file ArrowResultSet.h.

                                    {
  return std::is_same<T, arrow::Date32Builder>::value ||
         std::is_same<T, arrow::Date64Builder>::value;
}