List of usage examples for org.apache.commons.vfs2 FileExtensionSelector FileExtensionSelector
public FileExtensionSelector(final String... extensions)
From source file:org.pentaho.hadoop.shim.common.format.avro.PentahoAvroInputFormat.java
private DataFileStream<GenericRecord> createDataFileStream(String schemaFileName, String fileName) throws Exception { DatumReader<GenericRecord> datumReader; if (schemaFileName != null && schemaFileName.length() > 0) { datumReader = new GenericDatumReader<GenericRecord>(readAvroSchema(schemaFileName)); } else {// w ww . j av a2s. c om datumReader = new GenericDatumReader<GenericRecord>(); } FileObject fileObject = KettleVFS.getFileObject(fileName); if (fileObject.isFile()) { return new DataFileStream<GenericRecord>(fileObject.getContent().getInputStream(), datumReader); } else { FileObject[] avroFiles = fileObject.findFiles(new FileExtensionSelector("avro")); if (!Utils.isEmpty(avroFiles)) { return new DataFileStream<GenericRecord>(avroFiles[0].getContent().getInputStream(), datumReader); } return null; } }