67 lines
1.9 KiB
Java
67 lines
1.9 KiB
Java
package com.xinwei.utils;
|
||
|
||
import java.sql.Connection;
|
||
import java.sql.SQLException;
|
||
|
||
import javax.sql.DataSource;
|
||
|
||
import org.apache.commons.lang3.StringUtils;
|
||
import org.hibernate.Hibernate;
|
||
import org.hibernate.dialect.H2Dialect;
|
||
import org.hibernate.dialect.MySQL5InnoDBDialect;
|
||
import org.hibernate.dialect.Oracle10gDialect;
|
||
|
||
public class Hibernates {
|
||
|
||
public static final String DATETIME_TYPE = "org.jadira.usertype.dateandtime.joda.PersistentDateTime";
|
||
|
||
/**
|
||
* Initialize the lazy property value.
|
||
*
|
||
* eg.
|
||
* Hibernates.initLazyProperty(user.getGroups());
|
||
*/
|
||
public static void initLazyProperty(Object proxyedPropertyValue) {
|
||
Hibernate.initialize(proxyedPropertyValue);
|
||
}
|
||
|
||
/**
|
||
* 从DataSoure中取出connection, 根据connection的metadata中的jdbcUrl判断Dialect类型.
|
||
* 仅支持Oracle, H2, MySql，如需更多数据库类型，请仿照此类自行编写。
|
||
*/
|
||
public static String getDialect(DataSource dataSource) {
|
||
String jdbcUrl = getJdbcUrlFromDataSource(dataSource);
|
||
|
||
// 根据jdbc url判断dialect
|
||
if (StringUtils.contains(jdbcUrl, ":h2:")) {
|
||
return H2Dialect.class.getName();
|
||
} else if (StringUtils.contains(jdbcUrl, ":mysql:")) {
|
||
return MySQL5InnoDBDialect.class.getName();
|
||
} else if (StringUtils.contains(jdbcUrl, ":oracle:")) {
|
||
return Oracle10gDialect.class.getName();
|
||
} else {
|
||
throw new IllegalArgumentException("Unknown Database of " + jdbcUrl);
|
||
}
|
||
}
|
||
|
||
private static String getJdbcUrlFromDataSource(DataSource dataSource) {
|
||
Connection connection = null;
|
||
try {
|
||
connection = dataSource.getConnection();
|
||
if (connection == null) {
|
||
throw new IllegalStateException("Connection returned by DataSource [" + dataSource + "] was null");
|
||
}
|
||
return connection.getMetaData().getURL();
|
||
} catch (SQLException e) {
|
||
throw new RuntimeException("Could not get database url", e);
|
||
} finally {
|
||
if (connection != null) {
|
||
try {
|
||
connection.close();
|
||
} catch (SQLException e) {
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|