com.jchanghong.db.dialect.DialectFactory Maven / Gradle / Ivy
The newest version!
package com.jchanghong.db.dialect;
import com.jchanghong.core.util.ClassLoaderUtil;
import com.jchanghong.core.util.StrUtil;
import com.jchanghong.db.dialect.impl.AnsiSqlDialect;
import com.jchanghong.db.dialect.impl.H2Dialect;
import com.jchanghong.db.dialect.impl.MysqlDialect;
import com.jchanghong.db.dialect.impl.OracleDialect;
import com.jchanghong.db.dialect.impl.PostgresqlDialect;
import com.jchanghong.db.dialect.impl.SqlServer2012Dialect;
import com.jchanghong.db.dialect.impl.Sqlite3Dialect;
import com.jchanghong.log.StaticLog;
import javax.sql.DataSource;
import java.sql.Connection;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
/**
* 方言工厂类
*
* @author loolly
*
*/
public class DialectFactory {
/** JDBC 驱动 MySQL */
public static final String DRIVER_MYSQL = "com.mysql.jdbc.Driver";
/** JDBC 驱动 MySQL，在6.X版本中变动驱动类名，且使用SPI机制 */
public static final String DRIVER_MYSQL_V6 = "com.mysql.cj.jdbc.Driver";
/** JDBC 驱动 Oracle */
public static final String DRIVER_ORACLE = "oracle.jdbc.OracleDriver";
/** JDBC 驱动 Oracle，旧版使用 */
public static final String DRIVER_ORACLE_OLD = "oracle.jdbc.driver.OracleDriver";
/** JDBC 驱动 PostgreSQL */
public static final String DRIVER_POSTGRESQL = "org.postgresql.Driver";
/** JDBC 驱动 SQLLite3 */
public static final String DRIVER_SQLLITE3 = "org.sqlite.JDBC";
/** JDBC 驱动 SQLServer */
public static final String DRIVER_SQLSERVER = "com.microsoft.sqlserver.jdbc.SQLServerDriver";
/** JDBC 驱动 Hive */
public static final String DRIVER_HIVE = "org.apache.hadoop.hive.jdbc.HiveDriver";
/** JDBC 驱动 Hive2 */
public static final String DRIVER_HIVE2 = "org.apache.hive.jdbc.HiveDriver";
/** JDBC 驱动 H2 */
public static final String DRIVER_H2 = "org.h2.Driver";
/** JDBC 驱动 Derby */
public static final String DRIVER_DERBY = "org.apache.derby.jdbc.AutoloadedDriver";
/** JDBC 驱动 HSQLDB */
public static final String DRIVER_HSQLDB = "org.hsqldb.jdbc.JDBCDriver";
/** JDBC 驱动 达梦7 */
public static final String DRIVER_DM7 = "dm.jdbc.driver.DmDriver";
private static final Map DIALECT_POOL = new ConcurrentHashMap<>();
private DialectFactory() {
}
/**
* 根据驱动名创建方言
* 驱动名是不分区大小写完全匹配的
*
* @param driverName JDBC驱动类名
* @return 方言
*/
public static Dialect newDialect(String driverName) {
final Dialect dialect = internalNewDialect(driverName);
StaticLog.debug("Use Dialect: [{}].", dialect.getClass().getSimpleName());
return dialect;
}
/**
* 根据驱动名创建方言
* 驱动名是不分区大小写完全匹配的
*
* @param driverName JDBC驱动类名
* @return 方言
*/
private static Dialect internalNewDialect(String driverName) {
if (StrUtil.isNotBlank(driverName)) {
if (DRIVER_MYSQL.equalsIgnoreCase(driverName) || DRIVER_MYSQL_V6.equalsIgnoreCase(driverName)) {
return new MysqlDialect();
} else if (DRIVER_ORACLE.equalsIgnoreCase(driverName) || DRIVER_ORACLE_OLD.equalsIgnoreCase(driverName)) {
return new OracleDialect();
} else if (DRIVER_SQLLITE3.equalsIgnoreCase(driverName)) {
return new Sqlite3Dialect();
} else if (DRIVER_POSTGRESQL.equalsIgnoreCase(driverName)) {
return new PostgresqlDialect();
} else if (DRIVER_H2.equalsIgnoreCase(driverName)) {
return new H2Dialect();
} else if (DRIVER_SQLSERVER.equalsIgnoreCase(driverName)) {
return new SqlServer2012Dialect();
}
}
// 无法识别可支持的数据库类型默认使用ANSI方言，可兼容大部分SQL语句
return new AnsiSqlDialect();
}
/**
* 通过JDBC URL等信息识别JDBC驱动名
*
* @param nameContainsProductInfo 包含数据库标识的字符串
* @return 驱动
*/
public static String identifyDriver(String nameContainsProductInfo) {
if (StrUtil.isBlank(nameContainsProductInfo)) {
return null;
}
// 全部转为小写，忽略大小写
nameContainsProductInfo = StrUtil.cleanBlank(nameContainsProductInfo.toLowerCase());
String driver = null;
if (nameContainsProductInfo.contains("mysql")) {
driver = ClassLoaderUtil.isPresent(DRIVER_MYSQL_V6) ? DRIVER_MYSQL_V6 : DRIVER_MYSQL;
} else if (nameContainsProductInfo.contains("oracle")) {
driver = ClassLoaderUtil.isPresent(DRIVER_ORACLE) ? DRIVER_ORACLE : DRIVER_ORACLE_OLD;
} else if (nameContainsProductInfo.contains("postgresql")) {
driver = DRIVER_POSTGRESQL;
} else if (nameContainsProductInfo.contains("sqlite")) {
driver = DRIVER_SQLLITE3;
} else if (nameContainsProductInfo.contains("sqlserver")) {
driver = DRIVER_SQLSERVER;
} else if (nameContainsProductInfo.contains("hive")) {
driver = DRIVER_HIVE;
} else if (nameContainsProductInfo.contains("h2")) {
driver = DRIVER_H2;
} else if (nameContainsProductInfo.contains("derby")) {
// 嵌入式Derby数据库
driver = DRIVER_DERBY;
} else if (nameContainsProductInfo.contains("hsqldb")) {
// HSQLDB
driver = DRIVER_HSQLDB;
} else if (nameContainsProductInfo.contains("dm")) {
// 达梦7
driver = DRIVER_DM7;
}
return driver;
}
/**
* 获取共享方言
* @param ds 数据源，每一个数据源对应一个唯一方言
* @return {@link Dialect}方言
*/
public static Dialect getDialect(DataSource ds) {
Dialect dialect = DIALECT_POOL.get(ds);
if(null == dialect) {
// 数据源作为锁的意义在于：不同数据源不会导致阻塞，相同数据源获取方言时可保证互斥
//noinspection SynchronizationOnLocalVariableOrMethodParameter
synchronized (ds) {
dialect = DIALECT_POOL.get(ds);
if(null == dialect) {
dialect = newDialect(ds);
DIALECT_POOL.put(ds, dialect);
}
}
}
return dialect;
}
/**
* 创建方言
*
* @param ds 数据源
* @return 方言
*/
public static Dialect newDialect(DataSource ds) {
return newDialect(DriverUtil.identifyDriver(ds));
}
/**
* 创建方言
*
* @param conn 数据库连接对象
* @return 方言
*/
public static Dialect newDialect(Connection conn) {
return newDialect(DriverUtil.identifyDriver(conn));
}
}