org.opentripplanner.ext.legacygraphqlapi.datafetchers.LegacyGraphQLserviceTimeRangeImpl Maven / Gradle / Ivy

Go to download

package org.opentripplanner.ext.legacygraphqlapi.datafetchers;

import graphql.schema.DataFetcher;
import graphql.schema.DataFetchingEnvironment;
import org.opentripplanner.ext.legacygraphqlapi.LegacyGraphQLRequestContext;
import org.opentripplanner.ext.legacygraphqlapi.generated.LegacyGraphQLDataFetchers;
import org.opentripplanner.transit.service.TransitService;

public class LegacyGraphQLserviceTimeRangeImpl
  implements LegacyGraphQLDataFetchers.LegacyGraphQLServiceTimeRange {

  @Override
  public DataFetcher end() {
    return environment -> getTransitService(environment).getTransitServiceEnds().toEpochSecond();
  }

  @Override
  public DataFetcher start() {
    return environment -> getTransitService(environment).getTransitServiceStarts().toEpochSecond();
  }

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