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

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

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

revision 386 by torben, Fri Oct 2 13:44:31 2009 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;
10    import java.util.logging.Level;
11  import java.util.logging.Logger;  import java.util.logging.Logger;
12    
13  import com.gargoylesoftware.htmlunit.*;  import com.gargoylesoftware.htmlunit.BrowserVersion;
14  import com.gargoylesoftware.htmlunit.html.*;  import com.gargoylesoftware.htmlunit.Page;
15    import com.gargoylesoftware.htmlunit.RefreshHandler;
16    import com.gargoylesoftware.htmlunit.WebClient;
17    import com.gargoylesoftware.htmlunit.html.DomNodeList;
18    import com.gargoylesoftware.htmlunit.html.HtmlElement;
19    import com.gargoylesoftware.htmlunit.html.HtmlPage;
20    
21    import dk.thoerup.circuitbreaker.CircuitBreaker;
22    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 17  public class TimetableFetcher { Line 30  public class TimetableFetcher {
30                  }                  }
31                                    
32          }          }
33                    
34            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());
41            
42            private boolean useTempSite;
43            
44            public TimetableFetcher(boolean tmpSite, int cacheTimeout) {
45                    useTempSite = tmpSite;
46                    
47                    cache = new TimeoutMap<String,List<TimetableBean>>(cacheTimeout);
48                    stationCache = new TimeoutMap<String,Integer>( 3*60*60*1000 );
49            }
50            
51            
52            List<TimetableBean> cachedLookupTimetable(String trainID, String type) throws Exception {
53                    String key = trainID+type;
54                    List<TimetableBean> list = cache.get(key);
55                    
56                    if (list == null) {
57                            list = lookupTimetable(trainID,type);
58                            cache.put(key, list);
59                    } else {
60                            Statistics.getInstance().incrementTimetableCacheHits();
61                            logger.info("Timetable: Cache hit " + trainID);
62                    }
63                    return list;
64            }
65            
66          List<TimetableBean> lookupTimetable(String trainID, String type) throws Exception {          List<TimetableBean> lookupTimetable(String trainID, String type) throws Exception {
67                    if (useTempSite == false ){
68                            return lookupTimetableRealSite(trainID, type);
69                    } else {
70                            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 {            
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();              final WebClient webClient = new WebClient(BrowserVersion.FIREFOX_3);
97              webClient.setTimeout(2500);              webClient.setTimeout(2500);
98              webClient.setJavaScriptEnabled(false);                      webClient.setJavaScriptEnabled(false);        
99              webClient.setRefreshHandler( new NullRefreshHandler() );              webClient.setRefreshHandler( new NullRefreshHandler() );
100              webClient.setCssEnabled(false);              webClient.setCssEnabled(false);
101                            
102                            
103              final HtmlPage page = webClient.getPage(url);              HtmlunitInvocation wrapper = new HtmlunitInvocation(webClient, url);
104                CircuitBreaker breaker = CircuitBreakerManager.getManager().getCircuitBreaker("banedk");
105                            
106                HtmlPage page = (HtmlPage) breaker.invoke(wrapper);
107                        
108                            
109              boolean currentStation = false;              boolean currentStation = false;
110              boolean currentStationSaved = false;              boolean currentStationSaved = false;
# Line 58  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 {
176                  logger.warning("No time table found, trainID=" + trainID + " type=" + type);                  logger.warning("No time table found, trainID=" + trainID + " type=" + type);
177              }              }
178                webClient.closeAllWindows();
179                                    
180                  return timetableList;                  return timetableList;
181          }          }
Legend:
Removed from v.386  
changed lines
  Added in v.976
  ViewVC Help
Powered by ViewVC 1.1.20