MyBatis如何实现动态数据源切换？

MyBatis如何实现动态数据源切换在现代应用中特别是微服务架构中使用多个数据库的情况越来越常见。MyBatis是一个流行的Java持久层框架它允许我们方便地与多种数据库进行交互。在某些情况下我们可能需要动态切换数据源例如在多租户系统中。本文将介绍如何在MyBatis中实现动态数据源切换并提供相关的Java代码示例。1. 基本概念动态数据源切换的关键在于使用一个数据源路由器DataSource Router。这个路由器决定了在运行时使用哪个数据源。通过ThreadLocal来存储当前线程使用的数据源信息从而实现动态切换。2. 项目结构我们将创建一个简单的项目结构如下src ├── main │ ├── java │ │ └── com │ │ └── example │ │ ├── config │ │ │ └── DataSourceConfig.java │ │ ├── datasource │ │ │ ├── DynamicDataSource.java │ │ │ └── DataSourceContext.java │ │ └── mapper │ │ └── UserMapper.java │ └── resources │ └── mybatis-config.xml └── ...3. 环境配置首先确保你的pom.xml或者build.gradle中引入了MyBatis和相关数据库驱动的依赖。Maven 示例dependenciesdependencygroupIdorg.mybatis/groupIdartifactIdmybatis/artifactIdversion3.5.7/version/dependencydependencygroupIdmysql/groupIdartifactIdmysql-connector-java/artifactIdversion8.0.26/version/dependencydependencygroupIdorg.springframework/groupIdartifactIdspring-context/artifactIdversion5.3.10/version/dependency!-- 其他依赖 --/dependencies4. 数据源配置4.1 DataSourceConfig在DataSourceConfig.java中配置多个数据源。packagecom.example.config;importorg.apache.ibatis.session.SqlSessionFactory;importorg.mybatis.spring.SqlSessionFactoryBean;importorg.springframework.beans.factory.annotation.Qualifier;importorg.springframework.boot.autoconfigure.jdbc.DataSourceProperties;importorg.springframework.context.annotation.Bean;importorg.springframework.context.annotation.Configuration;importorg.springframework.context.annotation.Primary;importorg.springframework.jdbc.datasource.lookup.JndiDataSourceLookup;importorg.springframework.jdbc.datasource.routing.AbstractRoutingDataSource;importorg.springframework.orm.mybatis.MyBatisSqlSessionFactoryBean;importjavax.sql.DataSource;importjava.util.HashMap;importjava.util.Map;ConfigurationpublicclassDataSourceConfig{BeanPrimarypublicDataSourcedataSource(){AbstractRoutingDataSourceroutingDataSourcenewDynamicDataSource();MapObject,ObjecttargetDataSourcesnewHashMap();targetDataSources.put(dataSource1,createDataSource(jdbc:mysql://localhost:3306/db1,user1,pass1));targetDataSources.put(dataSource2,createDataSource(jdbc:mysql://localhost:3306/db2,user2,pass2));routingDataSource.setTargetDataSources(targetDataSources);routingDataSource.setDefaultTargetDataSource(targetDataSources.get(dataSource1));returnroutingDataSource;}privateDataSourcecreateDataSource(Stringurl,Stringusername,Stringpassword){DataSourcePropertiespropertiesnewDataSourceProperties();properties.setUrl(url);properties.setUsername(username);properties.setPassword(password);returnproperties.initializeDataSourceBuilder().build();}BeanpublicSqlSessionFactorysqlSessionFactory(Qualifier(dataSource)DataSourcedataSource)throwsException{SqlSessionFactoryBeansessionFactorynewSqlSessionFactoryBean();sessionFactory.setDataSource(dataSource);returnsessionFactory.getObject();}}4.2 动态数据源实现在DynamicDataSource.java中实现数据源选择的逻辑。packagecom.example.datasource;importorg.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;publicclassDynamicDataSourceextendsAbstractRoutingDataSource{OverrideprotectedObjectdetermineCurrentLookupKey(){returnDataSourceContext.getCurrentDataSource();}}4.3 数据源上下文在DataSourceContext.java中使用ThreadLocal管理当前数据源。packagecom.example.datasource;publicclassDataSourceContext{privatestaticfinalThreadLocalStringCONTEXTnewThreadLocal();publicstaticvoidsetCurrentDataSource(StringdataSource){CONTEXT.set(dataSource);}publicstaticStringgetCurrentDataSource(){returnCONTEXT.get();}publicstaticvoidclear(){CONTEXT.remove();}}5. 使用场景示例5.1 Mapper接口在UserMapper.java中示范如何使用Mapper。packagecom.example.mapper;importcom.example.model.User;importorg.apache.ibatis.annotations.Select;publicinterfaceUserMapper{Select(SELECT * FROM users WHERE id #{id})UserfindById(intid);}5.2 切换数据源在具体的方法中可以选择使用哪个数据源。importcom.example.datasource.DataSourceContext;importcom.example.mapper.UserMapper;importorg.springframework.beans.factory.annotation.Autowired;importorg.springframework.stereotype.Service;importorg.springframework.transaction.annotation.Transactional;ServicepublicclassUserService{AutowiredprivateUserMapperuserMapper;TransactionalpublicUsergetUserFromDataSource1(intid){DataSourceContext.setCurrentDataSource(dataSource1);returnuserMapper.findById(id);// 此时使用 dataSource1}TransactionalpublicUsergetUserFromDataSource2(intid){DataSourceContext.setCurrentDataSource(dataSource2);returnuserMapper.findById(id);// 此时使用 dataSource2}}6. 最后小结下哈通过以上步骤我们实现了在MyBatis中动态切换数据源的功能。关键在于实现AbstractRoutingDataSource和使用ThreadLocal来存储当前数据源。此外根据需求灵活切换数据源能够满足多租户系统等复杂场景的需求。随着业务的发展可能会面临更复杂的动态数据源切换需求比如基于用户、请求等条件来决定使用哪个数据源。但以上的基本架构为实现这种需求打下了良好的基础。希望本文能为你在项目中的动态数据源切换提供帮助