org.opentripplanner.updater.trip.GtfsRealtimeFileTripUpdateSource Maven / Gradle / Ivy

Go to download
package org.opentripplanner.updater.trip;

import com.google.transit.realtime.GtfsRealtime;
import com.google.transit.realtime.GtfsRealtime.FeedEntity;
import com.google.transit.realtime.GtfsRealtime.FeedMessage;
import com.google.transit.realtime.GtfsRealtime.TripUpdate;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/** Reads the GTFS-RT from a local file. */
public class GtfsRealtimeFileTripUpdateSource implements TripUpdateSource {

  private static final Logger LOG = LoggerFactory.getLogger(GtfsRealtimeFileTripUpdateSource.class);

  private final File file;
  /**
   * Default agency id that is used for the trip ids in the TripUpdates
   */
  private final String feedId;
  /**
   * True iff the last list with updates represent all updates that are active right now, i.e. all
   * previous updates should be disregarded
   */
  private boolean fullDataset = true;

  public GtfsRealtimeFileTripUpdateSource(Parameters config) {
    this.feedId = getFeedId();
    this.file = new File(config.getFile());
  }

  @Override
  public List getUpdates() {
    FeedMessage feedMessage = null;
    List feedEntityList = null;
    List updates = null;
    fullDataset = true;
    try {
      InputStream is = new FileInputStream(file);

      // Decode message
      feedMessage = FeedMessage.PARSER.parseFrom(is);
      feedEntityList = feedMessage.getEntityList();

      // Change fullDataset value if this is an incremental update
      if (
        feedMessage.hasHeader() &&
        feedMessage.getHeader().hasIncrementality() &&
        feedMessage
          .getHeader()
          .getIncrementality()
          .equals(GtfsRealtime.FeedHeader.Incrementality.DIFFERENTIAL)
      ) {
        fullDataset = false;
      }

      // Create List of TripUpdates
      updates = new ArrayList<>(feedEntityList.size());
      for (FeedEntity feedEntity : feedEntityList) {
        if (feedEntity.hasTripUpdate()) {
          updates.add(feedEntity.getTripUpdate());
        }
      }
    } catch (Exception e) {
      LOG.warn("Failed to parse gtfs-rt feed at {}", file, e);
    }
    return updates;
  }

  @Override
  public boolean getFullDatasetValueOfLastUpdates() {
    return fullDataset;
  }

  @Override
  public String getFeedId() {
    return this.feedId;
  }

  public String toString() {
    return "GtfsRealtimeFileTripUpdateSource(" + file + ")";
  }

  public interface Parameters {
    String getFeedId();

    String getFile();
  }
}