/[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 1021 by torben, Mon Aug 30 13:59:54 2010 UTC revision 1091 by torben, Tue Sep 21 15:46:02 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;
7  import java.util.logging.Level;  import java.util.logging.Level;
8  import java.util.logging.Logger;  import java.util.logging.Logger;
# Line 9  import javax.servlet.annotation.WebServl Line 12  import javax.servlet.annotation.WebServl
12  import javax.servlet.http.HttpServlet;  import javax.servlet.http.HttpServlet;
13  import javax.servlet.http.HttpServletRequest;  import javax.servlet.http.HttpServletRequest;
14  import javax.servlet.http.HttpServletResponse;  import javax.servlet.http.HttpServletResponse;
15  import javax.xml.parsers.DocumentBuilder;  
16  import javax.xml.parsers.DocumentBuilderFactory;  import javax.xml.transform.Source;
17  import javax.xml.transform.OutputKeys;  import javax.xml.transform.Templates;
18  import javax.xml.transform.Transformer;  import javax.xml.transform.Transformer;
19  import javax.xml.transform.TransformerFactory;  import javax.xml.transform.TransformerFactory;
 import javax.xml.transform.dom.DOMSource;  
20  import javax.xml.transform.stream.StreamResult;  import javax.xml.transform.stream.StreamResult;
21    import javax.xml.transform.stream.StreamSource;
22    
23  import org.w3c.dom.DOMImplementation;  import org.simpleframework.xml.Serializer;
24  import org.w3c.dom.Document;  import org.simpleframework.xml.core.Persister;
 import org.w3c.dom.Element;  
25    
26    import dk.thoerup.android.traininfo.common.DepartureBean;
27  import dk.thoerup.circuitbreaker.CircuitBreakerException;  import dk.thoerup.circuitbreaker.CircuitBreakerException;
 import dk.thoerup.traininfoservice.StationDAO;  
28  import dk.thoerup.traininfoservice.Statistics;  import dk.thoerup.traininfoservice.Statistics;
29    
30  /**  /**
# Line 35  public class DepartureServlet extends Ht Line 37  public class DepartureServlet extends Ht
37          Logger logger = Logger.getLogger( DepartureServlet.class.getName() );          Logger logger = Logger.getLogger( DepartureServlet.class.getName() );
38    
39          DepartureFetcher fetcher;          DepartureFetcher fetcher;
40      TransformerFactory transformerFactory = TransformerFactory.newInstance();  
41          DocumentBuilderFactory docBuilderFactory = DocumentBuilderFactory.newInstance();  
42            
43            TransformerFactory xslTransFact;
44            Templates xslTemplate;
45                    
46    
47          @Override          @Override
48          public void init() throws ServletException {          public void init() throws ServletException {
49                  super.init();                  super.init();
50    
51                  boolean useTempSite =  Boolean.parseBoolean( getServletContext().getInitParameter("usetempsite") );                  boolean useAzureSite =  Boolean.parseBoolean( getServletContext().getInitParameter("useazuresite") );
52                  int cacheTimeout =  Integer.parseInt( getServletContext().getInitParameter("cache_timeout") );                  int cacheTimeout =  Integer.parseInt( getServletContext().getInitParameter("cache_timeout") );
53                  logger.info( "DepartureServlet, use temp site=" + useTempSite + ", cache=" + cacheTimeout);                  int replyTimeout =  Integer.parseInt( getServletContext().getInitParameter("reply_timeout") );
54                  fetcher = new DepartureFetcher(useTempSite, cacheTimeout);                  logger.info( "DepartureServlet, use azure site=" + useAzureSite + ", cache=" + cacheTimeout);
55                    fetcher = new DepartureFetcher(useAzureSite, cacheTimeout, replyTimeout);
56                    
57                    xslTransFact = TransformerFactory.newInstance();
58                    
59                    String xslPath = getServletContext().getRealPath("/departures.xsl");
60                    Source stylesheet = new StreamSource( new File(xslPath) );
61                    
62                    try {
63                            xslTemplate = xslTransFact.newTemplates(stylesheet);
64                    } catch (Exception e) {
65                            throw new ServletException(e);
66                    }
67          }          }
68    
69          @Override          @Override
# Line 70  public class DepartureServlet extends Ht Line 87  public class DepartureServlet extends Ht
87    
88                  try {                  try {
89                          beans = fetcher.cachedLookupDepartures(station, arrival);                          beans = fetcher.cachedLookupDepartures(station, arrival);
90                            beans.arrival = arrival;
91                  } catch (java.io.IOException ioe) {                  } catch (java.io.IOException ioe) {
92                          logger.warning("Read failed, station="+station + ". " + ioe.getMessage() );                          logger.warning("Read failed, station="+station + ". " + ioe.getMessage() );
93                          Statistics.getInstance().incrementDepartureErrors();                          Statistics.getInstance().incrementDepartureErrors();
# Line 92  public class DepartureServlet extends Ht Line 110  public class DepartureServlet extends Ht
110                                    
111                  if (format.equalsIgnoreCase("xml")) {                  if (format.equalsIgnoreCase("xml")) {
112                          resp.setContentType("text/xml");                          resp.setContentType("text/xml");
113                          resp.getWriter().print( formatXml(beans, arrival) );                          resp.getWriter().print( formatXml(beans) );
114                  } else if (format.equalsIgnoreCase("html")) {                  } else if (format.equalsIgnoreCase("html")) {
115                                                    /*
116                          String advStr = req.getParameter("advanced");                          String advStr = req.getParameter("advanced");
117                          boolean advanced = advStr != null ? Boolean.parseBoolean(advStr) : false;                          boolean advanced = advStr != null ? Boolean.parseBoolean(advStr) : false;
118                                                    
119                          req.setAttribute("advanced", advanced);                          req.setAttribute("advanced", advanced);
120                          req.setAttribute("departurebeans", beans);                          req.setAttribute("departurebeans", beans);
121                          req.setAttribute("stationID", station );                          req.setAttribute("stationID", station );
122                          getServletContext().getRequestDispatcher("/ViewDepartures.jsp").forward(req,resp);                          getServletContext().getRequestDispatcher("/ViewDepartures.jsp").forward(req,resp);*/
123                            resp.setContentType("text/html");
124                            resp.getWriter().print(  xmlToHtml(formatXml(beans))  );
125                  } else {                  } else {
126                          resp.sendError(400, "Unknown format");                                            resp.sendError(400, "Unknown format");                  
127                  }                  }
128    
129          }          }
130    
131          protected String formatXml(DepartureBean beans, boolean arrival) throws ServletException{          protected String formatXml(DepartureBean beans) throws ServletException{
132                  String xml = "";                  
133                  try {                  Serializer serializer = new Persister();
                         DocumentBuilder builder = docBuilderFactory.newDocumentBuilder();  
                         DOMImplementation impl = builder.getDOMImplementation();  
134    
135                                            ByteArrayOutputStream out = new ByteArrayOutputStream();
                         Document doc = impl.createDocument(null,null,null);  
                 Element root = doc.createElement("departureinfo");  
                 root.setAttribute("station", beans.getStationName());  
                 root.setAttribute("arrival", Boolean.toString(arrival) );  
                   
                 for (String notif : beans.notifications) {  
                         Element notElem = doc.createElement("notification");  
                         notElem.setTextContent(notif);  
                         root.appendChild(notElem);  
                 }  
                   
                 for (DepartureEntry departure : beans.departureEntries) {  
                         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();  
136    
137                    try {
138                            serializer.write(beans, out);
139                    } catch (Exception e) {
140                            throw new ServletException(e);
141                    }
142                    
143                    return out.toString();
144            }
145            
146            protected String xmlToHtml(String input) throws ServletException {
147    
148                    try {                                  
149                            Transformer trans = xslTemplate.newTransformer();                      
150                    
151                            Source xml = new StreamSource( new ByteArrayInputStream(input.getBytes() ));
152                            ByteArrayOutputStream out = new ByteArrayOutputStream();
153                            trans.transform(xml, new StreamResult(out));            
154                    
155                            return out.toString();
156                  } catch (Exception e) {                  } catch (Exception e) {
157                          throw new ServletException(e);                          throw new ServletException(e);
158                  }                  }
                 return xml;  
159          }          }
160    
161  }  }

Legend:
Removed from v.1021  
changed lines
  Added in v.1091

  ViewVC Help
Powered by ViewVC 1.1.20