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

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

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

revision 462 by torben, Wed Oct 21 08:43:45 2009 UTC revision 1396 by torben, Thu Apr 28 10:13:36 2011 UTC
# Line 1  Line 1 
1  package dk.thoerup.traininfoservice.banedk;  package dk.thoerup.traininfoservice.banedk;
2    
3  import java.io.IOException;  import java.io.IOException;
4  import java.util.List;  import java.io.StringWriter;
5    
6  import java.util.logging.Level;  import java.util.logging.Level;
7  import java.util.logging.Logger;  import java.util.logging.Logger;
8    
9  import javax.servlet.ServletException;  import javax.servlet.ServletException;
10    import javax.servlet.annotation.WebServlet;
11  import javax.servlet.http.HttpServlet;  import javax.servlet.http.HttpServlet;
12  import javax.servlet.http.HttpServletRequest;  import javax.servlet.http.HttpServletRequest;
13  import javax.servlet.http.HttpServletResponse;  import javax.servlet.http.HttpServletResponse;
14    
15  import dk.thoerup.curcuitbreaker.CircuitBreakerException;  import org.simpleframework.xml.Serializer;
16    import org.simpleframework.xml.core.Persister;
17    
18    import dk.thoerup.android.traininfo.common.DepartureBean;
19    import dk.thoerup.android.traininfo.common.TimetableBean;
20    import dk.thoerup.circuitbreaker.CircuitBreakerException;
21    import dk.thoerup.traininfoservice.Statistics;
22    import dk.thoerup.traininfoservice.TraininfoSettings;;
23    
24  /**  /**
25   * Servlet implementation class TimetableServlet   * Servlet implementation class TimetableServlet
26   */   */
27    @WebServlet(urlPatterns={"/TimetableServlet"})
28  public class TimetableServlet extends HttpServlet {  public class TimetableServlet extends HttpServlet {
29          private static final long serialVersionUID = 1L;          private static final long serialVersionUID = 1L;
30                
31                    
32          Logger logger = Logger.getLogger( TimetableServlet.class.getName() );          Logger logger = Logger.getLogger( TimetableServlet.class.getName() );
33                    
34          TimetableFetcher fetcher = new TimetableFetcher();          TimetableFetcher fetcher;
35            
36            
37            @Override
38            public void init() throws ServletException {
39                    super.init();
40                    
41    
42                    TraininfoSettings settings = (TraininfoSettings) getServletContext().getAttribute("settings");
43                    fetcher = new TimetableFetcher(settings);
44            }
45                    
46          @Override          @Override
47          protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {          protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
48                  if (request.getParameter("train") == null) {                  if (request.getParameter("train") == null) {
49                          response.sendError(400, "not enough parameters");                          response.sendError(400, "not enough parameters (train)");
50                          return;                          return;
51                  }                  }
52                                    if (request.getParameter("type") == null) {
53                  String train = request.getParameter("train").trim();                          response.sendError(400, "not enough parameters (type)");
                   
                 String sections[] = train.split(" ");  
                 if (sections.length != 2) {  
                         response.sendError(400, "Invalid parameter");  
54                          return;                          return;
55                  }                  }
56                    Statistics.getInstance().incrementTimetableLookups();
57                    
58                    String train = request.getParameter("train").trim();
59                    String type = request.getParameter("type").trim();
60                            
61                                    
62                  String type = lookupType(sections[0]);                  TimetableBean timetable = null;
                 String trainID = sections[1];  
63                  try                  try
64                  {                  {
65                          List<TimetableBean> timetable = fetcher.cachedLookupTimetable(trainID, type);                          timetable = fetcher.cachedLookupTimetable(train, type);
                   
                         String responseXML = formatTimetable(timetable);  
66                                    
                         response.setContentType("text/xml");  
                         response.getWriter().print(responseXML);  
67                  } catch (java.io.IOException ioe) {                  } catch (java.io.IOException ioe) {
68                          logger.warning("(timetable) Read failed, train="+train + ". " + ioe.getMessage());                          logger.warning("(timetable) Read failed, type=" + type + ", train="+train + ". " + ioe.getMessage());
69                          response.sendError(500, "Backend timed out");                          Statistics.getInstance().incrementTimetableErrors();
70                            //timetable = generateErrorBean(1);
71                            response.sendError( 500, "Backend didn't answer" );
72                  } catch (CircuitBreakerException cbe) {                  } catch (CircuitBreakerException cbe) {
73                          logger.warning("Circuitbreaker - failing fast");                          logger.warning("Circuitbreaker - failing fast, type=" + type + ", train="+train);
74                          response.sendError(500);                          Statistics.getInstance().incrementTimetableErrors();                    
75                            timetable = generateErrorBean(1);
76                  }catch (Exception e) {                  }catch (Exception e) {
77                          logger.log(Level.WARNING, "failure looking up timetable: train=" +train, e);                          logger.log(Level.WARNING, "failure looking up timetable: type=" + type + ",train=" +train, e);
78                            Statistics.getInstance().incrementTimetableErrors();
79                          response.sendError(500);                          response.sendError(500);
80                  }                  }
81                                    
82          }                  if (timetable != null) {
83                                    String responseXML = formatTimetable(timetable);
         String lookupType(String trainCode) {  
                 trainCode = trainCode.toLowerCase();  
84                                    
85                  if (trainCode.equals("a") || trainCode.equals("b") || trainCode.equals("bx") || trainCode.equals("c") ||                          response.setContentType("text/xml");
86                                  trainCode.equals("e") || trainCode.equals("f") || trainCode.equals("h")) {                          response.getWriter().print(responseXML);
                         return "S2";  
                 } else {  
                         return "FJRN";  
87                  }                  }
88                    
89          }          }
90                    
91          String formatTimetable(List<TimetableBean> timetable) {          protected TimetableBean generateErrorBean(int code) {
92                  StringBuffer sb = new StringBuffer();                  TimetableBean bean = new TimetableBean();              
93                                    bean.errorCode = code;
94                  sb.append("<timetable>");                  return bean;
95                            }
96                  for (TimetableBean bean : timetable) {          
97                          sb.append("<entry");          String formatTimetable(TimetableBean timetable) throws ServletException {
98                          if (bean.isCurrent())                  Serializer serializer = new Persister();
99                                  sb.append(" current=\"true\"");  
100                          sb.append(">");                  StringWriter out = new StringWriter();
101                            
102                          sb.append("<station>").append( bean.getStation() ).append("</station>");                  try {
103                          sb.append("<arrival>").append( bean.getArrival() ).append("</arrival>");                          serializer.write(timetable, out);
104                          sb.append("<departure>").append( bean.getDeparture() ).append("</departure>");                  } catch (Exception e) {
105                                                    throw new ServletException(e);
                         sb.append("</entry>");  
106                  }                  }
107                                    
108                  sb.append("</timetable>");                  return out.toString();
                   
                 return sb.toString();  
109          }          }
110    
111  }  }
Legend:
Removed from v.462  
changed lines
  Added in v.1396
  ViewVC Help
Powered by ViewVC 1.1.20