traininfoservice/banedk/TimetableServlet.java

package dk.thoerup.traininfoservice.banedk;

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

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

/**
 * Servlet implementation class TimetableServlet
 */
public class TimetableServlet extends HttpServlet {
        private static final long serialVersionUID = 1L;
       
        
        Logger logger = Logger.getLogger( TimetableServlet.class.getName() );
        
        TimetableFetcher fetcher = new TimetableFetcher();
        
        protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
                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.lookupTimetable(trainID, type);
                
                        String responseXML = formatTimetable(timetable);
                
                        response.setContentType("text/xml");
                        response.getWriter().print(responseXML);
                } catch (Exception e) {
                        logger.warning("failure looking up timetable: train=" +train);
                        throw new ServletException(e);
                }
                
        }
        
        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			import java.util.logging.Logger;
6
7			import javax.servlet.ServletException;
8			import javax.servlet.http.HttpServlet;
9			import javax.servlet.http.HttpServletRequest;
10			import javax.servlet.http.HttpServletResponse;
11
12			/**
13			* Servlet implementation class TimetableServlet
14			*/
15			public class TimetableServlet extends HttpServlet {
16			private static final long serialVersionUID = 1L;
17
18
19			Logger logger = Logger.getLogger( TimetableServlet.class.getName() );
20
21			TimetableFetcher fetcher = new TimetableFetcher();
22
23			protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
24			String train = request.getParameter("train").trim();
25
26			String sections[] = train.split(" ");
27			if (sections.length != 2) {
28			response.sendError(400, "Invalid parameter");
29			return;
30			}
31
32			String type = lookupType(sections[0]);
33			String trainID = sections[1];
34			try
35			{
36			List<TimetableBean> timetable = fetcher.lookupTimetable(trainID, type);
37
38			String responseXML = formatTimetable(timetable);
39
40			response.setContentType("text/xml");
41			response.getWriter().print(responseXML);
42			} catch (Exception e) {
43			logger.warning("failure looking up timetable: train=" +train);
44			throw new ServletException(e);
45			}
46
47			}
48
49			String lookupType(String trainCode) {
50			trainCode = trainCode.toLowerCase();
51
52			if (trainCode.equals("a") \|\| trainCode.equals("b") \|\| trainCode.equals("bx") \|\| trainCode.equals("c") \|\|
53			trainCode.equals("e") \|\| trainCode.equals("f") \|\| trainCode.equals("h")) {
54			return "S2";
55			} else {
56			return "FJRN";
57			}
58			}
59
60			String formatTimetable(List<TimetableBean> timetable) {
61			StringBuffer sb = new StringBuffer();
62
63			sb.append("<timetable>");
64
65			for (TimetableBean bean : timetable) {
66			sb.append("<entry");
67			if (bean.isCurrent())
68			sb.append(" current=\"true\"");
69			sb.append(">");
70
71			sb.append("<station>").append( bean.getStation() ).append("</station>");
72			sb.append("<arrival>").append( bean.getArrival() ).append("</arrival>");
73			sb.append("<departure>").append( bean.getDeparture() ).append("</departure>");
74
75			sb.append("</entry>");
76			}
77
78			sb.append("</timetable>");
79
80			return sb.toString();
81			}
82
83			}