Search in sources :

Example 1 with HadoopReadOptions

use of org.apache.parquet.HadoopReadOptions in project parquet-mr by apache.

the class InternalParquetRecordReader method initialize.

public void initialize(ParquetFileReader reader, ParquetReadOptions options) {
    // copy custom configuration to the Configuration passed to the ReadSupport
    Configuration conf = new Configuration();
    if (options instanceof HadoopReadOptions) {
        conf = ((HadoopReadOptions) options).getConf();
    }
    for (String property : options.getPropertyNames()) {
        conf.set(property, options.getProperty(property));
    }
    // initialize a ReadContext for this file
    this.reader = reader;
    FileMetaData parquetFileMetadata = reader.getFooter().getFileMetaData();
    this.fileSchema = parquetFileMetadata.getSchema();
    Map<String, String> fileMetadata = parquetFileMetadata.getKeyValueMetaData();
    ReadSupport.ReadContext readContext = readSupport.init(new InitContext(conf, toSetMultiMap(fileMetadata), fileSchema));
    this.columnIOFactory = new ColumnIOFactory(parquetFileMetadata.getCreatedBy());
    this.requestedSchema = readContext.getRequestedSchema();
    this.columnCount = requestedSchema.getPaths().size();
    this.recordConverter = readSupport.prepareForRead(conf, fileMetadata, fileSchema, readContext);
    this.strictTypeChecking = options.isEnabled(STRICT_TYPE_CHECKING, true);
    this.total = reader.getRecordCount();
    this.unmaterializableRecordCounter = new UnmaterializableRecordCounter(options, total);
    this.filterRecords = options.useRecordFilter();
    reader.setRequestedSchema(requestedSchema);
    LOG.info("RecordReader initialized will read a total of {} records.", total);
}
Also used : ReadSupport(org.apache.parquet.hadoop.api.ReadSupport) InitContext(org.apache.parquet.hadoop.api.InitContext) HadoopReadOptions(org.apache.parquet.HadoopReadOptions) Configuration(org.apache.hadoop.conf.Configuration) FileMetaData(org.apache.parquet.hadoop.metadata.FileMetaData) ColumnIOFactory(org.apache.parquet.io.ColumnIOFactory)

Aggregations

Configuration (org.apache.hadoop.conf.Configuration)1 HadoopReadOptions (org.apache.parquet.HadoopReadOptions)1 InitContext (org.apache.parquet.hadoop.api.InitContext)1 ReadSupport (org.apache.parquet.hadoop.api.ReadSupport)1 FileMetaData (org.apache.parquet.hadoop.metadata.FileMetaData)1 ColumnIOFactory (org.apache.parquet.io.ColumnIOFactory)1