如何在JpaTransactionManager事务开启前执行SQL适配PolarDB-X

核心问题分析 #

直接替换JpaTransactionManager为DataSourceTransactionManager会导致JPA事务失效，因为JPA的事务上下文需要绑定EntityManager，而DataSourceTransactionManager仅管理JDBC层面事务，无法关联JPA的实体管理器，最终导致数据无法持久化。

我们需要在JPA事务开启阶段执行SET drds_transaction_policy = '2PC'，同时保留JPA事务的上下文绑定，以下是两种可行方案：

方案一：自定义JpaTransactionManager #

通过继承JpaTransactionManager并重写事务初始化方法，在事务启动后立即执行PolarDB-X的事务策略配置：

1. 实现自定义事务管理器 #

import org.springframework.orm.jpa.JpaTransactionManager;
import org.springframework.orm.jpa.JpaTransactionObject;
import org.springframework.transaction.TransactionDefinition;
import org.springframework.transaction.TransactionException;
import javax.persistence.EntityManager;
import java.sql.Connection;
import java.sql.Statement;
import java.sql.SQLException;

public class PolarDBXJpaTransactionManager extends JpaTransactionManager {

    public PolarDBXJpaTransactionManager() {
        super();
    }

    public PolarDBXJpaTransactionManager(javax.persistence.EntityManagerFactory emf) {
        super(emf);
    }

    @Override
    protected void doBegin(Object transaction, TransactionDefinition definition) throws TransactionException {
        // 先执行父类的事务初始化逻辑，确保JPA上下文绑定完成
        super.doBegin(transaction, definition);
        
        // 获取当前事务绑定的EntityManager
        JpaTransactionObject txObject = (JpaTransactionObject) transaction;
        EntityManager em = txObject.getEntityManagerHolder().getEntityManager();
        
        try {
            // 从EntityManager中解包获取JDBC连接
            Connection conn = em.unwrap(Connection.class);
            try (Statement stmt = conn.createStatement()) {
                stmt.executeUpdate("SET drds_transaction_policy = '2PC'");
            }
        } catch (SQLException e) {
            throw new TransactionException("配置PolarDB-X 2PC事务策略失败", e);
        }
    }
}

2. 配置启用自定义事务管理器 #

Java配置方式

@Configuration
@EnableTransactionManagement
public class JpaConfig {

    @Autowired
    private EntityManagerFactory entityManagerFactory;

    @Bean
    public PlatformTransactionManager transactionManager() {
        return new PolarDBXJpaTransactionManager(entityManagerFactory);
    }
}

XML配置方式

<bean id="transactionManager" class="com.yourpackage.PolarDBXJpaTransactionManager">
    <constructor-arg ref="entityManagerFactory"/>
</bean>

方案二：包装DataSource实现自动配置 #

通过自定义DataSource包装类，在获取连接时判断当前是否处于活跃事务中，自动执行事务策略配置，无需修改事务管理器：

1. 实现PolarDB-X专用数据源 #

import org.springframework.jdbc.datasource.DelegatingDataSource;
import org.springframework.transaction.support.TransactionSynchronizationManager;
import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;

public class PolarDBXDataSource extends DelegatingDataSource {

    public PolarDBXDataSource(DataSource targetDataSource) {
        super(targetDataSource);
    }

    @Override
    public Connection getConnection() throws SQLException {
        Connection conn = super.getConnection();
        // 仅在活跃事务中执行配置
        if (TransactionSynchronizationManager.isActualTransactionActive()) {
            try (Statement stmt = conn.createStatement()) {
                stmt.executeUpdate("SET drds_transaction_policy = '2PC'");
            }
        }
        return conn;
    }

    @Override
    public Connection getConnection(String username, String password) throws SQLException {
        Connection conn = super.getConnection(username, password);
        if (TransactionSynchronizationManager.isActualTransactionActive()) {
            try (Statement stmt = conn.createStatement()) {
                stmt.executeUpdate("SET drds_transaction_policy = '2PC'");
            }
        }
        return conn;
    }
}

2. 配置启用包装后的数据源 #

@Bean
public DataSource dataSource() {
    // 初始化基础数据源（例如HikariCP）
    HikariDataSource hikariDS = new HikariDataSource();
    hikariDS.setJdbcUrl("jdbc:mysql://your-polardb-x-endpoint:3306/your_db");
    hikariDS.setUsername("your_username");
    hikariDS.setPassword("your_password");
    
    // 用PolarDBXDataSource包装基础数据源
    return new PolarDBXDataSource(hikariDS);
}

方案对比 #

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