Spring Boot 配置 Mybatis 读写分离

JPA 的读写分离配置不能应用在 Mybatis 上, 所以 Mybatis 要单独处理

为了不影响原有代码, 使用了增加拦截器的方式, 在拦截器里根据 SQL 的 CRUD 来路由到不同的数据源

需要单独增加Mybatis的配置

java 复制代码
@Configuration
public class MyBatisConfig {
    @Bean
    public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception {
        SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
        sessionFactory.setDataSource(dataSource);

        // 注册插件
        sessionFactory.setPlugins(new Interceptor[]{new DataSourceInterceptor()});

        return sessionFactory.getObject();
    }
}

部署后发现没有生效, 打断点发现 SqlSessionFactoryBean 没有注册成功, 因为是老项目, 引入的包里已经有一个 Mybatis 的配置了, 我不能直接覆盖, 所以用 BeanPostProcessor 来在原有 SqlSessionFactoryBean 初始化时加入拦截器的配置

java 复制代码
@Configuration
public class MyBatisConfig {
    @Bean
    @ConditionalOnBean(name="routingDataSource")
    public BeanPostProcessor sqlSessionFactoryBeanPostProcessor(@Qualifier("routingDataSource") DataSource routingDataSource) {
        return new BeanPostProcessor() {
            @Override
            public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
                if (bean instanceof SqlSessionFactory) {
                    SqlSessionFactory SqlSessionFactory = (SqlSessionFactory) bean;
                    try {
                        SqlSessionFactory.getConfiguration().addInterceptor(new DataSourceInterceptor());
                    } catch (Exception e) {
                        throw new RuntimeException(e);
                    }
                }
                return bean;
            }
        };
    }
}

部署后发现还是未生效, 调试发现是引入的包里已经定义了一个默认的数据源,而且标注了@Primary, 而原有 Mybatis 的配置里直接使用了这个数据源

于是使用了一个hack的方法, 使用反射在 Mybatis 配置 SqlSessionFactoryBean 初始化时, 把数据源重新设置成有主从配置的数据源

java 复制代码
@Configuration
public class MyBatisConfig {
    @Bean
    @ConditionalOnBean(name="routingDataSource")
    public BeanPostProcessor sqlSessionFactoryBeanPostProcessor(@Qualifier("routingDataSource") DataSource routingDataSource) {
        return new BeanPostProcessor() {
@Override
            public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
                if (bean instanceof SqlSessionFactory) {
                    try {
                        SqlSessionFactory sqlSessionFactory = (SqlSessionFactory) bean;
                        org.apache.ibatis.session.Configuration configuration = sqlSessionFactory.getConfiguration();

                        // 使用反射或其他方式修改配置中的数据源
                        Field dataSourceField = configuration.getEnvironment().getClass().getDeclaredField("dataSource");
                        dataSourceField.setAccessible(true);
                        dataSourceField.set(configuration.getEnvironment(), routingDataSource);
                    } catch (Exception e) {
                        throw new BeansException("Failed to modify SqlSessionFactory", e) {};
                    }

                }
                return bean;
            }
        };
    }
}

再次部署测试通过

相关推荐
无心水28 分钟前
深入Java线程池:BlockingQueue实现全景解析与实战指南
java·后端·面试
Java水解32 分钟前
JAVA面试题大全(200+道题目)
java·后端·面试
Predestination王瀞潞34 分钟前
Java EE开发技术(第四章:过滤器和监听器银行管理系统-下)
java·java-ee
摇滚侠37 分钟前
Spring Boot3零基础教程,Profile 环境隔离用法,笔记55
java·spring boot·笔记
编程岁月37 分钟前
java面试-0305-java线程调度方法?sleep()和wait()区别?
java·开发语言·面试
DoveLx1 小时前
如何集成ElasticSearch到Spring Boot+MySQL项目中
java·elasticsearch
lazily-c1 小时前
Web后端开发总结
java·学习·web·1024程序员节
凭君语未可2 小时前
深度解析Java的多态特性
java·开发语言
华仔啊2 小时前
JVM参数到底配在哪?7大场景全解,新手不再迷茫!
java·jvm
penguin_bark3 小时前
C++ 异步编程(future、promise、packaged_task、async)
java·开发语言·c++