--- android/TrainInfoService/src/dk/thoerup/traininfoservice/banedk/DepartureFetcher.java 2010/02/08 18:42:00 587 +++ android/TrainInfoService/src/dk/thoerup/traininfoservice/banedk/DepartureFetcher.java 2010/02/08 19:12:15 588 @@ -1,8 +1,5 @@ package dk.thoerup.traininfoservice.banedk; -import java.sql.Connection; -import java.sql.ResultSet; -import java.sql.Statement; import java.util.ArrayList; import java.util.Collections; import java.util.List; @@ -16,7 +13,8 @@ import dk.thoerup.circuitbreaker.CircuitBreaker; import dk.thoerup.circuitbreaker.CircuitBreakerManager; -import dk.thoerup.traininfoservice.DBConnection; +import dk.thoerup.traininfoservice.StationBean; +import dk.thoerup.traininfoservice.StationDAO; public class DepartureFetcher { @@ -24,6 +22,8 @@ Map> cache; + StationDAO stationDao = new StationDAO(); + private boolean useTempSite; public DepartureFetcher(boolean tempSite, int cacheTimeout) { @@ -51,37 +51,21 @@ public List lookupDepartures(int stationID) throws Exception { List departureList = new ArrayList(); - Connection conn = null; - try - { - conn = DBConnection.getConnection(); - - String SQL = "SELECT stationcode_fjrn, stationcode_stog FROM trainstations WHERE id=" + stationID; - Statement stmt = conn.createStatement(); - ResultSet rs = stmt.executeQuery(SQL); - - if (rs.next()) { - String code = rs.getString( 1 ); - if (! rs.wasNull() ) { - List list = lookupDepartures(code, "FJRN"); - departureList.addAll(list); - } - - code = rs.getString(2); - if (! rs.wasNull() ) { - List list = lookupDepartures(code, "S2"); - departureList.addAll(list); - } - Collections.sort( departureList ); - - } - - } finally { - if (conn != null && !conn.isClosed() ) { - conn.close(); - } + StationBean station = stationDao.getById(stationID); + + if (station.getRegional() != null) { + List list = lookupDepartures(station.getRegional(), "FJRN"); + departureList.addAll(list); } + if (station.getStrain() != null) { + List list = lookupDepartures(station.getStrain(), "S2"); + departureList.addAll(list); + } + + Collections.sort( departureList ); + + return departureList; }