thoerup/traininfoservice/LocateStations.java

package dk.thoerup.traininfoservice;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.sql.SQLException;
import java.util.logging.Level;
import java.util.logging.Logger;

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

import org.simpleframework.xml.Serializer;
import org.simpleframework.xml.core.Persister;

/**
 * Servlet implementation class LocateStations
 */
@WebServlet(urlPatterns={"/LocateStations"})
public class LocateStations extends HttpServlet {
        private static final long serialVersionUID = 1L;

        Logger logger = Logger.getLogger( LocateStations.class.toString() );

        StationDAO stationDao = new StationDAO();


        protected String transformToIntList(String input) {             
                String strings[] = input.split(",");

                StringBuffer sb = new StringBuffer();
                sb.append("(");
                for (int i = 0; i<strings.length; i++) {
                        if (i>0) {
                                sb.append(",");
                        }
                        sb.append( Integer.parseInt(strings[i])); //by doing the integer conversion we ensure that it really is a integer
                }
                sb.append(")");
                return sb.toString();           
        }


        protected StationBean getStations(HttpServletRequest req) throws SQLException {
                StationBean stations = null;
                if (req.getParameter("latitude") != null && req.getParameter("latitude") != null) {
                        
                        Statistics.getInstance().incrementStationLookupsLocation();
                        
                        double latitude = Double.parseDouble( req.getParameter("latitude") );
                        double longitude = Double.parseDouble( req.getParameter("longitude") );
                        stations = stationDao.getByLocation(latitude, longitude);

                } else if (req.getParameter("name") != null) {
                        Statistics.getInstance().incrementStationLookupsName();
                        String name = req.getParameter("name").trim();
                        stations = stationDao.getByName(name);

                } else if (req.getParameter("list") != null) {
                        Statistics.getInstance().incrementStationLookupsFavorites();
                        String list = transformToIntList( req.getParameter("list"));
                        stations = stationDao.getByList(list);
                }
                return stations;
        }


        protected String formatStations(StationBean stations) throws ServletException {

                Serializer serializer = new Persister();

                ByteArrayOutputStream out = new ByteArrayOutputStream();

                try {
                        serializer.write(stations, out);
                } catch (Exception e) {
                        throw new ServletException(e);
                }
                
                return out.toString();
        }

        @Override
        protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {


                try {
                        StationBean stations = getStations(request);


                        if (stations != null){
                                String xml = formatStations(stations);

                                response.setContentType("text/xml");
                                response.getWriter().print(xml);
                        } else {
                                response.sendError(400, "not enough parameters");               
                        }


                } catch (Exception e) {                         
                        logger.log(Level.SEVERE, "Exception while finding stations", e);
                        response.sendError(500);                                                
                } 
        }

}
1	package dk.thoerup.traininfoservice;
2
3	import java.io.ByteArrayOutputStream;
4	import java.io.IOException;
5	import java.sql.SQLException;
6	import java.util.logging.Level;
7	import java.util.logging.Logger;
8
9	import javax.servlet.ServletException;
10	import javax.servlet.annotation.WebServlet;
11	import javax.servlet.http.HttpServlet;
12	import javax.servlet.http.HttpServletRequest;
13	import javax.servlet.http.HttpServletResponse;
14
15	import org.simpleframework.xml.Serializer;
16	import org.simpleframework.xml.core.Persister;
17
18	/**
19	* Servlet implementation class LocateStations
20	*/
21	@WebServlet(urlPatterns={"/LocateStations"})
22	public class LocateStations extends HttpServlet {
23	private static final long serialVersionUID = 1L;
24
25	Logger logger = Logger.getLogger( LocateStations.class.toString() );
26
27	StationDAO stationDao = new StationDAO();
28
29
30	protected String transformToIntList(String input) {
31	String strings[] = input.split(",");
32
33	StringBuffer sb = new StringBuffer();
34	sb.append("(");
35	for (int i = 0; i<strings.length; i++) {
36	if (i>0) {
37	sb.append(",");
38	}
39	sb.append( Integer.parseInt(strings[i])); //by doing the integer conversion we ensure that it really is a integer
40	}
41	sb.append(")");
42	return sb.toString();
43	}
44
45
46	protected StationBean getStations(HttpServletRequest req) throws SQLException {
47	StationBean stations = null;
48	if (req.getParameter("latitude") != null && req.getParameter("latitude") != null) {
49
50	Statistics.getInstance().incrementStationLookupsLocation();
51
52	double latitude = Double.parseDouble( req.getParameter("latitude") );
53	double longitude = Double.parseDouble( req.getParameter("longitude") );
54	stations = stationDao.getByLocation(latitude, longitude);
55
56	} else if (req.getParameter("name") != null) {
57	Statistics.getInstance().incrementStationLookupsName();
58	String name = req.getParameter("name").trim();
59	stations = stationDao.getByName(name);
60
61	} else if (req.getParameter("list") != null) {
62	Statistics.getInstance().incrementStationLookupsFavorites();
63	String list = transformToIntList( req.getParameter("list"));
64	stations = stationDao.getByList(list);
65	}
66	return stations;
67	}
68
69
70	protected String formatStations(StationBean stations) throws ServletException {
71
72	Serializer serializer = new Persister();
73
74	ByteArrayOutputStream out = new ByteArrayOutputStream();
75
76	try {
77	serializer.write(stations, out);
78	} catch (Exception e) {
79	throw new ServletException(e);
80	}
81
82	return out.toString();
83	}
84
85	@Override
86	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
87
88
89	try {
90	StationBean stations = getStations(request);
91
92
93	if (stations != null){
94	String xml = formatStations(stations);
95
96	response.setContentType("text/xml");
97	response.getWriter().print(xml);
98	} else {
99	response.sendError(400, "not enough parameters");
100	}
101
102
103	} catch (Exception e) {
104	logger.log(Level.SEVERE, "Exception while finding stations", e);
105	response.sendError(500);
106	}
107	}
108
109	}