--- android/TrainInfoService/src/dk/thoerup/traininfoservice/StationDAO.java	2010/07/17 06:01:09	997
+++ android/TrainInfoService/src/dk/thoerup/traininfoservice/StationDAO.java	2010/09/21 05:55:31	1082
@@ -1,21 +1,23 @@
 package dk.thoerup.traininfoservice;
 
+import java.sql.Array;
 import java.sql.Connection;
 import java.sql.PreparedStatement;
 import java.sql.ResultSet;
 import java.sql.SQLException;
 import java.sql.Statement;
-import java.util.ArrayList;
-import java.util.List;
 import java.util.logging.Logger;
 
+import dk.thoerup.android.traininfo.common.StationBean;
+import dk.thoerup.android.traininfo.common.StationBean.StationEntry;
+
 public class StationDAO {
 	final static int LOCATION_LIMIT = 8;
 	static final Logger logger = Logger.getLogger(StationDAO.class.getName());
 	
 	
-	private StationBean convertSingleRow(ResultSet res) throws SQLException {
-		StationBean station = new StationBean();
+	private StationEntry convertSingleRow(ResultSet res) throws SQLException {
+		StationEntry station = new StationEntry();
 
 		station.setId( res.getInt(1) );
 		station.setName( res.getString(2) );
@@ -26,28 +28,32 @@
 		station.setMetro( res.getString(7) );
 		station.setAddress( res.getString(8) );
 		station.setCalcdist( (int)res.getDouble(9) );
+		
+		station.setIsRegional( station.getRegional() != null );
+		station.setIsStrain( station.getStrain() != null );
+		station.setIsMetro( station.getMetro() != null );
 
 		return station;
 	}
 
-	private List<StationBean> convertResultset(ResultSet res) throws SQLException {
-		List<StationBean> stations = new ArrayList<StationBean>();
+	private StationBean convertResultset(ResultSet res) throws SQLException {
+		StationBean stations = new StationBean();
 		while (res.next()) {
-			stations.add( convertSingleRow(res) );
+			stations.entries.add( convertSingleRow(res) );
 		}
 		return stations;
 
 	}
 
 
-	public StationBean getById(int id) throws SQLException {
+	public StationEntry getById(int id) throws SQLException {
 		String SQL = "SELECT id,name,latitude,longitude,stationcode_fjrn,stationcode_stog,stationcode_metro,address,0.0 " +
 		"FROM trainstations WHERE id=" + id + " AND enabled=true";
 
 		Connection conn = null;
 		Statement stmt = null;
 		ResultSet res = null;
-		StationBean result;
+		StationEntry result;
 
 		try {
 			conn = DBConnection.getConnection();
@@ -67,6 +73,50 @@
 
 		return result;
 	}
+	
+	public StationBean dumpAll() throws SQLException {
+		
+		String SQL = "SELECT id,name,latitude,longitude,stationcode_fjrn,stationcode_stog,stationcode_metro,address,0.0,aliases " +
+				"FROM trainstations WHERE enabled = true ORDER BY id";
+		
+		Connection conn = null;
+		Statement stmt = null;
+		ResultSet res = null;		
+
+		
+		try {
+			conn = DBConnection.getConnection();
+
+			stmt = conn.createStatement();
+			res = stmt.executeQuery(SQL);		
+						
+			// Does mostly the same as convertResultset()
+			StationBean stations = new StationBean();
+			while (res.next()) {
+				StationEntry entry = convertSingleRow(res);
+				
+				Array arr = res.getArray(10);
+				if (arr != null) {
+					String[] aliases = (String[]) arr.getArray();
+					entry.setAliases(aliases);
+				}
+				
+				stations.entries.add( entry );
+				
+			}
+			return stations;
+			
+
+		} finally {
+			if (res != null)
+				res.close();
+			if (stmt != null)
+				stmt.close();
+			if (conn != null)
+				conn.close();
+		}
+		
+	}
 
 	/*
 	 * this code requires theses statements are run on database in order to do ILIKE searches against aliases (which is defines as array of varchar(64) )
@@ -74,14 +124,14 @@
 	 *     'select $2 ilike $1' language sql strict immutable;
 	 *     create operator ~~~ (procedure = rlike, leftarg = text, rightarg = text, commutator = ~~);
 	 */
-	public List<StationBean> getByName(String name) throws SQLException {
+	public StationBean getByName(String name) throws SQLException {
 		String SQL = "SELECT id,name,latitude,longitude,stationcode_fjrn,stationcode_stog, stationcode_metro, address, 0.0 " + 
 		"FROM trainstations " + 
 		"WHERE (name ILIKE ? OR ? ~~~ ANY(aliases)) AND enabled = true " +
 		"ORDER BY name ";
 
 
-		List<StationBean> result;
+		StationBean result;
 		Connection conn = null;
 		PreparedStatement stmt = null;
 		ResultSet res = null;
@@ -114,7 +164,7 @@
 	// calculate the distance:
 	//     sqrt( power(abs(latitude-?)*111320, 2) + power(abs(longitude-?)*63000,2) )::int as calcdist 
 
-	public List<StationBean> getByLocationWorker(double latitude, double longitude, boolean geolimit) throws SQLException {
+	public StationBean getByLocationWorker(double latitude, double longitude, boolean geolimit) throws SQLException {
 		
 		String limitExpression = (geolimit == true) ? "AND abs(latitude-?)<0.4 AND abs(longitude-?)<0.75 " : "";
 		
@@ -127,7 +177,7 @@
 
 
 		
-		List<StationBean> result;
+		StationBean result;
 		Connection conn = null;
 		PreparedStatement stmt = null;
 		ResultSet res = null;
@@ -154,10 +204,10 @@
 		return result;
 	}
 	
-	public List<StationBean> getByLocation(double latitude, double longitude) throws SQLException {
-		List<StationBean> result = getByLocationWorker(latitude, longitude, true);
+	public StationBean getByLocation(double latitude, double longitude) throws SQLException {
+		StationBean result = getByLocationWorker(latitude, longitude, true);
 		
-		if (result.size() < LOCATION_LIMIT) { //failover
+		if (result.entries.size() < LOCATION_LIMIT) { //failover
 			logger.info("getByLocation failover: " +latitude + "," + longitude);
 			
 			result = getByLocationWorker(latitude, longitude, false);
@@ -168,7 +218,7 @@
 	
 	
 
-	public List<StationBean> getByList(String list) throws SQLException {
+	public StationBean getByList(String list) throws SQLException {
 		String SQL = "SELECT id,name,latitude,longitude,stationcode_fjrn,stationcode_stog,stationcode_metro, address,0.0 " + 
 		"FROM trainstations " + 
 		"WHERE id IN " + list + " AND enabled = true " +
@@ -177,7 +227,7 @@
 		Connection conn = null;
 		Statement stmt = null;
 		ResultSet res = null;
-		List<StationBean> result;
+		StationBean result;
 
 		try {
 			conn = DBConnection.getConnection();
@@ -225,7 +275,7 @@
 		"WHERE name = ?  AND enabled = true " +
 		"LIMIT 1 ";
 
-		List<StationBean> result;
+		StationBean result;
 		Connection conn = null;
 		PreparedStatement stmt = null;
 		ResultSet res = null;
@@ -247,8 +297,8 @@
 				conn.close();
 		}
 
-		if (result.size() == 1) {
-			return result.get(0).getId();
+		if (result.entries.size() == 1) {
+			return result.entries.get(0).getId();
 		} else {
 			return -1;
 		}