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;

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