mirror of
https://github.com/Alfresco/alfresco-community-repo.git
synced 2025-06-30 18:15:39 +00:00
71584: Merged V4.2-BUG-FIX (4.2.3) to HEAD-BUG-FIX (4.3/Cloud) 70325: Merged V4.1-BUG-FIX (4.1.9) to V4.2-BUG-FIX (4.2.3) 70145: MNT-8629: Merged DEV to V4.1-BUG-FIX (4.1.9) 70077: MNT-8629: Support requested for DB2 when schema name does not match DB user name. - Take into account schema name during update git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@74685 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261
131 lines
3.3 KiB
Java
131 lines
3.3 KiB
Java
/*
|
|
* Copyright (C) 2005-2014 Alfresco Software Limited.
|
|
*
|
|
* This file is part of Alfresco
|
|
*
|
|
* Alfresco is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Lesser General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Alfresco is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
* along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
package org.alfresco.util;
|
|
|
|
import java.sql.Connection;
|
|
import java.sql.DatabaseMetaData;
|
|
import java.sql.ResultSet;
|
|
|
|
import org.apache.commons.logging.Log;
|
|
import org.apache.commons.logging.LogFactory;
|
|
import org.hibernate.cfg.Configuration;
|
|
import org.springframework.orm.hibernate3.LocalSessionFactoryBean;
|
|
|
|
/**
|
|
* Helper class to collect all of our DatabaseMetaData interpretations in one place.
|
|
*
|
|
* @author sfrensley
|
|
*
|
|
*/
|
|
public class DatabaseMetaDataHelper {
|
|
|
|
private static Log logger = LogFactory.getLog(DatabaseMetaDataHelper.class);
|
|
|
|
private Configuration cfg;
|
|
|
|
public void setLocalSessionFactory(LocalSessionFactoryBean localSessionFactory)
|
|
{
|
|
this.cfg = localSessionFactory.getConfiguration();
|
|
}
|
|
|
|
/**
|
|
* Trys to determine the schema name from the DatabaseMetaData obtained from the Connection.
|
|
* @param connection A database connection
|
|
* @return
|
|
*/
|
|
private String getSchemaFromConnection(Connection connection)
|
|
{
|
|
|
|
if (connection == null) {
|
|
logger.error("Unable to determine schema due to null connection.");
|
|
return null;
|
|
}
|
|
|
|
ResultSet schemas = null;
|
|
|
|
try
|
|
{
|
|
final DatabaseMetaData dbmd = connection.getMetaData();
|
|
|
|
// Assume that if there are schemas, we want the one named after the connection user or the one called "dbo" (MS
|
|
// SQL hack)
|
|
String schema = null;
|
|
schemas = dbmd.getSchemas();
|
|
while (schemas.next())
|
|
{
|
|
final String thisSchema = schemas.getString("TABLE_SCHEM");
|
|
if (thisSchema.equals(dbmd.getUserName()) || thisSchema.equalsIgnoreCase("dbo"))
|
|
{
|
|
schema = thisSchema;
|
|
break;
|
|
}
|
|
}
|
|
return schema;
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
logger.error("Unable to determine current schema.",e);
|
|
}
|
|
finally
|
|
{
|
|
if (schemas != null)
|
|
{
|
|
try
|
|
{
|
|
schemas.close();
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
//noop
|
|
}
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public String getSchema(Connection connection)
|
|
{
|
|
String schema = null;
|
|
|
|
if (this.cfg != null)
|
|
{
|
|
String tmpSchema = this.cfg.getProperty("hibernate.default_schema");
|
|
if (tmpSchema != null && tmpSchema.trim().length() > 0)
|
|
{
|
|
schema = tmpSchema;
|
|
}
|
|
}
|
|
|
|
// if hibernate.default_schema was specified as a system property, then override previous value
|
|
String tmpSchema = System.getProperty("hibernate.default_schema");
|
|
if (tmpSchema != null && tmpSchema.length() > 0)
|
|
{
|
|
schema = tmpSchema;
|
|
}
|
|
|
|
if (schema == null)
|
|
{
|
|
schema = getSchemaFromConnection(connection);
|
|
}
|
|
|
|
return schema;
|
|
}
|
|
|
|
}
|