Examples with IcebergSplitContainer org.apache.iceberg.mr.mapreduce.IcebergSplitContainer used on opensource projects

Example 1 with IcebergSplitContainer

use of org.apache.iceberg.mr.mapreduce.IcebergSplitContainer in project hive by apache.

the class HiveIcebergInputFormat method getRecordReader.

@Override
public RecordReader<Void, Container<Record>> getRecordReader(InputSplit split, JobConf job, Reporter reporter) throws IOException {
    job.set(InputFormatConfig.SELECTED_COLUMNS, job.get(ColumnProjectionUtils.READ_COLUMN_NAMES_CONF_STR, ""));
    if (HiveConf.getBoolVar(job, HiveConf.ConfVars.HIVE_VECTORIZATION_ENABLED) && Utilities.getIsVectorized(job)) {
        Preconditions.checkArgument(MetastoreUtil.hive3PresentOnClasspath(), "Vectorization only supported for Hive 3+");
        job.setEnum(InputFormatConfig.IN_MEMORY_DATA_MODEL, InputFormatConfig.InMemoryDataModel.HIVE);
        job.setBoolean(InputFormatConfig.SKIP_RESIDUAL_FILTERING, true);
        IcebergSplit icebergSplit = ((IcebergSplitContainer) split).icebergSplit();
        // bogus cast for favouring code reuse over syntax
        return (RecordReader) HIVE_VECTORIZED_RECORDREADER_CTOR.newInstance(new IcebergInputFormat<>(), icebergSplit, job, reporter);
    } else {
        return super.getRecordReader(split, job, reporter);
    }
}