基于Spring Boot 动态主从数据库路由

本文简单的介绍一下基于Spring boot框架动态连接多数据源的实现,在配置文件中添加多个数据源,采用主从配置的方式,配置zjsz、cloud两个数据库。

项目的目录结构:

application.properties

核心主要是dynamic和aop目录

1.首先配置数据源信息,在spring5.x之后开始默认使用HikariCP来做数据源

@Configuration
public class DataSourceConfig {

    @Bean(name = "masterDataSource")
    @ConfigurationProperties(prefix = "spirng.master.datasource")
    public HikariDataSource masterDataSource() {
        return new HikariDataSource();
    }


    @Bean(name = "slaveDataSource")
    @ConfigurationProperties(prefix = "spirng.slave.datasource")
    public HikariDataSource slaveDataSource() {
        return new HikariDataSource();
    }

    //动态数据源
    @Bean(name = "dynamicDataSource")
    //解决互相依赖关系
    @DependsOn({ "masterDataSource", "slaveDataSource"})
    @Primary
    public DataSource getDataSource() {
        DynamicDataSource dataSource = new DynamicDataSource();
        dataSource.setTargetDataSources(targetDataSources());
        return dataSource;
    }

    private Map<Object, Object> targetDataSources() {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put(DataSourceType.MASTER.getType(), masterDataSource());
        targetDataSources.put(DataSourceType.SLAVE.getType(), slaveDataSource());
        return targetDataSources;
    }
}
public enum DataSourceType {
    SLAVE("slave", "从库"),
    MASTER("master", "主库");

    private String type;
    private String name;

    DataSourceType(String type, String name) {
        this.type = type;
        this.name = name;
    }

    public String getType() {
        return type;
    }

    public void setType(String type) {
        this.type = type;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }
}

2.自定义Datasource注解

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
@Documented
@Inherited
public @interface DataSource {

    /**
     * 数据库路由
     */
    DataSourceType value() default DataSourceType.MASTER;
}

3.使用ThreadLocal来保证从@DataSource获取的路由key是线程安全的

public class DynamicDataSourceHolder {

    private static final ThreadLocal<String> holder = new ThreadLocal<>();

    public static void putDataSource(DataSourceType dataSourceType) {
        holder.set(dataSourceType.getType());
    }

    public static String getDataSouce() {
        return holder.get();
    }
}

 

4.继承AbstractRoutingDataSource并重写determineCurrentLookupKey法法

使用AbstractRoutingDataSource 的实现类,进行灵活的切换,可以通过AOP或者手动编程设置当前的DataSource,这样的编写方式比较好。

public class DynamicDataSource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        if (DynamicDataSourceHolder.getDataSouce() != null) {
            return DynamicDataSourceHolder.getDataSouce();
        }
        return DataSourceType.MASTER.getType();
    }
}

5.使用AOP进行拦截带有@DataSource注解的方法,并获取注解的值,进行动态数据源操作

@Aspect
@Component
public class DataSourceAspect implements Ordered {

//    @Pointcut("execution(public * com.aop.jdbc.impl..*.*(..))")
//    public void dynamic() {
//
//    }


    @Before(value = "@annotation(dataSource)")
    public void dataSourcePoint(JoinPoint joinPoint, DataSource dataSource) {
        DynamicDataSourceHolder.putDataSource(dataSource.value());
    }

    @Override
    public int getOrder() {
        return -1;
    }
}

6.使用方法

@Service("testService")
public class TestServiceImpl implements ITestService {

    @Autowired
    private OwnCodeMapper ownCodeMapper;

    @DataSource(DataSourceType.MASTER)
    @Override
    public String save() {
        System.out.println("=======save======进入主库操作========save======");
        System.out.println("删除影响行数:" + this.ownCodeMapper.deleteByPrimaryKey(2));
        return null;
    }

    @DataSource(DataSourceType.SLAVE)
    @Override
    public String query() {
        System.out.println("=======query======进入从库操作=========query=====");
        System.out.println("查询结果:" + this.ownCodeMapper.selectByPrimaryKey(36));
        return null;
    }
}

到此就完成了动态主从路由的代码开发。

源码地址,大家帮忙star哦!!

  • 4
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
dynamic-datasource-spring-boot-starter是一个基于Spring Boot的快速集成多数据源的启动器,它可以帮助我们方便地实现多数据源的配置和切换。你可以在其官方GitHub页面找到该项目的源代码及文档。 该启动器提供了一系列的功能和特性,适用于多种应用场景。下面是其中一些常用的应用场景: 1. 多数据库读写分离:你可以使用dynamic-datasource-spring-boot-starter来配置多个主从数据库,实现读写分离的功能。通过配置数据源的读写策略,可以根据不同的需求将读操作路由到从数据库,写操作路由数据库。 2. 多租户应用:该启动器提供了租户隔离的功能,可以根据不同的租户动态切换数据源。通过配置租户信息和对应的数据源,可以实现在一个应用中管理多个租户的数据。 3. 多数据源切换:使用该启动器,你可以在运行时动态切换数据源。通过使用Spring的AbstractRoutingDataSource,你可以根据需要在不同的线程或请求中切换数据源。 你可以根据自己的需求在项目中引入dynamic-datasource-spring-boot-starter的依赖,并参考官方文档来进行详细的配置和使用。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* *3* [使用dynamic-datasource-spring-boot-starter做多数据源及源码分析](https://blog.csdn.net/weixin_58045199/article/details/125673946)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 100%"] [ .reference_list ]

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值