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