Java tutorial
/* * Copyright 2014 Webpagebytes * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.webpagebytes.plugins; import java.beans.PropertyDescriptor; import java.lang.reflect.Field; import java.sql.Clob; import java.sql.Connection; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.SQLException; import java.sql.Timestamp; import java.sql.Types; import java.util.ArrayList; import java.util.Date; import java.util.HashMap; import java.util.HashSet; import java.util.List; import java.util.Map; import java.util.Set; import java.util.logging.Level; import java.util.logging.Logger; import org.apache.commons.dbcp.BasicDataSource; import com.webpagebytes.cms.cmsdata.WPBAdminFieldKey; import com.webpagebytes.cms.cmsdata.WPBAdminFieldStore; import com.webpagebytes.cms.cmsdata.WPBAdminFieldTextStore; import com.webpagebytes.cms.exception.WPBException; import com.webpagebytes.cms.exception.WPBSerializerException; /** * WPBSQLlDataStoreDao is a class that allows to read, write, delete and query records * from SQL like databases that run in the same operating system instance as the current * web server. */ public class WPBSQLDataStoreDao { private static final Logger log = Logger.getLogger(WPBSQLDataStoreDao.class.getName()); public enum WBSQLQueryOperator { LESS_THAN, GREATER_THAN, EQUAL, NOT_EQUAL, LESS_THAN_OR_EQUAL, GREATER_THAN_OR_EQUAL }; public enum WBSQLSortDirection { NO_SORT, ASCENDING, DESCENDING }; private BasicDataSource dataSource = new BasicDataSource(); private Map<String, String> dbProps = new HashMap<String, String>(); private static final String QUERY_RECORD = "SELECT * FROM %s WHERE %s=?"; private static final String QUERY_ALL_RECORDS = "SELECT * FROM %s"; public WPBSQLDataStoreDao(Map<String, String> dbProps) { this.dbProps.putAll(dbProps); dataSource.setDriverClassName(dbProps.get("driverClass")); dataSource.setUrl(dbProps.get("connectionUrl")); dataSource.setUsername(dbProps.get("userName")); dataSource.setPassword(dbProps.get("password")); String testOnBorrow = dbProps.get("testOnBorrow"); if (testOnBorrow != null && testOnBorrow.trim().toLowerCase().equals("true")) { dataSource.setTestOnBorrow(true); } String validationQuery = dbProps.get("validationQuery"); if (validationQuery != null) { dataSource.setValidationQuery(validationQuery); } } private Connection getConnection() throws SQLException { return dataSource.getConnection(); } /* * Given an instance of an object that has a particular property this method will set the object property with the * provided value. It assumes that the object has the setter method for the specified interface * @param object The object instance on which the property will be set * @param property The property name that will be set. This means that there is a setter public method defined for the * object instance * @param propertyValue The new value for the property that will be set * @throws WBSerializerException If the object property was not set with success */ public void setObjectProperty(Object object, String property, Object propertyValue) throws WPBSerializerException { try { PropertyDescriptor pd = new PropertyDescriptor(property, object.getClass()); pd.getWriteMethod().invoke(object, propertyValue); } catch (Exception e) { log.log(Level.SEVERE, "cannot setObjectProperty on " + property, e); throw new WPBSerializerException("Cannot set property for object", e); } } public Object getObjectProperty(Object object, String property) throws WPBSerializerException { try { PropertyDescriptor pd = new PropertyDescriptor(property, object.getClass()); return pd.getReadMethod().invoke(object); } catch (Exception e) { throw new WPBSerializerException("Cannot set property for object", e); } } public <T> boolean hasClassProperty(Class<T> kind, String property) throws WPBSerializerException { try { PropertyDescriptor pd = new PropertyDescriptor(property, kind); return pd.getReadMethod() != null; } catch (Exception e) { throw new WPBSerializerException("Cannot set property for object", e); } } private <T> T copyResultSetToObject(ResultSet resultSet, Class<T> kind) throws SQLException, WPBSerializerException { try { T result = kind.newInstance(); Field[] fields = kind.getDeclaredFields(); for (Field field : fields) { field.setAccessible(true); boolean storeField = (field.getAnnotation(WPBAdminFieldKey.class) != null) || (field.getAnnotation(WPBAdminFieldStore.class) != null) || (field.getAnnotation(WPBAdminFieldTextStore.class) != null); if (storeField) { String fieldName = field.getName(); String fieldNameUpperCase = field.getName().toUpperCase(); PropertyDescriptor pd = new PropertyDescriptor(fieldName, kind); // get the field type if (field.getType() == Long.class) { Long value = resultSet.getLong(fieldNameUpperCase); pd.getWriteMethod().invoke(result, value); } else if (field.getType() == String.class) { String value = resultSet.getString(fieldNameUpperCase); pd.getWriteMethod().invoke(result, value); } else if (field.getType() == Integer.class) { Integer value = resultSet.getInt(fieldNameUpperCase); pd.getWriteMethod().invoke(result, value); } else if (field.getType() == Date.class) { Timestamp ts = resultSet.getTimestamp(fieldNameUpperCase); Date value = new Date(ts.getTime()); pd.getWriteMethod().invoke(result, value); } } } return result; } catch (Exception e) { throw new WPBSerializerException("Cannot deserialize from Result Set", e); } } private <T> String getSQLStringForInsert(T obj, Set<String> fieldsToIgnore) { String sqlTemplate = "INSERT INTO %s (%s) values (%s)"; Class<? extends Object> kind = obj.getClass(); String tableName = kind.getSimpleName(); String listColumns = ""; String listParams = ""; Field[] fields = kind.getDeclaredFields(); for (Field field : fields) { boolean storeField = (field.getAnnotation(WPBAdminFieldKey.class) != null) || (field.getAnnotation(WPBAdminFieldStore.class) != null) || (field.getAnnotation(WPBAdminFieldTextStore.class) != null); if (storeField) { String fieldName = field.getName(); String fieldNameUpperCase = field.getName().toUpperCase(); if (fieldsToIgnore != null && fieldsToIgnore.contains(fieldName)) { continue; } listColumns = listColumns.concat(fieldNameUpperCase).concat(","); listParams = listParams.concat("?,"); } } if (listColumns.endsWith(",")) { listColumns = listColumns.substring(0, listColumns.length() - 1); } if (listParams.endsWith(",")) { listParams = listParams.substring(0, listParams.length() - 1); } return String.format(sqlTemplate, tableName, listColumns, listParams); } private String getSQLStringForUpdate(Object object, String keyFieldName) throws WPBSerializerException { String sqlTemplate = "UPDATE %s SET %s WHERE %s=?"; Class<? extends Object> kind = object.getClass(); String tableName = kind.getSimpleName(); String listColumns = ""; Field[] fields = kind.getDeclaredFields(); for (Field field : fields) { if (field.getAnnotation(WPBAdminFieldKey.class) != null) { continue; } boolean storeField = (field.getAnnotation(WPBAdminFieldKey.class) != null) || (field.getAnnotation(WPBAdminFieldStore.class) != null) || (field.getAnnotation(WPBAdminFieldTextStore.class) != null); if (storeField) { String fieldNameUpperCase = field.getName().toUpperCase(); listColumns = listColumns.concat(fieldNameUpperCase).concat("=?,"); } } if (listColumns.endsWith(",")) { listColumns = listColumns.substring(0, listColumns.length() - 1); } return String.format(sqlTemplate, tableName, listColumns, keyFieldName.toUpperCase()); } private <T> String getSQLStringForDelete(Class<T> kind, String keyFieldName) throws WPBSerializerException { String sqlTemplate = "DELETE FROM %s WHERE %s=?"; String tableName = kind.getSimpleName(); return String.format(sqlTemplate, tableName, keyFieldName); } private int buildStatementForInsertUpdate(Object obj, Set<String> ignoreFields, PreparedStatement preparedStatement, Connection connection) throws SQLException, WPBSerializerException { Class<? extends Object> kind = obj.getClass(); Field[] fields = kind.getDeclaredFields(); int fieldIndex = 0; for (int i = 0; i < fields.length; i++) { Field field = fields[i]; field.setAccessible(true); boolean storeField = (field.getAnnotation(WPBAdminFieldKey.class) != null) || (field.getAnnotation(WPBAdminFieldStore.class) != null) || (field.getAnnotation(WPBAdminFieldTextStore.class) != null); if (storeField) { String fieldName = field.getName(); if (ignoreFields != null && ignoreFields.contains(fieldName)) { continue; } fieldIndex = fieldIndex + 1; Object value = null; try { PropertyDescriptor pd = new PropertyDescriptor(fieldName, kind); value = pd.getReadMethod().invoke(obj); } catch (Exception e) { throw new WPBSerializerException("Cannot get property value", e); } if (field.getType() == Long.class) { Long valueLong = (Long) value; if (valueLong != null) { preparedStatement.setLong(fieldIndex, valueLong); } else { preparedStatement.setNull(fieldIndex, Types.BIGINT); } } else if (field.getType() == String.class) { String valueString = (String) value; if (field.getAnnotation(WPBAdminFieldStore.class) != null || field.getAnnotation(WPBAdminFieldKey.class) != null) { if (valueString != null) { preparedStatement.setString(fieldIndex, valueString); } else { preparedStatement.setNull(fieldIndex, Types.VARCHAR); } } else if (field.getAnnotation(WPBAdminFieldTextStore.class) != null) { if (valueString != null) { Clob clob = connection.createClob(); clob.setString(1, valueString); preparedStatement.setClob(fieldIndex, clob); } else { preparedStatement.setNull(fieldIndex, Types.CLOB); } } } else if (field.getType() == Integer.class) { Integer valueInt = (Integer) value; if (valueInt != null) { preparedStatement.setInt(fieldIndex, valueInt); } else { preparedStatement.setNull(fieldIndex, Types.INTEGER); } } else if (field.getType() == Date.class) { Date date = (Date) value; if (date != null) { java.sql.Timestamp sqlDate = new java.sql.Timestamp(date.getTime()); preparedStatement.setTimestamp(fieldIndex, sqlDate); } else { preparedStatement.setNull(fieldIndex, Types.DATE); } } } } return fieldIndex; } public <T> T getRecord(Class<T> kind, String keyFieldName, Object keyValue) throws SQLException, WPBException { Connection con = getConnection(); ResultSet resultSet = null; PreparedStatement statement = null; try { String table = kind.getSimpleName(); statement = con.prepareStatement(String.format(QUERY_RECORD, table, keyFieldName)); setPrepareStatementParameter(statement, 1, keyValue); resultSet = statement.executeQuery(); if (resultSet.next()) { T obj = copyResultSetToObject(resultSet, kind); return obj; } else { return null; } } catch (SQLException e) { throw e; } finally { if (statement != null) { statement.close(); } if (resultSet != null) { resultSet.close(); } con.close(); } } private String operatorToString(WBSQLQueryOperator operator) { String operation = ""; switch (operator) { case EQUAL: operation = "="; break; case NOT_EQUAL: operation = "!="; break; case GREATER_THAN: operation = ">"; break; case LESS_THAN: operation = "<"; break; case GREATER_THAN_OR_EQUAL: operation = ">="; break; case LESS_THAN_OR_EQUAL: operation = "<="; break; } return operation; } private String sortDirectionToString(WBSQLSortDirection sortDir) { String dir = ""; switch (sortDir) { case ASCENDING: dir = "ASC"; break; case DESCENDING: dir = "DESC"; break; case NO_SORT: dir = ""; } return dir; } private <T> List<T> advanceQuery(Class<T> kind, Set<String> propertyNames, Map<String, WBSQLQueryOperator> operators, Map<String, Object> values, String sortProperty, WBSQLSortDirection sortDirection) throws SQLException, WPBSerializerException { List<String> propertiesList = new ArrayList<String>(); propertiesList.addAll(propertyNames); String condition = ""; for (String property : propertiesList) { if (!hasClassProperty(kind, property)) { throw new SQLException("queryProperty value does not match a field of given Class kind"); } if (condition.length() > 0) { condition = condition.concat("AND"); } condition = condition.concat("(").concat(property).concat(operatorToString(operators.get(property))) .concat("?").concat(")"); } String sort = ""; if (sortProperty != null) { if (!hasClassProperty(kind, sortProperty)) { throw new SQLException("sortProperty value does not match a field of given Class kind"); } if (sortDirection != null && sortDirection != WBSQLSortDirection.NO_SORT) { sort = String.format("ORDER BY %s %s", sortProperty, sortDirectionToString(sortDirection)); } } String tableName = kind.getSimpleName(); String queryString = ""; if (condition.length() > 0) { queryString = String.format("SELECT * FROM %s WHERE %s %s", tableName, condition, sort); } else { queryString = String.format("SELECT * FROM %s %s", tableName, sort); } Connection connection = getConnection(); PreparedStatement preparedStatement = null; try { List<T> records = new ArrayList<T>(); preparedStatement = connection.prepareStatement(queryString); for (int i = 0; i < propertiesList.size(); i++) { setPrepareStatementParameter(preparedStatement, i + 1, values.get(propertiesList.get(i))); } ResultSet resultSet = preparedStatement.executeQuery(); while (resultSet.next()) { T obj = copyResultSetToObject(resultSet, kind); records.add(obj); } resultSet.close(); return records; } catch (SQLException e) { throw e; } finally { if (preparedStatement != null) { preparedStatement.close(); } if (connection != null) { connection.close(); } } } private <T> boolean advanceDelete(Class<T> kind, Set<String> propertyNames, Map<String, WBSQLQueryOperator> operators, Map<String, Object> values) throws SQLException, WPBSerializerException { List<String> propertiesList = new ArrayList<String>(); propertiesList.addAll(propertyNames); String condition = ""; for (String property : propertiesList) { if (!hasClassProperty(kind, property)) { throw new SQLException("queryProperty value does not match a field of given Class kind"); } if (condition.length() > 0) { condition = condition.concat("AND"); } condition = condition.concat("(").concat(property).concat(operatorToString(operators.get(property))) .concat("?").concat(")"); } String tableName = kind.getSimpleName(); String queryString = String.format("DELETE FROM %s WHERE %s", tableName, condition); Connection connection = getConnection(); PreparedStatement preparedStatement = null; try { preparedStatement = connection.prepareStatement(queryString); for (int i = 0; i < propertiesList.size(); i++) { setPrepareStatementParameter(preparedStatement, i + 1, values.get(propertiesList.get(i))); } return preparedStatement.execute(); } catch (SQLException e) { throw e; } finally { if (preparedStatement != null) { preparedStatement.close(); } if (connection != null) { connection.close(); } } } public <T> List<T> query(Class<T> kind, Set<String> propertyNames, Map<String, WBSQLQueryOperator> operators, Map<String, Object> values) throws SQLException, WPBSerializerException { return advanceQuery(kind, propertyNames, operators, values, null, WBSQLSortDirection.NO_SORT); } public <T> boolean deleteRecords(Class<T> kind, Set<String> propertyNames, Map<String, WBSQLQueryOperator> operators, Map<String, Object> values) throws SQLException, WPBSerializerException { return advanceDelete(kind, propertyNames, operators, values); } @SuppressWarnings("rawtypes") public boolean deleteRecords(Class kind) throws SQLException { String tableName = kind.getSimpleName(); String queryString = String.format("DELETE FROM %s", tableName); Connection connection = getConnection(); PreparedStatement preparedStatement = null; try { preparedStatement = connection.prepareStatement(queryString); return preparedStatement.execute(); } catch (SQLException e) { throw e; } finally { if (preparedStatement != null) { preparedStatement.close(); } if (connection != null) { connection.close(); } } } public <T> List<T> queryWithSort(Class<T> kind, Set<String> propertyNames, Map<String, WBSQLQueryOperator> operators, Map<String, Object> values, String sortProperty, WBSQLSortDirection sortDirection) throws SQLException, WPBSerializerException { return advanceQuery(kind, propertyNames, operators, values, sortProperty, sortDirection); } public <T> List<T> getAllRecords(Class<T> kind) throws SQLException, WPBException { Connection con = getConnection(); PreparedStatement statement = null; List<T> objects = new ArrayList<T>(); try { String table = kind.getSimpleName(); statement = con.prepareStatement(String.format(QUERY_ALL_RECORDS, table)); ResultSet resultSet = statement.executeQuery(); while (resultSet.next()) { T obj = copyResultSetToObject(resultSet, kind); objects.add(obj); } resultSet.close(); } catch (SQLException e) { throw e; } finally { if (statement != null) { statement.close(); } con.close(); } return objects; } public <T> T addRecordWithKey(T object, String keyFieldName) throws SQLException, WPBSerializerException { Connection connection = getConnection(); PreparedStatement preparedStatement = null; try { Set<String> ignoreFields = new HashSet<String>(); String sqlStatement = getSQLStringForInsert(object, ignoreFields); connection.setAutoCommit(true); preparedStatement = connection.prepareStatement(sqlStatement); buildStatementForInsertUpdate(object, ignoreFields, preparedStatement, connection); preparedStatement.execute(); return object; } catch (SQLException e) { throw e; } finally { if (preparedStatement != null) { preparedStatement.close(); } connection.close(); } } public <T> T addRecord(T object, String keyFieldName) throws SQLException, WPBSerializerException { Connection connection = getConnection(); PreparedStatement preparedStatement = null; try { Set<String> ignoreFields = new HashSet<String>(); ignoreFields.add(keyFieldName); String sqlStatement = getSQLStringForInsert(object, ignoreFields); connection.setAutoCommit(true); preparedStatement = connection.prepareStatement(sqlStatement); buildStatementForInsertUpdate(object, ignoreFields, preparedStatement, connection); preparedStatement.execute(); ResultSet resultKey = preparedStatement.getGeneratedKeys(); if (resultKey.next()) { Long key = resultKey.getLong(1); setObjectProperty(object, keyFieldName, key); } return object; } catch (SQLException e) { throw e; } finally { if (preparedStatement != null) { preparedStatement.close(); } connection.close(); } } public void updateRecord(Object object, String keyFieldName) throws SQLException, WPBSerializerException { Connection connection = getConnection(); PreparedStatement preparedStatement = null; try { Set<String> ignoreFields = new HashSet<String>(); ignoreFields.add(keyFieldName); String sqlStatement = getSQLStringForUpdate(object, keyFieldName); connection.setAutoCommit(true); preparedStatement = connection.prepareStatement(sqlStatement); int fieldsCount = buildStatementForInsertUpdate(object, ignoreFields, preparedStatement, connection); Object keyValue = getObjectProperty(object, keyFieldName); setPrepareStatementParameter(preparedStatement, fieldsCount + 1, keyValue); preparedStatement.execute(); } catch (SQLException e) { throw e; } finally { if (preparedStatement != null) { preparedStatement.close(); } connection.close(); } } public <T> void deleteRecord(Class<T> kind, String fieldName, Object keyValue) throws SQLException, WPBSerializerException { Connection connection = getConnection(); PreparedStatement preparedStatement = null; try { String sqlStatement = getSQLStringForDelete(kind, fieldName); connection.setAutoCommit(true); preparedStatement = connection.prepareStatement(sqlStatement); setPrepareStatementParameter(preparedStatement, 1, keyValue); preparedStatement.execute(); } catch (SQLException e) { throw e; } finally { if (preparedStatement != null) { preparedStatement.close(); } connection.close(); } } private void setPrepareStatementParameter(PreparedStatement preparedStatement, int index, Object param) throws SQLException, WPBSerializerException { if (param.getClass().equals(Integer.class)) { preparedStatement.setInt(index, (Integer) param); } else if (param.getClass().equals(Long.class)) { preparedStatement.setLong(index, (Long) param); } else if (param.getClass().equals(String.class)) { preparedStatement.setString(index, (String) param); } else { throw new WPBSerializerException("Unsupported key type"); } } }