Spring AOP实现多数据源动态切换

关于Spring AOP实现多数据源动态切换的攻略，我提供如下完整的步骤：

一、添加依赖

在Maven工程的pom.xml文件中，添加如下的Spring AOP和JDBC依赖：

<dependencies>
    <dependency>
        <groupId>org.springframework</groupId>
        <artifactId>spring-aop</artifactId>
        <version>5.2.9.RELEASE</version>
    </dependency>

    <dependency>
        <groupId>org.springframework</groupId>
        <artifactId>spring-jdbc</artifactId>
        <version>5.2.9.RELEASE</version>
    </dependency>
</dependencies>

二、创建多数据源

在程序中定义多个数据源对象。例如，我们可以创建两个MySQL数据源：dataSource1和dataSource2。并对应创建两个JdbcTemplate对象。

@Bean
public DataSource dataSource1() {
    DriverManagerDataSource dataSource = new DriverManagerDataSource();
    dataSource.setDriverClassName("com.mysql.jdbc.Driver");
    dataSource.setUrl("jdbc:mysql://localhost:3306/db1");
    dataSource.setUsername("user1");
    dataSource.setPassword("user1password");  
    return dataSource;
}

@Bean
public DataSource dataSource2() {
    DriverManagerDataSource dataSource = new DriverManagerDataSource();
    dataSource.setDriverClassName("com.mysql.jdbc.Driver");
    dataSource.setUrl("jdbc:mysql://localhost:3306/db2");
    dataSource.setUsername("user2");
    dataSource.setPassword("user2password");  
    return dataSource;
}

@Bean
public JdbcTemplate jdbcTemplate1() {
    return new JdbcTemplate(dataSource1());
}

@Bean
public JdbcTemplate jdbcTemplate2() {
    return new JdbcTemplate(dataSource2());
}

三、编写AOP切面

定义一个AOP切面类，用于对Dao层的方法进行切面处理。并通过@Around注解拦截方法，然后在方法执行前后进行数据源的切换。

@Component
@Aspect
public class DataSourceAspect {

    @Pointcut("execution(* org.example.dao..*.*(..))")
    public void daoAspect() {
    }

    @Around("daoAspect()")
    public Object around(ProceedingJoinPoint pjp) throws Throwable {
        // 获取方法参数
        Object[] args = pjp.getArgs();

        // 判断方法名
        String methodName = pjp.getSignature().getName();
        if (methodName.startsWith("find")) {
            DataSourceContextHolder.setDataSource(DataSourceContextHolder.DataSourceType.DS1);
        } else {
            DataSourceContextHolder.setDataSource(DataSourceContextHolder.DataSourceType.DS2);
        }

        try {
            // 执行方法
            return pjp.proceed(args);
        } catch (Throwable throwable) {
            throw throwable;
        } finally {
            // 清除数据源信息
            DataSourceContextHolder.clearDataSource();
        }
    }

}

在上面的代码中，我们通过对方法名进行判断，决定使用哪个数据源。然后在执行方法前，将线程中的数据源切换到对应的数据源，并执行方法。执行完毕后，清空线程中的数据源信息，保证下一次执行下一次的数据源切换。

四、创建数据源上下文

定义一个线程安全的数据源上下文类DataSourceContextHolder，它可以存储每个线程相关的数据源信息，使用ThreadLocal实现：

public class DataSourceContextHolder {

    private static final ThreadLocal<DataSourceType> contextHolder = new ThreadLocal<>();

    public static void setDataSource(DataSourceType dataSource) {
        contextHolder.set(dataSource);
    }

    public static DataSourceType getDataSource() {
        return contextHolder.get();
    }

    public static void clearDataSource() {
        contextHolder.remove();
    }

    public enum DataSourceType {
        DS1, DS2;
    }

}

五、反复测试

最后，我们可以使用Spring提供的JdbcTemplate进行简单的查询操作，测试切面的效果：

@Autowired
private JdbcTemplate jdbcTemplate1;

@Autowired
private JdbcTemplate jdbcTemplate2;

public void test() {
    String sql = "SELECT COUNT(1) FROM table1";
    long count1 = jdbcTemplate1.queryForObject(sql, Long.class);
    System.out.println("count1: " + count1);

    sql = "SELECT COUNT(1) FROM table2";
    long count2 = jdbcTemplate2.queryForObject(sql, Long.class);
    System.out.println("count2: " + count2);
}

可以看到，如果调用的方法是以“find”开头，就会使用dataSource1上的数据源。否则使用dataSource2上的数据源。

这就是Spring AOP实现多数据源动态切换的完整攻略。