Springboot-HikariCP 實(shí)現(xiàn)動(dòng)態(tài)數(shù)據(jù)源切換

gitHub項(xiàng)目地址：https://github.com/shizhenshuang123/hikaricp-demo
開發(fā)環(huán)境：
jdk：JDK1.8+
gradle：Gradle4.6+
Spring：2.1.3.RELEASE+
當(dāng)數(shù)據(jù)庫數(shù)據(jù)達(dá)到一定數(shù)量的時(shí)候，數(shù)據(jù)庫的響應(yīng)將會(huì)有所緩慢，一般都會(huì)采取一些措施。例如，讀寫分離、分表分庫、主從服務(wù)、緩存技術(shù)等等。這里采用的是多庫多表。
test庫中user_info表：

image.png

test2庫中user_info表：
image.png

導(dǎo)入依賴
// https://mvnrepository.com/artifact/org.springframework.boot/spring-boot-starter-web
    compile group: 'org.springframework.boot', name: 'spring-boot-starter-web', version: '2.1.3.RELEASE'
    // https://mvnrepository.com/artifact/org.springframework.boot/spring-boot-starter-jdbc
    compile group: 'org.springframework.boot', name: 'spring-boot-starter-jdbc', version: '2.1.3.RELEASE'
    // https://mvnrepository.com/artifact/org.projectlombok/lombok
    compileOnly group: 'org.projectlombok', name: 'lombok', version: '1.18.6'
    // https://mvnrepository.com/artifact/mysql/mysql-connector-java
    compile group: 'mysql', name: 'mysql-connector-java', version: '8.0.15'
    //mybatis插件
    compile group: 'org.mybatis.generator', name: 'mybatis-generator-core', version: '1.3.5'
    compile group: 'tk.mybatis', name: 'mapper', version: '4.0.2'
    // https://mvnrepository.com/artifact/org.mybatis.spring.boot/mybatis-spring-boot-starter
    compile group: 'org.mybatis.spring.boot', name: 'mybatis-spring-boot-starter', version: '2.0.0'

HikariCP在spring-boot-starter-jdbc中已經(jīng)被引入（Spring默認(rèn)數(shù)據(jù)源）


image.png
項(xiàng)目結(jié)構(gòu)
image.png
配置文件配置（application.yaml）默認(rèn)數(shù)據(jù)源信息
spring:
  datasource:
    url: jdbc:mysql://127.0.0.1:3306/test?useUnicode=true&characterEncoding=utf-8&zeroDateTimeBehavior=CONVERT_TO_NULL&serverTimezone=Asia/Shanghai&allowMultiQueries=true
    username: root
    password: admin
    sql-script-encoding: UTF-8
讀取配置信息DataSourceConfig.java，創(chuàng)建默認(rèn)的數(shù)據(jù)源
package utry.hikaricp.config;

import com.zaxxer.hikari.HikariDataSource;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.transaction.annotation.EnableTransactionManagement;
import utry.hikaricp.DataSourceProvider;
import utry.hikaricp.info.DataSourceInfo;
import utry.hikaricp.source.MyHikariDataSource;

/**
 * @author szs
 * @date 2019/3/14 14:22
 */
@Configuration
@EnableTransactionManagement
public class DataSourceConfig {

    @Value("${spring.datasource.url}")
    private String url;

    @Value("${spring.datasource.username}")
    private String userName;

    @Value("${spring.datasource.password}")
    private String password;

    @Bean
    @ConditionalOnMissingBean(MyHikariDataSource.class)
    public MyHikariDataSource dataSource() {
        DataSourceInfo info = new DataSourceInfo();
        info.setUrl(url);
        info.setUsername(userName);
        info.setPassword(password);
        HikariDataSource dataSource = DataSourceProvider.create(info);
        MyHikariDataSource hikariCPDataSource = new MyHikariDataSource();
        hikariCPDataSource.updateDataSourceMap("1", dataSource);
        return hikariCPDataSource;
    }
}
通過DataSourceProvider創(chuàng)建HikariDataSource
package utry.hikaricp;

import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import utry.hikaricp.info.DataSourceInfo;

/**
 * @author szs
 * @date 2019/3/14 15:07
 */
public class DataSourceProvider {
    public static HikariDataSource create(DataSourceInfo sourceInfo) {
        HikariConfig hikariConfig = new HikariConfig();
        hikariConfig.setUsername(sourceInfo.getUsername());
        hikariConfig.setPassword(sourceInfo.getPassword());
        hikariConfig.setJdbcUrl(sourceInfo.getUrl());
        hikariConfig.setDriverClassName("com.mysql.jdbc.Driver");
        return new HikariDataSource(hikariConfig);
    }
}
最核心的數(shù)據(jù)源的類MyHikariDataSource.java
package utry.hikaricp.source;

import com.zaxxer.hikari.HikariDataSource;
import org.springframework.jdbc.datasource.AbstractDataSource;

import javax.sql.DataSource;
import java.io.Closeable;
import java.io.IOException;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

/**
 * @author szs
 * @date 2019/3/14 14:30
 */
public class MyHikariDataSource extends AbstractDataSource {

    private static Map<String, HikariDataSource> dataSourceMap = new HashMap<>(1);

    @Override
    public Connection getConnection() throws SQLException {
        return dataSourceMap.get("1").getConnection();
    }

    @Override
    public Connection getConnection(String username, String password) throws SQLException {
        return dataSourceMap.get("1").getConnection(username, password);
    }

    private void destroy(){
        DataSource dataSource = dataSourceMap.get("1");
        Closeable closeable = (Closeable) dataSource;
        if (closeable != null) {
            try {
                closeable.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    public void updateDataSourceMap(String key, HikariDataSource value) {
        destroy();
        dataSourceMap.put(key, value);
    }
}

只要將上面這個(gè)類中的DataSourceMap集合中的HikariDataSource修改就能實(shí)現(xiàn)數(shù)據(jù)源的切換了，調(diào)用updateDataSourceMap()方法修改即可（修改之前，先關(guān)閉之前的數(shù)據(jù)源，這里寫死的map中key=1，讀者可以實(shí)現(xiàn)動(dòng)態(tài)設(shè)置）

實(shí)體類UserInfo.java
package utry.hikaricp.model;

public class UserInfo {
    private Integer id;

    private String remarks;

    public UserInfo() {
    }

    public UserInfo(Integer id, String remarks) {
        this.id = id;
        this.remarks = remarks;
    }

    public Integer getId() {
        return id;
    }

    public void setId(Integer id) {
        this.id = id;
    }

    public String getRemarks() {
        return remarks;
    }

    public void setRemarks(String remarks) {
        this.remarks = remarks;
    }
}
HikariCpController.java兩個(gè)接口：
  1. 更新數(shù)據(jù)源
  2. 查詢用戶信息
package utry.hikaricp.controller;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import utry.hikaricp.factory.HikariCPDataSourceFactory;
import utry.hikaricp.mapper.UserInfoMapper;
import utry.hikaricp.model.UserInfo;

/**
 * @author szs
 * @date 2019/3/18 11:50
 */
@RestController
public class HikariCpController {

    @Autowired
    private UserInfoMapper mapper;

    @Autowired
    private HikariCPDataSourceFactory factory;

    @RequestMapping("update")
    public void info(String username, String password, String url) {
        factory.reload(username, password, url);
    }

    @RequestMapping("get")
    public UserInfo get() {
        try {
            UserInfo userInfo = mapper.selectByPrimaryKey(1);
            return userInfo;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }
}
啟動(dòng)項(xiàng)目
image.png

  1. 調(diào)用get接口，返回的數(shù)據(jù)是test庫中的


    image.png

  2. 調(diào)用update請(qǐng)求更新數(shù)據(jù)源


    image.png

    調(diào)用成功之后，可以看到，之前的數(shù)據(jù)源已經(jīng)被shutdown，初始化了一個(gè)新的數(shù)據(jù)源


    image.png

  3. 再次調(diào)用get請(qǐng)求，可以看到獲取的test2數(shù)據(jù)庫中的數(shù)據(jù)


    image.png
最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點(diǎn)，簡(jiǎn)書系信息發(fā)布平臺(tái)，僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容