RuoYi-Vue-Plus（动态添加移除数据源）

### 数据源动态加载方法 在 RuoYi-Vue-Plus 中，为了支持从数据库中动态加载数据源的功能，通常采用 Spring Boot 的 `AbstractRoutingDataSource` 或者自定义的数据源切换逻辑来实现。以下是具体的实现方式： #### 1. **配置文件中的初始数据源** 在项目的 `application.yml` 文件中，可以预先设置默认的核心数据源（Core DataSource）。该核心数据源用于连接到存储其他动态数据源信息的表。 ```yaml spring: datasource: core: jdbc-url: jdbc:mysql://localhost:3306/ry-vue?useUnicode=true&characterEncoding=UTF-8&serverTimezone=Asia/Shanghai&allowPublicKeyRetrieval=true username: root password: 12345678 ``` 此部分配置为核心数据源的信息[^1]。 --- #### 2. **创建动态数据源管理类** 通过继承 `AbstractRoutingDataSource` 类或者手动维护一个线程安全的 Map 来保存动态数据源实例。 ##### 动态数据源工具类代码示例： ```java import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; public class DynamicDataSource extends AbstractRoutingDataSource { private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>(); public void setDataSourceKey(String key) { CONTEXT_HOLDER.set(key); } @Override protected Object determineCurrentLookupKey() { return CONTEXT_HOLDER.get(); } } ``` 在此处实现了基于当前线程上下文决定使用哪个数据源的关键逻辑[^2]。 --- #### 3. **初始化动态数据源** 当应用启动时，可以通过读取数据库中的配置表（例如 `sys_datasource` 表），将其中记录的所有数据源信息注册到动态数据源容器中。 ##### 初始化逻辑伪代码： ```java @Service public class DataSourceInitService { @Autowired private DruidDynamicDataSource dynamicDataSource; @PostConstruct public void initDatasourcesFromDB() { List<SysDataSourceEntity> sysDataSources = dataSourceRepository.findAll(); // 假设有一个 Repository 查询所有数据源 sysDataSources.forEach(sysDataSource -> { HikariDataSource ds = createDataSource(sysDataSource.getUrl(), sysDataSource.getUsername(), sysDataSource.getPassword()); dynamicDataSource.addDataSource(sysDataSource.getName(), ds); // 将新数据源加入动态池 }); } private HikariDataSource createDataSource(String url, String username, String password) { HikariConfig config = new HikariConfig(); config.setJdbcUrl(url); config.setUsername(username); config.setPassword(password); return new HikariDataSource(config); } } ``` 这段代码展示了如何从数据库中获取多个数据源并将其注入到动态数据源管理器中[^3]。 --- #### 4. **切换单个请求使用的数据源** 在实际业务场景下，可能需要根据某些条件（如租户 ID、模块名等）动态切换数据源。这一步骤通常是通过拦截器或 AOP 实现的。 ##### 切换数据源的拦截器示例： ```java @Component public class DataSourceInterceptor implements HandlerInterceptor { @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { String tenantId = request.getHeader("X-Tenant-ID"); // 获取租户ID或其他标识符 if (StringUtils.hasText(tenantId)) { DynamicDataSourceContextHolder.setDataSourceKey(tenantId); // 设置对应的数据源键值 } return true; } @Override public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception { DynamicDataSourceContextHolder.clearDataSourceKey(); // 清理线程变量 } } ``` 以上代码片段说明了如何利用 HTTP 请求头传递参数完成数据源的选择操作[^4]。 --- #### 总结 综上所述，在 RuoYi-Vue-Plus 项目中实现从数据库加载数据源的主要思路包括以下几个方面： 1. 使用静态配置作为核心数据源； 2. 创建动态数据源管理组件，并扩展其行为以适应多数据源需求； 3. 在程序运行期间实时更新可用数据源列表； 4. 结合具体业务规则灵活调整每次查询所依赖的具体数据源。 这种设计模式不仅增强了系统的灵活性，还能够很好地满足分布式架构下的复杂应用场景。 ---