thoerup/traininfoservice/DumpResultSet.java

package dk.thoerup.traininfoservice;

import java.io.IOException;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.Statement;
import java.util.Date;

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


public class DumpResultSet extends HttpServlet {

    public DumpResultSet() {
        super();
    }


        private static final long serialVersionUID = 1L;

        String dumpResultset(String sql) throws ServletException {
                StringBuilder sb = new StringBuilder();
                
                Connection conn = null;
                Statement stmt = null;
                ResultSet rs = null;
                
                try {
                        int count = 0;
                        
                        conn =  DBConnection.getConnection();
                        stmt = conn.createStatement();
                        
                        rs = stmt.executeQuery(sql);
                        ResultSetMetaData meta = rs.getMetaData();
                        int columns = meta.getColumnCount();
                        
                        sb.append("<h2>").append(sql).append("</h2>");
                        sb.append("<table border=1><tr>");
                        for (int i=1; i<=columns; i++) {
                                sb.append("<th>").append( meta.getColumnName(i)).append("</th>");
                        }
                        sb.append("</tr>\n\n");
                        while(rs.next())
                        {
                                count++;
                                
                                sb.append("<tr>");
                                for (int i=1; i<=columns; i++) {
                                        String value = rs.getString(i);
                                        if (value != null) {
                                                value = value.replace("\n", "<br>");
                                        }
                                        sb.append("<td>").append( value ).append("</td>");
                                }
                                sb.append("</tr>\n");
                        }
                        sb.append("</table>");
                        sb.append("Rowcount: ").append(count).append("<br>\n");
                        sb.append("Generated: ").append(new Date()).append("<br>\n");
                } catch (Exception e) { 
                        throw new ServletException(e);
                } finally {

                        try {
                                if (rs != null && !rs.isClosed())
                                        rs.close();
                        } catch (Exception e) {}
                        try {
                                if (stmt != null && !stmt.isClosed())
                                        stmt.close();
                        } catch (Exception e) {}
                        
                        try {
                                if (conn != null && !conn.isClosed())
                                        conn.close();
                        } catch (Exception e) {}
                }
                return sb.toString();
        }
        
        String dumpUpdate(String query) throws ServletException {
                StringBuilder sb = new StringBuilder();

                Connection conn = null;
                Statement stmt = null;
                try {
                        conn =  DBConnection.getConnection();
                        stmt = conn.createStatement();
                        stmt.execute(query);
                        int count = stmt.getUpdateCount();
                        
                        sb.append("<h2>").append(query).append("</h2>");
                        sb.append("Affected rows: ").append(count);
                } catch (Exception e) { 
                        throw new ServletException(e);
                } finally {
                        try {
                                if (stmt != null && !stmt.isClosed())
                                        stmt.close();
                        } catch (Exception e) {}
                        
                        try {
                                if (conn != null && !conn.isClosed())
                                        conn.close();
                        } catch (Exception e) {}
                }
                
                return sb.toString();
        }
        
        
        @Override
        protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
                String dump = req.getParameter("dump");
                
                String res = null;
        
                if (dump != null) {
                        if (dump.equals("all")) {
                                res = dumpResultset("SELECT * FROM trainstations ORDER BY id");
                        } else if (dump.equals("coords")) {
                                res = dumpResultset("SELECT * FROM trainstations WHERE latitude IS NULL OR longitude IS NULL ORDER BY id");                     
                        } else if (dump.equals("duplicate")) {
                                res = dumpResultset("SELECT name,count(*) FROM trainstations GROUP BY name HAVING count(*) > 1"); 
                        } else if (dump.equals("allfull")) {
                                res = dumpResultset("select *, " + 
                                                "'<a href=\"http://maps.google.dk/?q=' ||latitude || ',' || longitude || '\">Maps</a>' AS maps, " + 
                                                "'<a href=\"http://www.bane.dk/visStation.asp?ArtikelID=4275&W=FJRN&S=' || stationcode_fjrn  || '\">Bane.dk Fjern</a>' as banedk1, " + 
                                                "'<a href=\"http://www.bane.dk/visStation.asp?ArtikelID=4275&W=S2&S=' || stationcode_stog  || '\">Bane.dk stog</a>' as banedk2 " + 
                                                "FROM trainstations ORDER BY id");
                        } else if (dump.equals("disabled")) {
                                res = dumpResultset("SELECT * FROM trainstations WHERE enabled = false ORDER BY id");
                        } else if (dump.equals("noaddress")) {
                                res = dumpResultset("SELECT * FROM trainstations WHERE address IS NULL or address = '' ");
                        } else if (dump.equals("aliases")) {
                                res = dumpResultset("SELECT * FROM trainstations WHERE aliases IS NOT null");
                        } else if (dump.equals("updatecoords")) {
                                res = dumpUpdate("UPDATE trainstations SET earth_coord = ll_to_earth(latitude,longitude)");
                        } 
                }
                if (res == null) {
                        res = "<a href=DumpResultSet?dump=all>All</a><br>" +
                                        "<a href=DumpResultSet?dump=allfull>All with links</a><br>" +
                                        "<a href=DumpResultSet?dump=coords>Missing coords</a><br>" +
                                        "<a href=DumpResultSet?dump=duplicate>Duplicate stations</a><br>" +
                                        "<a href=DumpResultSet?dump=disabled>Disabled stations</a><br>" +
                                        "<a href=DumpResultSet?dump=noaddress>No address</a><br>" +
                                        "<a href=DumpResultSet?dump=aliases>Has aliases</a><br>" +
                                        "<a href=DumpResultSet?dump=updatecoords>update coords</a><br>";
                }
                
                resp.setContentType("text/html");
                resp.getWriter().println(res);
        }

}
1	torben	296	package dk.thoerup.traininfoservice;
2
3			import java.io.IOException;
4			import java.sql.Connection;
5			import java.sql.ResultSet;
6			import java.sql.ResultSetMetaData;
7			import java.sql.Statement;
8	torben	827	import java.util.Date;
9	torben	296
10			import javax.servlet.ServletException;
11			import javax.servlet.http.HttpServlet;
12			import javax.servlet.http.HttpServletRequest;
13			import javax.servlet.http.HttpServletResponse;
14
15	torben	1080
16	torben	296	public class DumpResultSet extends HttpServlet {
17
18	torben	298	public DumpResultSet() {
19			super();
20			}
21	torben	296
22	torben	298
23	torben	296	private static final long serialVersionUID = 1L;
24
25			String dumpResultset(String sql) throws ServletException {
26			StringBuilder sb = new StringBuilder();
27
28			Connection conn = null;
29			Statement stmt = null;
30			ResultSet rs = null;
31	torben	298
32	torben	296	try {
33	torben	304	int count = 0;
34
35	torben	296	conn = DBConnection.getConnection();
36			stmt = conn.createStatement();
37
38			rs = stmt.executeQuery(sql);
39			ResultSetMetaData meta = rs.getMetaData();
40			int columns = meta.getColumnCount();
41
42			sb.append("<h2>").append(sql).append("</h2>");
43	torben	298	sb.append("<table border=1><tr>");
44	torben	296	for (int i=1; i<=columns; i++) {
45			sb.append("<th>").append( meta.getColumnName(i)).append("</th>");
46			}
47	torben	304	sb.append("</tr>\n\n");
48	torben	296	while(rs.next())
49			{
50	torben	304	count++;
51
52	torben	296	sb.append("<tr>");
53			for (int i=1; i<=columns; i++) {
54	torben	494	String value = rs.getString(i);
55			if (value != null) {
56			value = value.replace("\n", "<br>");
57			}
58			sb.append("<td>").append( value ).append("</td>");
59	torben	296	}
60	torben	304	sb.append("</tr>\n");
61	torben	296	}
62			sb.append("</table>");
63	torben	827	sb.append("Rowcount: ").append(count).append("<br>\n");
64			sb.append("Generated: ").append(new Date()).append("<br>\n");
65	torben	296	} catch (Exception e) {
66			throw new ServletException(e);
67			} finally {
68
69			try {
70			if (rs != null && !rs.isClosed())
71			rs.close();
72			} catch (Exception e) {}
73			try {
74			if (stmt != null && !stmt.isClosed())
75			stmt.close();
76			} catch (Exception e) {}
77
78			try {
79			if (conn != null && !conn.isClosed())
80			conn.close();
81			} catch (Exception e) {}
82			}
83			return sb.toString();
84			}
85
86	torben	733	String dumpUpdate(String query) throws ServletException {
87			StringBuilder sb = new StringBuilder();
88
89			Connection conn = null;
90			Statement stmt = null;
91			try {
92			conn = DBConnection.getConnection();
93			stmt = conn.createStatement();
94			stmt.execute(query);
95			int count = stmt.getUpdateCount();
96
97			sb.append("<h2>").append(query).append("</h2>");
98			sb.append("Affected rows: ").append(count);
99			} catch (Exception e) {
100			throw new ServletException(e);
101			} finally {
102			try {
103			if (stmt != null && !stmt.isClosed())
104			stmt.close();
105			} catch (Exception e) {}
106
107			try {
108			if (conn != null && !conn.isClosed())
109			conn.close();
110			} catch (Exception e) {}
111			}
112
113			return sb.toString();
114			}
115
116
117
118	torben	296	@Override
119			protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
120			String dump = req.getParameter("dump");
121
122	torben	298	String res = null;
123	torben	296
124	torben	298	if (dump != null) {
125			if (dump.equals("all")) {
126	torben	304	res = dumpResultset("SELECT * FROM trainstations ORDER BY id");
127	torben	298	} else if (dump.equals("coords")) {
128	torben	304	res = dumpResultset("SELECT * FROM trainstations WHERE latitude IS NULL OR longitude IS NULL ORDER BY id");
129	torben	298	} else if (dump.equals("duplicate")) {
130			res = dumpResultset("SELECT name,count() FROM trainstations GROUP BY name HAVING count() > 1");
131	torben	304	} else if (dump.equals("allfull")) {
132			res = dumpResultset("select *, " +
133	torben	497	"'<a href=\"http://maps.google.dk/?q=' \|\|latitude \|\| ',' \|\| longitude \|\| '\">Maps</a>' AS maps, " +
134	torben	314	"'<a href=\"http://www.bane.dk/visStation.asp?ArtikelID=4275&W=FJRN&S=' \|\| stationcode_fjrn \|\| '\">Bane.dk Fjern</a>' as banedk1, " +
135			"'<a href=\"http://www.bane.dk/visStation.asp?ArtikelID=4275&W=S2&S=' \|\| stationcode_stog \|\| '\">Bane.dk stog</a>' as banedk2 " +
136			"FROM trainstations ORDER BY id");
137	torben	377	} else if (dump.equals("disabled")) {
138			res = dumpResultset("SELECT * FROM trainstations WHERE enabled = false ORDER BY id");
139	torben	492	} else if (dump.equals("noaddress")) {
140			res = dumpResultset("SELECT * FROM trainstations WHERE address IS NULL or address = '' ");
141	torben	715	} else if (dump.equals("aliases")) {
142			res = dumpResultset("SELECT * FROM trainstations WHERE aliases IS NOT null");
143	torben	733	} else if (dump.equals("updatecoords")) {
144			res = dumpUpdate("UPDATE trainstations SET earth_coord = ll_to_earth(latitude,longitude)");
145	torben	961	}
146	torben	296	}
147	torben	298	if (res == null) {
148	torben	813	res = "<a href=DumpResultSet?dump=all>All</a><br>" +
149			"<a href=DumpResultSet?dump=allfull>All with links</a><br>" +
150			"<a href=DumpResultSet?dump=coords>Missing coords</a><br>" +
151			"<a href=DumpResultSet?dump=duplicate>Duplicate stations</a><br>" +
152			"<a href=DumpResultSet?dump=disabled>Disabled stations</a><br>" +
153			"<a href=DumpResultSet?dump=noaddress>No address</a><br>" +
154			"<a href=DumpResultSet?dump=aliases>Has aliases</a><br>" +
155	torben	961	"<a href=DumpResultSet?dump=updatecoords>update coords</a><br>";
156	torben	298	}
157
158	torben	296	resp.setContentType("text/html");
159			resp.getWriter().println(res);
160			}
161
162			}