/* This program is free software: you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public License
as published by the Free Software Foundation, either version 3 of
the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>. */
package org.opentripplanner.updater.bike_rental;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;
import java.util.prefs.Preferences;
import javax.xml.parsers.ParserConfigurationException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.opentripplanner.updater.PreferencesConfigurable;
import org.opentripplanner.routing.bike_rental.BikeRentalStation;
import org.opentripplanner.routing.graph.Graph;
import org.opentripplanner.util.HttpUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.xml.sax.SAXException;
// TODO This class could probably inherit from GenericJSONBikeRentalDataSource
public class CityBikesBikeRentalDataSource implements BikeRentalDataSource, PreferencesConfigurable {
private static final Logger log = LoggerFactory.getLogger(BixiBikeRentalDataSource.class);
private String url;
ArrayList<BikeRentalStation> stations = new ArrayList<BikeRentalStation>();
public CityBikesBikeRentalDataSource() {
}
@Override
public boolean update() {
try {
InputStream stream = HttpUtils.getData(url);
if (stream == null) {
log.warn("Failed to get data from url " + url);
return false;
}
Reader reader = new BufferedReader(new InputStreamReader(stream,
Charset.forName("UTF-8")));
StringBuilder builder = new StringBuilder();
char[] buffer = new char[4096];
int charactersRead;
while ((charactersRead = reader.read(buffer, 0, buffer.length)) > 0) {
builder.append(buffer, 0, charactersRead);
}
String data = builder.toString();
parseJson(data);
} catch (IOException e) {
log.warn("Error reading bike rental feed from " + url, e);
return false;
} catch (ParserConfigurationException e) {
throw new RuntimeException(e);
} catch (SAXException e) {
log.warn("Error parsing bike rental feed from " + url + "(bad XML of some sort)", e);
return false;
}
return true;
}
private void parseJson(String data) throws ParserConfigurationException, SAXException,
IOException {
ArrayList<BikeRentalStation> out = new ArrayList<BikeRentalStation>();
// Jackson ObjectMapper to read in JSON
// TODO: test against real data
ObjectMapper mapper = new ObjectMapper();
for (JsonNode stationNode : mapper.readTree(data)) {
BikeRentalStation brStation = new BikeRentalStation();
// We need string IDs but they are in JSON as numbers. Avoid null from textValue(). See pull req #1450.
brStation.id = String.valueOf(stationNode.get("id").intValue());
brStation.x = stationNode.get("lng").doubleValue() / 1000000.0;
brStation.y = stationNode.get("lat").doubleValue() / 1000000.0;
brStation.name = stationNode.get("name").textValue();
brStation.bikesAvailable = stationNode.get("bikes").intValue();
brStation.spacesAvailable = stationNode.get("free").intValue();
if (brStation != null && brStation.id != null) {
out.add(brStation);
}
}
synchronized (this) {
stations = out;
}
}
@Override
public synchronized List<BikeRentalStation> getStations() {
return stations;
}
public String getUrl() {
return url;
}
public void setUrl(String url) {
this.url = url;
}
@Override
public String toString() {
return getClass().getName() + "(" + url + ")";
}
@Override
public void configure(Graph graph, Preferences preferences) {
String url = preferences.get("url", null);
if (url == null)
throw new IllegalArgumentException("Missing mandatory 'url' configuration.");
setUrl(url);
}
}