org.opentripplanner.datastore.file.DirectoryDataSource Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of otp Show documentation

The OpenTripPlanner multimodal journey planning system

There is a newer version: 2.6.0

package org.opentripplanner.datastore.file;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import org.apache.commons.io.FileUtils;
import org.opentripplanner.datastore.api.CompositeDataSource;
import org.opentripplanner.datastore.api.DataSource;
import org.opentripplanner.datastore.api.FileType;

/**
 * This is a adapter to wrap a file directory and all files in it as a composite data source.
 * Sub-directories are ignored.
 */
public class DirectoryDataSource extends AbstractFileDataSource implements CompositeDataSource {

  public DirectoryDataSource(File path, FileType type) {
    super(path, type);
  }

  @Override
  @SuppressWarnings("ConstantConditions")
  public Collection content() {
    Collection content = new ArrayList<>();
    if (file.exists()) {
      for (File file : file.listFiles()) {
        // Skip any nested directories
        if (file.isDirectory()) {
          continue;
        }
        // In general the file type at a child level is not used, but we tag
        // each file with the same type as the parent directory.
        // Examples: GTFS or NETEX.
        content.add(new FileDataSource(file, type));
      }
    }
    return content;
  }

  @SuppressWarnings("ResultOfMethodCallIgnored")
  @Override
  public DataSource entry(String filename) {
    if (!file.exists()) {
      file.mkdirs();
    }
    return new FileDataSource(new File(file, filename), type);
  }

  @Override
  public void delete() {
    try {
      if (file.exists()) {
        FileUtils.deleteDirectory(file);
      }
    } catch (IOException e) {
      throw new RuntimeException("Failed to delete " + path() + ": " + e.getLocalizedMessage(), e);
    }
  }

  @Override
  public void close() {
    /* Nothing to close */
  }
}