Example usage for org.apache.commons.vfs2 FileExtensionSelector FileExtensionSelector

List of usage examples for org.apache.commons.vfs2 FileExtensionSelector FileExtensionSelector

Introduction

In this page you can find the example usage for org.apache.commons.vfs2 FileExtensionSelector FileExtensionSelector.

Prototype

public FileExtensionSelector(final String... extensions) 

Source Link

Document

Creates a new selector for the given extensions.

Usage

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;
    }
}