Springboot 2.0 动态切换数据库,多数据源的实现

要制作快速开发框架需要动态切换数据库做增删改查操作,本文介绍在 Springboot 2.0 中制作的工具类
package com.cc.dynamicdb.utils;

import com.cc.dynamicdb.model.DBConfig;
import com.zaxxer.hikari.HikariDataSource;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @ClassName:DynamicDBUtils
 * @Description:依赖类:DBConfig
 * @Author:chanchaw
 * @Date:2019/11/13 15:45
 * @Version:1.0
 **/
@Service
public class DynamicDBUtils {

    /**
     * 查询返回记录集
     * @param dbConfig 链接数据库用到的5个参数构成的对象,例如:{ server:49.234.90.38,port:3306,dbName:atools,userName:root,userPwd:!Xdf123456 }
     * @param sql 要执行的SQL语句字串,已经将参数拼接在内了,不用传查询参数进来
     * @return
     */
    public List<Map<String,Object>> queryForList(DBConfig dbConfig,String sql){
        HikariDataSource dataSource = customDataSource(dbConfig);
        NamedParameterJdbcTemplate jdbcTemplate = new NamedParameterJdbcTemplate(dataSource);

        List<Map<String,Object>> list = jdbcTemplate.queryForList(sql,new HashMap<>());

        dataSource.close();
        return list;
    }


    /**
     * 查询返回记录集
     * @param dbConfig 链接数据库用到的5个参数构成的对象,例如:{ server:49.234.90.38,port:3306,dbName:atools,userName:root,userPwd:!Xdf123456 }
     * @param sql 查询SQL语句,带有占位符,例如:SELECT * FROM projects WHERE 1=1 AND iid>:iid; 最后的 :iid 是参数占位符
     * @param params Map<String,Object> 类型的参数,其中包含1 - N 个参数,键名称要和第二个参数的占位符保持一致
     * @return
     */
    public List<Map<String,Object>> queryForList(DBConfig dbConfig,String sql,Map<String,Object> params){
        HikariDataSource dataSource = customDataSource(dbConfig);
        NamedParameterJdbcTemplate jdbcTemplate = new NamedParameterJdbcTemplate(dataSource);

        List<Map<String,Object>> list = jdbcTemplate.queryForList(sql,new MapSqlParameterSource(params));

        dataSource.close();
        return list;
    }

    /**
     * 增/删/改 数据
     * @param dbConfig
     * @param sql
     * @return
     */
    public int update(DBConfig dbConfig,String sql){
        HikariDataSource dataSource = customDataSource(dbConfig);
        NamedParameterJdbcTemplate jdbcTemplate = new NamedParameterJdbcTemplate(dataSource);

        int ret = jdbcTemplate.update(sql,new HashMap<>());
        dataSource.close();
        return ret;
    }


    /**
     * 增/删/改 数据
     * @param dbConfig dbConfig 链接数据库用到的5个参数构成的对象,例如:{ server:49.234.90.38,port:3306,dbName:atools,userName:root,userPwd:!Xdf123456 }
     * @param sql 查询SQL语句,带有占位符,例如:SELECT * FROM projects WHERE 1=1 AND iid>:iid; 最后的 :iid 是参数占位符
     * @param params Map<String,Object> 类型的参数,其中包含1 - N 个参数,键名称要和第二个参数的占位符保持一致
     * @return
     */
    public int update(DBConfig dbConfig,String sql,Map<String,Object> params){
        HikariDataSource dataSource = customDataSource(dbConfig);
        NamedParameterJdbcTemplate jdbcTemplate = new NamedParameterJdbcTemplate(dataSource);

        int ret = jdbcTemplate.update(sql,new MapSqlParameterSource(params));
        dataSource.close();
        return ret;
    }



    public HikariDataSource customDataSource(DBConfig dbConfig) {

        final HikariDataSource customDataConfig = new HikariDataSource();

        customDataConfig.setJdbcUrl(dbConfig.getUrl());

        // 不设置驱动 - 采用默认驱动
        if( dbConfig.getDriver() == null||dbConfig.getDriver().length()<=0 ){
            customDataConfig.setDriverClassName("com.mysql.cj.jdbc.Driver");
        }else{
            customDataConfig.setDriverClassName(dbConfig.getDriver());
        }



        customDataConfig.setUsername(dbConfig.getDbUserName());
        customDataConfig.setPassword(dbConfig.getDbUserPwd());

        customDataConfig.setMaximumPoolSize(2);
        customDataConfig.setMinimumIdle(1);
        customDataConfig.setAutoCommit(true);
        //customDataConfig.setConnectionTestQuery("SELECT 1 FROM DUAL");
        customDataConfig.addDataSourceProperty("cachePrepStmts", "true");
        customDataConfig.addDataSourceProperty("prepStmtCacheSize", "250");
        customDataConfig.addDataSourceProperty("prepStmtCacheSqlLimit", "2048");

        return customDataConfig;
    }
}

还有个依赖类

package com.cc.dynamicdb.model;

/**
 * @ClassName:DBConfig
 * @Description:TODO
 * @Author:chanchaw
 * @Date:2019/11/13 8:16
 * @Version:1.0
 **/

public class DBConfig {
    private String server;
    private int port;
    private String dbName;
    private String driver;

    private String dbUserName;
    private String dbUserPwd;

    public String getServer() {
        return server;
    }

    public void setServer(String server) {
        this.server = server;
    }

    public int getPort() {
        return port;
    }

    public void setPort(int port) {
        this.port = port;
    }

    public String getDbName() {
        return dbName;
    }

    public void setDbName(String dbName) {
        this.dbName = dbName;
    }

    public String getDriver() {
        return driver;
    }

    public void setDriver(String driver) {
        this.driver = driver;
    }

    public String getDbUserName() {
        return dbUserName;
    }

    public void setDbUserName(String dbUserName) {
        this.dbUserName = dbUserName;
    }

    public String getDbUserPwd() {
        return dbUserPwd;
    }

    public void setDbUserPwd(String dbUserPwd) {
        this.dbUserPwd = dbUserPwd;
    }

    @Override
    public String toString() {
        return "DBConfig{" +
                "server='" + server + '\'' +
                ", port=" + port +
                ", dbName='" + dbName + '\'' +
                ", driver='" + driver + '\'' +
                ", dbUserName='" + dbUserName + '\'' +
                ", dbUserPwd='" + dbUserPwd + '\'' +
                '}';
    }

    /**
     * 构建数据库链接字串
     * @return
     */
    public String getUrl(){
        String ret = "jdbc:mysql://" + server + ":" + port + "/" + dbName + "?userUnicode=true&characterEncoding=utf-8&serverTimezone=UTC&useLegacyDatetimeCode=false";
        return ret;
    }
}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
在 Spring Boot 2.0 中配置多数据可以通过以下步骤实现: 1. 添加相关依赖:在 `pom.xml` 文件中添加 Spring Boot 对多数据的支持依赖,例如 `spring-boot-starter-data-jpa` 和 `mysql-connector-java`。 2. 配置数据:在 `application.properties` 或 `application.yml` 文件中配置数据信息。例如,定义两个数据的配置: ```properties # 第一个数据 spring.datasource.url=jdbc:mysql://localhost:3306/db1 spring.datasource.username=user1 spring.datasource.password=password1 spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver # 第二个数据 spring.datasource.secondary.url=jdbc:mysql://localhost:3306/db2 spring.datasource.secondary.username=user2 spring.datasource.secondary.password=password2 spring.datasource.secondary.driver-class-name=com.mysql.cj.jdbc.Driver ``` 3. 定义多数据配置类:创建一个配置类,用于配置多个数据和实体管理器工厂。可以使用 `@Configuration` 和 `@EnableJpaRepositories` 注解来自定义数据配置。 ```java @Configuration @EnableJpaRepositories(basePackages = "com.example.repository1", entityManagerFactoryRef = "entityManagerFactory1") public class DataSource1Config { // 配置第一个数据的相关配置,例如 DataSource、EntityManagerFactory、TransactionManager 等 @Bean(name = "dataSource1") @ConfigurationProperties(prefix = "spring.datasource") public DataSource dataSource() { return DataSourceBuilder.create().build(); } // ... } @Configuration @EnableJpaRepositories(basePackages = "com.example.repository2", entityManagerFactoryRef = "entityManagerFactory2") public class DataSource2Config { // 配置第二个数据的相关配置,例如 DataSource、EntityManagerFactory、TransactionManager 等 @Bean(name = "dataSource2") @ConfigurationProperties(prefix = "spring.datasource.secondary") public DataSource dataSource() { return DataSourceBuilder.create().build(); } // ... } ``` 4. 配置实体管理器工厂:在每个数据的配置类中,需要配置对应的实体管理器工厂。 ```java @Configuration @EnableJpaRepositories(basePackages = "com.example.repository1", entityManagerFactoryRef = "entityManagerFactory1") public class DataSource1Config { // ... @Bean(name = "entityManagerFactory1") public LocalContainerEntityManagerFactoryBean entityManagerFactory( EntityManagerFactoryBuilder builder, @Qualifier("dataSource1") DataSource dataSource) { return builder .dataSource(dataSource) .packages("com.example.entity1") .build(); } // ... } @Configuration @EnableJpaRepositories(basePackages = "com.example.repository2", entityManagerFactoryRef = "entityManagerFactory2") public class DataSource2Config { // ... @Bean(name = "entityManagerFactory2") public LocalContainerEntityManagerFactoryBean entityManagerFactory( EntityManagerFactoryBuilder builder, @Qualifier("dataSource2") DataSource dataSource) { return builder .dataSource(dataSource) .packages("com.example.entity2") .build(); } // ... } ``` 5. 使用多个数据:在需要使用数据的地方,可以通过 `@Qualifier` 注解指定要使用的数据。 ```java @Service public class MyService { @Autowired @Qualifier("entityManagerFactory1") private EntityManagerFactory entityManagerFactory1; @Autowired @Qualifier("entityManagerFactory2") private EntityManagerFactory entityManagerFactory2; // ... } ``` 以上就是在 Spring Boot 2.0 中配置多数据的基本步骤。需要注意配置类和数据的命名,以及注解的使用。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值