- 当前位置：首页 >系统运维 >SpringBoot+AOP构建多数据源的切换实践
SpringBoot+AOP构建多数据源的切换实践
发布时间：2025-11-05 11:22:41 来源：云智核 作者：人工智能
针对微服务架构中常用的建多设计模块，通常我们都会需要使用到druid作为我们的数据实践数据连接池，当架构发生扩展的切换时候 ，通常面对的建多数据存储服务器也会渐渐增加，从原本的数据实践单库架构逐渐扩展为复杂的多库架构。
当在业务层需要涉及到查询多种同数据库的切换场景下，我们通常需要在执行sql的建多时候动态指定对应的datasource。
而Spring的数据实践AbstractRoutingDataSource则正好为我们提供了这一功能点，下边我将通过一个简单的切换基于springboot+aop的案例来实现如何通过自定义注解切换不同的数据源进行读数据操作，同时也将结合部分源码的建多内容进行讲解。
首先我们需要自定义一个专门用于申明当前java应用程序所需要使用到哪些数据源信息：
package mutidatasource.annotation; import mutidatasource.config.DataSourceConfigRegister; import mutidatasource.enums.SupportDatasourceEnum; import org.springframework.context.annotation.Import; import org.springframework.stereotype.Component; import java.lang.annotation.*; /** * 注入数据源 * * @author idea * @data 2020/3/7 */ @Target({ElementType.METHOD,数据实践ElementType.TYPE}) @Retention(RetentionPolicy.RUNTIME) @Documented @Import(DataSourceConfigRegister.class) public @interface AppDataSource { SupportDatasourceEnum[] datasourceType(); }这里为了方便，我将测试中使用的切换数据源地址都配置在来enum里面，如果后边需要灵活处理的建多话，可以将这些配置信息抽取出来放在一些配置中心上边。数据实践
package mutidatasource.enums; import lombok.AllArgsConstructor; import lombok.Getter; import lombok.NoArgsConstructor; /** * 目前支持的切换数据源信息 * * @author idea * @data 2020/3/7 */ @AllArgsConstructor @Getter public enum SupportDatasourceEnum { PROD_DB("jdbc:mysql://localhost:3306/db-prod?useUnicode=true&characterEncoding=utf8","root","root","db-prod"), DEV_DB("jdbc:mysql://localhost:3306/db-dev?useUnicode=true&characterEncoding=utf8","root","root","db-dev"), PRE_DB("jdbc:mysql://localhost:3306/db-pre?useUnicode=true&characterEncoding=utf8","root","root","db-pre"); String url; String username; String password; String databaseName; @Override public String toString() { return super.toString().toLowerCase(); } }之所以要创建这个@AppDataSource注解，是要在springboot的亿华云计算启动类上边进行标注：
package mutidatasource; import mutidatasource.annotation.AppDataSource; import mutidatasource.enums.SupportDatasourceEnum; import org.springframework.boot.SpringApplication; import org.springframework.boot.autoconfigure.SpringBootApplication; /** * @author idea * @data 2020/3/7 */ @SpringBootApplication @AppDataSource(datasourceType = {SupportDatasourceEnum.DEV_DB, SupportDatasourceEnum.PRE_DB, SupportDatasourceEnum.PROD_DB}) public class SpringApplicationDemo { public static void main(String[] args) { SpringApplication.run(SpringApplicationDemo.class); } }借助springboot的ImportSelector 自定义一个注册器来获取启动类头部的注解所指定的数据源类型：
package mutidatasource.config; import lombok.extern.slf4j.Slf4j; import mutidatasource.annotation.AppDataSource; import mutidatasource.core.DataSourceContextHolder; import mutidatasource.enums.SupportDatasourceEnum; import org.springframework.context.annotation.ImportSelector; import org.springframework.core.annotation.AnnotationAttributes; import org.springframework.core.type.AnnotationMetadata; import org.springframework.stereotype.Component; /** * @author idea * @data 2020/3/7 */ @Slf4j @Component public class DataSourceConfigRegister implements ImportSelector { @Override public String[] selectImports(AnnotationMetadata annotationMetadata) { AnnotationAttributes attributes = AnnotationAttributes.fromMap(annotationMetadata.getAnnotationAttributes(AppDataSource.class.getName())); System.out.println("####### datasource import #######"); if (null != attributes) { Object object = attributes.get("datasourceType"); SupportDatasourceEnum[] supportDatasourceEnums = (SupportDatasourceEnum[]) object; for (SupportDatasourceEnum supportDatasourceEnum : supportDatasourceEnums) { DataSourceContextHolder.addDatasource(supportDatasourceEnum); } } return new String[0]; } }好的，现在我们已经能够获取到对应的数据源类型信息了，这里你会看到一个叫做DataSourceContextHolder的角色。这个对象主要是用于对每个请求线程的数据源信息做统一的分配和管理。
在多并发场景下，为了防止不同线程请求的数据源出现“互窜”情况，通常我们都会使用到threadlocal来做处理。为每一个线程都分配一个指定的，属于其内部的副本变量，当当前线程结束之前，记得将对应的线程副本也进行销毁。
package mutidatasource.core; import mutidatasource.enums.SupportDatasourceEnum; import java.util.HashSet; /** * @author idea * @data 2020/3/7 */ public class DataSourceContextHolder { private static final HashSet<SupportDatasourceEnum> dataSourceSet = new HashSet<>(); private static final ThreadLocal<String> databaseHolder = new ThreadLocal<>(); public static void setDatabaseHolder(SupportDatasourceEnum supportDatasourceEnum) { databaseHolder.set(supportDatasourceEnum.toString()); } /** * 取得当前数据源 * * @return */ public static String getDatabaseHolder() { return databaseHolder.get(); } /** * 添加数据源 * * @param supportDatasourceEnum */ public static void addDatasource(SupportDatasourceEnum supportDatasourceEnum) { dataSourceSet.add(supportDatasourceEnum); } /** * 获取当期应用所支持的所有数据源 * * @return */ public static HashSet<SupportDatasourceEnum> getDataSourceSet() { return dataSourceSet; } /** * 清除上下文数据 */ public static void clear() { databaseHolder.remove(); } }spring内部的AbstractRoutingDataSource动态路由数据源里面有一个抽象方法叫做
determineCurrentLookupKey，这个方法适用于提供给开发者自定义对应数据源的查询key。
package mutidatasource.core; import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; /** * @author idea * @data 2020/3/7 */ public class DynamicDataSource extends AbstractRoutingDataSource { @Override protected Object determineCurrentLookupKey() { String dataSource = DataSourceContextHolder.getDatabaseHolder(); return dataSource; } }这里我使用的druid数据源，所以配置数据源的配置类如下：这里面我默认该应用配置类PROD数据源，WordPress模板用于测试使用。
package mutidatasource.core; import com.alibaba.druid.pool.DruidDataSource; import lombok.extern.slf4j.Slf4j; import mutidatasource.enums.SupportDatasourceEnum; import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Primary; import org.springframework.stereotype.Component; import javax.sql.DataSource; import java.util.HashMap; import java.util.HashSet; /** * @author idea * @data 2020/3/7 */ @Slf4j @Component public class DynamicDataSourceConfiguration { @Bean @Primary @ConditionalOnMissingBean public DataSource dataSource() { System.out.println("init datasource"); DynamicDataSource dynamicDataSource = new DynamicDataSource(); //设置原始数据源 HashMap<Object, Object> dataSourcesMap = new HashMap<>(); HashSet<SupportDatasourceEnum> dataSet = DataSourceContextHolder.getDataSourceSet(); for (SupportDatasourceEnum supportDatasourceEnum : dataSet) { DataSource dataSource = this.createDataSourceProperties(supportDatasourceEnum); dataSourcesMap.put(supportDatasourceEnum.toString(), dataSource); } dynamicDataSource.setTargetDataSources(dataSourcesMap); dynamicDataSource.setDefaultTargetDataSource(createDataSourceProperties(SupportDatasourceEnum.PRE_DB)); return dynamicDataSource; } private synchronized DataSource createDataSourceProperties(SupportDatasourceEnum supportDatasourceEnum) { DruidDataSource druidDataSource = new DruidDataSource(); druidDataSource.setUrl(supportDatasourceEnum.getUrl()); druidDataSource.setUsername(supportDatasourceEnum.getUsername()); druidDataSource.setPassword(supportDatasourceEnum.getPassword()); //具体配置 druidDataSource.setMaxActive(100); druidDataSource.setInitialSize(5); druidDataSource.setMinIdle(1); druidDataSource.setMaxWait(30000); //间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 druidDataSource.setTimeBetweenConnectErrorMillis(60000); return druidDataSource; } }好了现在一个基础的数据源注入已经可以了，那么我们该如何借助注解来实现动态切换数据源的操作呢？
为此，我设计了一个叫做UsingDataSource的注解，通过利用该注解来识别当前线程所需要使用的数据源操作：
package mutidatasource.annotation; import mutidatasource.enums.SupportDatasourceEnum; import java.lang.annotation.*; /** * @author idea * @data 2020/3/7 */ @Target({ElementType.METHOD,ElementType.TYPE}) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface UsingDataSource { SupportDatasourceEnum type() ; }然后，借助了spring的aop来做切面拦截：
package mutidatasource.core; import lombok.extern.slf4j.Slf4j; import mutidatasource.annotation.UsingDataSource; import org.aspectj.lang.JoinPoint; import org.aspectj.lang.ProceedingJoinPoint; import org.aspectj.lang.Signature; import org.aspectj.lang.annotation.*; import org.aspectj.lang.reflect.MethodSignature; import org.springframework.context.annotation.Configuration; import org.springframework.core.annotation.AnnotationUtils; import org.springframework.core.annotation.Order; import org.springframework.stereotype.Component; import java.lang.reflect.Method; import java.util.Arrays; /** * @author idea * @data 2020/3/7 */ @Slf4j @Aspect @Configuration public class DataSourceAspect { public DataSourceAspect(){ System.out.println("this is init"); } @Pointcut("@within(mutidatasource.annotation.UsingDataSource) || " + "@annotation(mutidatasource.annotation.UsingDataSource)") public void pointCut(){ } @Before("pointCut() && @annotation(usingDataSource)") public void doBefore(UsingDataSource usingDataSource){ log.debug("select dataSource---"+usingDataSource.type()); DataSourceContextHolder.setDatabaseHolder(usingDataSource.type()); } @After("pointCut()") public void doAfter(){ DataSourceContextHolder.clear(); } }测试类如下所示：
package mutidatasource.controller; import lombok.extern.slf4j.Slf4j; import mutidatasource.annotation.UsingDataSource; import mutidatasource.enums.SupportDatasourceEnum; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.jdbc.core.JdbcTemplate; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; /** * @author idea * @data 2020/3/8 */ @RestController @RequestMapping(value = "/test") @Slf4j public class TestController { @Autowired private JdbcTemplate jdbcTemplate; @GetMapping(value = "/testDev") @UsingDataSource(type=SupportDatasourceEnum.DEV_DB) public void testDev() { showData(); } @GetMapping(value = "/testPre") @UsingDataSource(type=SupportDatasourceEnum.PRE_DB) public void testPre() { showData(); } private void showData() { jdbcTemplate.queryForList("select * from test1").forEach(row -> log.info(row.toString())); } }最后 启动springboot服务，通过使用注解即可测试对应功能。
关于AbstractRoutingDataSource 动态路由数据源的注入原理，
可以看到这个内部类里面包含了多种用于做数据源映射的map数据结构。
在该类的最底部，有一个determineCurrentLookupKey函数，也就是上边我们所提及的使用于查询当前数据源key的方法。
具体代码如下：
/** * Retrieve the current target DataSource. Determines the * {@link #determineCurrentLookupKey() current lookup key}, performs * a lookup in the {@link #setTargetDataSources targetDataSources} map, * falls back to the specified * {@link #setDefaultTargetDataSource default target DataSource} if necessary. * @see #determineCurrentLookupKey() */ protected DataSource determineTargetDataSource() { Assert.notNull(this.resolvedDataSources, "DataSource router not initialized"); //这里面注入我们当前线程使用的数据源 Object lookupKey = determineCurrentLookupKey(); //在初始化数据源的时候需要我们去给resolvedDataSources进行注入 DataSource dataSource = this.resolvedDataSources.get(lookupKey); if (dataSource == null && (this.lenientFallback || lookupKey == null)) { dataSource = this.resolvedDefaultDataSource; } if (dataSource == null) { throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]"); } return dataSource; } /** * Determine the current lookup key. This will typically be * implemented to check a thread-bound transaction context. * <p>Allows for arbitrary keys. The returned key needs * to match the stored lookup key type, as resolved by the * {@link #resolveSpecifiedLookupKey} method. */ @Nullable protected abstract Object determineCurrentLookupKey();而在该类的afterPropertiesSet里面，又有对于初始化数据源的注入操作，免费源码下载这里面的targetDataSources 正是上文中我们对在初始化数据源时候注入的信息。
@Override public void afterPropertiesSet() { if (this.targetDataSources == null) { throw new IllegalArgumentException("Property targetDataSources is required"); } this.resolvedDataSources = new HashMap<>(this.targetDataSources.size()); this.targetDataSources.forEach((key, value) -> { Object lookupKey = resolveSpecifiedLookupKey(key); DataSource dataSource = resolveSpecifiedDataSource(value); this.resolvedDataSources.put(lookupKey, dataSource); }); if (this.defaultTargetDataSource != null) { this.resolvedDefaultDataSource = resolveSpecifiedDataSource(this.defaultTargetDataSource); } }- 大家都为在UBUNTU中使用聊天工具而烦燥吗？大家还为找QQ而在网上不停的找方案吗？什么lumaqq啊，EVAQQ啊！假如让你感觉 装LUMAQQ感觉不方便怎么办？假如用apt-get install eva成功安装上EVA却用不了怎么办，而且EVA又不怎么稳定怎么办？现在Gaim搞定，大家都知道GAIM中有很聊天工具可以一起使用集合。同时在一个界面上使用，方便吧！但是MSN是直接可以用的（系统装好后），而QQ需要装一个叫openq的插件，下载地址：http://textwiki.org/wiki/OpenQ。假如没有这个东西的话你怎么也上不了QQ，总会跳出来说明 你的QQ密码有误，所以这个东西是必需要要用的哦！本人下载的是一个文件。而不是DEB包，所以右键归档成OpenQ.tar.gz接着安装就OK安装方法：[root@localhost xxx]# tar zxvf openq.tar.gz[root@localhost xxx]# cd openq[beinan@localhost openq-0.3.2]#./configure --libdir=/usr/lib[beinan@localhost openq-0.3.2]# make[beinan@localhost openq-0.3.2]# make install安装好后OpenQ，我们查看一下 /usr/lib/gaim 目录中，会有两个libqq两个相关的文件；[root@localhost openq-0.3.2]# ls -lh /usr/lib/gaim/libqq*-rwxr-xr-x 1 root root 675 8月 31 11:08 /usr/lib/gaim/libqq.la-rwxr-xr-x 1 root root 8.7M 8月 31 11:08 /usr/lib/gaim/libqq.so到这儿就OK啦！
- U盘锁如何解锁？（解决U盘锁无法打开的问题）
- 如何完整地卸载系统——详细教程（从根本上删除系统并恢复设备到出厂状态的方法）
- 金士顿内存条的性能与品质如何？（探究金士顿内存条的可靠性、速度和适用范围）
- 15-3567CPU的性能评测与分析（一款强劲且高效的处理器）
- 探索O2O模式下的做市场发展前景（以做市场O2O为核心，助推传统商业转型）
- 微星GP拆机教程
- 苹果电脑如何装Windows系统（教程分享及操作步骤）
- 减少电热毯辐射的有效方法（保护身体健康关注电热毯辐射问题）
- 碳纤维膜的手感体验（轻盈舒适，质感卓越的碳纤维材质手感）
- 拉黑卖家（探究拉黑卖家的利与弊，以及对消费者的影响）
- 解读1.8GHzCPU的性能和应用领域（探索1.8GHzCPU的关键特点及其适用范围）
- 全面评估OppoR9（性能强劲、拍照出色、设计精致，OppoR9是一款令人惊艳的智能手机）
- AM3底座（探索AM3底座的技术特点与优势）
- 安装小企鹅输入法复制代码代码如下:Alt+F2运行以下命令关闭iBus启动Fcitx:复制代码代码如下:复制代码代码如下:sudo service lightdm restart复制代码代码如下:复制代码代码如下:复制代码代码如下:复制代码代码如下:复制代码代码如下:复制代码代码如下:复制代码代码如下:复制代码代码如下:复制代码代码如下:复制代码代码如下:复制代码代码如下:复制代码代码如下:im-switch -s fcitx -z defaultsudo im-switch -s fcitx -z default第六步:注销或重启就生效在右上角的输入法设置中输入法 ==>搜狗拼音或输入法的“配置”中设置这个输入法的翻页不是安pagedown或pageup,而是加号键跟减号键(其实不加shift）
- 打印机驱动安装教程（从下载到配置，轻松解决打印机驱动问题）
- 游戏U盘使用教程（轻松携带游戏，畅玩游戏世界）
- 「探索heilox20（「挑战极限，体验科技魅力」）
- 剪影电脑追踪打码教程（学会使用剪影电脑追踪功能，轻松完成打码任务）
- 金立S10（颠覆传统游戏体验，金立S10给你带来全新感受）
随便看看
IT资讯网益华科技服务器租用源码下载源码库IT技术网企商汇亿华云香港云服务器技术快报益华科技编程之道益华IT技术论坛云站无忧亿华科技益强资讯优选码力社亿华云极客码头亿华灵动亿华云计算码上建站多维IT资讯益强数据堂IT资讯网益强前沿资讯亿华智造智能时代思维库极客编程益强科技亿华互联汇智坊科技前瞻益强科技益强智未来运维纵横益强编程舍云智核益强IT技术网益强智囊团
- Copyright © 2025 Powered by SpringBoot+AOP构建多数据源的切换实践,云智核 滇ICP备2023006006号-28sitemap