mybatis动态切换数据源怎么实现 - 问答

实现MyBatis动态切换数据源可以通过以下几个步骤：

创建多个数据源：首先，你需要在Spring配置文件中配置多个数据源，例如配置两个数据源dataSource1和dataSource2。
创建数据源切换器：创建一个数据源切换器，用于在运行时切换数据源。可以通过ThreadLocal或者AOP等方式来实现。
配置MyBatis的SqlSessionFactory：在MyBatis的配置文件中，配置多个SqlSessionFactory，并分别指定不同的数据源。
创建Mapper接口：为每个数据源创建对应的Mapper接口。
创建Mapper映射文件：为每个Mapper接口创建对应的Mapper映射文件，分别指定不同的数据源。
配置动态数据源路由器：在Spring配置文件中配置动态数据源路由器，用于根据不同的条件选择相应的数据源。
使用动态数据源：在需要使用不同数据源的地方，通过调用动态数据源路由器的方法来切换数据源。

下面是一个简单的示例代码：

public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDataSource();
    }
}

public class DataSourceContextHolder {
    private static final ThreadLocal<String> dataSourceHolder = new ThreadLocal<>();

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

    public static String getDataSource() {
        return dataSourceHolder.get();
    }

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

@Configuration
public class DataSourceConfig {
    @Bean
    @Primary
    @ConfigurationProperties("spring.datasource.datasource1")
    public DataSource dataSource1() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties("spring.datasource.datasource2")
    public DataSource dataSource2() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    public DynamicDataSource dynamicDataSource() {
        DynamicDataSource dataSource = new DynamicDataSource();
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put("dataSource1", dataSource1());
        targetDataSources.put("dataSource2", dataSource2());
        dataSource.setTargetDataSources(targetDataSources);
        dataSource.setDefaultTargetDataSource(dataSource1());
        return dataSource;
    }

    @Bean
    public SqlSessionFactory sqlSessionFactory(DynamicDataSource dynamicDataSource) throws Exception {
        SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
        sessionFactory.setDataSource(dynamicDataSource);
        return sessionFactory.getObject();
    }
}

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

@Aspect
@Component
public class DataSourceAspect {
    @Before("@annotation(dataSource)")
    public void beforeSwitchDataSource(JoinPoint joinPoint, DataSource dataSource) {
        DataSourceContextHolder.setDataSource(dataSource.value());
    }

    @After("@annotation(dataSource)")
    public void afterSwitchDataSource(JoinPoint joinPoint, DataSource dataSource) {
        DataSourceContextHolder.clearDataSource();
    }
}

@Mapper
public interface UserMapper {
    @DataSource("dataSource1")
    List<User> getAllUsersFromDataSource1();

    @DataSource("dataSource2")
    List<User> getAllUsersFromDataSource2();
}

@Service
public class UserService {
    @Autowired
    private UserMapper userMapper;

    public List<User> getAllUsersFromDataSource1() {
        return userMapper.getAllUsersFromDataSource1();
    }

    public List<User> getAllUsersFromDataSource2() {
        return userMapper.getAllUsersFromDataSource2();
    }
}

以上代码中，通过@Configuration注解配置了多个数据源，并使用@Primary注解指定了默认的数据源。在DynamicDataSource类中，使用AbstractRoutingDataSource来实现动态切换数据源。利用ThreadLocal来存储当前线程的数据源选择。在DataSourceAspect类中，使用AOP来切换数据源。在UserMapper接口中，使用@DataSource注解指定使用哪个数据源。在UserService类中，调用UserMapper接口的方法来操作数据库。最后，在需要切换数据源的地方，使用DataSourceContextHolder.setDataSource()方法来切换数据源。

0 赞

0 踩