读写数据源路由下自定义事务隔离级别异常问题求解

问题的核心矛盾在于：HibernateJpaDialect的prepareConnection属性开启时（默认值），Spring会提前获取数据库连接并设置隔离级别，导致数据源路由逻辑（基于ThreadLocal上下文）无法触发；关闭该属性后路由生效，但Spring无法干预连接的隔离级别设置。以下是两种可行的解决思路：

方案一：自定义JpaDialect，在事务阶段绑定数据源上下文 #

通过继承HibernateJpaDialect，重写事务开始逻辑，确保在Spring获取连接并设置隔离级别前，先完成数据源路由上下文的绑定。

1. 实现自定义JpaDialect #

public class RoutingHibernateJpaDialect extends HibernateJpaDialect {

    @Override
    public Object beginTransaction(EntityManager entityManager, TransactionDefinition definition) 
            throws PersistenceException, SQLException, TransactionException {
        
        // 1. 根据事务属性绑定数据源上下文
        if (definition.isReadOnly()) {
            DataSourceContextHolder.setSlave();
        } else {
            DataSourceContextHolder.setMaster();
        }

        // 2. 调用父类方法，此时已绑定上下文，获取连接时会路由到正确数据源
        Object transactionData = super.beginTransaction(entityManager, definition);

        // 3. 手动处理自定义隔离级别（父类已处理，但显式保证逻辑）
        if (definition.getIsolationLevel() != TransactionDefinition.ISOLATION_DEFAULT) {
            Connection connection = getJdbcConnection(entityManager, true);
            try {
                connection.setTransactionIsolation(definition.getIsolationLevel());
            } catch (SQLException ex) {
                throw new TransactionSystemException("Failed to set transaction isolation level", ex);
            } finally {
                releaseJdbcConnection(connection, entityManager);
            }
        }

        return transactionData;
    }

    @Override
    public void cleanupTransaction(Object transactionData) {
        super.cleanupTransaction(transactionData);
        // 事务结束后清理上下文，避免线程污染
        DataSourceContextHolder.clear();
    }
}

2. 配置JPA使用自定义Dialect #

保持prepareConnection为默认的true，确保Spring能干预连接的隔离级别设置：

@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactory() {
    LocalContainerEntityManagerFactoryBean em = new LocalContainerEntityManagerFactoryBean();
    em.setDataSource(routingDataSource());
    em.setPackagesToScan("com.your.package.entity");

    HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
    em.setJpaVendorAdapter(vendorAdapter);

    // 绑定自定义Dialect
    RoutingHibernateJpaDialect jpaDialect = new RoutingHibernateJpaDialect();
    em.setJpaDialect(jpaDialect);

    // 设置Hibernate连接释放模式为ON_CLOSE，匹配异常提示要求
    em.setJpaPropertyMap(Map.of(
        "hibernate.connection.release_mode", "on_close",
        "hibernate.hbm2ddl.auto", "update"
    ));

    return em;
}

3. 确保路由数据源与上下文类正常工作 #

你的AbstractRoutingDataSource实现需正确读取ThreadLocal上下文：

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

// 线程上下文持有类
public class DataSourceContextHolder {
    private static final ThreadLocal<String> CONTEXT = new ThreadLocal<>();

    public static void setMaster() {
        CONTEXT.set("master");
    }

    public static void setSlave() {
        CONTEXT.set("slave");
    }

    public static String getDataSourceType() {
        return CONTEXT.get();
    }

    public static void clear() {
        CONTEXT.remove();
    }
}

方案二：高优先级切面预绑定数据源上下文 #

通过Spring AOP切面，在事务切面执行前（设置更高优先级），根据@Transactional注解的属性绑定数据源上下文，确保Spring获取连接时路由逻辑已生效。

1. 实现数据源路由切面 #

@Aspect
@Component
@Order(Ordered.HIGHEST_PRECEDENCE) // 确保优先级高于事务切面
public class DataSourceRoutingAspect {

    @Around("@annotation(transactional)")
    public Object routeDataSource(ProceedingJoinPoint joinPoint, Transactional transactional) throws Throwable {
        try {
            // 根据事务只读属性绑定上下文
            if (transactional.readOnly()) {
                DataSourceContextHolder.setSlave();
            } else {
                DataSourceContextHolder.setMaster();
            }
            return joinPoint.proceed();
        } finally {
            DataSourceContextHolder.clear();
        }
    }
}

2. 保持JPA默认配置 #

无需修改HibernateJpaDialect的prepareConnection属性（默认开启），只需确保Hibernate连接释放模式设置为ON_CLOSE：

# application.properties
spring.jpa.properties.hibernate.connection.release_mode=on_close

关键注意事项 #

1. 事务与非事务场景区分：上述方案仅覆盖事务内的路由，非事务的只读查询需额外通过切面或方法注解绑定上下文。
2. 线程上下文清理：务必在事务/方法结束后清理ThreadLocal变量，防止线程池复用导致的数据源路由错误。
3. 隔离级别验证：测试自定义隔离级别时，可通过数据库日志或连接元数据确认隔离级别是否正确应用到目标数据源。

内容的提问来源于stack exchange，提问作者martin kimani