vaadin-framework/src/com/itmill/toolkit/demo/reservation/SampleDB.java

package com.itmill.toolkit.demo.reservation;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Date;
import java.util.Iterator;
import java.util.List;

import com.itmill.toolkit.data.Container;
import com.itmill.toolkit.data.Item;
import com.itmill.toolkit.data.util.QueryContainer;

public class SampleDB {
	public class User {
		public static final String TABLE = "USER";
		public static final String PROPERTY_ID_ID = TABLE + "_ID";
		public static final String PROPERTY_ID_FULLNAME = TABLE + "_FULLNAME";
		public static final String PROPERTY_ID_EMAIL = TABLE + "_EMAIL";
		public static final String PROPERTY_ID_PASSWORD = TABLE + "_PASSWORD";
		public static final String PROPERTY_ID_DELETED = TABLE + "_DELETED";
	}

	public class Resource {
		public static final String TABLE = "RESOURCE";
		public static final String PROPERTY_ID_ID = TABLE + "_ID";
		public static final String PROPERTY_ID_NAME = TABLE + "_NAME";
		public static final String PROPERTY_ID_DESCRIPTION = TABLE
				+ "_DESCRIPTION";
		public static final String PROPERTY_ID_LOCATIONX = TABLE
				+ "_LOCATION_X";
		public static final String PROPERTY_ID_LOCATIONY = TABLE
				+ "_LOCATION_Y";
		public static final String PROPERTY_ID_CATEGORY = TABLE + "_CATEGORY";
		public static final String PROPERTY_ID_DELETED = TABLE + "_DELETED";
	}

	public class Reservation {
		public static final String TABLE = "RESERVATION";
		public static final String PROPERTY_ID_ID = TABLE + "_ID";
		public static final String PROPERTY_ID_DESCRIPTION = TABLE
				+ "_DESCRIPTION";
		public static final String PROPERTY_ID_RESOURCE_ID = TABLE
				+ "_RESOURCE_ID";
		public static final String PROPERTY_ID_RESERVED_BY_ID = TABLE
				+ "_RESERVED_BY_USER_ID";
		public static final String PROPERTY_ID_RESERVED_FROM = TABLE
				+ "_RESERVED_FROM";
		public static final String PROPERTY_ID_RESERVED_TO = TABLE
				+ "_RESERVED_TO";
	}

	// TODO -> param
	private static final String DB_URL = "jdbc:hsqldb:file:reservation.db";

	private static final String CREATE_TABLE_USER = "CREATE TABLE "
			+ User.TABLE + " (" + " " + User.PROPERTY_ID_ID
			+ " INTEGER IDENTITY" + ", " + User.PROPERTY_ID_FULLNAME
			+ " VARCHAR(100) NOT NULL" + ", " + User.PROPERTY_ID_EMAIL
			+ " VARCHAR(50) NOT NULL" + ", " + User.PROPERTY_ID_PASSWORD
			+ " VARCHAR(20) NOT NULL" + ", " + User.PROPERTY_ID_DELETED
			+ " BOOLEAN DEFAULT false NOT NULL" + ", UNIQUE("
			+ User.PROPERTY_ID_FULLNAME + "), UNIQUE(" + User.PROPERTY_ID_EMAIL
			+ ") )";
	private static final String CREATE_TABLE_RESOURCE = "CREATE TABLE "
			+ Resource.TABLE + " (" + " " + Resource.PROPERTY_ID_ID
			+ " INTEGER IDENTITY" + ", " + Resource.PROPERTY_ID_NAME
			+ " VARCHAR(30) NOT NULL" + ", " + Resource.PROPERTY_ID_DESCRIPTION
			+ " VARCHAR(100)" + ", " + Resource.PROPERTY_ID_LOCATIONX
			+ " DOUBLE" + ", " + Resource.PROPERTY_ID_LOCATIONY + " DOUBLE"
			+ ", " + Resource.PROPERTY_ID_CATEGORY + " VARCHAR(30)" + ", "
			+ Resource.PROPERTY_ID_DELETED + " BOOLEAN DEFAULT false NOT NULL"
			+ ", UNIQUE(" + Resource.PROPERTY_ID_NAME + "))";
	private static final String CREATE_TABLE_RESERVATION = "CREATE TABLE "
			+ Reservation.TABLE + " (" + " " + Reservation.PROPERTY_ID_ID
			+ " INTEGER IDENTITY" + ", " + Reservation.PROPERTY_ID_RESOURCE_ID
			+ " INTEGER" + ", " + Reservation.PROPERTY_ID_RESERVED_BY_ID
			+ " INTEGER" + ", " + Reservation.PROPERTY_ID_RESERVED_FROM
			+ " TIMESTAMP NOT NULL" + ", "
			+ Reservation.PROPERTY_ID_RESERVED_TO + " TIMESTAMP NOT NULL"
			+ ", " + Reservation.PROPERTY_ID_DESCRIPTION + " VARCHAR(100)"
			+ ", FOREIGN KEY (" + Reservation.PROPERTY_ID_RESOURCE_ID
			+ ") REFERENCES " + Resource.TABLE + "(" + Resource.PROPERTY_ID_ID
			+ "), FOREIGN KEY (" + Reservation.PROPERTY_ID_RESERVED_BY_ID
			+ ") REFERENCES " + User.TABLE + "(" + User.PROPERTY_ID_ID + "))";

	private Connection connection = null;

	/**
	 * Create database.
	 */
	public SampleDB() {
		this(false);
	}

	public SampleDB(boolean recreate) {
		// connect to SQL database
		connect();

		if (recreate) {
			dropTables();
		}

		// initialize SQL database
		createTables();

		// test by executing sample JDBC query
		testDatabase();
	}

	private void dropTables() {
		try {
			update("DROP TABLE " + Reservation.TABLE);
		} catch (SQLException IGNORED) {
			// IGNORED, assuming it was not there
		}
		try {
			update("DROP TABLE " + Resource.TABLE);
		} catch (SQLException IGNORED) {
			// IGNORED, assuming it was not there
		}
		try {
			update("DROP TABLE " + User.TABLE);
		} catch (SQLException IGNORED) {
			// IGNORED, assuming it was not there
		}
	}

	/**
	 * Connect to SQL database. In this sample we use HSQLDB and an toolkit
	 * named database in implicitly created into system memory.
	 * 
	 */
	private void connect() {
		try {
			Class.forName("org.hsqldb.jdbcDriver").newInstance();
			connection = DriverManager.getConnection(DB_URL);
		} catch (Exception e) {
			throw new RuntimeException(e);
		}
	}

	/**
	 * use for SQL commands CREATE, DROP, INSERT and UPDATE
	 * 
	 * @param expression
	 * @throws SQLException
	 */
	private void update(String expression) throws SQLException {
		Statement st = null;
		st = connection.createStatement();
		int i = st.executeUpdate(expression);
		if (i == -1) {
			System.out.println("SampleDatabase error : " + expression);
		}
		st.close();
	}

	/**
	 * Create test table and few rows. Issue note: using capitalized column
	 * names as HSQLDB returns column names in capitalized form with this demo.
	 * 
	 */
	private void createTables() {
		try {
			String stmt = null;
			stmt = CREATE_TABLE_RESOURCE;
			update(stmt);
		} catch (SQLException e) {
			if (e.toString().indexOf("Table already exists") == -1)
				throw new RuntimeException(e);
		}
		try {
			String stmt = null;
			stmt = CREATE_TABLE_USER;
			update(stmt);
		} catch (SQLException e) {
			if (e.toString().indexOf("Table already exists") == -1)
				throw new RuntimeException(e);
		}
		try {
			String stmt = null;
			stmt = CREATE_TABLE_RESERVATION;
			update(stmt);
		} catch (SQLException e) {
			if (e.toString().indexOf("Table already exists") == -1)
				throw new RuntimeException(e);
		}
	}

	/**
	 * Test database connection with simple SELECT command.
	 * 
	 */
	private String testDatabase() {
		String result = null;
		try {
			Statement stmt = connection.createStatement(
					ResultSet.TYPE_SCROLL_INSENSITIVE,
					ResultSet.CONCUR_UPDATABLE);
			ResultSet rs = stmt.executeQuery("SELECT COUNT(*) FROM "
					+ Resource.TABLE);
			rs.next();
			result = "rowcount for table test is " + rs.getObject(1).toString();
			stmt.close();
		} catch (SQLException e) {
			throw new RuntimeException(e);
		}
		return result;
	}

	public Connection getConnection() {
		return connection;
	}

	public Container getCategories() {
		// TODO where deleted=?
		String q = "SELECT DISTINCT(" + Resource.PROPERTY_ID_CATEGORY
				+ ") FROM " + Resource.TABLE + " ORDER BY "
				+ Resource.PROPERTY_ID_CATEGORY;
		try {
			return new QueryContainer(q, connection,
					ResultSet.TYPE_SCROLL_INSENSITIVE,
					ResultSet.CONCUR_READ_ONLY);
		} catch (SQLException e) {
			throw new RuntimeException(e);
		}

	}

	public Container getResources(String category) {
		// TODO where deleted=?
		String q = "SELECT * FROM " + Resource.TABLE;
		if (category != null) {
			q += " WHERE " + Resource.PROPERTY_ID_CATEGORY + "='" + category
					+ "'"; // FIXME ->
			// PreparedStatement!
		}

		try {
			return new QueryContainer(q, connection,
					ResultSet.TYPE_SCROLL_INSENSITIVE,
					ResultSet.CONCUR_READ_ONLY);
		} catch (SQLException e) {
			throw new RuntimeException(e);
		}

	}

	public Container getReservations(List resources) {
		// TODO where reserved_by=?
		// TODO where from=?
		// TODO where to=?
		// TODO where deleted=?
		String q = "SELECT * FROM " + Reservation.TABLE + "," + Resource.TABLE;
		q += " WHERE " + Reservation.PROPERTY_ID_RESOURCE_ID + "="
				+ Resource.PROPERTY_ID_ID;
		if (resources != null && resources.size() > 0) {
			StringBuilder s = new StringBuilder();
			for (Iterator it = resources.iterator(); it.hasNext();) {
				if (s.length() > 0) {
					s.append(",");
				}
				s.append(((Item) it.next())
						.getItemProperty(Resource.PROPERTY_ID_ID));
			}
			q += " HAVING " + Reservation.PROPERTY_ID_RESOURCE_ID + " IN (" + s
					+ ")";
		}
		q += " ORDER BY " + Reservation.PROPERTY_ID_RESERVED_FROM;
		try {
			QueryContainer qc = new QueryContainer(q, connection,
					ResultSet.TYPE_SCROLL_INSENSITIVE,
					ResultSet.CONCUR_READ_ONLY);
			if (qc.size() < 1) {
				return null;
			} else {
				return qc;
			}
		} catch (SQLException e) {
			throw new RuntimeException(e);
		}
	}

	public void addReservation(Item resource, int reservedById,
			Date reservedFrom, Date reservedTo, String description) {
		if (reservedFrom.after(reservedTo)) {
			Date tmp = reservedTo;
			reservedTo = reservedFrom;
			reservedFrom = tmp;
		}
		int resourceId = ((Integer) resource.getItemProperty(
				Resource.PROPERTY_ID_ID).getValue()).intValue();
		String q = "INSERT INTO " + Reservation.TABLE + " ("
				+ Reservation.PROPERTY_ID_RESOURCE_ID + ","
				+ Reservation.PROPERTY_ID_RESERVED_BY_ID + ","
				+ Reservation.PROPERTY_ID_RESERVED_FROM + ","
				+ Reservation.PROPERTY_ID_RESERVED_TO + ","
				+ Reservation.PROPERTY_ID_DESCRIPTION + ")"
				+ "VALUES (?,?,?,?,?)";
		synchronized (DB_URL) {
			try {
				if (!isAvailableResource(resourceId, reservedFrom, reservedTo)) {
					throw new ResourceNotAvailableException(
							"The resource is not available at that time.");
				}
				PreparedStatement p = connection.prepareStatement(q);
				p.setInt(1, resourceId);
				p.setInt(2, reservedById);
				p.setTimestamp(3,
						new java.sql.Timestamp(reservedFrom.getTime()));
				p.setTimestamp(4, new java.sql.Timestamp(reservedTo.getTime()));
				p.setString(5, description);
				p.execute();
			} catch (Exception e) {
				throw new RuntimeException(e);
			}
		}
	}

	public boolean isAvailableResource(int resourceId, Date reservedFrom,
			Date reservedTo) {
		// TODO where deleted=?
		if (reservedFrom.after(reservedTo)) {
			Date tmp = reservedTo;
			reservedTo = reservedFrom;
			reservedFrom = tmp;
		}
		String checkQ = "SELECT count(*) FROM " + Reservation.TABLE + " WHERE "
				+ Reservation.PROPERTY_ID_RESOURCE_ID + "=? AND (("
				+ Reservation.PROPERTY_ID_RESERVED_FROM + ">=? AND "
				+ Reservation.PROPERTY_ID_RESERVED_FROM + "<=?) OR ("
				+ Reservation.PROPERTY_ID_RESERVED_TO + ">=? AND "
				+ Reservation.PROPERTY_ID_RESERVED_TO + "<=?) OR ("
				+ Reservation.PROPERTY_ID_RESERVED_FROM + "<=? AND "
				+ Reservation.PROPERTY_ID_RESERVED_TO + ">=?)" + ")";
		try {
			PreparedStatement p = connection.prepareStatement(checkQ);
			p.setInt(1, resourceId);
			p.setTimestamp(2, new java.sql.Timestamp(reservedFrom.getTime()));
			p.setTimestamp(3, new java.sql.Timestamp(reservedTo.getTime()));
			p.setTimestamp(4, new java.sql.Timestamp(reservedFrom.getTime()));
			p.setTimestamp(5, new java.sql.Timestamp(reservedTo.getTime()));
			p.setTimestamp(6, new java.sql.Timestamp(reservedFrom.getTime()));
			p.setTimestamp(7, new java.sql.Timestamp(reservedTo.getTime()));
			p.execute();
			ResultSet rs = p.getResultSet();
			if (rs.next() && rs.getInt(1) > 0) {
				return false;
			}
		} catch (Exception e) {
			throw new RuntimeException(e);
		}
		return true;
	}

	public Container getUsers() {
		// TODO where deleted=?
		String q = "SELECT * FROM " + User.TABLE + " ORDER BY "
				+ User.PROPERTY_ID_FULLNAME;
		try {
			QueryContainer qc = new QueryContainer(q, connection,
					ResultSet.TYPE_SCROLL_INSENSITIVE,
					ResultSet.CONCUR_READ_ONLY);
			return qc;
		} catch (SQLException e) {
			throw new RuntimeException(e);
		}
	}

	public void generateResources() {
		/*
		 * map.addMarker("Old Mill", new Point.Double(60.452224f,22.299929f));
		 * map.addMarker("ICT", new Point.Double(60.449007f,22.295508f));
		 * map.addMarker("DataCity", new Point.Double(60.448329f,22.295165f));
		 * map.addMarker("BioCity", new Point.Double(60.449451f,22.293105f));
		 * map.addMarker("PharmaCity", new Point.Double(60.44888f,22.292032f));
		 * map.addMarker("Intelligate", new Point.Double(60.450403f,22.29495f));
		 * map.addMarker("Paviljonki", new Point.Double(60.445408f,22.290831f));
		 * map.addMarker("Trivium", new Point.Double(60.44641962165445f,
		 * 22.301753170493f)); map.addMarker("Linja-auto asema", new
		 * Point.Double(60.457049f,22.267957f));
		 */
		/*
		 * Object[][] resources = { { "IT Mill Toolkit Manual", "the manual",
		 * "Books", new Double(60.452224),new Double(22.299929)}, { "IT Mill
		 * Toolkit for Dummies", "the hardcover version", "Books", new
		 * Double(60.452224),new Double(22.299929) }, { "Sony", "Old Sony video
		 * projector", "AV equipment", new Double(60.449007),new
		 * Double(22.295508) }, { "Sanyo", "Brand new hd-ready video projector",
		 * "AV equipment", new Double(60.452224),new Double(22.299929) }, {
		 * "Room 7", "Conference room in the lobby", "Conference rooms", new
		 * Double(60.449451),new Double(22.292032) }, { "Nintendo Wii", "Teh
		 * uber fun", "Entertainment", new Double(60.445408),new
		 * Double(22.290831) }, { "Playstation", "We don't actually have one",
		 * "Entertainment", new Double(60.44641962165445),new
		 * Double(22.301753170493) }, { "Luokkahuone", "Classroom right next to
		 * IT Mill", "Conference rooms", new Double(60.44888),new
		 * Double(22.292032) } };
		 */
		Object[][] resources = {
				// Turku
				{ "01 Ford Mondeo", "w/ company logo", "Turku",
						new Double(60.510857), new Double(22.275424) },
				{ "02 Citroen Jumper", "w/ company logo. 12m3 storage space.",
						"Turku", new Double(60.452171), new Double(22.2995) },
				{ "03 Saab 93", "Cabriolet<br/>Keys from the rental desk.",
						"Turku", new Double(60.4507), new Double(22.295551) },
				{ "04 Volvo S60", "Key from the rental desk.", "Turku",
						new Double(60.434722), new Double(22.224398) },
				{ "05 Smart fourtwo", "Cabrio. Keys from infodesk.", "Turku",
						new Double(60.508970), new Double(22.264790) },
				// Helsinki
				{ "06 Smart fourtwo", "Cabrio. Keys from infodesk.",
						"Helsinki", new Double(60.17175), new Double(24.939029) },
				{ "07 Smart fourtwo", "Cabrio. Keys from infodesk.",
						"Helsinki", new Double(60.17175), new Double(24.939029) },
				{ "08 Smart fourtwo", "Cabrio. Keys from infodesk.",
						"Helsinki", new Double(60.166579),
						new Double(24.953899) },
				{ "09 Volvo S60", "Keys from infodesk.", "Helsinki",
						new Double(60.317832), new Double(24.967289) },
				{ "10 Saab 93", "Keys from infodesk.", "Helsinki",
						new Double(60.249193), new Double(25.045921) },
				// Silicon Valley
				{ "11 Ford Mustang", "Keys from Acme clerk.", "Silicon Valley",
						new Double(37.615853), new Double(-122.386384) },
				{ "12 Ford Mondeo", "Keys from infodesk.", "Silicon Valley",
						new Double(37.365028), new Double(-121.922654) },
				{ "13 Ford Mustang", "GT Cal Special. Keys from guard.",
						"Silicon Valley", new Double(37.403812),
						new Double(-121.977425) },
				{ "14 Ford Mustang", "GT Cal Special. Keys from guard.",
						"Silicon Valley", new Double(37.403812),
						new Double(-121.977425) },
				{ "15 Land Rover", "Keys from infodesk.", "Silicon Valley",
						new Double(37.365028), new Double(-121.922654) },
				// Paris
				{ "17 Peugeot 308", "Keys from infodesk.", "Paris",
						new Double(48.844756), new Double(2.372784) },
				{ "18 Citroën C6", "Keys from rental desk.", "Paris",
						new Double(49.007253), new Double(2.545025) },
				{ "19 Citroën C6", "Keys from infodesk.", "Paris",
						new Double(48.729061), new Double(2.368087) },
				{ "20 Peugeot 308", "Keys from ticket sales.", "Paris",
						new Double(48.880931), new Double(2.356988) },
				{ "21 Peugeot 308", "Keys from ticket sales.", "Paris",
						new Double(48.876479), new Double(2.358161) },
				// STHLM
				{ "22 Volvo S60", "Keys from infodesk.", "Stockholm",
						new Double(59.350414), new Double(18.106574) },
				{ "23 Saab 93", "Keys from infodesk.", "Stockholm",
						new Double(59.355905), new Double(17.946784) },
				{ "24 Smart fourtwo", "Keys from infodesk.", "Stockholm",
						new Double(59.315939), new Double(18.095904) },
				{ "25 Smart fourtwo", "Keys from infodesk.", "Stockholm",
						new Double(59.330716), new Double(18.058702) },
				// Boston
				{ "26 Ford Mustang", "Keys from infodesk.", "Boston",
						new Double(42.366588), new Double(-71.020955) },
				{ "27 Smart fourtwo", "Keys from infodesk.", "Boston",
						new Double(42.365419), new Double(-71.061748) },
				{ "28 Volvo S60", "Keys from Seaport Hotel reception.",
						"Boston", new Double(42.34811), new Double(-71.041127) },
				{ "29 Smart fourtwo", "Keys from Seaport Hotel reception.",
						"Boston", new Double(42.348072), new Double(-71.041315) },

		};

		String q = "INSERT INTO " + Resource.TABLE + "("
				+ Resource.PROPERTY_ID_NAME + ","
				+ Resource.PROPERTY_ID_DESCRIPTION + ","
				+ Resource.PROPERTY_ID_CATEGORY + ","
				+ Resource.PROPERTY_ID_LOCATIONX + ","
				+ Resource.PROPERTY_ID_LOCATIONY + ")" + " VALUES (?,?,?,?,?)";
		try {
			PreparedStatement stmt = connection.prepareStatement(q);
			for (int i = 0; i < resources.length; i++) {
				int j = 0;
				stmt.setString(j + 1, (String) resources[i][j++]);
				stmt.setString(j + 1, (String) resources[i][j++]);
				stmt.setString(j + 1, (String) resources[i][j++]);
				stmt.setDouble(j + 1, ((Double) resources[i][j++])
						.doubleValue());
				stmt.setDouble(j + 1, ((Double) resources[i][j++])
						.doubleValue());
				stmt.execute();
			}
		} catch (SQLException e) {
			throw new RuntimeException(e);
		}
	}

	public void generateDemoUser() {
		String q = "INSERT INTO USER (" + User.PROPERTY_ID_FULLNAME + ","
				+ User.PROPERTY_ID_EMAIL + "," + User.PROPERTY_ID_PASSWORD
				+ ") VALUES (?,?,?)";
		try {
			PreparedStatement stmt = connection.prepareStatement(q);
			stmt.setString(1, "Demo User");
			stmt.setString(2, "demo.user@itmill.com");
			stmt.setString(3, "demo");
			stmt.execute();
		} catch (SQLException e) {
			throw new RuntimeException(e);
		}

	}

}