Springboot项目实现Mysql多数据源切换的完整实例

下面是完整的攻略说明：

1. 前言

在实际开发中，一个服务可能需要涉及多个数据库，为了不同的业务之间数据互不干扰，我们需要对不同的业务使用不同的数据库。Spring Boot提供了良好的支持，使得我们很容易地实现多数据源切换。本文将介绍如何使用Spring Boot来实现Mysql多数据源切换。

2. 配置多数据源

在Spring Boot中，要使用多数据源，需要配置多个数据源和一个主数据源。在配置多个数据源时，我们需要提供各自的JDBC配置，并指定数据源的名称。在Spring Boot应用程序启动时，它会自动将多个数据源添加到上下文中，并且我们可以根据数据源名称来访问它们。

下面是一个例子，配置了两个数据源"db1"和"db2"：

spring:
  datasource:
    db1:
      url: jdbc:mysql://localhost:3306/db1
      username: root
      password: password1
      driver-class-name: com.mysql.jdbc.Driver
    db2:
      url: jdbc:mysql://localhost:3306/db2
      username: root
      password: password2
      driver-class-name: com.mysql.jdbc.Driver

3. 配置多数据源访问

在配置多数据源之后，需要有一个机制来选择特定的数据源。这可以通过将选定的数据源放入一个ThreadLocal变量中来实现。我们可以写一个AOP切面，在每个方法调用前根据需要的数据源名称选择数据源，然后在方法返回后清除选择的数据源。

下面是一个例子，定义了一个DynamicDataSource类，它实现了数据源选择的逻辑：

public class DynamicDataSource extends AbstractRoutingDataSource {

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

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

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

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

DynamicDataSource继承了AbstractRoutingDataSource，它实现了determineCurrentLookupKey方法来选择当前的数据源。setDataSource方法用于设置当前数据源名称，clearDataSource方法用于清除当前数据源名称。

然后，在我们需要使用多数据源的方法上使用@DS注解，它可以将指定的数据源名称放入DynamicDataSource中。

@Target({ ElementType.METHOD, ElementType.TYPE })
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DS {
    String value() default "db1";
}

我们还需要写一个AOP切面，用于在方法调用前根据需要的数据源名称选择数据源。切面的实现类如下：

@Component
@Aspect
public class DataSourceAspect {

    @Around("@annotation(com.example.multidatasource.annotation.DS) && execution(* com.example.multidatasource.service..*(..))")
    public Object dynamicDataSource(ProceedingJoinPoint joinPoint) throws Throwable {
        MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
        DS ds = methodSignature.getMethod().getAnnotation(DS.class);
        if(ds != null) {
            DynamicDataSource.setDataSource(ds.value());
        }
        try {
            return joinPoint.proceed();
        }
        finally {
            DynamicDataSource.clearDataSource();
        }
    }
}

@Around表明它是一个环绕通知切面，切面的切点为使用@DS注解的所有方法。环绕通知切面会拦截并执行方法，我们可以在执行方法前后进行操作。方法参数joinPoint包含了执行方法的相关信息。在该切面中，我们先获取@DS注解，然后根据注解值设置指定的数据源，最后调用切点方法。方法执行完成后，我们清除ThreadLocal中的数据源名称。

4. 测试多数据源

下面是一个例子，实现了根据用户名查找用户，使用了两个数据源"db1"和"db2"：

@Service
public class UserServiceImpl implements UserService {

    @Autowired
    private UserMapper userMapper1;

    @Autowired
    private UserMapper userMapper2;

    @DS("db1")
    @Override
    public User getUserByName1(String name) {
        return userMapper1.getUserByName(name);
    }

    @DS("db2")
    @Override
    public User getUserByName2(String name) {
        return userMapper2.getUserByName(name);
    }
}

在getUserByName1方法和getUserByName2方法上使用了@DS注解，分别指定了使用的数据源。UserMapper是一个简单的Mybatis Mapper， 查询用户信息的方法如下：

public interface UserMapper {
    User getUserByName(String name);
}

5. 总结

通过以上配置和代码实现，我们就可以在Spring Boot中实现Mysql多数据源切换的功能了。在获取数据源时，使用@DS注解就可以执行选择的数据源了。当然，实际项目中还有很多细节需要考虑，比如使用连接池、事务处理等。