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

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

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

android/TrainInfoService/src/dk/thoerup/traininfoservice/banedk/DepartureServlet.java revision 425 by torben, Thu Oct 8 20:46:40 2009 UTC android/TrainInfoServiceGoogle/src/dk/thoerup/traininfoservice/banedk/DepartureServlet.java revision 1080 by torben, Mon Sep 20 20:11:55 2010 UTC
# Line 1  Line 1 
1  package dk.thoerup.traininfoservice.banedk;  package dk.thoerup.traininfoservice.banedk;
2    
3    import java.io.ByteArrayInputStream;
4    import java.io.ByteArrayOutputStream;
5    import java.io.File;
6  import java.io.IOException;  import java.io.IOException;
 import java.sql.Connection;  
 import java.sql.ResultSet;  
 import java.sql.Statement;  
 import java.util.List;  
7  import java.util.logging.Level;  import java.util.logging.Level;
8  import java.util.logging.Logger;  import java.util.logging.Logger;
9    
# Line 12  import javax.servlet.ServletException; Line 11  import javax.servlet.ServletException;
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  import javax.xml.parsers.DocumentBuilder;  
15  import javax.xml.parsers.DocumentBuilderFactory;  import javax.xml.transform.Source;
 import javax.xml.transform.OutputKeys;  
16  import javax.xml.transform.Transformer;  import javax.xml.transform.Transformer;
17  import javax.xml.transform.TransformerFactory;  import javax.xml.transform.TransformerFactory;
 import javax.xml.transform.dom.DOMSource;  
18  import javax.xml.transform.stream.StreamResult;  import javax.xml.transform.stream.StreamResult;
19    import javax.xml.transform.stream.StreamSource;
20    
21  import org.w3c.dom.DOMImplementation;  import org.simpleframework.xml.Serializer;
22  import org.w3c.dom.Document;  import org.simpleframework.xml.core.Persister;
 import org.w3c.dom.Element;  
23    
24  import dk.thoerup.curcuitbreaker.CircuitBreakerException;  import dk.thoerup.android.traininfo.common.DepartureBean;
25  import dk.thoerup.traininfoservice.DBConnection;  import dk.thoerup.circuitbreaker.CircuitBreakerException;
26    import dk.thoerup.traininfoservice.Statistics;
27    
28  /**  /**
29   * Servlet implementation class DepartureServlet   * Servlet implementation class DepartureServlet
30   */   */
31    
32  public class DepartureServlet extends HttpServlet {  public class DepartureServlet extends HttpServlet {
33          private static final long serialVersionUID = 1L;          private static final long serialVersionUID = 1L;
34                    
35          Logger logger = Logger.getLogger( DepartureServlet.class.getName() );          Logger logger = Logger.getLogger( DepartureServlet.class.getName() );
36    
37          DepartureFetcher fetcher = new DepartureFetcher();          DepartureFetcher fetcher;
38      TransformerFactory transformerFactory = TransformerFactory.newInstance();  
39          DocumentBuilderFactory docBuilderFactory = DocumentBuilderFactory.newInstance();  
40            
41            String xslPath;
42                    
43    
44          @Override          @Override
45          protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {          public void init() throws ServletException {
46                    super.init();
47    
48                    boolean useAzureSite =  Boolean.parseBoolean( getServletContext().getInitParameter("useazuresite") );
49                    int cacheTimeout =  Integer.parseInt( getServletContext().getInitParameter("cache_timeout") );
50                    int replyTimeout =  Integer.parseInt( getServletContext().getInitParameter("reply_timeout") );
51                    logger.info( "DepartureServlet, use azure site=" + useAzureSite + ", cache=" + cacheTimeout);
52                    fetcher = new DepartureFetcher(useAzureSite, cacheTimeout, replyTimeout);
53                    
54                    xslPath = getServletContext().getRealPath("/departures.xsl");
55            }
56    
57            @Override
58            protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {          
59                    if (req.getParameter("station") == null) {
60                            resp.sendError(400, "not enough parameters");
61                            return;
62                    }
63                    
64                    boolean arrival = false;
65                    try {
66                            arrival = Integer.parseInt( req.getParameter("arrival")) != 0;
67                    } catch (Exception e) {}
68                    
69                    Statistics.getInstance().incrementDepartureLookups();
70                    
71                  int station = Integer.parseInt( req.getParameter("station") );                  int station = Integer.parseInt( req.getParameter("station") );
72                  String format = req.getParameter("format");                  String format = req.getParameter("format");
73    
74                  List<DepartureBean> beans;                  DepartureBean beans;            
   
                 String stationName = getStationName(station);  
75    
76                  try {                  try {
77                          beans = fetcher.cachedLookupDepartures(station);                          beans = fetcher.cachedLookupDepartures(station, arrival);
78                  } catch (java.net.SocketTimeoutException ste) {                          beans.arrival = arrival;
79                          logger.warning("Read timed out, station="+station);                  } catch (java.io.IOException ioe) {
80                            logger.warning("Read failed, station="+station + ". " + ioe.getMessage() );
81                            Statistics.getInstance().incrementDepartureErrors();
82                          resp.sendError(500);                          resp.sendError(500);
83                          return;                          return;
84                  } catch (CircuitBreakerException cbe) {                  } catch (CircuitBreakerException cbe) {
85                          logger.warning("Circuitbreaker - failing fast");                          logger.warning("Circuitbreaker - failing fast");
86                            Statistics.getInstance().incrementDepartureErrors();
87                          resp.sendError(500);                          resp.sendError(500);
88                          return;                          return;
89                  } catch (Throwable e) {                                  } catch (Exception e) {                
90                          logger.log(Level.WARNING, "Unknown exception, station=" +station, e);                          logger.log(Level.WARNING, "Unknown exception, station=" +station, e);
91                            Statistics.getInstance().incrementDepartureErrors();
92                          resp.sendError(500);                          resp.sendError(500);
93                          return;                          return;
94                  }                  }
# Line 70  public class DepartureServlet extends Ht Line 98  public class DepartureServlet extends Ht
98                                    
99                  if (format.equalsIgnoreCase("xml")) {                  if (format.equalsIgnoreCase("xml")) {
100                          resp.setContentType("text/xml");                          resp.setContentType("text/xml");
101                          resp.getWriter().print( formatXml(beans, stationName) );                          resp.getWriter().print( formatXml(beans) );
102                  } else if (format.equalsIgnoreCase("html")) {                  } else if (format.equalsIgnoreCase("html")) {
103                          req.setAttribute("stationname", stationName );                          /*
104                            String advStr = req.getParameter("advanced");
105                            boolean advanced = advStr != null ? Boolean.parseBoolean(advStr) : false;
106                            
107                            req.setAttribute("advanced", advanced);
108                          req.setAttribute("departurebeans", beans);                          req.setAttribute("departurebeans", beans);
109                          req.setAttribute("stationID", station );                          req.setAttribute("stationID", station );
110                          getServletContext().getRequestDispatcher("/ViewDepartures.jsp").forward(req,resp);                          getServletContext().getRequestDispatcher("/ViewDepartures.jsp").forward(req,resp);*/
111                            resp.setContentType("text/html");
112                            resp.getWriter().print(  xmlToHtml(formatXml(beans))  );
113                  } else {                  } else {
114                          throw new ServletException("Unknown format");                          resp.sendError(400, "Unknown format");                  
115                  }                  }
116    
117          }          }
118    
119          protected String formatXml(List<DepartureBean> beans, String stationName) throws ServletException{          protected String formatXml(DepartureBean beans) throws ServletException{
120                  String xml = "";                  
121                  try {                  Serializer serializer = new Persister();
                         DocumentBuilder builder = docBuilderFactory.newDocumentBuilder();  
                         DOMImplementation impl = builder.getDOMImplementation();  
   
                           
                         Document doc = impl.createDocument(null,null,null);  
                 Element root = doc.createElement("departureinfo");  
                 root.setAttribute("station", stationName);  
                 for (DepartureBean departure : beans) {  
                         Element train = doc.createElement("train");  
                           
                         Element time = doc.createElement("time");  
                         time.setTextContent( departure.getTime() );  
                         train.appendChild(time);  
                           
                         Element updated = doc.createElement("updated");  
                         updated.setTextContent( String.valueOf(departure.getUpdated()) );  
                         train.appendChild(updated);  
                           
                         Element trainNumber = doc.createElement("trainnumber");  
                         trainNumber.setTextContent( departure.getTrainNumber() );  
                         train.appendChild(trainNumber);  
                           
                         Element destination = doc.createElement("destination");  
                         destination.setTextContent( departure.getDestination());  
                         train.appendChild(destination);  
                           
                         Element origin = doc.createElement("origin");  
                         origin.setTextContent( departure.getOrigin() );  
                         train.appendChild(origin);  
                           
                         Element location= doc.createElement("location");  
                         location.setTextContent( departure.getLocation() );  
                         train.appendChild(location);  
                           
                         Element status = doc.createElement("status");  
                         status.setTextContent( departure.getStatus() );  
                         train.appendChild(status);  
                           
                         Element note = doc.createElement("note");  
                         note.setTextContent( departure.getNote() );  
                         train.appendChild(note);  
                           
                         root.appendChild(train);  
                 }  
                   
                 doc.appendChild(root);  
                   
                   
                 DOMSource domSource = new DOMSource(doc);  
   
                 Transformer transformer = transformerFactory.newTransformer();  
                 //transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes");  
                 transformer.setOutputProperty(OutputKeys.METHOD, "xml");  
                 transformer.setOutputProperty(OutputKeys.ENCODING,"ISO-8859-1");  
                 transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");  
                 transformer.setOutputProperty(OutputKeys.INDENT, "yes");  
                 java.io.StringWriter sw = new java.io.StringWriter();  
                 StreamResult sr = new StreamResult(sw);  
                 transformer.transform(domSource, sr);  
                 xml = sw.toString();  
122    
123                    ByteArrayOutputStream out = new ByteArrayOutputStream();
124    
125                    try {
126                            serializer.write(beans, out);
127                  } catch (Exception e) {                  } catch (Exception e) {
128                          throw new ServletException(e);                          throw new ServletException(e);
129                  }                  }
130                  return xml;                  
131                    return out.toString();
132          }          }
133            
134            protected String xmlToHtml(String input)  {
135    
         protected String getStationName(int stationID) {  
                 String station = "";  
   
                 Connection conn = null;  
136                  try {                  try {
137                          conn = DBConnection.getConnection();                          Source stylesheet = new StreamSource( new File(xslPath) );
138                          Statement stmt = conn.createStatement();                  
139                          ResultSet rs = stmt.executeQuery("SELECT name FROM trainstations WHERE id=" + stationID);                  
140                          if (rs.next()) {                          TransformerFactory transFact = TransformerFactory.newInstance();
141                                  station = rs.getString(1);                  
142                          }                          Transformer trans = transFact.newTransformer(stylesheet);
143                    
144                  } catch (Exception e) {                          Source xml = new StreamSource( new ByteArrayInputStream(input.getBytes() ));
145                  } finally {                          ByteArrayOutputStream out = new ByteArrayOutputStream();
146                          try {                          trans.transform(xml, new StreamResult(out));            
147                                  if (conn != null && !conn.isClosed())                  
148                                          conn.close();                          return out.toString();
149                          } catch (Exception e) {}                  } catch (Exception e) {
150                            return e.toString();
151                  }                  }
   
                 return station;  
152          }          }
153    
154  }  }
Legend:
Removed from v.425  
changed lines
  Added in v.1080
  ViewVC Help
Powered by ViewVC 1.1.20