org.opentripplanner.apis.gtfs.datafetchers.TripOnServiceDateImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of application Show documentation
The newest version!
package org.opentripplanner.apis.gtfs.datafetchers;

import graphql.schema.DataFetcher;
import graphql.schema.DataFetchingEnvironment;
import java.time.Instant;
import java.time.LocalDate;
import java.util.List;
import javax.annotation.Nullable;
import org.opentripplanner.apis.gtfs.GraphQLRequestContext;
import org.opentripplanner.apis.gtfs.generated.GraphQLDataFetchers;
import org.opentripplanner.model.Timetable;
import org.opentripplanner.model.TripTimeOnDate;
import org.opentripplanner.transit.model.network.TripPattern;
import org.opentripplanner.transit.model.timetable.Trip;
import org.opentripplanner.transit.model.timetable.TripOnServiceDate;
import org.opentripplanner.transit.service.TransitService;
import org.opentripplanner.utils.time.ServiceDateUtils;

public class TripOnServiceDateImpl implements GraphQLDataFetchers.GraphQLTripOnServiceDate {

  @Override
  public DataFetcher serviceDate() {
    return env -> getSource(env).getServiceDate();
  }

  @Override
  public DataFetcher end() {
    return environment -> {
      var arguments = getFromTripTimesArguments(environment);
      if (arguments.timetable() == null) {
        return null;
      }
      return TripTimeOnDate.lastFromTripTimes(
        arguments.timetable(),
        arguments.trip(),
        arguments.serviceDate(),
        arguments.midnight()
      );
    };
  }

  @Override
  public DataFetcher start() {
    return environment -> {
      var arguments = getFromTripTimesArguments(environment);
      if (arguments.timetable() == null) {
        return null;
      }
      return TripTimeOnDate.firstFromTripTimes(
        arguments.timetable(),
        arguments.trip(),
        arguments.serviceDate(),
        arguments.midnight()
      );
    };
  }

  @Override
  public DataFetcher> stopCalls() {
    return environment -> {
      var arguments = getFromTripTimesArguments(environment);
      if (arguments.timetable() == null) {
        return List.of();
      }
      return TripTimeOnDate.fromTripTimes(
        arguments.timetable(),
        arguments.trip(),
        arguments.serviceDate(),
        arguments.midnight()
      );
    };
  }

  @Override
  public DataFetcher trip() {
    return this::getTrip;
  }

  @Nullable
  private Timetable getTimetable(
    DataFetchingEnvironment environment,
    Trip trip,
    LocalDate serviceDate
  ) {
    TransitService transitService = getTransitService(environment);
    TripPattern tripPattern = transitService.findPattern(trip, serviceDate);
    // no matching pattern found
    if (tripPattern == null) {
      return null;
    }

    return transitService.findTimetable(tripPattern, serviceDate);
  }

  private TransitService getTransitService(DataFetchingEnvironment environment) {
    return environment.getContext().transitService();
  }

  private Trip getTrip(DataFetchingEnvironment environment) {
    return getSource(environment).getTrip();
  }

  private TripOnServiceDate getSource(DataFetchingEnvironment environment) {
    return environment.getSource();
  }

  private FromTripTimesArguments getFromTripTimesArguments(DataFetchingEnvironment environment) {
    TransitService transitService = getTransitService(environment);
    Trip trip = getTrip(environment);
    var serviceDate = getSource(environment).getServiceDate();

    Instant midnight = ServiceDateUtils.asStartOfService(
      serviceDate,
      transitService.getTimeZone()
    ).toInstant();
    Timetable timetable = getTimetable(environment, trip, serviceDate);
    return new FromTripTimesArguments(trip, serviceDate, midnight, timetable);
  }

  private record FromTripTimesArguments(
    Trip trip,
    LocalDate serviceDate,
    Instant midnight,
    @Nullable Timetable timetable
  ) {}
}