thoerup/traininfoservice/LoadStations.java

package dk.thoerup.traininfoservice;

import java.io.IOException;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

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 dk.thoerup.traininfoservice.db.DBConnection;

@Deprecated
@WebServlet(urlPatterns={"/LoadStations"})

public class LoadStations extends HttpServlet {
        private static final long serialVersionUID = 1L;

        public LoadStations() {
                super();
        }

        protected void insertOrUpdate(Connection conn, String type, String code, String name)  throws SQLException{
                String sql = "SELECT id FROM trainstations WHERE stationcode=?";
                PreparedStatement stmt = conn.prepareStatement(sql);
                stmt.setString(1, code);
                
                ResultSet rs = stmt.executeQuery();
                
                boolean stog = type.equalsIgnoreCase("S2");
                
                if (rs.next()) {
                        int id = rs.getInt(1);
                        stmt = conn.prepareStatement("UPDATE trainstations SET name=?, stog=?, updated=now() WHERE id=?");
                        stmt.setString(1, name);
                        stmt.setBoolean(2, stog);
                        stmt.setInt(3, id);
                        stmt.execute();
                } else {
                        stmt = conn.prepareStatement("INSERT INTO trainstations (name,stationcode,stog,updated) VALUES (?,?,?,now())");
                        stmt.setString(1, name);
                        stmt.setString(2, code);
                        stmt.setBoolean(3, stog);
                        stmt.execute();
                }
                
        }

        @Override
        protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
                String url = "http://rafiki.t-hoerup.dk/tog/stations.php";

                String stationStr = DownloadUtil.getContentString(url, 5000, "ISO-8859-1");

                String stations[] = stationStr.split("\n");
                Connection conn = null; 
                try {
                        conn = DBConnection.getConnection();            
                        for (String stationLine : stations) {
                                String fields[] = stationLine.split(";");
                                
                                
                                insertOrUpdate(conn, fields[0], fields[1], fields[2]);
                        }
                } catch (Exception e) {
                        throw new ServletException(e);
                } finally {
                        try {
                                if (conn != null && !conn.isClosed())
                                        conn.close();
                        } catch (SQLException e) {}
                }                       
                PrintWriter out = resp.getWriter();
                out.print("ok");
        }

}
1	package dk.thoerup.traininfoservice;
2
3	import java.io.IOException;
4	import java.io.PrintWriter;
5	import java.sql.Connection;
6	import java.sql.PreparedStatement;
7	import java.sql.ResultSet;
8	import java.sql.SQLException;
9
10	import javax.servlet.ServletException;
11	import javax.servlet.annotation.WebServlet;
12	import javax.servlet.http.HttpServlet;
13	import javax.servlet.http.HttpServletRequest;
14	import javax.servlet.http.HttpServletResponse;
15
16	import dk.thoerup.traininfoservice.db.DBConnection;
17
18	@Deprecated
19	@WebServlet(urlPatterns={"/LoadStations"})
20
21	public class LoadStations extends HttpServlet {
22	private static final long serialVersionUID = 1L;
23
24	public LoadStations() {
25	super();
26	}
27
28	protected void insertOrUpdate(Connection conn, String type, String code, String name) throws SQLException{
29	String sql = "SELECT id FROM trainstations WHERE stationcode=?";
30	PreparedStatement stmt = conn.prepareStatement(sql);
31	stmt.setString(1, code);
32
33	ResultSet rs = stmt.executeQuery();
34
35	boolean stog = type.equalsIgnoreCase("S2");
36
37	if (rs.next()) {
38	int id = rs.getInt(1);
39	stmt = conn.prepareStatement("UPDATE trainstations SET name=?, stog=?, updated=now() WHERE id=?");
40	stmt.setString(1, name);
41	stmt.setBoolean(2, stog);
42	stmt.setInt(3, id);
43	stmt.execute();
44	} else {
45	stmt = conn.prepareStatement("INSERT INTO trainstations (name,stationcode,stog,updated) VALUES (?,?,?,now())");
46	stmt.setString(1, name);
47	stmt.setString(2, code);
48	stmt.setBoolean(3, stog);
49	stmt.execute();
50	}
51
52	}
53
54	@Override
55	protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
56	String url = "http://rafiki.t-hoerup.dk/tog/stations.php";
57
58	String stationStr = DownloadUtil.getContentString(url, 5000, "ISO-8859-1");
59
60	String stations[] = stationStr.split("\n");
61	Connection conn = null;
62	try {
63	conn = DBConnection.getConnection();
64	for (String stationLine : stations) {
65	String fields[] = stationLine.split(";");
66
67
68	insertOrUpdate(conn, fields[0], fields[1], fields[2]);
69	}
70	} catch (Exception e) {
71	throw new ServletException(e);
72	} finally {
73	try {
74	if (conn != null && !conn.isClosed())
75	conn.close();
76	} catch (SQLException e) {}
77	}
78	PrintWriter out = resp.getWriter();
79	out.print("ok");
80	}
81
82	}