org.opentripplanner.updater.bike_rental.datasources.BicimadBikeRentalDataSource 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.updater.bike_rental.datasources;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.opentripplanner.routing.bike_rental.BikeRentalStation;
import org.opentripplanner.updater.bike_rental.BikeRentalDataSource;
import org.opentripplanner.updater.bike_rental.datasources.params.BikeRentalDataSourceParameters;
import org.opentripplanner.util.HttpUtils;
import org.opentripplanner.util.NonLocalizedString;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;

/**
 * Implementation of a BikeRentalDataSource for the BICIMAD API.
 *
 * @see BikeRentalDataSource
 */
class BicimadBikeRentalDataSource implements BikeRentalDataSource {

  private static final Logger log = LoggerFactory.getLogger(GenericJsonBikeRentalDataSource.class);

  private static final String jsonExternalParsePath = "data";

  private static final String jsonInternalParsePath = "stations";

  private final String url;

  List stations = new ArrayList<>();

  public BicimadBikeRentalDataSource(BikeRentalDataSourceParameters config) {
    this.url = config.getUrl();
  }

  @Override
  public boolean update() {
    try {
      InputStream data;

      URL url2 = new URL(url);

      String proto = url2.getProtocol();
      if (proto.equals("http") || proto.equals("https")) {
        data = HttpUtils.getData(url);
      }
      else {
        // Local file probably, try standard java
        data = url2.openStream();
      }

      if (data == null) {
        log.warn("Failed to get data from url " + url);
        return false;
      }
      parseJSON(data);
      data.close();
    }
    catch (IllegalArgumentException e) {
      log.warn("Error parsing bike rental feed from " + url, e);
      return false;
    }
    catch (JsonProcessingException e) {
      log.warn("Error parsing bike rental feed from " + url + "(bad JSON of some sort)", e);
      return false;
    }
    catch (IOException e) {
      log.warn("Error reading bike rental feed from " + url, e);
      return false;
    }
    return true;
  }

  private void parseJSON(InputStream dataStream) throws IllegalArgumentException, IOException {

    ArrayList out = new ArrayList<>();

    String rentalString = convertStreamToString(dataStream);

    JsonNode rootNode = getJsonNode(rentalString, jsonExternalParsePath);
    rootNode = getJsonNode(rootNode.asText(), jsonInternalParsePath);
    for (int i = 0; i < rootNode.size(); i++) {
      JsonNode node = rootNode.get(i);
      if (node == null) {
        continue;
      }
      BikeRentalStation brstation = makeStation(node);
            if (brstation != null) { out.add(brstation); }
    }
    synchronized (this) {
      stations = out;
    }
  }

  private String convertStreamToString(java.io.InputStream is) {
    java.util.Scanner scanner = null;
    String result;
    try {

      scanner = new java.util.Scanner(is).useDelimiter("\\A");
      result = scanner.hasNext() ? scanner.next() : "";
      scanner.close();
    }
    finally {
            if (scanner != null) { scanner.close(); }
    }
    return result;

  }

  private JsonNode getJsonNode(String rentalString, String jsonParsePath) throws IOException {
    ObjectMapper mapper = new ObjectMapper();
    JsonNode rootNode = mapper.readTree(rentalString);

    if (!jsonParsePath.equals("")) {
      String delimiter = "/";
      String[] parseElement = jsonParsePath.split(delimiter);
      for (int i = 0; i < parseElement.length; i++) {
        rootNode = rootNode.path(parseElement[i]);
      }

      if (rootNode.isMissingNode()) {
        throw new IllegalArgumentException("Could not find jSON elements " + jsonParsePath);
      }
    }
    return rootNode;
  }

  public BikeRentalStation makeStation(JsonNode node) {

    if (!node.path("activate").asText().equals("1")) {
      return null;
    }
    BikeRentalStation station = new BikeRentalStation();
    station.id = String.format("%d", node.path("id").asInt());
    station.x = node.path("longitude").asDouble();
    station.y = node.path("latitude").asDouble();
    station.name = new NonLocalizedString(node.path("name").asText());
    station.bikesAvailable = node.path("dock_bikes").asInt();
    station.spacesAvailable = node.path("free_bases").asInt();
    return station;
  }

  @Override
  public synchronized List getStations() {
    return stations;
  }

  @Override
  public String toString() {
    return getClass().getName() + "(" + url + ")";
  }

  public String getUrl() {
    return url;
  }
}