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

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

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

revision 862 by torben, Wed Jun 16 10:03:13 2010 UTC revision 1303 by torben, Tue Apr 19 11:52:39 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  import java.util.logging.Level;  import java.util.logging.Level;
6  import java.util.logging.Logger;  import java.util.logging.Logger;
7    
8  import javax.servlet.ServletException;  import javax.servlet.ServletException;
9    import javax.servlet.annotation.WebServlet;
10  import javax.servlet.http.HttpServlet;  import javax.servlet.http.HttpServlet;
11  import javax.servlet.http.HttpServletRequest;  import javax.servlet.http.HttpServletRequest;
12  import javax.servlet.http.HttpServletResponse;  import javax.servlet.http.HttpServletResponse;
13  import javax.xml.parsers.DocumentBuilder;  import javax.xml.transform.Templates;
 import javax.xml.parsers.DocumentBuilderFactory;  
 import javax.xml.transform.OutputKeys;  
 import javax.xml.transform.Transformer;  
14  import javax.xml.transform.TransformerFactory;  import javax.xml.transform.TransformerFactory;
 import javax.xml.transform.dom.DOMSource;  
 import javax.xml.transform.stream.StreamResult;  
15    
16  import org.w3c.dom.DOMImplementation;  import org.simpleframework.xml.Serializer;
17  import org.w3c.dom.Document;  import org.simpleframework.xml.core.Persister;
 import org.w3c.dom.Element;  
18    
19    import dk.thoerup.android.traininfo.common.DepartureBean;
20  import dk.thoerup.circuitbreaker.CircuitBreakerException;  import dk.thoerup.circuitbreaker.CircuitBreakerException;
 import dk.thoerup.traininfoservice.StationDAO;  
21  import dk.thoerup.traininfoservice.Statistics;  import dk.thoerup.traininfoservice.Statistics;
22    import dk.thoerup.traininfoservice.TraininfoSettings;
23    import dk.thoerup.traininfoservice.banedk.DepartureFetcher.FetchTrainType;
24    import dk.thoerup.traininfoservice.db.StationDAO;
25    
26  /**  /**
27   * Servlet implementation class DepartureServlet   * Servlet implementation class DepartureServlet
28   */   */
29    @WebServlet(urlPatterns={"/DepartureServlet"})
30  public class DepartureServlet extends HttpServlet {  public class DepartureServlet extends HttpServlet {
31          private static final long serialVersionUID = 1L;          private static final long serialVersionUID = 1L;
32                    
33          Logger logger = Logger.getLogger( DepartureServlet.class.getName() );          Logger logger = Logger.getLogger( DepartureServlet.class.getName() );
34    
35          DepartureFetcher fetcher;          DepartureFetcher fetcher;
36      TransformerFactory transformerFactory = TransformerFactory.newInstance();  
37          DocumentBuilderFactory docBuilderFactory = DocumentBuilderFactory.newInstance();  
38            
39            TransformerFactory xslTransFact;
40            Templates xslTemplate;
41                    
42    
43          @Override          @Override
44          public void init() throws ServletException {          public void init() throws ServletException {
45                  super.init();                  super.init();
46    
47                  boolean useTempSite =  Boolean.parseBoolean( getServletContext().getInitParameter("usetempsite") );  
48                  int cacheTimeout =  Integer.parseInt( getServletContext().getInitParameter("cache_timeout") );                  TraininfoSettings settings = (TraininfoSettings) getServletContext().getAttribute("settings");
49                  logger.info( "DepartureServlet, use temp site=" + useTempSite + ", cache=" + cacheTimeout);                  fetcher = new DepartureFetcher(settings);
50                  fetcher = new DepartureFetcher(useTempSite, cacheTimeout);                  
51                    /*
52                    xslTransFact = TransformerFactory.newInstance();
53                    
54                    String xslPath = getServletContext().getRealPath("/departures.xsl");
55                    Source stylesheet = new StreamSource( new File(xslPath) );
56                    
57                    try {
58                            xslTemplate = xslTransFact.newTemplates(stylesheet);
59                    } catch (Exception e) {
60                            throw new ServletException(e);
61                    }*/
62          }          }
63    
64          @Override          @Override
65          protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {                    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {          
66                  if (req.getParameter("station") == null) {                  if (req.getParameter("station") == null) {
67                          resp.sendError(400, "not enough parameters");                          resp.sendError(400, "not enough parameters (station)");
68                          return;                          return;
69                  }                  }
70                    if (req.getParameter("format") == null) {
71                            resp.sendError(400, "not enough parameters (format)");
72                            return;                
73                    }
74                                    
75                  boolean arrival = false;                  boolean arrival = false;
76                  try {                  try {
# Line 65  public class DepartureServlet extends Ht Line 82  public class DepartureServlet extends Ht
82                  int station = Integer.parseInt( req.getParameter("station") );                  int station = Integer.parseInt( req.getParameter("station") );
83                  String format = req.getParameter("format");                  String format = req.getParameter("format");
84    
85                  List<DepartureBean> beans;                  DepartureBean beans;
86                    
87                  String stationName = StationDAO.getStationName(station);                  DepartureFetcher.FetchTrainType type = FetchTrainType.BOTH;
88                    if ( req.getParameter("type") != null) {
89                            type = FetchTrainType.valueOf( req.getParameter("type") );
90                    }
91    
92                  try {                  try {
93                          beans = fetcher.cachedLookupDepartures(station, arrival);                          beans = fetcher.cachedLookupDepartures(station, arrival, type);
94                            beans.arrival = arrival;
95                  } catch (java.io.IOException ioe) {                  } catch (java.io.IOException ioe) {
96                          logger.warning("Read failed, station="+station + ". " + ioe.getMessage() );                          logger.warning("Read failed, station="+station + ". " + ioe.getMessage() );
97                          Statistics.getInstance().incrementDepartureErrors();                          Statistics.getInstance().incrementDepartureErrors();
98                          resp.sendError(500);                          resp.sendError(500);
99                          return;                          return;
100                  } catch (CircuitBreakerException cbe) {                  } catch (CircuitBreakerException cbe) {
101                          logger.warning("Circuitbreaker - failing fast");                          logger.warning("Circuitbreaker - failing fast, station=" +station);
102                          Statistics.getInstance().incrementDepartureErrors();                          Statistics.getInstance().incrementDepartureErrors();
103                          resp.sendError(500);                          resp.sendError(500);
104                          return;                          return;
105                    } catch (StationDAO.NostationException nse) {
106                            logger.log(Level.WARNING, "Station not in Database, station=" +station);
107                            Statistics.getInstance().incrementDepartureErrors();
108                            resp.sendError(400, "invalid station ID");
109                            return;                
110                  } catch (Exception e) {                                  } catch (Exception e) {                
111                          logger.log(Level.WARNING, "Unknown exception, station=" +station, e);                          logger.log(Level.WARNING, "Unknown exception, station=" +station, e);
112                          Statistics.getInstance().incrementDepartureErrors();                          Statistics.getInstance().incrementDepartureErrors();
# Line 93  public class DepartureServlet extends Ht Line 119  public class DepartureServlet extends Ht
119                                    
120                  if (format.equalsIgnoreCase("xml")) {                  if (format.equalsIgnoreCase("xml")) {
121                          resp.setContentType("text/xml");                          resp.setContentType("text/xml");
122                          resp.getWriter().print( formatXml(beans, stationName) );                          resp.getWriter().print( formatXml(beans) );
123                  } else if (format.equalsIgnoreCase("html")) {                  } else if (format.equalsIgnoreCase("html")) {
124                                                    
125                          String advStr = req.getParameter("advanced");                          String advStr = req.getParameter("advanced");
126                          boolean advanced = advStr != null ? Boolean.parseBoolean(advStr) : false;                          boolean advanced = advStr != null ? Boolean.parseBoolean(advStr) : false;
127                                                    
128                          req.setAttribute("advanced", advanced);                          req.setAttribute("advanced", advanced);
                         req.setAttribute("stationname", stationName );  
129                          req.setAttribute("departurebeans", beans);                          req.setAttribute("departurebeans", beans);
130                          req.setAttribute("stationID", station );                          req.setAttribute("stationID", station );
131                          getServletContext().getRequestDispatcher("/ViewDepartures.jsp").forward(req,resp);                          getServletContext().getRequestDispatcher("/ViewDepartures.jsp").forward(req,resp);
132                            /*resp.setContentType("text/html");
133                            resp.getWriter().print(  xmlToHtml(formatXml(beans))  );*/
134                  } else {                  } else {
135                          resp.sendError(400, "Unknown format");                                            resp.sendError(400, "Unknown format");                  
136                  }                  }
137    
138          }          }
139    
140          protected String formatXml(List<DepartureBean> beans, String stationName) throws ServletException{          protected String formatXml(DepartureBean beans) throws ServletException{
141                  String xml = "";                  
142                  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);  
                           
                         Element type = doc.createElement("type");  
                         type.setTextContent( departure.getType() );  
                         train.appendChild(type);  
                           
                         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", "1");  
                 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();  
143    
144                    StringWriter out = new StringWriter();
145    
146                    try {
147                            serializer.write(beans, out);
148                  } catch (Exception e) {                  } catch (Exception e) {
149                          throw new ServletException(e);                          throw new ServletException(e);
150                  }                  }
151                  return xml;                  
152                    return out.toString();
153          }          }
154            /*
155            protected String xmlToHtml(String input) throws ServletException {
156    
157                    try {                                  
158                            Transformer trans = xslTemplate.newTransformer();
159                            
160                    
161                            Source xml = new StreamSource( new StringReader(input));
162                            StringWriter out = new StringWriter();
163                            trans.transform(xml, new StreamResult(out));            
164                    
165                            return out.toString();
166                    } catch (Exception e) {
167                            throw new ServletException(e);
168                    }
169            }*/
170    
171  }  }
Legend:
Removed from v.862  
changed lines
  Added in v.1303
  ViewVC Help
Powered by ViewVC 1.1.20