|
|
|
|
@ -20,182 +20,39 @@ package org.wso2.carbon.device.application.mgt.core.dao.common;
|
|
|
|
|
|
|
|
|
|
import org.apache.commons.logging.Log;
|
|
|
|
|
import org.apache.commons.logging.LogFactory;
|
|
|
|
|
import org.wso2.carbon.device.application.mgt.common.exception.IllegalTransactionStateException;
|
|
|
|
|
import org.wso2.carbon.device.application.mgt.common.exception.TransactionManagementException;
|
|
|
|
|
import org.wso2.carbon.device.application.mgt.common.exception.UnsupportedDatabaseEngineException;
|
|
|
|
|
import org.wso2.carbon.device.application.mgt.core.config.datasource.DataSourceConfig;
|
|
|
|
|
import org.wso2.carbon.device.application.mgt.core.config.datasource.JNDILookupDefinition;
|
|
|
|
|
import org.wso2.carbon.device.application.mgt.core.dao.impl.GenericAppManagementDAO;
|
|
|
|
|
import org.wso2.carbon.device.application.mgt.core.util.ApplicationManagerConstants;
|
|
|
|
|
import org.wso2.carbon.device.application.mgt.core.util.ConnectionManagerUtil;
|
|
|
|
|
import org.wso2.carbon.device.application.mgt.core.util.ConnectionManagerUtil.DatabaseType;
|
|
|
|
|
|
|
|
|
|
import javax.sql.DataSource;
|
|
|
|
|
import java.sql.Connection;
|
|
|
|
|
import java.sql.SQLException;
|
|
|
|
|
import java.util.Hashtable;
|
|
|
|
|
import java.util.List;
|
|
|
|
|
|
|
|
|
|
public class ApplicationManagementDAOFactory {
|
|
|
|
|
|
|
|
|
|
public static final String H2 = "H2";
|
|
|
|
|
private DatabaseType databaseType;
|
|
|
|
|
private static DataSource dataSource;
|
|
|
|
|
private static String databaseEngine;
|
|
|
|
|
|
|
|
|
|
private static ThreadLocal<Connection> currentConnection = new ThreadLocal<Connection>();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private static final Log log = LogFactory.getLog(ApplicationManagementDAOFactory.class);
|
|
|
|
|
|
|
|
|
|
public ApplicationManagementDAOFactory(DataSourceConfig dataSourceConfig) {
|
|
|
|
|
dataSource = ConnectionManagerUtil.resolveDataSource(dataSourceConfig);
|
|
|
|
|
ConnectionManagerUtil.setDataSource(dataSource);
|
|
|
|
|
String databaseEngine = H2;
|
|
|
|
|
try {
|
|
|
|
|
databaseEngine = dataSource.getConnection().getMetaData().getDatabaseProductName();
|
|
|
|
|
} catch (SQLException e) {
|
|
|
|
|
log.error("Error occurred while retrieving config.datasource connection", e);
|
|
|
|
|
}
|
|
|
|
|
this.databaseType = DatabaseType.lookup(databaseEngine);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public ApplicationManagementDAO getApplicationManagementDAO(){
|
|
|
|
|
switch (databaseType) {
|
|
|
|
|
default:
|
|
|
|
|
return new GenericAppManagementDAO();
|
|
|
|
|
public static ApplicationManagementDAO getApplicationManagementDAO(){
|
|
|
|
|
if (databaseEngine != null) {
|
|
|
|
|
switch (databaseEngine) {
|
|
|
|
|
case ApplicationManagerConstants.DataBaseTypes.DB_TYPE_H2:
|
|
|
|
|
case ApplicationManagerConstants.DataBaseTypes.DB_TYPE_MYSQL:
|
|
|
|
|
return new GenericAppManagementDAO();
|
|
|
|
|
default:
|
|
|
|
|
throw new UnsupportedDatabaseEngineException("Unsupported database engine : " + databaseEngine);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
throw new IllegalStateException("Database engine has not initialized properly.");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
public static void init(DataSourceConfig config) {
|
|
|
|
|
dataSource = resolveDataSource(config);
|
|
|
|
|
try {
|
|
|
|
|
databaseEngine = dataSource.getConnection().getMetaData().getDatabaseProductName();
|
|
|
|
|
} catch (SQLException e) {
|
|
|
|
|
log.error("Error occurred while retrieving config.datasource connection", e);
|
|
|
|
|
}
|
|
|
|
|
ConnectionManagerUtil.resolveDataSource(config);
|
|
|
|
|
databaseEngine = ConnectionManagerUtil.getDatabaseType();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static void init(DataSource dtSource) {
|
|
|
|
|
dataSource = dtSource;
|
|
|
|
|
try {
|
|
|
|
|
databaseEngine = dataSource.getConnection().getMetaData().getDatabaseProductName();
|
|
|
|
|
} catch (SQLException e) {
|
|
|
|
|
log.error("Error occurred while retrieving config.datasource connection", e);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static void beginTransaction() throws TransactionManagementException {
|
|
|
|
|
Connection conn = currentConnection.get();
|
|
|
|
|
if (conn != null) {
|
|
|
|
|
throw new IllegalTransactionStateException("A transaction is already active within the context of " +
|
|
|
|
|
"this particular thread. Therefore, calling 'beginTransaction/openConnection' while another " +
|
|
|
|
|
"transaction is already active is a sign of improper transaction handling");
|
|
|
|
|
}
|
|
|
|
|
try {
|
|
|
|
|
conn = dataSource.getConnection();
|
|
|
|
|
conn.setAutoCommit(false);
|
|
|
|
|
currentConnection.set(conn);
|
|
|
|
|
} catch (SQLException e) {
|
|
|
|
|
throw new TransactionManagementException("Error occurred while retrieving config.datasource connection", e);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static void openConnection() throws SQLException {
|
|
|
|
|
Connection conn = currentConnection.get();
|
|
|
|
|
if (conn != null) {
|
|
|
|
|
throw new IllegalTransactionStateException("A transaction is already active within the context of " +
|
|
|
|
|
"this particular thread. Therefore, calling 'beginTransaction/openConnection' while another " +
|
|
|
|
|
"transaction is already active is a sign of improper transaction handling");
|
|
|
|
|
}
|
|
|
|
|
conn = dataSource.getConnection();
|
|
|
|
|
currentConnection.set(conn);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static Connection getConnection() throws SQLException {
|
|
|
|
|
Connection conn = currentConnection.get();
|
|
|
|
|
if (conn == null) {
|
|
|
|
|
throw new IllegalTransactionStateException("No connection is associated with the current transaction. " +
|
|
|
|
|
"This might have ideally been caused by not properly initiating the transaction via " +
|
|
|
|
|
"'beginTransaction'/'openConnection' methods");
|
|
|
|
|
}
|
|
|
|
|
return conn;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static void commitTransaction() {
|
|
|
|
|
Connection conn = currentConnection.get();
|
|
|
|
|
if (conn == null) {
|
|
|
|
|
throw new IllegalTransactionStateException("No connection is associated with the current transaction. " +
|
|
|
|
|
"This might have ideally been caused by not properly initiating the transaction via " +
|
|
|
|
|
"'beginTransaction'/'openConnection' methods");
|
|
|
|
|
}
|
|
|
|
|
try {
|
|
|
|
|
conn.commit();
|
|
|
|
|
} catch (SQLException e) {
|
|
|
|
|
log.error("Error occurred while committing the transaction", e);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static void rollbackTransaction() {
|
|
|
|
|
Connection conn = currentConnection.get();
|
|
|
|
|
if (conn == null) {
|
|
|
|
|
throw new IllegalTransactionStateException("No connection is associated with the current transaction. " +
|
|
|
|
|
"This might have ideally been caused by not properly initiating the transaction via " +
|
|
|
|
|
"'beginTransaction'/'openConnection' methods");
|
|
|
|
|
}
|
|
|
|
|
try {
|
|
|
|
|
conn.rollback();
|
|
|
|
|
} catch (SQLException e) {
|
|
|
|
|
log.warn("Error occurred while roll-backing the transaction", e);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static void closeConnection() {
|
|
|
|
|
Connection conn = currentConnection.get();
|
|
|
|
|
if (conn == null) {
|
|
|
|
|
throw new IllegalTransactionStateException("No connection is associated with the current transaction. " +
|
|
|
|
|
"This might have ideally been caused by not properly initiating the transaction via " +
|
|
|
|
|
"'beginTransaction'/'openConnection' methods");
|
|
|
|
|
}
|
|
|
|
|
try {
|
|
|
|
|
conn.close();
|
|
|
|
|
} catch (SQLException e) {
|
|
|
|
|
log.warn("Error occurred while close the connection");
|
|
|
|
|
}
|
|
|
|
|
currentConnection.remove();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Resolve data source from the data source definition
|
|
|
|
|
*
|
|
|
|
|
* @param config data source configuration
|
|
|
|
|
* @return data source resolved from the data source definition
|
|
|
|
|
*/
|
|
|
|
|
private static DataSource resolveDataSource(DataSourceConfig config) {
|
|
|
|
|
DataSource dataSource = null;
|
|
|
|
|
if (config == null) {
|
|
|
|
|
throw new RuntimeException(
|
|
|
|
|
"Device Management Repository data source configuration " + "is null and " +
|
|
|
|
|
"thus, is not initialized");
|
|
|
|
|
}
|
|
|
|
|
JNDILookupDefinition jndiConfig = config.getJndiLookupDefinition();
|
|
|
|
|
if (jndiConfig != null) {
|
|
|
|
|
if (log.isDebugEnabled()) {
|
|
|
|
|
log.debug("Initializing Device Management Repository data source using the JNDI " +
|
|
|
|
|
"Lookup Definition");
|
|
|
|
|
}
|
|
|
|
|
List<JNDILookupDefinition.JNDIProperty> jndiPropertyList =
|
|
|
|
|
jndiConfig.getJndiProperties();
|
|
|
|
|
if (jndiPropertyList != null) {
|
|
|
|
|
Hashtable<Object, Object> jndiProperties = new Hashtable<Object, Object>();
|
|
|
|
|
for (JNDILookupDefinition.JNDIProperty prop : jndiPropertyList) {
|
|
|
|
|
jndiProperties.put(prop.getName(), prop.getValue());
|
|
|
|
|
}
|
|
|
|
|
dataSource = ApplicationManagementDAOUtil.lookupDataSource(jndiConfig.getJndiName(), jndiProperties);
|
|
|
|
|
} else {
|
|
|
|
|
dataSource = ApplicationManagementDAOUtil.lookupDataSource(jndiConfig.getJndiName(), null);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return dataSource;
|
|
|
|
|
ConnectionManagerUtil.setDataSource(dtSource);
|
|
|
|
|
databaseEngine = ConnectionManagerUtil.getDatabaseType();
|
|
|
|
|
}
|
|
|
|
|
}
|
Harshan Liyanage
7 years ago