package com.wgcloud.config;
|
|
import org.apache.ibatis.mapping.DatabaseIdProvider;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.stereotype.Component;
|
|
import javax.sql.DataSource;
|
import java.sql.Connection;
|
import java.sql.SQLException;
|
import java.util.Properties;
|
|
/**
|
* @version v3.3
|
* @ClassName:MyDatabaseIdProvider.java
|
* @author: http://www.wgstart.com
|
* @date: 2021年12月23日
|
* @Description: 自定义一个数据库标识提供类，实现DatabaseIdProvider接口
|
* @Copyright: 2019-2021 wgcloud. All rights reserved.
|
*/
|
@Component
|
public class MyDatabaseIdProvider implements DatabaseIdProvider {
|
|
private static Logger logger = LoggerFactory.getLogger(MyDatabaseIdProvider.class);
|
|
@Override
|
public void setProperties(Properties p) {
|
logger.info(p.getProperty("MySQL"));
|
}
|
|
/**
|
* 用于mybatis里的时间查询条件使用，没有其他用途
|
*
|
* @param dataSource
|
* @return
|
* @throws SQLException
|
*/
|
@Override
|
public String getDatabaseId(DataSource dataSource) throws SQLException {
|
Connection conn = dataSource.getConnection();
|
String dbName = conn.getMetaData().getDatabaseProductName();
|
String dbAlias = "";
|
switch (dbName) {
|
case "MySQL":
|
dbAlias = "mysql";
|
break;
|
case "PostgreSQL":
|
dbAlias = "postgresql";
|
break;
|
case "MariaDB":
|
dbAlias = "mysql";
|
break;
|
case "Oracle":
|
dbAlias = "oracle";
|
break;
|
default:
|
break;
|
}
|
return dbAlias;
|
}
|
}
|
