JPA：如何设置MySQL会话变量？

Question

我需要为我的应用程序设置一个mysql会话变量，以便按照预期使用MariaDB Galera集群。SQL调用是:set SESSION wsrep_sync_wait=1。当应用程序使用数据库时，应始终设置该值。我使用EclipseLink作为JPA提供程序。

我的问题是：实现这一点的最佳途径是什么？

public class SessionCustomizerImpl implements org.eclipse.persistence.config.SessionCustomizer {

    private final static String WSREP_SYNC_WAIT_CHECK_SQL = "SHOW SESSION VARIABLES LIKE 'wsrep_sync_wait'";
    private final static String WSREP_SYNC_WAIT_SET_SQL = "SET SESSION wsrep_sync_wait = 1";

    @Override
    public void customize(Session session) throws Exception {
        Vector result = session.executeSQL(WSREP_SYNC_WAIT_CHECK_SQL);
        if ((result != null) && !result.isEmpty()) {
            session.executeNonSelectingSQL(WSREP_SYNC_WAIT_SET_SQL);
            // Galera connection detected; wsrep_sync_wait set to 1
        } else {
            // No Galera connection detected; wsrep_sync_wait not set
        }

    }
}

这对我不起作用。从EntityManager查询会话变量将返回0值。

选项2:EntityManager工厂

每次创建新的EntityManager时，都会执行SQL。

public class SyncWaitEntityManagerFactory implements Factory<EntityManager> {
    private final EntityManagerFactory emf;

    @Inject
    public SyncWaitEntityManagerFactory(EntityManagerFactory emf) {
        this.emf = emf;
    }

    @Override
    public EntityManager provide() {
        final EntityManager em = emf.createEntityManager();
        // set it
        em.getTransaction().begin();
        em.createNativeQuery("SET SESSION wsrep_sync_wait = 1").executeUpdate();
        em.getTransaction().commit();

        return em;
    }

    @Override
    public void dispose(EntityManager instance) {
        if (instance.isOpen()) {
            instance.close();
        }
    }

}

    String jdbcUrl = "jdbc:mysql://db.example.test:3306/"+ JDBC_DB 
            +"?sessionVariables=wsrep_sync_wait=1";
    Properties p = new Properties();
    p.put("javax.persistence.jdbc.url", jdbcUrl);
    p.put("javax.persistence.jdbc.user", JDBC_USER);
    p.put("javax.persistence.jdbc.password", JDBC_PASSWORD);
    EntityManagerFactory emf = Persistence.createEntityManagerFactory("myPU", p);

    EntityManager entityManager = emf.createEntityManager();

不错的解决方案。为我工作；无需努力，无需交易。缺点：我不能捕捉异常（例如：首先检查变量是否存在，然后设置它--允许在不支持/使用这个特定变量的系统上部署代码）。

Answer 1

您还可以在每次调用getConnection（）时使用一个方面来执行查询，这基本上是针对每个事务（在调用之后设置方面，以便我们有一个有效的连接对象）：

@Component
@Aspect
public class CustomConnectionPreparer implements ConnectionPreparer
{
    @AfterReturning(pointcut = "execution(* *.getConnection(..))", returning = "connection")
    public Connection prepare(Connection connection) throws SQLException {
        // execute the query (also exception handling)
        try (Statement statement = connection.createStatement()) {
            statement.execute("SELECT 1");
        } catch (SQLException e) {
            throw e;
        }
    
        return connection;
    }
}