traininfoservice/banedk/TimetableServlet.java

package dk.thoerup.traininfoservice.banedk;

import java.io.IOException;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import dk.thoerup.circuitbreaker.CircuitBreakerException;
import dk.thoerup.traininfoservice.Statistics;

/**
 * Servlet implementation class TimetableServlet
 */
public class TimetableServlet extends HttpServlet {
        private static final long serialVersionUID = 1L;
       
        
        Logger logger = Logger.getLogger( TimetableServlet.class.getName() );
        
        TimetableFetcher fetcher;
        
        @Override
        public void init() throws ServletException {
                super.init();
                
                boolean useTempSite =  Boolean.parseBoolean( getServletContext().getInitParameter("usetempsite") );
                int cacheTimeout =  Integer.parseInt( getServletContext().getInitParameter("cache_timeout") );
                logger.info( "TimetableServlet, use temp site=" + useTempSite + ", cache=" + cacheTimeout);
                fetcher = new TimetableFetcher(useTempSite,cacheTimeout);
        }
        
        @Override
        protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
                if (request.getParameter("train") == null) {
                        response.sendError(400, "not enough parameters");
                        return;
                }
                Statistics.getInstance().incrementTimetableLookups();
                
                String train = request.getParameter("train").trim();
                String trainID;
                String type = request.getParameter("type");
                
                if (type != null) {
                         trainID = train;
                } else { //remove this failover sometime after release of client 0.6.4 
                        String sections[] = train.split(" ");
                        if (sections.length != 2) {
                                response.sendError(400, "Invalid parameter");
                                return;
                        }
                                
                        type = lookupType(sections[0]);
                        trainID = sections[1];
                }
                
                try
                {
                        List<TimetableBean> timetable = fetcher.cachedLookupTimetable(trainID, type);
                
                        String responseXML = formatTimetable(timetable);
                
                        response.setContentType("text/xml");
                        response.getWriter().print(responseXML);
                } catch (java.io.IOException ioe) {
                        logger.warning("(timetable) Read failed, train="+train + ". " + ioe.getMessage());
                        Statistics.getInstance().incrementTimetableErrors();
                        response.sendError(500, "Backend timed out");
                } catch (CircuitBreakerException cbe) {
                        logger.warning("Circuitbreaker - failing fast");
                        Statistics.getInstance().incrementTimetableErrors();
                        response.sendError(500);
                }catch (Exception e) {
                        logger.log(Level.WARNING, "failure looking up timetable: train=" +train, e);
                        Statistics.getInstance().incrementTimetableErrors();
                        response.sendError(500);
                }
                
        }
        /*remove /outcomment/deprecate after 0.6.4 release */
        String lookupType(String trainCode) {
                trainCode = trainCode.toLowerCase();
                
                if (trainCode.equals("a") || trainCode.equals("b") || trainCode.equals("bx") || trainCode.equals("c") || 
                                trainCode.equals("e") || trainCode.equals("f") || trainCode.equals("h")) {
                        return "S2";
                } else {
                        return "FJRN";
                }
        }
        
        String formatTimetable(List<TimetableBean> timetable) {
                StringBuffer sb = new StringBuffer();
                
                sb.append("<timetable>");
                
                for (TimetableBean bean : timetable) {
                        sb.append("<entry");
                        if (bean.isCurrent())
                                sb.append(" current=\"true\"");
                        sb.append(">");
                        
                        sb.append("<station>").append( bean.getStation() ).append("</station>");
                        sb.append("<arrival>").append( bean.getArrival() ).append("</arrival>");
                        sb.append("<departure>").append( bean.getDeparture() ).append("</departure>");
                        
                        sb.append("</entry>");
                }
                
                sb.append("</timetable>");
                
                return sb.toString();
        }

}
1	torben	350	package dk.thoerup.traininfoservice.banedk;
2
3			import java.io.IOException;
4			import java.util.List;
5	torben	378	import java.util.logging.Level;
6	torben	350	import java.util.logging.Logger;
7
8			import javax.servlet.ServletException;
9			import javax.servlet.http.HttpServlet;
10			import javax.servlet.http.HttpServletRequest;
11			import javax.servlet.http.HttpServletResponse;
12
13	torben	468	import dk.thoerup.circuitbreaker.CircuitBreakerException;
14	torben	711	import dk.thoerup.traininfoservice.Statistics;
15	torben	421
16	torben	350	/**
17			* Servlet implementation class TimetableServlet
18			*/
19			public class TimetableServlet extends HttpServlet {
20			private static final long serialVersionUID = 1L;
21
22
23			Logger logger = Logger.getLogger( TimetableServlet.class.getName() );
24
25	torben	581	TimetableFetcher fetcher;
26	torben	350
27	torben	425	@Override
28	torben	581	public void init() throws ServletException {
29			super.init();
30
31	torben	584	boolean useTempSite = Boolean.parseBoolean( getServletContext().getInitParameter("usetempsite") );
32			int cacheTimeout = Integer.parseInt( getServletContext().getInitParameter("cache_timeout") );
33			logger.info( "TimetableServlet, use temp site=" + useTempSite + ", cache=" + cacheTimeout);
34			fetcher = new TimetableFetcher(useTempSite,cacheTimeout);
35	torben	581	}
36
37			@Override
38	torben	350	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
39	torben	458	if (request.getParameter("train") == null) {
40			response.sendError(400, "not enough parameters");
41			return;
42			}
43	torben	711	Statistics.getInstance().incrementTimetableLookups();
44	torben	458
45	torben	350	String train = request.getParameter("train").trim();
46	torben	697	String trainID;
47			String type = request.getParameter("type");
48	torben	350
49	torben	697	if (type != null) {
50			trainID = train;
51			} else { //remove this failover sometime after release of client 0.6.4
52			String sections[] = train.split(" ");
53			if (sections.length != 2) {
54			response.sendError(400, "Invalid parameter");
55			return;
56			}
57
58			type = lookupType(sections[0]);
59			trainID = sections[1];
60	torben	350	}
61	torben	697
62	torben	350	try
63			{
64	torben	387	List<TimetableBean> timetable = fetcher.cachedLookupTimetable(trainID, type);
65	torben	350
66			String responseXML = formatTimetable(timetable);
67
68			response.setContentType("text/xml");
69			response.getWriter().print(responseXML);
70	torben	462	} catch (java.io.IOException ioe) {
71			logger.warning("(timetable) Read failed, train="+train + ". " + ioe.getMessage());
72	torben	716	Statistics.getInstance().incrementTimetableErrors();
73	torben	378	response.sendError(500, "Backend timed out");
74	torben	421	} catch (CircuitBreakerException cbe) {
75			logger.warning("Circuitbreaker - failing fast");
76	torben	716	Statistics.getInstance().incrementTimetableErrors();
77	torben	421	response.sendError(500);
78	torben	451	}catch (Exception e) {
79	torben	378	logger.log(Level.WARNING, "failure looking up timetable: train=" +train, e);
80	torben	716	Statistics.getInstance().incrementTimetableErrors();
81	torben	378	response.sendError(500);
82	torben	350	}
83
84			}
85	torben	697	/remove /outcomment/deprecate after 0.6.4 release /
86	torben	350	String lookupType(String trainCode) {
87			trainCode = trainCode.toLowerCase();
88
89			if (trainCode.equals("a") \|\| trainCode.equals("b") \|\| trainCode.equals("bx") \|\| trainCode.equals("c") \|\|
90			trainCode.equals("e") \|\| trainCode.equals("f") \|\| trainCode.equals("h")) {
91			return "S2";
92			} else {
93			return "FJRN";
94			}
95			}
96
97			String formatTimetable(List<TimetableBean> timetable) {
98			StringBuffer sb = new StringBuffer();
99
100			sb.append("<timetable>");
101
102			for (TimetableBean bean : timetable) {
103			sb.append("<entry");
104			if (bean.isCurrent())
105			sb.append(" current=\"true\"");
106			sb.append(">");
107
108			sb.append("<station>").append( bean.getStation() ).append("</station>");
109			sb.append("<arrival>").append( bean.getArrival() ).append("</arrival>");
110			sb.append("<departure>").append( bean.getDeparture() ).append("</departure>");
111
112			sb.append("</entry>");
113			}
114
115			sb.append("</timetable>");
116
117			return sb.toString();
118			}
119
120			}