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

#include <hdfs-sequence-scanner.h>

Inheritance diagram for impala::HdfsSequenceScanner:
Collaboration diagram for impala::HdfsSequenceScanner:

Classes

struct  RecordLocation
 Struct for record locations and lens in compressed blocks. More...
 
struct  SeqFileHeader
 Data that is fixed across headers. This struct is shared between scan ranges. More...
 

Public Member Functions

 HdfsSequenceScanner (HdfsScanNode *scan_node, RuntimeState *state)
 
virtual ~HdfsSequenceScanner ()
 
virtual Status Prepare (ScannerContext *context)
 Implementation of HdfsScanner interface. More...
 
virtual void Close ()
 
virtual Status ProcessSplit ()
 

Static Public Member Functions

static llvm::Function * Codegen (HdfsScanNode *, const std::vector< ExprContext * > &conjunct_ctxs)
 Codegen writing tuples and evaluating predicates. More...
 
static Status IssueInitialRanges (HdfsScanNode *scan_node, const std::vector< HdfsFileDesc * > &files)
 Issue the initial ranges for all sequence container files. More...
 

Static Public Attributes

static const uint8_t SEQFILE_VERSION_HEADER [4] = {'S', 'E', 'Q', 6}
 
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

virtual FileHeaderAllocateFileHeader ()
 Implementation of sequence container super class methods. More...
 
virtual Status ReadFileHeader ()
 
virtual Status InitNewRange ()
 Reset internal state for a new scan range. More...
 
virtual Status ProcessRange ()
 
virtual THdfsFileFormat::type file_format () const
 Returns type of scanner: e.g. rcfile, seqfile. More...
 
Status ReadSync ()
 
Status SkipToSync (const uint8_t *sync, int sync_size)
 
bool finished ()
 
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)
 
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

FileHeaderheader_
 File header for this scan range. This is not owned by the parent scan node. More...
 
bool only_parsing_header_
 If true, this scanner object is only for processing the header. More...
 
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...
 
Status parse_status_
 
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...
 

Static Protected Attributes

static const int SYNC_HASH_SIZE = 16
 Size of the sync hash field. More...
 
static const int HEADER_SIZE = 1024
 
static const int SYNC_MARKER = -1
 Sync indicator. More...
 

Private Member Functions

Status ReadBlockHeader ()
 
Status ProcessBlockCompressedScanRange ()
 
Status ReadCompressedBlock ()
 
Status ProcessDecompressedBlock ()
 
Status GetRecord (uint8_t **record_ptr, int64_t *record_len)
 
virtual void LogRowParseError (int row_idx, std::stringstream *)
 

Private Attributes

boost::scoped_ptr
< DelimitedTextParser
delimited_text_parser_
 Helper class for picking fields and rows from delimited text. More...
 
std::vector< FieldLocationfield_locations_
 
std::vector< RecordLocationrecord_locations_
 
int current_block_length_
 Length of the current sequence file block (or record). More...
 
int current_key_length_
 Length of the current key. This is specified as 4 bytes in the format description. More...
 
uint8_t * unparsed_data_buffer_
 Buffer for data read from HDFS or from decompressing the HDFS data. More...
 
int64_t num_buffered_records_in_compressed_block_
 Number of buffered records unparsed_data_buffer_ from block compressed data. More...
 
uint8_t * next_record_in_compressed_block_
 Next record from block compressed data. More...
 

Static Private Attributes

static const int MAX_BLOCK_SIZE = (1024 * 1024 * 1024)
 
static const char *const SEQFILE_VALUE_CLASS_NAME
 

Detailed Description

Definition at line 156 of file hdfs-sequence-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

HdfsSequenceScanner::HdfsSequenceScanner ( HdfsScanNode scan_node,
RuntimeState state 
)

Definition at line 40 of file hdfs-sequence-scanner.cc.

HdfsSequenceScanner::~HdfsSequenceScanner ( )
virtual

Definition at line 46 of file hdfs-sequence-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 impala::HdfsParquetScanner::Close().

BaseSequenceScanner::FileHeader * HdfsSequenceScanner::AllocateFileHeader ( )
protectedvirtual

Implementation of sequence container super class methods.

Implements impala::BaseSequenceScanner.

Definition at line 99 of file hdfs-sequence-scanner.cc.

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(), impala::HdfsParquetScanner::Close(), impala::HdfsTextScanner::FillByteBufferGzip(), impala::HdfsAvroScanner::ProcessRange(), ReadCompressedBlock(), impala::HdfsParquetScanner::BaseColumnReader::ReadDataPage(), and impala::HdfsRCFileScanner::ResetRowGroup().

Function * HdfsSequenceScanner::Codegen ( HdfsScanNode ,
const std::vector< ExprContext * > &  conjunct_ctxs 
)
static
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 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 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 impala::HdfsParquetScanner::AssembleRows(), impala::HdfsScanner::AttachPool(), impala::HdfsTextScanner::FinishScanRange(), ProcessDecompressedBlock(), impala::HdfsParquetScanner::ProcessFooter(), impala::HdfsTextScanner::ProcessRange(), impala::HdfsAvroScanner::ProcessRange(), ProcessRange(), impala::HdfsRCFileScanner::ProcessRange(), and impala::HdfsParquetScanner::ProcessSplit().

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 impala::HdfsParquetScanner::AssembleRows(), impala::HdfsAvroScanner::DecodeAvroData(), impala::HdfsRCFileScanner::ProcessRange(), impala::HdfsScanner::WriteCompleteTuple(), impala::HdfsScanner::WriteEmptyTuples(), and impala::HdfsTextScanner::WriteFields().

virtual THdfsFileFormat::type impala::HdfsSequenceScanner::file_format ( ) const
inlineprotectedvirtual

Returns type of scanner: e.g. rcfile, seqfile.

Implements impala::BaseSequenceScanner.

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

bool impala::BaseSequenceScanner::finished ( )
inlineprotectedinherited
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 impala::HdfsParquetScanner::AssembleRows(), impala::HdfsTextScanner::FinishScanRange(), ProcessDecompressedBlock(), impala::HdfsParquetScanner::ProcessFooter(), impala::HdfsTextScanner::ProcessRange(), impala::HdfsAvroScanner::ProcessRange(), ProcessRange(), and impala::HdfsRCFileScanner::ProcessRange().

Status HdfsSequenceScanner::GetRecord ( uint8_t **  record_ptr,
int64_t *  record_len 
)
inlineprivate
Status HdfsScanner::InitializeWriteTuplesFn ( HdfsPartitionDescriptor partition,
THdfsFileFormat::type  type,
const std::string &  scanner_name 
)
protectedinherited

Initializes write_tuples_fn_ to the jitted function if codegen is possible.

  • partition - partition descriptor for this scanner/scan range
  • type - type for this scanner
  • scanner_name - debug string name for this scanner (e.g. HdfsTextScanner)

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

References impala::HdfsPartitionDescriptor::escape_char(), impala::HdfsScanNode::GetCodegenFn(), impala::ExecNode::id(), impala::HdfsScanNode::IncNumScannersCodegenDisabled(), impala::HdfsScanNode::IncNumScannersCodegenEnabled(), impala::Status::OK, impala::HdfsScanner::scan_node_, impala::TupleDescriptor::string_slots(), impala::HdfsScanNode::tuple_desc(), and impala::HdfsScanner::write_tuples_fn_.

Referenced by InitNewRange(), and impala::HdfsTextScanner::ResetScanner().

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 impala::HdfsParquetScanner::AssembleRows(), impala::HdfsAvroScanner::DecodeAvroData(), impala::HdfsRCFileScanner::ProcessRange(), impala::HdfsScanner::WriteCompleteTuple(), and impala::HdfsTextScanner::WriteFields().

Status BaseSequenceScanner::IssueInitialRanges ( HdfsScanNode scan_node,
const std::vector< HdfsFileDesc * > &  files 
)
staticinherited
void HdfsSequenceScanner::LogRowParseError ( int  row_idx,
std::stringstream *   
)
privatevirtual

Appends the current file and line to the RuntimeState's error log. row_idx is 0-based (in current batch) where the parse error occurred.

Reimplemented from impala::HdfsScanner.

Definition at line 493 of file hdfs-sequence-scanner.cc.

References record_locations_.

Status HdfsSequenceScanner::ProcessRange ( )
protectedvirtual

Process the current range until the end or an error occurred. Note this might be called multiple times if we skip over bad data. This function should read from the underlying ScannerContext materializing tuples to the context. When this function is called, it is guaranteed to be at the start of a data block (i.e. right after the sync marker).

Implements impala::BaseSequenceScanner.

Definition at line 271 of file hdfs-sequence-scanner.cc.

References impala::HdfsScanner::CommitRows(), impala::HdfsScanner::context_, COUNTER_ADD, delimited_text_parser_, impala::ScannerContext::Stream::eof(), field_locations_, impala::BaseSequenceScanner::finished(), impala::HdfsScanner::GetMemory(), GetRecord(), impala::BaseSequenceScanner::header_, impala::BaseSequenceScanner::FileHeader::is_compressed, impala::HdfsSequenceScanner::SeqFileHeader::is_row_compressed, impala::ScanNode::materialize_tuple_timer(), impala::HdfsScanNode::materialized_slots(), num_buffered_records_in_compressed_block_, impala::Status::OK, impala::HdfsScanner::parse_status_, pool, ProcessBlockCompressedScanRange(), impala::ExecNode::ReachedLimit(), impala::ScannerContext::Stream::ReadInt(), impala::BaseSequenceScanner::ReadSync(), record_locations_, impala::HdfsScanner::ReportTupleParseError(), RETURN_IF_ERROR, RETURN_IF_FALSE, impala::ScanNode::rows_read_counter(), impala::HdfsScanner::scan_node_, SCOPED_TIMER, impala::HdfsScanner::stream_, impala::BaseSequenceScanner::SYNC_MARKER, impala::HdfsScanner::template_tuple_, impala::HdfsScanner::tuple_, UNLIKELY, impala::HdfsScanner::WriteCompleteTuple(), and impala::HdfsScanner::WriteEmptyTuples().

Status BaseSequenceScanner::ProcessSplit ( )
virtualinherited

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 100 of file base-sequence-scanner.cc.

References impala::RuntimeState::abort_on_error(), impala::ObjectPool::Add(), impala::HdfsScanNode::AddDiskIoRanges(), impala::BaseSequenceScanner::AllocateFileHeader(), impala::BaseSequenceScanner::bytes_skipped_counter_, impala::BaseSequenceScanner::CloseFileRanges(), COUNTER_ADD, impala::ScannerContext::Stream::eof(), impala::ScannerContext::Stream::file_offset(), impala::ScannerContext::Stream::filename(), impala::BaseSequenceScanner::finished_, impala::HdfsScanNode::GetFileDesc(), impala::HdfsScanNode::GetFileMetadata(), impala::BaseSequenceScanner::header_, impala::BaseSequenceScanner::FileHeader::header_size, impala::HdfsScanner::InitNewRange(), impala::BaseSequenceScanner::FileHeader::is_compressed, impala::Status::IsCancelled(), impala::Status::IsMemLimitExceeded(), impala::RuntimeState::LogError(), impala::Status::msg(), impala::RuntimeState::obj_pool(), impala::Status::OK, impala::Status::ok(), impala::BaseSequenceScanner::only_parsing_header_, impala::HdfsScanner::parse_status_, impala::BaseSequenceScanner::ProcessRange(), impala::BaseSequenceScanner::ReadFileHeader(), RETURN_IF_ERROR, RETURN_IF_FALSE, impala::HdfsScanner::scan_node_, impala::ScannerContext::Stream::set_contains_tuple_data(), impala::HdfsScanNode::SetFileMetadata(), impala::ScannerContext::Stream::SkipBytes(), impala::BaseSequenceScanner::SkipToSync(), impala::HdfsScanner::state_, impala::HdfsScanner::stream_, impala::BaseSequenceScanner::FileHeader::sync, and impala::BaseSequenceScanner::SYNC_HASH_SIZE.

Status HdfsSequenceScanner::ReadBlockHeader ( )
private
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 ProcessRange(), and impala::HdfsScanner::WriteAlignedTuples().

Status BaseSequenceScanner::SkipToSync ( const uint8_t *  sync,
int  sync_size 
)
protectedinherited
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(), InitNewRange(), and impala::HdfsTextScanner::ProcessSplit().

Status impala::HdfsScanner::UpdateDecompressor ( const std::string &  codec)
protectedinherited
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 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 ProcessRange(), and impala::HdfsScanner::WriteAlignedTuples().

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

Member Data Documentation

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(), ProcessDecompressedBlock(), impala::HdfsParquetScanner::ProcessSplit(), impala::HdfsScanner::StartNewRowBatch(), impala::HdfsTextScanner::WriteFields(), and impala::HdfsScanner::~HdfsScanner().

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().

int impala::HdfsSequenceScanner::current_block_length_
private

Length of the current sequence file block (or record).

Definition at line 243 of file hdfs-sequence-scanner.h.

Referenced by GetRecord(), and ReadBlockHeader().

int impala::HdfsSequenceScanner::current_key_length_
private

Length of the current key. This is specified as 4 bytes in the format description.

Definition at line 246 of file hdfs-sequence-scanner.h.

Referenced by GetRecord(), and ReadBlockHeader().

boost::scoped_ptr<MemPool> impala::HdfsScanner::data_buffer_pool_
protectedinherited
THdfsCompression::type impala::HdfsScanner::decompression_type_
protectedinherited
boost::scoped_ptr<DelimitedTextParser> impala::HdfsSequenceScanner::delimited_text_parser_
private

Helper class for picking fields and rows from delimited text.

Definition at line 222 of file hdfs-sequence-scanner.h.

Referenced by InitNewRange(), ProcessDecompressedBlock(), and ProcessRange().

std::vector<FieldLocation> impala::HdfsSequenceScanner::field_locations_
private

Definition at line 223 of file hdfs-sequence-scanner.h.

Referenced by Prepare(), ProcessDecompressedBlock(), and ProcessRange().

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.

const int BaseSequenceScanner::HEADER_SIZE = 1024
staticprotectedinherited

Estimate of header size in bytes. This is initial number of bytes to issue per file. If the estimate is too low, more bytes will be read as necessary.

Definition at line 121 of file base-sequence-scanner.h.

Referenced by impala::BaseSequenceScanner::IssueInitialRanges().

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 int impala::HdfsSequenceScanner::MAX_BLOCK_SIZE = (1024 * 1024 * 1024)
staticprivate

Maximum size of a compressed block. This is used to check for corrupted block size so we do not read the whole file before we detect the error.

Definition at line 187 of file hdfs-sequence-scanner.h.

Referenced by ReadCompressedBlock().

uint8_t* impala::HdfsSequenceScanner::next_record_in_compressed_block_
private

Next record from block compressed data.

Definition at line 255 of file hdfs-sequence-scanner.h.

Referenced by ProcessDecompressedBlock(), and ReadCompressedBlock().

int64_t impala::HdfsSequenceScanner::num_buffered_records_in_compressed_block_
private

Number of buffered records unparsed_data_buffer_ from block compressed data.

Definition at line 252 of file hdfs-sequence-scanner.h.

Referenced by InitNewRange(), ProcessBlockCompressedScanRange(), ProcessDecompressedBlock(), ProcessRange(), and ReadCompressedBlock().

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().

bool impala::BaseSequenceScanner::only_parsing_header_
protectedinherited
std::vector<RecordLocation> impala::HdfsSequenceScanner::record_locations_
private

Records are processed in batches. This vector stores batches of record locations that are being processed. TODO: better perf not to use vector?

Definition at line 240 of file hdfs-sequence-scanner.h.

Referenced by LogRowParseError(), Prepare(), ProcessDecompressedBlock(), and ProcessRange().

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(), impala::HdfsParquetScanner::AssembleRows(), impala::HdfsParquetScanner::BaseColumnReader::BaseColumnReader(), impala::HdfsTextScanner::Close(), impala::BaseSequenceScanner::Close(), impala::HdfsParquetScanner::Close(), impala::BaseSequenceScanner::CloseFileRanges(), impala::HdfsScanner::CommitRows(), impala::HdfsParquetScanner::CreateColumnReaders(), impala::HdfsParquetScanner::CreateReader(), impala::HdfsRCFileScanner::DebugString(), impala::HdfsAvroScanner::DecodeAvroData(), impala::HdfsTextScanner::FillByteBufferCompressedFile(), impala::HdfsTextScanner::FinishScanRange(), impala::HdfsParquetScanner::InitColumns(), impala::HdfsScanner::InitializeWriteTuplesFn(), impala::HdfsTextScanner::InitNewRange(), impala::HdfsAvroScanner::InitNewRange(), InitNewRange(), impala::HdfsRCFileScanner::InitNewRange(), impala::HdfsAvroScanner::ParseMetadata(), impala::HdfsTextScanner::Prepare(), impala::BaseSequenceScanner::Prepare(), impala::HdfsParquetScanner::Prepare(), impala::HdfsScanner::Prepare(), Prepare(), impala::HdfsRCFileScanner::Prepare(), ProcessBlockCompressedScanRange(), ProcessDecompressedBlock(), impala::HdfsParquetScanner::ProcessFooter(), impala::HdfsTextScanner::ProcessRange(), impala::HdfsAvroScanner::ProcessRange(), 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().

const char *const HdfsSequenceScanner::SEQFILE_VALUE_CLASS_NAME
staticprivate
Initial value:
=
"org.apache.hadoop.io.Text"

The value class name located in the SeqFile Header. This is always "org.apache.hadoop.io.Text"

Definition at line 191 of file hdfs-sequence-scanner.h.

Referenced by ReadFileHeader().

const uint8_t HdfsSequenceScanner::SEQFILE_VERSION_HEADER = {'S', 'E', 'Q', 6}
static

The four byte SeqFile version header present at the beginning of every SeqFile file: {'S', 'E', 'Q', 6}

Definition at line 160 of file hdfs-sequence-scanner.h.

Referenced by ReadFileHeader(), and impala::HdfsRCFileScanner::ReadFileHeader().

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(), impala::HdfsParquetScanner::CreateColumnReaders(), impala::HdfsRCFileScanner::DebugString(), impala::HdfsTextScanner::FillByteBuffer(), impala::HdfsTextScanner::FillByteBufferCompressedFile(), impala::HdfsTextScanner::FillByteBufferGzip(), impala::HdfsTextScanner::FindFirstTuple(), impala::HdfsTextScanner::FinishScanRange(), GetRecord(), impala::HdfsTextScanner::InitNewRange(), impala::HdfsRCFileScanner::InitNewRange(), impala::HdfsRCFileScanner::NextField(), impala::HdfsAvroScanner::ParseMetadata(), impala::BaseSequenceScanner::Prepare(), impala::HdfsScanner::Prepare(), ProcessBlockCompressedScanRange(), impala::HdfsParquetScanner::ProcessFooter(), impala::HdfsTextScanner::ProcessRange(), impala::HdfsAvroScanner::ProcessRange(), ProcessRange(), impala::HdfsRCFileScanner::ProcessRange(), impala::HdfsTextScanner::ProcessSplit(), impala::BaseSequenceScanner::ProcessSplit(), impala::HdfsParquetScanner::ProcessSplit(), ReadBlockHeader(), impala::HdfsRCFileScanner::ReadColumnBuffers(), ReadCompressedBlock(), impala::HdfsAvroScanner::ReadFileHeader(), ReadFileHeader(), impala::HdfsRCFileScanner::ReadFileHeader(), impala::HdfsRCFileScanner::ReadKeyBuffers(), impala::HdfsRCFileScanner::ReadNumColumnsMetadata(), impala::HdfsRCFileScanner::ReadRowGroupHeader(), impala::BaseSequenceScanner::ReadSync(), impala::HdfsScanner::ReportTupleParseError(), impala::BaseSequenceScanner::SkipToSync(), impala::HdfsParquetScanner::ValidateFileMetadata(), impala::HdfsAvroScanner::VerifyTypesMatch(), and impala::HdfsTextScanner::WriteFields().

const int impala::BaseSequenceScanner::SYNC_HASH_SIZE = 16
staticprotectedinherited
const int BaseSequenceScanner::SYNC_MARKER = -1
staticprotectedinherited

Sync indicator.

Definition at line 124 of file base-sequence-scanner.h.

Referenced by ProcessRange(), and impala::HdfsRCFileScanner::ProcessRange().

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(), impala::HdfsParquetScanner::AssembleRows(), impala::HdfsParquetScanner::CreateColumnReaders(), impala::HdfsAvroScanner::DecodeAvroData(), impala::HdfsAvroScanner::InitNewRange(), impala::HdfsScanner::Prepare(), 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
Tuple* impala::HdfsScanner::tuple_
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().

uint8_t* impala::HdfsSequenceScanner::unparsed_data_buffer_
private

Buffer for data read from HDFS or from decompressing the HDFS data.

Definition at line 249 of file hdfs-sequence-scanner.h.

Referenced by GetRecord(), and ReadCompressedBlock().

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(), ProcessDecompressedBlock(), and impala::HdfsTextScanner::WriteFields().


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