基于注解和Spring的多数据源配置和使用

在企业级应用开发中，尤其是大型分布式系统，常常需要连接多个数据库来满足不同的业务需求，例如，一个系统可能需要管理用户数据、订单数据以及日志数据，这些数据可能分散在不同的数据库中。在这种情况下，"多数据源"的配置就显得尤为重要。Spring框架提供了强大的支持，允许我们通过注解轻松地进行数据源的切换和管理，从而实现灵活的数据操作。本篇文章将深入探讨如何基于注解和Spring实现多数据源配置和使用。 我们需要理解"注解"在Java中的作用。注解是一种元数据，它提供了一种安全的方法来关联信息和代码（类、方法、变量等）。Spring框架广泛使用注解进行依赖注入、AOP（面向切面编程）以及其他功能的实现，使得代码更加简洁，更易于维护。 在Spring中，多数据源配置通常涉及以下几个关键步骤： 1. **数据源配置**：创建多个DataSource对象，每个对象对应一个数据库连接。可以使用`org.springframework.jdbc.datasource.DriverManagerDataSource`或`com.zaxxer.hikari.HikariDataSource`等数据源实现。例如，我们可以定义两个数据源，分别命名为`primaryDataSource`和`secondaryDataSource`。 2. **数据源切换**：通过`@Profile`注解实现不同环境下的数据源选择。在Spring Boot中，可以通过配置文件（如application.properties或application.yml）指定当前环境，然后在数据源配置类上使用`@Profile`注解，使其在特定环境中生效。 3. **使用`@Configuration`和`@Bean`注解**：在Spring配置类中，使用`@Configuration`注解标记该类为配置类，然后使用`@Bean`注解声明数据源对象。例如： ```java @Configuration public class DataSourceConfig { @Bean(name = "primaryDataSource") @ConfigurationProperties(prefix = "spring.datasource.primary") public DataSource primaryDataSource() { return DataSourceBuilder.create().build(); } @Bean(name = "secondaryDataSource") @ConfigurationProperties(prefix = "spring.datasource.secondary") public DataSource secondaryDataSource() { return DataSourceBuilder.create().build(); } } ``` 4. **数据源路由**：为了实现数据源的动态切换，我们需要一个数据源路由类（`AbstractRoutingDataSource`），这个类可以根据一定的规则（比如事务上下文、请求参数等）动态选择使用哪个数据源。自定义数据源路由类并重写`determineCurrentLookupKey`方法，例如： ```java @Configuration public class DynamicDataSourceConfig extends AbstractRoutingDataSource { @Override protected Object determineCurrentLookupKey() { // 这里根据业务逻辑返回数据源的标识，例如请求参数、线程局部变量等 return DataSourceContextHolder.getDataSource(); } @Autowired private Map<Object, Object> dataSources; @Override public void afterPropertiesSet() { if (this.dataSources == null) { throw new IllegalArgumentException("dataSources must not be null"); } this.setTargetDataSources(dataSources); super.afterPropertiesSet(); } } ``` 在这里，`DataSourceContextHolder`是一个自定义的类，用于存储当前线程所使用的数据源。 5. **配置事务管理器**：对于多数据源，我们需要配置多个事务管理器，并通过`@Transactional`注解指定使用哪个事务管理器。例如： ```java @Configuration @EnableTransactionManagement public class TransactionConfig { @Bean(name = "primaryTransactionManager") public PlatformTransactionManager primaryTransactionManager(@Qualifier("primaryDataSource") DataSource primaryDataSource) { return new JpaTransactionManager(primaryDataSource); } @Bean(name = "secondaryTransactionManager") public PlatformTransactionManager secondaryTransactionManager(@Qualifier("secondaryDataSource") DataSource secondaryDataSource) { return new JpaTransactionManager(secondaryDataSource); } } ``` 6. **业务代码中使用**：在服务层或DAO层，通过`@Resource`或`@Autowired`注解注入需要的数据源，或者使用`@Transactional`注解指定数据源。例如： ```java @Service public class UserService { @Autowired @Qualifier("primaryDataSource") private JdbcTemplate primaryJdbcTemplate; @Autowired @Qualifier("secondaryDataSource") private JdbcTemplate secondaryJdbcTemplate; // ... } ``` 7. **数据源切换上下文**：在实际操作中，需要一个机制来控制数据源的切换，例如使用线程局部变量`ThreadLocal`存储当前数据源，或者通过拦截器来设置数据源上下文。 通过以上步骤，我们可以实现基于注解和Spring的多数据源配置和使用。这种方法使得代码更加简洁，易于理解和维护，同时提供了灵活的数据源切换能力，适应了复杂的企业级应用需求。在实际项目中，还需要考虑数据源的负载均衡、读写分离、数据同步等问题，以确保系统的稳定性和性能。