org.opentripplanner.ext.legacygraphqlapi.datafetchers.LegacyGraphQLfareComponentImpl 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.ext.legacygraphqlapi.datafetchers;

import graphql.schema.DataFetcher;
import graphql.schema.DataFetchingEnvironment;
import java.util.stream.Collectors;
import org.opentripplanner.ext.legacygraphqlapi.LegacyGraphQLRequestContext;
import org.opentripplanner.ext.legacygraphqlapi.generated.LegacyGraphQLDataFetchers;
import org.opentripplanner.routing.core.FareComponent;
import org.opentripplanner.transit.model.network.Route;
import org.opentripplanner.transit.service.TransitService;

public class LegacyGraphQLfareComponentImpl
  implements LegacyGraphQLDataFetchers.LegacyGraphQLFareComponent {

  @Override
  public DataFetcher cents() {
    return environment -> getSource(environment).price().cents();
  }

  @Override
  public DataFetcher currency() {
    return environment -> getSource(environment).price().currency().getCurrencyCode();
  }

  @Override
  public DataFetcher fareId() {
    return environment -> getSource(environment).fareId().toString();
  }

  @Override
  public DataFetcher> routes() {
    return environment -> {
      TransitService transitService = getTransitService(environment);
      return getSource(environment)
        .routes()
        .stream()
        .map(transitService::getRouteForId)
        .collect(Collectors.toList());
    };
  }

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

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