Examples with ConvertToIndexedRecord org.talend.components.adapter.beam.transform.ConvertToIndexedRecord used on opensource projects

Example 1 with ConvertToIndexedRecord

use of org.talend.components.adapter.beam.transform.ConvertToIndexedRecord in project components by Talend.

the class SimpleRecordFormatAvroIO method read.

@Override
public PCollection<IndexedRecord> read(PBegin in) {
    // Reuseable coder.
    LazyAvroCoder<Object> lac = LazyAvroCoder.of();
    AvroHdfsFileSource source = AvroHdfsFileSource.of(doAs, path, lac);
    source.getExtraHadoopConfiguration().addFrom(getExtraHadoopConfiguration());
    source.setLimit(limit);
    PCollection<KV<AvroKey, NullWritable>> read = // 
    in.apply(Read.from(source)).setCoder(source.getDefaultOutputCoder());
    PCollection<AvroKey> pc1 = read.apply(Keys.<AvroKey>create());
    PCollection<Object> pc2 = pc1.apply(ParDo.of(new ExtractRecordFromAvroKey()));
    pc2 = pc2.setCoder(lac);
    PCollection<IndexedRecord> pc3 = pc2.apply(ConvertToIndexedRecord.<Object>of());
    return pc3;
}