org.opentripplanner.ext.transmodelapi.TransmodelAPI Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of otp Show documentation
Show all versions of otp Show documentation
The OpenTripPlanner multimodal journey planning system
package org.opentripplanner.ext.transmodelapi;
import com.fasterxml.jackson.databind.ObjectMapper;
import graphql.schema.GraphQLSchema;
import org.opentripplanner.ext.transmodelapi.mapping.TransitIdMapper;
import org.opentripplanner.ext.transmodelapi.support.GqlUtil;
import org.opentripplanner.routing.api.request.RoutingRequest;
import org.opentripplanner.routing.graph.Graph;
import org.opentripplanner.standalone.server.OTPServer;
import org.opentripplanner.standalone.server.Router;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.ws.rs.BadRequestException;
import javax.ws.rs.Consumes;
import javax.ws.rs.DefaultValue;
import javax.ws.rs.GET;
import javax.ws.rs.HeaderParam;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Callable;
import java.util.concurrent.CancellationException;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;
// TODO move to org.opentripplanner.api.resource, this is a Jersey resource class
@Path("/routers/{ignoreRouterId}/transmodel/index") // It would be nice to get rid of the final /index.
@Produces(MediaType.APPLICATION_JSON) // One @Produces annotation for all endpoints.
public class TransmodelAPI {
@SuppressWarnings("unused")
private static final Logger LOG = LoggerFactory.getLogger(TransmodelAPI.class);
private static GqlUtil gqlUtil;
private static GraphQLSchema schema;
private final Router router;
private final TransmodelGraph index;
private final ObjectMapper deserializer = new ObjectMapper();
/**
* @deprecated The support for multiple routers are removed from OTP2.
* See https://github.com/opentripplanner/OpenTripPlanner/issues/2760
*/
@Deprecated @PathParam("ignoreRouterId")
private String ignoreRouterId;
public TransmodelAPI(@Context OTPServer otpServer) {
this.router = otpServer.getRouter();
this.index = new TransmodelGraph(schema);
}
/**
* This method should be called BEFORE the Web-Container is started and load new
* instances of this class. This is a hack, and it would be better if the configuration
* was done more explicit and enforced, not relaying on a "static" setup method to be called.
*/
public static void setUp(
boolean hideFeedId,
Graph graph,
RoutingRequest defaultRoutingRequest
) {
if(hideFeedId) {
TransitIdMapper.setupFixedFeedId(graph.getAgencies());
}
gqlUtil = new GqlUtil(graph.getTimeZone());
schema = TransmodelGraphQLSchema.create(defaultRoutingRequest, gqlUtil);
}
/**
* Return 200 when service is loaded.
*/
@GET
@Path("/live")
public Response isAlive() {
return Response.status(Response.Status.NO_CONTENT).build();
}
@POST
@Path("/graphql")
@Consumes(MediaType.APPLICATION_JSON)
public Response getGraphQL(HashMap queryParameters, @HeaderParam("OTPMaxResolves") @DefaultValue("1000000") int maxResolves) {
if (queryParameters==null || !queryParameters.containsKey("query")) {
LOG.debug("No query found in body");
throw new BadRequestException("No query found in body");
}
String query = (String) queryParameters.get("query");
Object queryVariables = queryParameters.getOrDefault("variables", null);
String operationName = (String) queryParameters.getOrDefault("operationName", null);
Map variables;
if (queryVariables instanceof Map) {
variables = (Map) queryVariables;
} else if (queryVariables instanceof String && !((String) queryVariables).isEmpty()) {
try {
variables = deserializer.readValue((String) queryVariables, Map.class);
} catch (IOException e) {
throw new BadRequestException("Variables must be a valid json object");
}
} else {
variables = new HashMap<>();
}
return index.getGraphQLResponse(query, router, variables, operationName, maxResolves);
}
@POST
@Path("/graphql")
@Consumes("application/graphql")
public Response getGraphQL(String query, @HeaderParam("OTPMaxResolves") @DefaultValue("1000000") int maxResolves) {
return index.getGraphQLResponse(query, router, null, null, maxResolves);
}
@POST
@Path("/graphql/batch")
@Consumes(MediaType.APPLICATION_JSON)
public Response getGraphQLBatch(List> queries, @HeaderParam("OTPTimeout") @DefaultValue("10000") int timeout, @HeaderParam("OTPMaxResolves") @DefaultValue("1000000") int maxResolves) {
List> responses = new ArrayList<>();
List> futures = new ArrayList();
for (HashMap query : queries) {
Map variables;
if (query.get("variables") instanceof Map) {
variables = (Map) query.get("variables");
} else if (query.get("variables") instanceof String && ((String) query.get("variables")).length() > 0) {
try {
variables = deserializer.readValue((String) query.get("variables"), Map.class);
} catch (IOException e) {
throw new BadRequestException("Variables must be a valid json object");
}
} else {
variables = null;
}
String operationName = (String) query.getOrDefault("operationName", null);
futures.add(() -> index.getGraphQLExecutionResult((String) query.get("query"), router,
variables, operationName, maxResolves));
}
try {
List> results = index.threadPool.invokeAll(futures);
for (int i = 0; i < queries.size(); i++) {
HashMap response = new HashMap<>();
response.put("id", queries.get(i).get("id"));
response.put("payload", results.get(i).get());
responses.add(response);
}
} catch (CancellationException | ExecutionException | InterruptedException e) {
return Response.status(Response.Status.INTERNAL_SERVER_ERROR).build();
}
return Response.status(Response.Status.OK).entity(responses).build();
}
}