使用Spring的AbstractRoutingDataSource实现多数据源切换示例

以下是使用Spring的AbstractRoutingDataSource实现多数据源切换的完整攻略。

1. 引入依赖

首先需要在项目中引入Spring的相关依赖，其中包括Spring JDBC、Spring AOP和Spring Context等模块。最新版本的Spring依赖可以在Maven中心库中获取，或者可以到Spring官网查看最新的版本信息。

2. 配置数据源

在Spring的配置文件中，为每个数据源配置DataSource和JdbcTemplate的bean，并将这些bean注入到AbstractRoutingDataSource子类的相关属性中。

假设我们需要配置两个数据源，分别是DB1和DB2，那么在配置文件中应该如下所示：

<!-- 配置DB1数据源 -->
<bean id="db1DataSource" class="org.apache.commons.dbcp2.BasicDataSource">
    <property name="driverClassName" value="com.mysql.cj.jdbc.Driver"/>
    <property name="url" value="jdbc:mysql://localhost:3306/db1"/>
    <property name="username" value="username"/>
    <property name="password" value="password"/>
</bean>

<!-- 配置DB2数据源 -->
<bean id="db2DataSource" class="org.apache.commons.dbcp2.BasicDataSource">
    <property name="driverClassName" value="com.mysql.cj.jdbc.Driver"/>
    <property name="url" value="jdbc:mysql://localhost:3306/db2"/>
    <property name="username" value="username"/>
    <property name="password" value="password"/>
</bean>

<!-- 使用AbstractRoutingDataSource实现多数据源切换 -->
<bean id="dynamicDataSource" class="com.example.DynamicDataSource">
    <property name="targetDataSources">
        <map key-type="java.lang.String">
            <entry key="db1" value-ref="db1DataSource"/>
            <entry key="db2" value-ref="db2DataSource"/>
        </map>
    </property>
    <property name="defaultTargetDataSource" ref="db1DataSource"/>
</bean>

3. 实现AbstractRoutingDataSource子类

在项目中创建一个继承自AbstractRoutingDataSource的子类，并重写determineCurrentLookupKey方法，该方法会由Spring负责调用，用于获取当前数据源的key值。

例如，以下是DynamicDataSource类的示例代码：

public class DynamicDataSource extends AbstractRoutingDataSource {
    private static final ThreadLocal<String> dataSourceKey = new ThreadLocal<>();

    @Override
    protected Object determineCurrentLookupKey() {
        return dataSourceKey.get();
    }

    public static void setDataSourceKey(String dataSource) {
        dataSourceKey.set(dataSource);
    }

    public static void clearDataSourceKey() {
        dataSourceKey.remove();
    }
}

其中，determineCurrentLookupKey方法返回的是一个String类型的key值，这个值会被用于选择正确的数据源。在这个示例中，使用了一个ThreadLocal对象来保存当前线程使用的数据源的key值。

4. 切换数据源

当需要切换数据源时，只需要调用DynamicDataSource的静态方法setDataSourceKey来设置当前数据源的key值即可，例如：

DynamicDataSource.setDataSourceKey("db2");

示例1：动态切换数据源

以下是一个采用注解方式切换数据源的示例：

@Service
public class UserService {
    @Autowired
    private JdbcTemplate jdbcTemplate;

    @DataSource("db1")
    public void insertUser(User user) {
        String sql = "insert into t_user(id, name, age) values (?,?,?)";
        jdbcTemplate.update(sql, user.getId(), user.getName(), user.getAge());
    }

    @DataSource("db2")
    public User getUserById(int id) {
        String sql = "select * from t_user where id = ?";
        return jdbcTemplate.queryForObject(sql, new Object[]{id}, new UserRowMapper());
    }
}

其中，@DataSource注解用于标识当前方法需要使用的数据源。切换数据源的方式是在方法执行之前，使用切面来获取@DataSource注解中设置的数据源key值，并设置到DynamicDataSource的ThreadLocal对象中。

以下是切面的示例代码：

@Aspect
@Component
public class DataSourceAspect {
    @Pointcut("@annotation(com.example.DataSource)")
    public void dataSourcePointCut() {
    }

    @Around("dataSourcePointCut()")
    public Object around(ProceedingJoinPoint point) throws Throwable {
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();

        DataSource dataSource = method.getAnnotation(DataSource.class);
        if (dataSource != null) {
            DynamicDataSource.setDataSourceKey(dataSource.value());
        }

        try {
            return point.proceed();
        } finally {
            DynamicDataSource.clearDataSourceKey();
        }
    }
}

在以上示例中，定义了一个切面DataSourceAspect，并通过@Pointcut注解指定了需要拦截的方法，然后在@Around注解的方法中获取到@DataSource注解中设置的数据源key值，并设置到DynamicDataSource中，然后调用目标方法，最后清空ThreadLocal对象中的数据源key值。

示例2：通过参数切换数据源

以下是一个采用参数来切换数据源的示例：

@Service
public class UserService {
    @Autowired
    private JdbcTemplate jdbcTemplate;

    public void insertUser(User user, String dataSource) {
        DynamicDataSource.setDataSourceKey(dataSource);

        String sql = "insert into t_user(id, name, age) values (?,?,?)";
        jdbcTemplate.update(sql, user.getId(), user.getName(), user.getAge());

        DynamicDataSource.clearDataSourceKey();
    }

    public User getUserById(int id, String dataSource) {
        DynamicDataSource.setDataSourceKey(dataSource);

        String sql = "select * from t_user where id = ?";
        User user = jdbcTemplate.queryForObject(sql, new Object[]{id}, new UserRowMapper());

        DynamicDataSource.clearDataSourceKey();

        return user;
    }
}

在以上示例中，insertUser和getUserById方法中都增加了一个String类型的参数dataSource，用于指定数据源key值。在方法开始时，通过调用DynamicDataSource的静态方法setDataSourceKey来设置当前线程的数据源key值，在方法结束时，再调用DynamicDataSource的静态方法clearDataSourceKey来清空ThreadLocal对象中的数据源key值。

调用示例代码：

@Autowired
private UserService userService;

public void test() {
   User user1 = new User("1", "Tom", 18);
   User user2 = new User("2", "Jerry", 20);

   userService.insertUser(user1, "db1");
   userService.insertUser(user2, "db2");

   User user3 = userService.getUserById(1, "db1");
   User user4 = userService.getUserById(2, "db2");
}