spring aop + 自定义注解的形式实现多数据源切换

Spring AOP（面向切面编程）是Spring框架中的一个组件，它允许您以一种声明性的方式来处理横切关注点（如事务管理，日志记录等）。 通过使用AOP，可以将这些关注点从应用程序的主体中分离出来，从而实现代码的复用和灵活性。

在使用Spring框架中实现多数据源的切换时，可以使用自定义注解的形式来实现。首先，首先在应用程序的主体中定义两个数据源。 然后，可以定义一个自定义注解，用于标识哪些方法应该使用哪个数据源。例如，使用“@Primary”注解标记主要数据源，使用“@Secondary”注解标记辅助数据源。

然后，在Spring配置中定义一个AOP切面，该切面使用上述自定义注解来切换数据源。下面是这种方法的一个示例：

@Aspect
@Component
public class DataSourceAspect {

    @Around("@annotation(Primary)")
    public Object primaryDataSource(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {
        // 切换到主要数据源
        DynamicDataSourceContextHolder.setDataSource(DynamicDataSourceContextHolder.DATA_SOURCE_PRIMARY);

        try {
            return proceedingJoinPoint.proceed();
        } finally {
            // 切换回默认数据源
            DynamicDataSourceContextHolder.clearDataSource();
        }
    }

    @Around("@annotation(Secondary)")
    public Object secondaryDataSource(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {
        // 切换到辅助数据源
        DynamicDataSourceContextHolder.setDataSource(DynamicDataSourceContextHolder.DATA_SOURCE_SECONDARY);

        try {
            return proceedingJoinPoint.proceed();
        } finally {
            // 切换回默认数据源
            DynamicDataSourceContextHolder.clearDataSource();
        }
    }
}

在上面的代码中，我们可以看到“@Around”注解被用于定义一个环绕通知，该通知基于使用“@Primary”或“@Secondary”注解的方法进行拦截。 在方法执行之前，我们使用“DynamicDataSourceContextHolder”来将数据源设置为主要或辅助数据源。 在方法执行完成之后，我们将数据源切换回默认数据源。

最后，我们可以将“@Primary”和“@Secondary”注解带到相应的方法上，以切换不同的数据源，例如：

@Service
public class UserServiceImpl implements UserService {

    @Autowired
    private UserDao userDao;

    @Override
    @Primary
    public User getUserById(long id) {
        return userDao.getUserById(id);
    }

    @Override
    @Secondary
    public List<User> getAllUsers() {
        return userDao.getAllUsers();
    }
}

在上面的代码中，我们可以看到“@Primary”注解被用于getUserById（）方法，表示这个方法应该从主要数据源中读取数据。相反，getAllUsers（）方法被标记为“@Secondary”注解，表示这个方法应该从辅助数据源中读取数据。

通过这种方式，我们可以很容易地切换应用程序中的不同数据源，并且代码的重复率很低。这种方法适用于需要在应用程序的不同部分使用不同数据源的多租户应用程序。