Impala
Impalaistheopensource,nativeanalyticdatabaseforApacheHadoop.
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
impala::HdfsParquetScanner Class Reference

#include <hdfs-parquet-scanner.h>

Inheritance diagram for impala::HdfsParquetScanner:
Collaboration diagram for impala::HdfsParquetScanner:

Classes

class  BaseColumnReader
 
class  BoolColumnReader
 
class  ColumnReader
 
struct  FileVersion
 
struct  SchemaNode
 Internal representation of a column schema (including nested-type columns). More...
 

Public Member Functions

 HdfsParquetScanner (HdfsScanNode *scan_node, RuntimeState *state)
 
virtual ~HdfsParquetScanner ()
 
virtual Status Prepare (ScannerContext *context)
 One-time initialisation of state that is constant across scan ranges. More...
 
virtual void Close ()
 
virtual Status ProcessSplit ()
 

Static Public Member Functions

static Status IssueInitialRanges (HdfsScanNode *scan_node, const std::vector< HdfsFileDesc * > &files)
 

Static Public Attributes

static const int FILE_BLOCK_SIZE = 4096
 
static const char * LLVM_CLASS_NAME = "class.impala::HdfsScanner"
 

Protected Types

typedef int(* WriteTuplesFn )(HdfsScanner *, MemPool *, TupleRow *, int, FieldLocation *, int, int, int, int)
 

Protected Member Functions

Status InitializeWriteTuplesFn (HdfsPartitionDescriptor *partition, THdfsFileFormat::type type, const std::string &scanner_name)
 
void StartNewRowBatch ()
 Set batch_ to a new row batch and update tuple_mem_ accordingly. More...
 
int GetMemory (MemPool **pool, Tuple **tuple_mem, TupleRow **tuple_row_mem)
 
Status CommitRows (int num_rows)
 
void AddFinalRowBatch ()
 
void AttachPool (MemPool *pool, bool commit_batch)
 
bool IR_ALWAYS_INLINE EvalConjuncts (TupleRow *row)
 
int WriteEmptyTuples (RowBatch *row_batch, int num_tuples)
 
int WriteEmptyTuples (ScannerContext *context, TupleRow *tuple_row, int num_tuples)
 Write empty tuples and commit them to the context object. More...
 
int WriteAlignedTuples (MemPool *pool, TupleRow *tuple_row_mem, int row_size, FieldLocation *fields, int num_tuples, int max_added_tuples, int slots_per_tuple, int row_start_indx)
 
Status UpdateDecompressor (const THdfsCompression::type &compression)
 
Status UpdateDecompressor (const std::string &codec)
 
bool ReportTupleParseError (FieldLocation *fields, uint8_t *errors, int row_idx)
 
virtual void LogRowParseError (int row_idx, std::stringstream *)
 
bool WriteCompleteTuple (MemPool *pool, FieldLocation *fields, Tuple *tuple, TupleRow *tuple_row, Tuple *template_tuple, uint8_t *error_fields, uint8_t *error_in_row)
 
void ReportColumnParseError (const SlotDescriptor *desc, const char *data, int len)
 
void InitTuple (Tuple *template_tuple, Tuple *tuple)
 
Tuplenext_tuple (Tuple *t) const
 
TupleRownext_row (TupleRow *r) const
 
ExprContextGetConjunctCtx (int idx) const
 

Static Protected Member Functions

static llvm::Function * CodegenWriteCompleteTuple (HdfsScanNode *, LlvmCodeGen *, const std::vector< ExprContext * > &conjunct_ctxs)
 
static llvm::Function * CodegenWriteAlignedTuples (HdfsScanNode *, LlvmCodeGen *, llvm::Function *write_tuple_fn)
 

Protected Attributes

HdfsScanNodescan_node_
 The scan node that started this scanner. More...
 
RuntimeStatestate_
 RuntimeState for error reporting. More...
 
ScannerContextcontext_
 Context for this scanner. More...
 
ScannerContext::Streamstream_
 The first stream for context_. More...
 
std::vector< ExprContext * > conjunct_ctxs_
 
Tupletemplate_tuple_
 
int tuple_byte_size_
 Fixed size of each tuple, in bytes. More...
 
Tupletuple_
 Current tuple pointer into tuple_mem_. More...
 
RowBatchbatch_
 
uint8_t * tuple_mem_
 The tuple memory of batch_. More...
 
int num_errors_in_file_
 number of errors in current file More...
 
boost::scoped_ptr< TextConvertertext_converter_
 Helper class for converting text to other types;. More...
 
int32_t num_null_bytes_
 Number of null bytes in the tuple. More...
 
boost::scoped_ptr< Codecdecompressor_
 Decompressor class to use, if any. More...
 
THdfsCompression::type decompression_type_
 The most recently used decompression type. More...
 
boost::scoped_ptr< MemPooldata_buffer_pool_
 
RuntimeProfile::Counterdecompress_timer_
 Time spent decompressing bytes. More...
 
WriteTuplesFn write_tuples_fn_
 Jitted write tuples function pointer. Null if codegen is disabled. More...
 

Private Member Functions

Status AssembleRows (int row_group_idx)
 
Status ProcessFooter (bool *eosr)
 
Status CreateColumnReaders ()
 
BaseColumnReaderCreateReader (const SchemaNode &node)
 
Status InitColumns (int row_group_idx)
 
Status ValidateFileMetadata ()
 Validates the file metadata. More...
 
Status ValidateColumn (const BaseColumnReader &col_reader, int row_group_idx)
 
Status InitNewRange ()
 Part of the HdfsScanner interface, not used in Parquet. More...
 
Status CreateSchemaTree (const std::vector< parquet::SchemaElement > &schema, SchemaNode *node) const
 
Status CreateSchemaTree (const std::vector< parquet::SchemaElement > &schema, int max_def_level, int *idx, int *col_idx, SchemaNode *node) const
 Recursive implementation used internally by the above CreateSchemaTree() function. More...
 

Private Attributes

std::vector< BaseColumnReader * > column_readers_
 Column reader for each materialized columns for this file. More...
 
parquet::FileMetaData file_metadata_
 File metadata thrift object. More...
 
FileVersion file_version_
 Version of the application that wrote this file. More...
 
SchemaNode schema_
 The root schema node for this file. More...
 
const DiskIoMgr::ScanRangemetadata_range_
 Scan range for the metadata. More...
 
Status parse_status_
 Returned in ProcessSplit. More...
 
boost::scoped_ptr< MemPooldictionary_pool_
 
ScopedTimer< MonotonicStopWatchassemble_rows_timer_
 Timer for materializing rows. This ignores time getting the next buffer. More...
 
RuntimeProfile::Counternum_cols_counter_
 Number of cols that need to be read. More...
 

Static Private Attributes

static const int FOOTER_SIZE = 100 * 1024
 

Friends

class BaseColumnReader
 
template<typename T >
class ColumnReader
 
class BoolColumnReader
 

Detailed Description

This scanner parses Parquet files located in HDFS, and writes the content as tuples in the Impala in-memory representation of data, e.g. (tuples, rows, row batches). For the file format spec, see: github.com/Parquet/parquet-format Parquet (and other columnar formats) use scanner ranges differently than other formats. Each materialized column maps to a single ScanRange. For streaming reads, all the columns need to be read in parallel. This is done by issuing one ScanRange (in IssueInitialRanges()) for the file footer as the other scanners do. This footer range is processed in ProcessSplit(). ProcessSplit() then computes the column ranges and submits them to the IoMgr for immediate scheduling (so they don't surface in DiskIoMgr::GetNextRange()). Scheduling them immediately also guarantees they are all read at once. Like the other scanners, each parquet scanner object is one to one with a ScannerContext. Unlike the other scanners though, the context will have multiple streams, one for each column.

Definition at line 43 of file hdfs-parquet-scanner.h.

Member Typedef Documentation

typedef int(* impala::HdfsScanner::WriteTuplesFn)(HdfsScanner *, MemPool *, TupleRow *, int, FieldLocation *, int, int, int, int)
protectedinherited

Matching typedef for WriteAlignedTuples for codegen. Refer to comments for that function.

Definition at line 212 of file hdfs-scanner.h.

Constructor & Destructor Documentation

impala::HdfsParquetScanner::HdfsParquetScanner ( HdfsScanNode scan_node,
RuntimeState state 
)

Definition at line 125 of file hdfs-parquet-scanner.cc.

References assemble_rows_timer_.

impala::HdfsParquetScanner::~HdfsParquetScanner ( )
virtual

Definition at line 133 of file hdfs-parquet-scanner.cc.

Member Function Documentation

void HdfsScanner::AddFinalRowBatch ( )
protectedinherited

Attach all remaining resources from context_ to batch_ and send batch_ to the scan node. This must be called after all rows have been committed and no further resources are needed from context_ (in practice this will happen in each scanner subclass's Close() implementation).

Definition at line 145 of file hdfs-scanner.cc.

References impala::HdfsScanNode::AddMaterializedRowBatch(), impala::HdfsScanner::batch_, impala::HdfsScanner::context_, impala::ScannerContext::ReleaseCompletedResources(), and impala::HdfsScanner::scan_node_.

Referenced by impala::HdfsTextScanner::Close(), impala::BaseSequenceScanner::Close(), and Close().

void impala::HdfsScanner::AttachPool ( MemPool pool,
bool  commit_batch 
)
inlineprotectedinherited

Release all memory in 'pool' to batch_. If commit_batch is true, the row batch will be committed. commit_batch should be true if the attached pool is expected to be non-trivial (i.e. a decompression buffer) to minimize scanner mem usage.

Definition at line 256 of file hdfs-scanner.h.

References impala::MemPool::AcquireData(), impala::HdfsScanner::batch_, impala::HdfsScanner::CommitRows(), and impala::RowBatch::tuple_data_pool().

Referenced by impala::HdfsTextScanner::Close(), impala::BaseSequenceScanner::Close(), Close(), impala::HdfsTextScanner::FillByteBufferGzip(), impala::HdfsAvroScanner::ProcessRange(), impala::HdfsSequenceScanner::ReadCompressedBlock(), impala::HdfsParquetScanner::BaseColumnReader::ReadDataPage(), and impala::HdfsRCFileScanner::ResetRowGroup().

void HdfsParquetScanner::Close ( )
virtual

Release all resources the scanner has allocated. This is the last chance for the scanner to attach any resources to the ScannerContext object.

Reimplemented from impala::HdfsScanner.

Definition at line 449 of file hdfs-parquet-scanner.cc.

References impala::HdfsScanner::AddFinalRowBatch(), assemble_rows_timer_, impala::HdfsScanner::AttachPool(), impala::HdfsScanner::Close(), column_readers_, dictionary_pool_, impala::HdfsScanNode::RangeComplete(), and impala::HdfsScanner::scan_node_.

Function * HdfsScanner::CodegenWriteAlignedTuples ( HdfsScanNode ,
LlvmCodeGen ,
llvm::Function *  write_tuple_fn 
)
staticprotectedinherited

Codegen function to replace WriteAlignedTuples. WriteAlignedTuples is cross compiled to IR. This function loads the precompiled IR function, modifies it and returns the resulting function.

Definition at line 495 of file hdfs-scanner.cc.

References impala::LlvmCodeGen::codegen_timer(), impala::LlvmCodeGen::FinalizeFunction(), impala::LlvmCodeGen::GetFunction(), impala::LlvmCodeGen::ReplaceCallSites(), and SCOPED_TIMER.

Referenced by impala::HdfsTextScanner::Codegen(), and impala::HdfsSequenceScanner::Codegen().

Function * HdfsScanner::CodegenWriteCompleteTuple ( HdfsScanNode ,
LlvmCodeGen ,
const std::vector< ExprContext * > &  conjunct_ctxs 
)
staticprotectedinherited

Codegen function to replace WriteCompleteTuple. Should behave identically to WriteCompleteTuple.

Definition at line 296 of file hdfs-scanner.cc.

References impala::LlvmCodeGen::FnPrototype::AddArgument(), impala::TupleDescriptor::byte_size(), impala::LlvmCodeGen::codegen_timer(), impala::LlvmCodeGen::CodegenMemcpy(), impala::TextConverter::CodegenWriteSlot(), impala::HdfsScanNode::ComputeSlotMaterializationOrder(), impala::LlvmCodeGen::context(), impala::CodegenAnyVal::CreateCallWrapped(), impala::LlvmCodeGen::false_value(), impala::LlvmCodeGen::FinalizeFunction(), impala::TupleDescriptor::GenerateLlvmStruct(), impala::Status::GetDetail(), impala::LlvmCodeGen::GetFunction(), impala::LlvmCodeGen::GetIntConstant(), impala::LlvmCodeGen::GetType(), impala::CodegenAnyVal::GetVal(), impala::HdfsScanNode::hdfs_table(), impala::FieldLocation::LLVM_CLASS_NAME, impala::TupleRow::LLVM_CLASS_NAME, impala::Tuple::LLVM_CLASS_NAME, impala::HdfsScanner::LLVM_CLASS_NAME, impala::MemPool::LLVM_CLASS_NAME, impala::HdfsScanNode::materialized_slots(), impala::HdfsTableDescriptor::null_column_value(), impala::HdfsScanNode::num_materialized_partition_keys(), impala::TupleDescriptor::num_null_bytes(), impala::Status::ok(), impala::LlvmCodeGen::OptimizeFunctionWithExprs(), impala::HdfsScanNode::runtime_state(), SCOPED_TIMER, impala::LlvmCodeGen::true_value(), impala::HdfsScanNode::tuple_desc(), impala::HdfsScanNode::tuple_idx(), impala::ColumnType::type, impala::SlotDescriptor::type(), impala::TYPE_BOOLEAN, impala::TYPE_DECIMAL, impala::TYPE_INT, impala::TYPE_TIMESTAMP, and impala::TYPE_TINYINT.

Referenced by impala::HdfsTextScanner::Codegen(), and impala::HdfsSequenceScanner::Codegen().

Status HdfsScanner::CommitRows ( int  num_rows)
protectedinherited

Commit num_rows to the current row batch. If this completes, the row batch is enqueued with the scan node and StartNewRowBatch() is called. Returns Status::OK if the query is not cancelled and hasn't exceeded any mem limits. Scanner can call this with 0 rows to flush any pending resources (attached pools and io buffers) to minimize memory consumption.

Definition at line 124 of file hdfs-scanner.cc.

References impala::HdfsScanNode::AddMaterializedRowBatch(), impala::RowBatch::AtCapacity(), impala::HdfsScanner::batch_, impala::TupleDescriptor::byte_size(), impala::Status::CANCELLED, impala::ScannerContext::cancelled(), impala::RowBatch::capacity(), impala::RuntimeState::CheckQueryState(), impala::RowBatch::CommitRows(), impala::HdfsScanner::conjunct_ctxs_, impala::HdfsScanner::context_, impala::ExprContext::FreeLocalAllocations(), impala::ScannerContext::num_completed_io_buffers(), impala::RowBatch::num_rows(), impala::Status::OK, impala::ScannerContext::ReleaseCompletedResources(), RETURN_IF_ERROR, impala::HdfsScanner::scan_node_, impala::HdfsScanner::StartNewRowBatch(), impala::HdfsScanner::state_, impala::HdfsScanNode::tuple_desc(), and impala::HdfsScanner::tuple_mem_.

Referenced by AssembleRows(), impala::HdfsScanner::AttachPool(), impala::HdfsTextScanner::FinishScanRange(), impala::HdfsSequenceScanner::ProcessDecompressedBlock(), ProcessFooter(), impala::HdfsTextScanner::ProcessRange(), impala::HdfsAvroScanner::ProcessRange(), impala::HdfsSequenceScanner::ProcessRange(), impala::HdfsRCFileScanner::ProcessRange(), and ProcessSplit().

Status HdfsParquetScanner::CreateColumnReaders ( )
private
Status impala::HdfsParquetScanner::CreateSchemaTree ( const std::vector< parquet::SchemaElement > &  schema,
SchemaNode node 
) const
private

Unflattens the schema metadata from a Parquet file metadata and converts it to our SchemaNode representation. Returns the result in 'n' unless an error status is returned. Does not set the slot_desc field of any SchemaNode.

Referenced by ProcessFooter().

Status impala::HdfsParquetScanner::CreateSchemaTree ( const std::vector< parquet::SchemaElement > &  schema,
int  max_def_level,
int *  idx,
int *  col_idx,
SchemaNode node 
) const
private

Recursive implementation used internally by the above CreateSchemaTree() function.

bool IR_ALWAYS_INLINE impala::HdfsScanner::EvalConjuncts ( TupleRow row)
inlineprotectedinherited

Convenience function for evaluating conjuncts using this scanner's ExprContexts. This must always be inlined so we can correctly replace the call to ExecNode::EvalConjuncts() during codegen.

Definition at line 266 of file hdfs-scanner.h.

References impala::HdfsScanner::conjunct_ctxs_, and impala::ExecNode::EvalConjuncts().

Referenced by AssembleRows(), impala::HdfsAvroScanner::DecodeAvroData(), impala::HdfsRCFileScanner::ProcessRange(), impala::HdfsScanner::WriteCompleteTuple(), impala::HdfsScanner::WriteEmptyTuples(), and impala::HdfsTextScanner::WriteFields().

ExprContext * HdfsScanner::GetConjunctCtx ( int  idx) const
protectedinherited

Simple wrapper around conjunct_ctxs_. Used in the codegen'd version of WriteCompleteTuple() because it's easier than writing IR to access conjunct_ctxs_.

Definition at line 79 of file hdfs-scanner-ir.cc.

References impala::HdfsScanner::conjunct_ctxs_, and gen_ir_descriptions::idx.

int HdfsScanner::GetMemory ( MemPool **  pool,
Tuple **  tuple_mem,
TupleRow **  tuple_row_mem 
)
protectedinherited

Gets memory for outputting tuples into batch_. *pool is the mem pool that should be used for memory allocated for those tuples. *tuple_mem should be the location to output tuples, and *tuple_row_mem for outputting tuple rows. Returns the maximum number of tuples/tuple rows that can be output (before the current row batch is complete and a new one is allocated). Memory returned from this call is invalidated after calling CommitRows. Callers must call GetMemory again after calling this function.

Definition at line 115 of file hdfs-scanner.cc.

References impala::RowBatch::AddRow(), impala::HdfsScanner::batch_, impala::RowBatch::capacity(), impala::RowBatch::GetRow(), impala::RowBatch::num_rows(), impala::RowBatch::tuple_data_pool(), and impala::HdfsScanner::tuple_mem_.

Referenced by AssembleRows(), impala::HdfsTextScanner::FinishScanRange(), impala::HdfsSequenceScanner::ProcessDecompressedBlock(), ProcessFooter(), impala::HdfsTextScanner::ProcessRange(), impala::HdfsAvroScanner::ProcessRange(), impala::HdfsSequenceScanner::ProcessRange(), and impala::HdfsRCFileScanner::ProcessRange().

Status HdfsScanner::InitializeWriteTuplesFn ( HdfsPartitionDescriptor partition,
THdfsFileFormat::type  type,
const std::string &  scanner_name 
)
protectedinherited
Status impala::HdfsParquetScanner::InitNewRange ( )
inlineprivatevirtual

Part of the HdfsScanner interface, not used in Parquet.

Implements impala::HdfsScanner.

Definition at line 184 of file hdfs-parquet-scanner.h.

References impala::Status::OK.

void impala::HdfsScanner::InitTuple ( Tuple template_tuple,
Tuple tuple 
)
inlineprotectedinherited

Initialize a tuple. TODO: only copy over non-null slots. TODO: InitTuple is called frequently, avoid the if, perhaps via templatization.

Definition at line 355 of file hdfs-scanner.h.

References impala::HdfsScanner::num_null_bytes_, and impala::HdfsScanner::tuple_byte_size_.

Referenced by AssembleRows(), impala::HdfsAvroScanner::DecodeAvroData(), impala::HdfsRCFileScanner::ProcessRange(), impala::HdfsScanner::WriteCompleteTuple(), and impala::HdfsTextScanner::WriteFields().

void HdfsScanner::LogRowParseError ( int  row_idx,
std::stringstream *   
)
protectedvirtualinherited

Utility function to append an error message for an invalid row. This is called from ReportTupleParseError() row_idx is the index of the row in the current batch. Subclasses should override this function (i.e. text needs to join boundary rows). Since this is only in the error path, vtable overhead is acceptable.

Reimplemented in impala::HdfsSequenceScanner, and impala::HdfsTextScanner.

Definition at line 572 of file hdfs-scanner.cc.

Referenced by impala::HdfsScanner::ReportTupleParseError().

Tuple* impala::HdfsScanner::next_tuple ( Tuple t) const
inlineprotectedinherited
Status HdfsParquetScanner::Prepare ( ScannerContext context)
virtual
Status HdfsParquetScanner::ProcessFooter ( bool eosr)
private

Process the file footer and parse file_metadata_. This should be called with the last FOOTER_SIZE bytes in context_. *eosr is a return value. If true, the scan range is complete (e.g. select count(*))

Definition at line 916 of file hdfs-parquet-scanner.cc.

References impala::HdfsScanNode::AllocateScanRange(), impala::DiskIoMgr::BufferDescriptor::buffer(), impala::HdfsScanner::CommitRows(), impala::HdfsScanner::context_, COUNTER_ADD, CreateSchemaTree(), impala::DeserializeThriftMsg(), impala::DiskIoMgr::RequestRange::disk_id(), impala::ScannerContext::Stream::eosr(), impala::DiskIoMgr::ScanRange::expected_local(), impala::DiskIoMgr::RequestRange::file(), impala::HdfsFileDesc::file_length, file_metadata_, impala::ScannerContext::Stream::filename(), impala::DiskIoMgr::RequestRange::fs(), impala::ScannerContext::Stream::GetBuffer(), impala::Status::GetDetail(), impala::HdfsScanNode::GetFileDesc(), impala::HdfsScanner::GetMemory(), impala::RuntimeState::io_mgr(), impala::DiskIoMgr::BufferDescriptor::len(), impala::HdfsScanNode::MarkFileDescIssued(), impala::HdfsScanNode::materialized_slots(), impala::DiskIoMgr::max_read_buffer_size(), metadata_range_, impala::HdfsFileDesc::mtime, impala::Status::OK, impala::Status::ok(), impala::PARQUET_VERSION_NUMBER, pool, impala::DiskIoMgr::Read(), impala::HdfsScanNode::reader_context(), impala::DiskIoMgr::BufferDescriptor::Return(), RETURN_IF_ERROR, impala::ScanNode::rows_read_counter(), impala::HdfsScanNode::runtime_state(), impala::HdfsScanner::scan_node_, impala::ScannerContext::Stream::scan_range(), schema_, impala::HdfsScanner::stream_, impala::DiskIoMgr::ScanRange::try_cache(), UNLIKELY, ValidateFileMetadata(), and impala::HdfsScanner::WriteEmptyTuples().

Referenced by ProcessSplit().

Status HdfsParquetScanner::ProcessSplit ( )
virtual

Process an entire split, reading bytes from the context's streams. Context is initialized with the split data (e.g. template tuple, partition descriptor, etc). This function should only return on error or end of scan range.

Implements impala::HdfsScanner.

Definition at line 769 of file hdfs-parquet-scanner.cc.

References AssembleRows(), impala::HdfsScanner::batch_, column_readers_, impala::HdfsScanner::CommitRows(), impala::HdfsScanner::context_, COUNTER_SET, CreateColumnReaders(), file_metadata_, InitColumns(), num_cols_counter_, impala::Status::OK, ProcessFooter(), impala::ScannerContext::ReleaseCompletedResources(), RETURN_IF_ERROR, and impala::HdfsScanner::stream_.

bool HdfsScanner::ReportTupleParseError ( FieldLocation fields,
uint8_t *  errors,
int  row_idx 
)
protectedinherited

Utility function to report parse errors for each field. If errors[i] is nonzero, fields[i] had a parse error. row_idx is the idx of the row in the current batch that had the parse error Returns false if parsing should be aborted. In this case parse_status_ is set to the error. This is called from WriteAlignedTuples.

Definition at line 546 of file hdfs-scanner.cc.

References impala::RuntimeState::abort_on_error(), impala::ScannerContext::Stream::filename(), impala::RuntimeState::LogError(), impala::RuntimeState::LogHasSpace(), impala::HdfsScanner::LogRowParseError(), impala::HdfsScanNode::materialized_slots(), impala::HdfsScanner::num_errors_in_file_, impala::Status::ok(), impala::HdfsScanner::parse_status_, impala::HdfsScanner::ReportColumnParseError(), impala::RuntimeState::ReportFileErrors(), impala::HdfsScanner::scan_node_, impala::HdfsScanner::state_, and impala::HdfsScanner::stream_.

Referenced by impala::HdfsSequenceScanner::ProcessRange(), and impala::HdfsScanner::WriteAlignedTuples().

Status HdfsScanner::UpdateDecompressor ( const THdfsCompression::type &  compression)
protectedinherited

Update the decompressor_ object given a compression type or codec name. Depending on the old compression type and the new one, it may close the old decompressor and/or create a new one of different type.

Definition at line 513 of file hdfs-scanner.cc.

References impala::Codec::CreateDecompressor(), impala::HdfsScanner::data_buffer_pool_, impala::HdfsScanner::decompression_type_, impala::HdfsScanner::decompressor_, impala::Status::OK, RETURN_IF_ERROR, impala::HdfsScanner::scan_node_, impala::TupleDescriptor::string_slots(), and impala::HdfsScanNode::tuple_desc().

Referenced by impala::HdfsAvroScanner::InitNewRange(), impala::HdfsSequenceScanner::InitNewRange(), and impala::HdfsTextScanner::ProcessSplit().

Status impala::HdfsScanner::UpdateDecompressor ( const std::string &  codec)
protectedinherited
Status HdfsParquetScanner::ValidateFileMetadata ( )
private
int HdfsScanner::WriteAlignedTuples ( MemPool pool,
TupleRow tuple_row_mem,
int  row_size,
FieldLocation fields,
int  num_tuples,
int  max_added_tuples,
int  slots_per_tuple,
int  row_start_indx 
)
protectedinherited

Processes batches of fields and writes them out to tuple_row_mem.

  • 'pool' mempool to allocate from for auxiliary tuple memory
  • 'tuple_row_mem' preallocated tuple_row memory this function must use.
  • 'fields' must start at the beginning of a tuple.
  • 'num_tuples' number of tuples to process
  • 'max_added_tuples' the maximum number of tuples that should be added to the batch.
  • 'row_start_index' is the number of rows that have already been processed as part of WritePartialTuple. Returns the number of tuples added to the row batch. This can be less than num_tuples/tuples_till_limit because of failed conjuncts. Returns -1 if parsing should be aborted due to parse errors.

Definition at line 33 of file hdfs-scanner-ir.cc.

References impala::HdfsScanner::ReportTupleParseError(), impala::HdfsScanner::template_tuple_, impala::HdfsScanner::tuple_, impala::HdfsScanner::tuple_byte_size_, UNLIKELY, and impala::HdfsScanner::WriteCompleteTuple().

Referenced by impala::HdfsSequenceScanner::ProcessDecompressedBlock(), and impala::HdfsTextScanner::WriteFields().

bool HdfsScanner::WriteCompleteTuple ( MemPool pool,
FieldLocation fields,
Tuple tuple,
TupleRow tuple_row,
Tuple template_tuple,
uint8_t *  error_fields,
uint8_t *  error_in_row 
)
protectedinherited

Writes out all slots for 'tuple' from 'fields'. 'fields' must be aligned to the start of the tuple (e.g. fields[0] maps to slots[0]). After writing the tuple, it will be evaluated against the conjuncts.

  • error_fields is an out array. error_fields[i] will be set to true if the ith field had a parse error
  • error_in_row is an out bool. It is set to true if any field had parse errors Returns whether the resulting tuplerow passed the conjuncts. The parsing of the fields and evaluating against conjuncts is combined in this function. This is done so it can be possible to evaluate conjuncts as slots are materialized (on partial tuples). This function is replaced by a codegen'd function at runtime. This is the reason that the out error parameters are typed uint8_t instead of bool. We need to be able to match this function's signature identically for the codegen'd function. Bool's as out parameters can get converted to bytes by the compiler and rather than implicitly depending on that to happen, we will explicitly type them to bytes. TODO: revisit this

Definition at line 217 of file hdfs-scanner.cc.

References impala::HdfsScanner::EvalConjuncts(), impala::HdfsScanner::InitTuple(), impala::FieldLocation::len, impala::HdfsScanNode::materialized_slots(), impala::HdfsScanner::scan_node_, impala::TupleRow::SetTuple(), impala::HdfsScanner::text_converter_, impala::HdfsScanNode::tuple_idx(), and UNLIKELY.

Referenced by impala::HdfsSequenceScanner::ProcessRange(), and impala::HdfsScanner::WriteAlignedTuples().

int HdfsScanner::WriteEmptyTuples ( ScannerContext context,
TupleRow tuple_row,
int  num_tuples 
)
protectedinherited

Friends And Related Function Documentation

friend class BaseColumnReader
friend

Definition at line 116 of file hdfs-parquet-scanner.h.

friend class BoolColumnReader
friend

Definition at line 121 of file hdfs-parquet-scanner.h.

Referenced by CreateReader().

template<typename T >
friend class ColumnReader
friend

Definition at line 119 of file hdfs-parquet-scanner.h.

Member Data Documentation

ScopedTimer<MonotonicStopWatch> impala::HdfsParquetScanner::assemble_rows_timer_
private

Timer for materializing rows. This ignores time getting the next buffer.

Definition at line 147 of file hdfs-parquet-scanner.h.

Referenced by AssembleRows(), Close(), and HdfsParquetScanner().

RowBatch* impala::HdfsScanner::batch_
protectedinherited

The current row batch being populated. Creating new row batches, attaching context resources, and handing off to the scan node is handled by this class in CommitRows(), but AttachPool() must be called by scanner subclasses to attach any memory allocated by that subclass. All row batches created by this class are transferred to the scan node (i.e., all batches are ultimately owned by the scan node).

Definition at line 177 of file hdfs-scanner.h.

Referenced by impala::HdfsScanner::AddFinalRowBatch(), impala::HdfsScanner::AttachPool(), impala::HdfsScanner::CommitRows(), impala::HdfsScanner::GetMemory(), impala::HdfsScanner::next_row(), impala::HdfsSequenceScanner::ProcessDecompressedBlock(), ProcessSplit(), impala::HdfsScanner::StartNewRowBatch(), impala::HdfsTextScanner::WriteFields(), and impala::HdfsScanner::~HdfsScanner().

std::vector<BaseColumnReader*> impala::HdfsParquetScanner::column_readers_
private

Column reader for each materialized columns for this file.

Definition at line 125 of file hdfs-parquet-scanner.h.

Referenced by AssembleRows(), Close(), CreateColumnReaders(), InitColumns(), and ProcessSplit().

std::vector<ExprContext*> impala::HdfsScanner::conjunct_ctxs_
protectedinherited

ExprContext for each conjunct. Each scanner has its own ExprContexts so the conjuncts can be safely evaluated in parallel.

Definition at line 154 of file hdfs-scanner.h.

Referenced by impala::HdfsScanner::Close(), impala::HdfsScanner::CommitRows(), impala::HdfsScanner::EvalConjuncts(), impala::HdfsScanner::GetConjunctCtx(), and impala::HdfsScanner::Prepare().

boost::scoped_ptr<MemPool> impala::HdfsScanner::data_buffer_pool_
protectedinherited
THdfsCompression::type impala::HdfsScanner::decompression_type_
protectedinherited
boost::scoped_ptr<MemPool> impala::HdfsParquetScanner::dictionary_pool_
private

Pool to copy dictionary page buffer into. This pool is shared across all the pages in a column chunk.

Definition at line 144 of file hdfs-parquet-scanner.h.

Referenced by Close(), and impala::HdfsParquetScanner::BaseColumnReader::ReadDataPage().

const int impala::HdfsScanner::FILE_BLOCK_SIZE = 4096
staticinherited

Assumed size of an OS file block. Used mostly when reading file format headers, etc. This probably ought to be a derived number from the environment.

Definition at line 95 of file hdfs-scanner.h.

parquet::FileMetaData impala::HdfsParquetScanner::file_metadata_
private

File metadata thrift object.

Definition at line 128 of file hdfs-parquet-scanner.h.

Referenced by AssembleRows(), InitColumns(), ProcessFooter(), ProcessSplit(), ValidateColumn(), and ValidateFileMetadata().

FileVersion impala::HdfsParquetScanner::file_version_
private
const int impala::HdfsParquetScanner::FOOTER_SIZE = 100 * 1024
staticprivate

Size of the file footer. This is a guess. If this value is too little, we will need to issue another read.

Definition at line 113 of file hdfs-parquet-scanner.h.

const char * HdfsScanner::LLVM_CLASS_NAME = "class.impala::HdfsScanner"
staticinherited

Scanner subclasses must implement these static functions as well. Unfortunately, c++ does not allow static virtual functions. Issue the initial ranges for 'files'. HdfsFileDesc groups all the splits assigned to this scan node by file. This is called before any of the scanner subclasses are created to process splits in 'files'. The strategy on how to parse the scan ranges depends on the file format.

  • For simple text files, all the splits are simply issued to the io mgr and one split == one scan range.
  • For formats with a header, the metadata is first parsed, and then the ranges are issued to the io mgr. There is one scan range for the header and one range for each split.
  • For columnar formats, the header is parsed and only the relevant byte ranges should be issued to the io mgr. This is one range for the metadata and one range for each column, for each split. This function is how scanners can pick their strategy. void IssueInitialRanges(HdfsScanNode* scan_node, const std::vector<HdfsFileDesc*>& files); Codegen all functions for this scanner. The codegen'd function is specific to the scanner subclass but not specific to each scanner object. We don't want to codegen the functions for each scanner object. llvm::Function* Codegen(HdfsScanNode* scan_node);

Definition at line 137 of file hdfs-scanner.h.

Referenced by impala::HdfsScanner::CodegenWriteCompleteTuple().

const DiskIoMgr::ScanRange* impala::HdfsParquetScanner::metadata_range_
private

Scan range for the metadata.

Definition at line 137 of file hdfs-parquet-scanner.h.

Referenced by InitColumns(), ProcessFooter(), and ValidateColumn().

RuntimeProfile::Counter* impala::HdfsParquetScanner::num_cols_counter_
private

Number of cols that need to be read.

Definition at line 150 of file hdfs-parquet-scanner.h.

Referenced by Prepare(), and ProcessSplit().

int impala::HdfsScanner::num_errors_in_file_
protectedinherited

number of errors in current file

Definition at line 183 of file hdfs-scanner.h.

Referenced by impala::HdfsScanner::ReportTupleParseError().

int32_t impala::HdfsScanner::num_null_bytes_
protectedinherited

Number of null bytes in the tuple.

Definition at line 189 of file hdfs-scanner.h.

Referenced by impala::HdfsScanner::InitTuple().

Status impala::HdfsParquetScanner::parse_status_
private

Returned in ProcessSplit.

Definition at line 140 of file hdfs-parquet-scanner.h.

Referenced by AssembleRows(), and impala::HdfsParquetScanner::BoolColumnReader::ReadSlot().

HdfsScanNode* impala::HdfsScanner::scan_node_
protectedinherited

The scan node that started this scanner.

Definition at line 141 of file hdfs-scanner.h.

Referenced by impala::HdfsScanner::AddFinalRowBatch(), AssembleRows(), impala::HdfsParquetScanner::BaseColumnReader::BaseColumnReader(), impala::HdfsTextScanner::Close(), impala::BaseSequenceScanner::Close(), Close(), impala::BaseSequenceScanner::CloseFileRanges(), impala::HdfsScanner::CommitRows(), CreateColumnReaders(), CreateReader(), impala::HdfsRCFileScanner::DebugString(), impala::HdfsAvroScanner::DecodeAvroData(), impala::HdfsTextScanner::FillByteBufferCompressedFile(), impala::HdfsTextScanner::FinishScanRange(), InitColumns(), impala::HdfsScanner::InitializeWriteTuplesFn(), impala::HdfsTextScanner::InitNewRange(), impala::HdfsAvroScanner::InitNewRange(), impala::HdfsSequenceScanner::InitNewRange(), impala::HdfsRCFileScanner::InitNewRange(), impala::HdfsAvroScanner::ParseMetadata(), impala::HdfsTextScanner::Prepare(), impala::BaseSequenceScanner::Prepare(), Prepare(), impala::HdfsScanner::Prepare(), impala::HdfsSequenceScanner::Prepare(), impala::HdfsRCFileScanner::Prepare(), impala::HdfsSequenceScanner::ProcessBlockCompressedScanRange(), impala::HdfsSequenceScanner::ProcessDecompressedBlock(), ProcessFooter(), impala::HdfsTextScanner::ProcessRange(), impala::HdfsAvroScanner::ProcessRange(), impala::HdfsSequenceScanner::ProcessRange(), impala::HdfsRCFileScanner::ProcessRange(), impala::BaseSequenceScanner::ProcessSplit(), impala::HdfsParquetScanner::BaseColumnReader::ReadDataPage(), impala::HdfsRCFileScanner::ReadRowGroup(), impala::HdfsScanner::ReportColumnParseError(), impala::HdfsScanner::ReportTupleParseError(), impala::HdfsTextScanner::ResetScanner(), impala::HdfsAvroScanner::ResolveSchemas(), impala::HdfsScanner::StartNewRowBatch(), impala::HdfsScanner::UpdateDecompressor(), impala::HdfsAvroScanner::VerifyTypesMatch(), impala::HdfsScanner::WriteCompleteTuple(), impala::HdfsScanner::WriteEmptyTuples(), impala::HdfsTextScanner::WriteFields(), and impala::HdfsTextScanner::WritePartialTuple().

SchemaNode impala::HdfsParquetScanner::schema_
private

The root schema node for this file.

Definition at line 134 of file hdfs-parquet-scanner.h.

Referenced by CreateColumnReaders(), and ProcessFooter().

ScannerContext::Stream* impala::HdfsScanner::stream_
protectedinherited

The first stream for context_.

Definition at line 150 of file hdfs-scanner.h.

Referenced by impala::HdfsTextScanner::Close(), impala::BaseSequenceScanner::Close(), CreateColumnReaders(), impala::HdfsRCFileScanner::DebugString(), impala::HdfsTextScanner::FillByteBuffer(), impala::HdfsTextScanner::FillByteBufferCompressedFile(), impala::HdfsTextScanner::FillByteBufferGzip(), impala::HdfsTextScanner::FindFirstTuple(), impala::HdfsTextScanner::FinishScanRange(), impala::HdfsSequenceScanner::GetRecord(), impala::HdfsTextScanner::InitNewRange(), impala::HdfsRCFileScanner::InitNewRange(), impala::HdfsRCFileScanner::NextField(), impala::HdfsAvroScanner::ParseMetadata(), impala::BaseSequenceScanner::Prepare(), impala::HdfsScanner::Prepare(), impala::HdfsSequenceScanner::ProcessBlockCompressedScanRange(), ProcessFooter(), impala::HdfsTextScanner::ProcessRange(), impala::HdfsAvroScanner::ProcessRange(), impala::HdfsSequenceScanner::ProcessRange(), impala::HdfsRCFileScanner::ProcessRange(), impala::HdfsTextScanner::ProcessSplit(), impala::BaseSequenceScanner::ProcessSplit(), ProcessSplit(), impala::HdfsSequenceScanner::ReadBlockHeader(), impala::HdfsRCFileScanner::ReadColumnBuffers(), impala::HdfsSequenceScanner::ReadCompressedBlock(), impala::HdfsAvroScanner::ReadFileHeader(), impala::HdfsSequenceScanner::ReadFileHeader(), impala::HdfsRCFileScanner::ReadFileHeader(), impala::HdfsRCFileScanner::ReadKeyBuffers(), impala::HdfsRCFileScanner::ReadNumColumnsMetadata(), impala::HdfsRCFileScanner::ReadRowGroupHeader(), impala::BaseSequenceScanner::ReadSync(), impala::HdfsScanner::ReportTupleParseError(), impala::BaseSequenceScanner::SkipToSync(), ValidateFileMetadata(), impala::HdfsAvroScanner::VerifyTypesMatch(), and impala::HdfsTextScanner::WriteFields().

Tuple* impala::HdfsScanner::template_tuple_
protectedinherited

A partially materialized tuple with only partition key slots set. The non-partition key slots are set to NULL. The template tuple must be copied into tuple_ before any of the other slots are materialized. Pointer is NULL if there are no partition key slots. This template tuple is computed once for each file and valid for the duration of that file. It is owned by the HDFS scan node.

Definition at line 164 of file hdfs-scanner.h.

Referenced by impala::HdfsAvroScanner::AllocateFileHeader(), AssembleRows(), CreateColumnReaders(), impala::HdfsAvroScanner::DecodeAvroData(), impala::HdfsAvroScanner::InitNewRange(), impala::HdfsScanner::Prepare(), impala::HdfsSequenceScanner::ProcessRange(), impala::HdfsRCFileScanner::ProcessRange(), impala::HdfsAvroScanner::ResolveSchemas(), impala::HdfsScanner::WriteAlignedTuples(), impala::HdfsScanner::WriteEmptyTuples(), and impala::HdfsTextScanner::WriteFields().

boost::scoped_ptr<TextConverter> impala::HdfsScanner::text_converter_
protectedinherited
int impala::HdfsScanner::tuple_byte_size_
protectedinherited
uint8_t* impala::HdfsScanner::tuple_mem_
protectedinherited

The tuple memory of batch_.

Definition at line 180 of file hdfs-scanner.h.

Referenced by impala::HdfsScanner::CommitRows(), impala::HdfsScanner::GetMemory(), and impala::HdfsScanner::StartNewRowBatch().

WriteTuplesFn impala::HdfsScanner::write_tuples_fn_
protectedinherited

Jitted write tuples function pointer. Null if codegen is disabled.

Definition at line 215 of file hdfs-scanner.h.

Referenced by impala::HdfsScanner::InitializeWriteTuplesFn(), impala::HdfsSequenceScanner::ProcessDecompressedBlock(), and impala::HdfsTextScanner::WriteFields().


The documentation for this class was generated from the following files: