Слово pure не соответствует слову hibernate .
Я делюсь своими кодами.
Допустим, мы можем определить метод для использования Connection, полученный из EntityManager.
static <R> applyConnection(final EntityManager manager,
final Function<Connection, R> function) {
if (manager == null) {
throw new NullPointerException("manager is null");
}
if (function == null) {
throw new NullPointerException("function is null");
}
// we gonna fill here up
throw new RuntimeException("failed to work with a connection");
}
EclipseLink
Это довольно просто, как описано в ссылке выше.
- Обратите внимание, что
EntityManager должен быть присоединен к Transaction, иначе метод unwrap вернет null. (Не очень хороший ход.)
- Я не уверен, что ответственность за закрытие соединения.
// --------------------------------------------------------- EclipseLink
try {
final Connection connection = manager.unwrap(Connection.class);
if (connection != null) { // manage is not in any transaction
return function.apply(connection);
}
} catch (final PersistenceException pe) {
logger.log(FINE, pe, () -> "failed to unwrap as a connection");
}
Hibernate
Это должно быть сделано в основном со следующими кодами.
// using vendor specific APIs
final Session session = (Session) manager.unwrap(Session.class);
//return session.doReturningWork<R>(function::apply);
return session.doReturningWork(new ReturningWork<R>() {
@Override public R execute(final Connection connection) {
return function.apply(connection);
}
});
Что ж, нам (по крайней мере, мне) может не потребоваться никаких зависимостей от поставщика. Прокси приходит на помощь.
try {
// See? You shouldn't fire me, ass hole!!!
final Class<?> sessionClass
= Class.forName("org.hibernate.Session");
final Object session = manager.unwrap(sessionClass);
final Class<?> returningWorkClass
= Class.forName("org.hibernate.jdbc.ReturningWork");
final Method executeMethod
= returningWorkClass.getMethod("execute", Connection.class);
final Object workProxy = Proxy.newProxyInstance(
lookup().lookupClass().getClassLoader(),
new Class[]{returningWorkClass},
(proxy, method, args) -> {
if (method.equals(executeMethod)) {
final Connection connection = (Connection) args[0];
return function.apply(connection);
}
return null;
});
final Method doReturningWorkMethod = sessionClass.getMethod(
"doReturningWork", returningWorkClass);
return (R) doReturningWorkMethod.invoke(session, workProxy);
} catch (final ReflectiveOperationException roe) {
logger.log(Level.FINE, roe, () -> "failed to work with hibernate");
}
OpenJPA
Я не уверен, что OpenJPA уже использует способ с использованием unwrap(Connection.class), но это можно сделать способом, описанным в одной из приведенных выше ссылок.
Не ясно, ответственность за закрытие соединения. Документ (одна из приведенных выше ссылок) явно говорит, но я плохо разбираюсь в английском.
try {
final Class<?> k = Class.forName(
"org.apache.openjpa.persistence.OpenJPAEntityManager");
if (k.isInstance(manager)) {
final Method m = k.getMethod("getConnection");
try {
try (Connection c = (Connection) m.invoke(manager)) {
return function.apply(c);
}
} catch (final SQLException sqle) {
logger.log(FINE, sqle, () -> "failed to work with openjpa");
}
}
} catch (final ReflectiveOperationException roe) {
logger.log(Level.FINE, roe, () -> "failed to work with openjpa");
}
ПРИЛОЖЕНИЕ
static <U, R> R applyConnection(
final EntityManager manager,
final BiFunction<Connection, U, R> function, final U u) {
if (manager == null) {
throw new NullPointerException("manager is null");
}
if (function == null) {
throw new NullPointerException("function is null");
}
return applyConnection(manager, t -> function.apply(t, u));
}
static void acceptConnection(
final EntityManager manager, final Consumer<Connection> consumer) {
if (manager == null) {
throw new NullPointerException("manager is null");
}
if (consumer == null) {
throw new NullPointerException("consumer is null");
}
applyConnection(
manager,
t -> {
consumer.accept(t);
return null;
}
);
}
static <U> void acceptConnection(
final EntityManager manager,
final BiConsumer<Connection, U> consumer, final U u) {
if (manager == null) {
throw new NullPointerException("manager is null");
}
if (consumer == null) {
throw new NullPointerException("consumer is null");
}
acceptConnection(manager, t -> consumer.accept(t, u));
}