springboot mybatis(mybatis plus)实现不同数据库执行不同的SQL

 一.配置

第一种


    /**
     * 数据库配置
     *
     * @return 配置
     */
    @Bean
    public DatabaseIdProvider getDatabaseIdProvider() {
        DatabaseIdProvider databaseIdProvider = new VendorDatabaseIdProvider();
        Properties properties = new Properties();
        properties.setProperty("Oracle", "oracle");
        properties.setProperty("MySQL", "mysql");
        databaseIdProvider.setProperties(properties);
        return databaseIdProvider;
    }

第二种

 写一个实现类 实现DatabaseIdProvider类,实现它的两个方法,返回数据库类型

import org.apache.ibatis.mapping.DatabaseIdProvider;
import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Properties;
/**
 * 自定义一个数据库标示提供类,实现DatabaseIdProvider接口
 *
 */
@Component
public class MyDatabaseIdProvider implements DatabaseIdProvider {
    @Override
    public void setProperties(Properties p) {
        System.out.println(p.getProperty("Oracle"));
    }
    @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 "Oracle":
                dbAlias = "oracle";
                break;
            default:
                break;
        }
        return dbAlias;
    }
}

二.使用

1.在if中通过——databaseId来进行判断

2. 在 标签上的databaseId属性设置数据库类型,就可以根据数据库类型选择执行不同的SQL语句

三.原理

mybatis是根据数据源连接信息的metadata.getDatabaseProductName()来确定数据库类型,并加入环境变量中

 

  • 3
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
在 Spring Boot 中使用 Mybatis-Plus 获取数据库名称,可以先通过 `DataSource` 获取 `Connection` 对象,然后再从 `Connection` 对象中获取数据库名称。具体的代码如下: ```java import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import javax.sql.DataSource; import java.sql.Connection; import java.sql.SQLException; import java.util.regex.Matcher; import java.util.regex.Pattern; @Configuration public class MybatisPlusConfig { @Autowired private DataSource dataSource; /** * 配置 MybatisSqlSessionFactoryBean */ @Bean public MybatisSqlSessionFactoryBean mybatisSqlSessionFactoryBean() throws Exception { MybatisSqlSessionFactoryBean sessionFactory = new MybatisSqlSessionFactoryBean(); sessionFactory.setDataSource(dataSource); // 获取数据库名称 String dbName = getDatabaseName(dataSource); // 配置 Mybatis-Plus 的 GlobalConfig 和 Mybatis 配置文件 // ... return sessionFactory; } /** * 获取数据库名称 */ private String getDatabaseName(DataSource dataSource) throws SQLException { Connection connection = dataSource.getConnection(); String url = connection.getMetaData().getURL(); connection.close(); Pattern pattern = Pattern.compile("jdbc:mysql://[^/]+/([^?]+)"); Matcher matcher = pattern.matcher(url); if (matcher.find()) { return matcher.group(1); } else { throw new RuntimeException("Cannot get database name from JDBC URL"); } } } ``` 在上述代码中,我们通过 `dataSource.getConnection()` 获取了一个 `Connection` 对象,然后从中获取了 JDBC URL。接着,使用正则表达式从 JDBC URL 中提取出数据库名称,并返回。注意,在获取完 JDBC URL 后,需要立即关闭 `Connection` 对象,以释放资源。在实际使用中,需要根据不同数据库类型,使用不同的正则表达式来提取数据库名称。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值