SpringBoot通过代码配置数据源

<think>好的，我现在需要帮助用户找到一个在Spring Boot中通过代码配置数据源的示例。用户明确要求的是代码方式，而不是依赖配置文件。让我先回忆一下相关的知识，并参考提供的引用内容。 首先，根据引用[1]，提到了使用@Import和ImportBeanDefinitionRegistrar来动态注册Bean，这可能涉及到动态数据源的配置。引用[2]和[3]提到通过自定义数据源类以及使用AbstractRoutingDataSource来实现多数据源的动态切换。引用[4]提到了使用dynamic-datasource-spring-boot-starter这个依赖，可能是一个现成的解决方案。引用[5]则详细说明了AbstractRoutingDataSource的工作原理，通过覆写determineCurrentLookupKey方法来切换数据源。 用户需要的是代码示例，所以可能需要手动配置数据源而不是依赖自动配置。通常，Spring Boot的数据源配置可以通过application.properties或application.yml，但用户希望用代码方式，因此需要手动创建DataSource Bean。 首先，应该考虑如何手动创建多个数据源，并设置默认数据源。然后，利用AbstractRoutingDataSource来动态切换。根据引用[5]，AbstractRoutingDataSource需要一个Map来存储目标数据源，并设置默认数据源。同时，需要重写determineCurrentLookupKey方法，以决定当前使用哪个数据源。 另外，引用[2]中提到使用ThreadLocal来保存当前数据源的标识，这样可以在运行时动态切换。例如，通过一个工具类来设置和获取当前线程的数据源key。 现在，我需要将这些知识点整合成一个步骤清晰的代码示例。步骤可能包括： 1. 添加必要的依赖，比如HikariCP，JDBC驱动等。 2. 创建数据源配置类，手动创建多个数据源实例。 3. 创建动态数据源类继承AbstractRoutingDataSource，并配置数据源Map和默认数据源。 4. 使用ThreadLocal来管理当前数据源的key。 5. 配置Spring的@Bean，将动态数据源注册到容器中。 6. 可能需要排除自动配置的数据源，避免冲突。 需要注意的是，如果使用Spring Boot的自动配置，默认会创建一个数据源，因此需要排除自动配置，或者在配置类中覆盖默认的DataSource Bean。 参考引用[3]中的例子，他们可能在配置类中手动创建了多个数据源，并通过@Bean注解注入。而引用[5]中的AbstractRoutingDataSource是关键，用来动态路由。 现在，检查是否有遗漏的部分。比如事务管理，多数据源下的事务可能需要特别处理，但用户的问题可能暂时不需要这部分。另外，数据源的属性可能需要从外部读取，比如通过@Value注解注入，但用户要求代码方式，可能需要在代码中硬编码或通过其他方式获取配置。 可能还需要配置JdbcTemplate或者MyBatis等持久层框架，以使用动态数据源。但用户的问题可能仅关注数据源配置，所以示例可能不需要涉及这些。 最后，确保代码示例正确，并且符合Spring Boot的配置方式。例如，使用@Configuration类，正确创建Bean，处理依赖关系等。</think>在Spring Boot中通过代码配置数据源的核心实现分为以下步骤： ### 一、手动配置动态数据源 ```java @Configuration public class DataSourceConfig { // 创建主数据源 @Bean @Primary public DataSource primaryDataSource() { HikariDataSource dataSource = new HikariDataSource(); dataSource.setJdbcUrl("jdbc:mysql://localhost:3306/db1"); dataSource.setUsername("root"); dataSource.setPassword("123456"); return dataSource; } // 创建动态数据源路由 @Bean public AbstractRoutingDataSource dynamicDataSource() { DynamicDataSourceRouter router = new DynamicDataSourceRouter(); // 配置数据源集合 Map<Object, Object> targetDataSources = new HashMap<>(); targetDataSources.put("db1", primaryDataSource()); targetDataSources.put("db2", createDataSource("db2")); // 第二个数据源 router.setDefaultTargetDataSource(primaryDataSource()); router.setTargetDataSources(targetDataSources); return router; } // 动态创建其他数据源 private DataSource createDataSource(String dbName) { HikariDataSource ds = new HikariDataSource(); ds.setJdbcUrl("jdbc:mysql://localhost:3306/"+dbName); ds.setUsername("root"); ds.setPassword("123456"); return ds; } } ``` ### 二、实现动态路由逻辑 ```java public class DynamicDataSourceRouter extends AbstractRoutingDataSource { private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>(); @Override protected Object determineCurrentLookupKey() { return CONTEXT_HOLDER.get(); } public static void setDataSourceKey(String key) { CONTEXT_HOLDER.set(key); } public static void clearDataSourceKey() { CONTEXT_HOLDER.remove(); } } ``` ### 三、排除自动配置（重要） 在启动类添加： ```java @SpringBootApplication(exclude = {DataSourceAutoConfiguration.class}) ``` ### 四、使用示例 ```java @Service public class DataService { @Autowired private JdbcTemplate jdbcTemplate; public void queryFromDb2() { DynamicDataSourceRouter.setDataSourceKey("db2"); jdbcTemplate.query("SELECT * FROM table"); DynamicDataSourceRouter.clearDataSourceKey(); } } ``` ### 关键实现原理 1. 通过继承$AbstractRoutingDataSource$实现动态路由[^5] 2. 使用$ThreadLocal$保存当前线程数据源标识 3. 需要手动创建所有数据源实例 4. 必须排除Spring Boot的自动数据源配置