1 |
package dk.thoerup.traininfoservice.banedk; |
package dk.thoerup.traininfoservice.banedk; |
2 |
|
|
|
import java.io.ByteArrayInputStream; |
|
|
import java.io.ByteArrayOutputStream; |
|
|
import java.io.File; |
|
3 |
import java.io.IOException; |
import java.io.IOException; |
4 |
|
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 |
|
|
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; |
|
|
|
|
import javax.xml.transform.Source; |
|
13 |
import javax.xml.transform.Templates; |
import javax.xml.transform.Templates; |
|
import javax.xml.transform.Transformer; |
|
14 |
import javax.xml.transform.TransformerFactory; |
import javax.xml.transform.TransformerFactory; |
|
import javax.xml.transform.stream.StreamResult; |
|
|
import javax.xml.transform.stream.StreamSource; |
|
15 |
|
|
16 |
import org.simpleframework.xml.Serializer; |
import org.simpleframework.xml.Serializer; |
17 |
import org.simpleframework.xml.core.Persister; |
import org.simpleframework.xml.core.Persister; |
18 |
|
|
19 |
import dk.thoerup.android.traininfo.common.DepartureBean; |
import dk.thoerup.android.traininfo.common.DepartureBean; |
20 |
|
import dk.thoerup.android.traininfo.common.StationBean; |
21 |
import dk.thoerup.circuitbreaker.CircuitBreakerException; |
import dk.thoerup.circuitbreaker.CircuitBreakerException; |
|
import dk.thoerup.traininfoservice.StationDAO; |
|
22 |
import dk.thoerup.traininfoservice.Statistics; |
import dk.thoerup.traininfoservice.Statistics; |
23 |
|
import dk.thoerup.traininfoservice.TraininfoSettings; |
24 |
|
import dk.thoerup.traininfoservice.banedk.DepartureFetcher.FetchTrainType; |
25 |
|
import dk.thoerup.traininfoservice.db.StationDAO; |
26 |
|
|
27 |
/** |
/** |
28 |
* Servlet implementation class DepartureServlet |
* Servlet implementation class DepartureServlet |
45 |
public void init() throws ServletException { |
public void init() throws ServletException { |
46 |
super.init(); |
super.init(); |
47 |
|
|
48 |
boolean useAzureSite = Boolean.parseBoolean( getServletContext().getInitParameter("useazuresite") ); |
|
49 |
int cacheTimeout = Integer.parseInt( getServletContext().getInitParameter("cache_timeout") ); |
TraininfoSettings settings = (TraininfoSettings) getServletContext().getAttribute("settings"); |
50 |
int replyTimeout = Integer.parseInt( getServletContext().getInitParameter("reply_timeout") ); |
fetcher = new DepartureFetcher(settings); |
|
logger.info( "DepartureServlet, use azure site=" + useAzureSite + ", cache=" + cacheTimeout); |
|
|
fetcher = new DepartureFetcher(useAzureSite, cacheTimeout, replyTimeout); |
|
51 |
|
|
52 |
|
/* |
53 |
xslTransFact = TransformerFactory.newInstance(); |
xslTransFact = TransformerFactory.newInstance(); |
54 |
|
|
55 |
String xslPath = getServletContext().getRealPath("/departures.xsl"); |
String xslPath = getServletContext().getRealPath("/departures.xsl"); |
59 |
xslTemplate = xslTransFact.newTemplates(stylesheet); |
xslTemplate = xslTransFact.newTemplates(stylesheet); |
60 |
} catch (Exception e) { |
} catch (Exception e) { |
61 |
throw new ServletException(e); |
throw new ServletException(e); |
62 |
} |
}*/ |
63 |
} |
} |
64 |
|
|
65 |
@Override |
@Override |
66 |
protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException { |
protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException { |
67 |
if (req.getParameter("station") == null) { |
if (req.getParameter("station") == null) { |
68 |
resp.sendError(400, "not enough parameters"); |
resp.sendError(400, "not enough parameters (station)"); |
69 |
return; |
return; |
70 |
} |
} |
71 |
|
if (req.getParameter("format") == null) { |
72 |
|
resp.sendError(400, "not enough parameters (format)"); |
73 |
|
return; |
74 |
|
} |
75 |
|
|
76 |
boolean arrival = false; |
boolean arrival = false; |
77 |
try { |
try { |
83 |
int station = Integer.parseInt( req.getParameter("station") ); |
int station = Integer.parseInt( req.getParameter("station") ); |
84 |
String format = req.getParameter("format"); |
String format = req.getParameter("format"); |
85 |
|
|
86 |
DepartureBean beans; |
DepartureBean beans; |
87 |
|
|
88 |
|
DepartureFetcher.FetchTrainType type = FetchTrainType.BOTH; |
89 |
|
if ( req.getParameter("type") != null) { |
90 |
|
type = FetchTrainType.valueOf( req.getParameter("type") ); |
91 |
|
} |
92 |
|
|
93 |
|
|
94 |
try { |
try { |
95 |
beans = fetcher.cachedLookupDepartures(station, arrival); |
beans = fetcher.cachedLookupDepartures(station, arrival, type); |
96 |
beans.arrival = arrival; |
beans.arrival = arrival; |
97 |
} catch (java.io.IOException ioe) { |
} catch (java.io.IOException ioe) { |
98 |
logger.warning("Read failed, station="+station + ". " + ioe.getMessage() ); |
logger.warning("Read failed, station="+station + ". " + ioe.getMessage() ); |
99 |
Statistics.getInstance().incrementDepartureErrors(); |
Statistics.getInstance().incrementDepartureErrors(); |
100 |
resp.sendError(500); |
resp.sendError(500, "backend didnt answer"); |
101 |
return; |
return; |
102 |
} catch (CircuitBreakerException cbe) { |
} catch (CircuitBreakerException cbe) { |
103 |
logger.warning("Circuitbreaker - failing fast, station=" +station); |
logger.warning("Circuitbreaker - failing fast, station=" +station); |
104 |
Statistics.getInstance().incrementDepartureErrors(); |
Statistics.getInstance().incrementDepartureErrors(); |
105 |
resp.sendError(500); |
beans = generateErrorBean(1); |
|
return; |
|
106 |
} catch (StationDAO.NostationException nse) { |
} catch (StationDAO.NostationException nse) { |
107 |
logger.log(Level.WARNING, "Station not in Database, station=" +station); |
logger.log(Level.WARNING, "Station not in Database, station=" +station); |
108 |
Statistics.getInstance().incrementDepartureErrors(); |
Statistics.getInstance().incrementDepartureErrors(); |
122 |
resp.setContentType("text/xml"); |
resp.setContentType("text/xml"); |
123 |
resp.getWriter().print( formatXml(beans) ); |
resp.getWriter().print( formatXml(beans) ); |
124 |
} else if (format.equalsIgnoreCase("html")) { |
} else if (format.equalsIgnoreCase("html")) { |
125 |
/* |
|
126 |
String advStr = req.getParameter("advanced"); |
String advStr = req.getParameter("advanced"); |
127 |
boolean advanced = advStr != null ? Boolean.parseBoolean(advStr) : false; |
boolean advanced = advStr != null ? Boolean.parseBoolean(advStr) : false; |
128 |
|
|
129 |
req.setAttribute("advanced", advanced); |
req.setAttribute("advanced", advanced); |
130 |
req.setAttribute("departurebeans", beans); |
req.setAttribute("departurebeans", beans); |
131 |
req.setAttribute("stationID", station ); |
req.setAttribute("stationID", station ); |
132 |
getServletContext().getRequestDispatcher("/ViewDepartures.jsp").forward(req,resp);*/ |
getServletContext().getRequestDispatcher("/ViewDepartures.jsp").forward(req,resp); |
133 |
resp.setContentType("text/html"); |
/*resp.setContentType("text/html"); |
134 |
resp.getWriter().print( xmlToHtml(formatXml(beans)) ); |
resp.getWriter().print( xmlToHtml(formatXml(beans)) );*/ |
135 |
} else { |
} else { |
136 |
resp.sendError(400, "Unknown format"); |
resp.sendError(400, "Unknown format"); |
137 |
} |
} |
138 |
|
} |
139 |
|
|
140 |
|
protected DepartureBean generateErrorBean(int code) { |
141 |
|
DepartureBean bean = new DepartureBean(); |
142 |
|
bean.stationName=""; |
143 |
|
bean.errorCode = code; |
144 |
|
return bean; |
145 |
} |
} |
146 |
|
|
147 |
protected String formatXml(DepartureBean beans) throws ServletException{ |
protected String formatXml(DepartureBean beans) throws ServletException{ |
148 |
|
|
149 |
Serializer serializer = new Persister(); |
Serializer serializer = new Persister(); |
150 |
|
|
151 |
ByteArrayOutputStream out = new ByteArrayOutputStream(); |
StringWriter out = new StringWriter(); |
152 |
|
|
153 |
try { |
try { |
154 |
serializer.write(beans, out); |
serializer.write(beans, out); |
158 |
|
|
159 |
return out.toString(); |
return out.toString(); |
160 |
} |
} |
161 |
|
/* |
162 |
protected String xmlToHtml(String input) throws ServletException { |
protected String xmlToHtml(String input) throws ServletException { |
163 |
|
|
164 |
try { |
try { |
165 |
Transformer trans = xslTemplate.newTransformer(); |
Transformer trans = xslTemplate.newTransformer(); |
166 |
|
|
167 |
|
|
168 |
Source xml = new StreamSource( new ByteArrayInputStream(input.getBytes() )); |
Source xml = new StreamSource( new StringReader(input)); |
169 |
ByteArrayOutputStream out = new ByteArrayOutputStream(); |
StringWriter out = new StringWriter(); |
170 |
trans.transform(xml, new StreamResult(out)); |
trans.transform(xml, new StreamResult(out)); |
171 |
|
|
172 |
return out.toString(); |
return out.toString(); |
173 |
} catch (Exception e) { |
} catch (Exception e) { |
174 |
throw new ServletException(e); |
throw new ServletException(e); |
175 |
} |
} |
176 |
} |
}*/ |
177 |
|
|
178 |
} |
} |