/[projects]/android/TrainInfoService/src/dk/thoerup/traininfoservice/banedk/TimetableFetcher.java
ViewVC logotype

Diff of /android/TrainInfoService/src/dk/thoerup/traininfoservice/banedk/TimetableFetcher.java

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 591 by torben, Wed Feb 10 14:07:04 2010 UTC revision 976 by torben, Sat Jul 10 09:10:50 2010 UTC
# Line 3  package dk.thoerup.traininfoservice.bane Line 3  package dk.thoerup.traininfoservice.bane
3    
4  import java.io.IOException;  import java.io.IOException;
5  import java.net.URL;  import java.net.URL;
6    import java.sql.SQLException;
7  import java.util.ArrayList;  import java.util.ArrayList;
8  import java.util.List;  import java.util.List;
9  import java.util.Map;  import java.util.Map;
10    import java.util.logging.Level;
11  import java.util.logging.Logger;  import java.util.logging.Logger;
12    
13  import com.gargoylesoftware.htmlunit.BrowserVersion;  import com.gargoylesoftware.htmlunit.BrowserVersion;
# Line 18  import com.gargoylesoftware.htmlunit.htm Line 20  import com.gargoylesoftware.htmlunit.htm
20    
21  import dk.thoerup.circuitbreaker.CircuitBreaker;  import dk.thoerup.circuitbreaker.CircuitBreaker;
22  import dk.thoerup.circuitbreaker.CircuitBreakerManager;  import dk.thoerup.circuitbreaker.CircuitBreakerManager;
23    import dk.thoerup.traininfoservice.StationDAO;
24    import dk.thoerup.traininfoservice.Statistics;
25    
26  public class TimetableFetcher {  public class TimetableFetcher {
27                    
# Line 28  public class TimetableFetcher { Line 32  public class TimetableFetcher {
32          }          }
33                                    
34          Map<String, List<TimetableBean>> cache;          Map<String, List<TimetableBean>> cache;
35            Map<String, Integer> stationCache;
36    
37            StationDAO stationDao = new StationDAO();
38                    
39                    
40          Logger logger = Logger.getLogger(TimetableFetcher.class.getName());          Logger logger = Logger.getLogger(TimetableFetcher.class.getName());
# Line 38  public class TimetableFetcher { Line 45  public class TimetableFetcher {
45                  useTempSite = tmpSite;                  useTempSite = tmpSite;
46                                    
47                  cache = new TimeoutMap<String,List<TimetableBean>>(cacheTimeout);                  cache = new TimeoutMap<String,List<TimetableBean>>(cacheTimeout);
48                    stationCache = new TimeoutMap<String,Integer>( 3*60*60*1000 );
49          }          }
50                    
51                    
# Line 49  public class TimetableFetcher { Line 57  public class TimetableFetcher {
57                          list = lookupTimetable(trainID,type);                          list = lookupTimetable(trainID,type);
58                          cache.put(key, list);                          cache.put(key, list);
59                  } else {                  } else {
60                            Statistics.getInstance().incrementTimetableCacheHits();
61                          logger.info("Timetable: Cache hit " + trainID);                          logger.info("Timetable: Cache hit " + trainID);
62                  }                  }
63                  return list;                  return list;
# Line 61  public class TimetableFetcher { Line 70  public class TimetableFetcher {
70                          return new ArrayList<TimetableBean>(); // no timetable data on temp site                          return new ArrayList<TimetableBean>(); // no timetable data on temp site
71                  }                  }
72          }          }
73            
74            int getStationId(String name) {
75                    Integer id = stationCache.get(name);
76                    
77                    if (id == null) {
78                            try {
79                                    id = stationDao.getIdByName(name);
80                                    stationCache.put(name, id);
81                            } catch (SQLException e) {
82                                    logger.log(Level.SEVERE, "getStationId failed", e);
83                                    id = -1;
84                            }
85                    }
86    
87                    return id;
88            }
89    
90          List<TimetableBean> lookupTimetableRealSite(String trainID, String type) throws Exception {                      List<TimetableBean> lookupTimetableRealSite(String trainID, String type) throws Exception {            
91                  List<TimetableBean> timetableList = new ArrayList<TimetableBean>();                  List<TimetableBean> timetableList = new ArrayList<TimetableBean>();
92                                    
93                  String url = "http://www.bane.dk/visRute.asp?W=" + type + "&TogNr=" + trainID + "&artikelId=4276";                  //String url = "http://www.bane.dk/visRute.asp?W=" + type + "&TogNr=" + trainID + "&artikelId=4276";
94                                                    String url = "http://trafikinfo.bane.dk/TrafikInformation/Ruteplan/" + trainID;                
95    
96              final WebClient webClient = new WebClient(BrowserVersion.FIREFOX_3);              final WebClient webClient = new WebClient(BrowserVersion.FIREFOX_3);
97              webClient.setTimeout(2500);              webClient.setTimeout(2500);
# Line 75  public class TimetableFetcher { Line 100  public class TimetableFetcher {
100              webClient.setCssEnabled(false);              webClient.setCssEnabled(false);
101                            
102                            
103              BanedkInvocation wrapper = new BanedkInvocation(webClient, url);              HtmlunitInvocation wrapper = new HtmlunitInvocation(webClient, url);
104              CircuitBreaker breaker = CircuitBreakerManager.getManager().getCircuitBreaker("banedk");              CircuitBreaker breaker = CircuitBreakerManager.getManager().getCircuitBreaker("banedk");
105                            
106              HtmlPage page = (HtmlPage) breaker.invoke(wrapper);              HtmlPage page = (HtmlPage) breaker.invoke(wrapper);
# Line 103  public class TimetableFetcher { Line 128  public class TimetableFetcher {
128                          }                          }
129                                                    
130                          TimetableBean bean = new TimetableBean();                          TimetableBean bean = new TimetableBean();
131                          bean.setStation( fields.get(0).asText() );                          
132                            String station = fields.get(0).asText() ;
133                            if (station.equals("København"))
134                                    station = "København H"; //correct inconsistency in naming
135                            
136                            bean.setStation( station );
137                          bean.setArrival( fields.get(1).asText() );                          bean.setArrival( fields.get(1).asText() );
138                          bean.setDeparture( fields.get(2).asText() );                          bean.setDeparture( fields.get(2).asText() );
139                                                    
140                            boolean cancelled = fields.get(3).asText().equalsIgnoreCase("aflyst");
141                            bean.setCancelled(cancelled);
142                            
143                          if (currentStation == true && currentStationSaved == false ) {                          if (currentStation == true && currentStationSaved == false ) {
144                                  bean.setCurrent(currentStation);                                  bean.setCurrent(currentStation);
145                                  currentStationSaved = true;                                  currentStationSaved = true;
146                          }                          }
147                                                    
148                            bean.setStationId( getStationId( station ));
149                            
150                          timetableList.add(bean);                          timetableList.add(bean);
151                    }
152                    
153                    final String cancelledString = "Aflyst";
154                    for (int i=0;i<timetableList.size(); i++) { //handle cancelled labels
155                            final int lastIdx = (timetableList.size() - 1);
156                            
157                            TimetableBean current = timetableList.get(i);
158                            if (current.isCancelled()) {
159                                    if (i == 0) {
160                                            current.setDeparture(cancelledString);
161                                    } else if (i == lastIdx) {
162                                            current.setArrival(cancelledString);
163                                    } else if (i>0 && i<lastIdx) {
164                                            TimetableBean next = timetableList.get(i+1);
165                                            TimetableBean prev = timetableList.get(i-1);
166                                            
167                                            if (next.isCancelled())
168                                                    current.setDeparture(cancelledString);
169                                            if (prev.isCancelled())
170                                                    current.setArrival(cancelledString);
171                                    }
172                            }
173                  }                  }
174                                    
175              } else {              } else {
Legend:
Removed from v.591  
changed lines
  Added in v.976
  ViewVC Help
Powered by ViewVC 1.1.20