/*
* JBoss, Home of Professional Open Source.
* Copyright 2011, Red Hat, Inc., and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.as.jpa.container;
import java.io.IOException;
import java.io.Serializable;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.util.Map;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import org.jboss.as.jpa.service.PersistenceUnitServiceImpl;
import org.jboss.as.jpa.transaction.TransactionUtil;
import org.jboss.as.jpa.util.JPAServiceNames;
import org.jboss.as.server.CurrentServiceContainer;
import org.jboss.msc.service.ServiceContainer;
import org.jboss.msc.service.ServiceController;
import static org.jboss.as.jpa.JpaMessages.MESSAGES;
/**
* Transaction scoped entity manager will be injected into SLSB or SFSB beans. At bean invocation time, they
* will join the active transaction if one is present. Otherwise, they will simply be cleared at the end of
* the bean invocation.
* <p/>
* This is a proxy for the underlying persistent provider EntityManager.
*
* @author Scott Marlow
*/
public class TransactionScopedEntityManager extends AbstractEntityManager implements Serializable {
private static final long serialVersionUID = 455498111L;
private final String puScopedName; // Scoped name of the persistent unit
private final Map properties;
private transient EntityManagerFactory emf;
public TransactionScopedEntityManager(String puScopedName, Map properties, EntityManagerFactory emf) {
this.puScopedName = puScopedName;
this.properties = properties;
this.emf = emf;
}
@Override
protected EntityManager getEntityManager() {
EntityManager result;
boolean isInTx;
isInTx = TransactionUtil.isInTx();
if (isInTx) {
result = TransactionUtil.getOrCreateTransactionScopedEntityManager(emf, puScopedName, properties);
} else {
result = NonTxEmCloser.get(puScopedName);
if (result == null) {
result = EntityManagerUtil.createEntityManager(emf, properties);
NonTxEmCloser.add(puScopedName, result);
}
}
return result;
}
@Override
protected boolean isExtendedPersistenceContext() {
return false;
}
@Override
protected boolean isInTx() {
return TransactionUtil.isInTx();
}
/**
* Catch the application trying to close the container managed entity manager and throw an IllegalStateException
*/
@Override
public void close() {
// Transaction scoped entity manager will be closed when the (owning) component invocation completes
throw MESSAGES.cannotCloseTransactionContainerEntityManger();
}
private void readObject(java.io.ObjectInputStream in) throws IOException, ClassNotFoundException {
// read all non-transient fields
in.defaultReadObject();
final ServiceController<?> controller = currentServiceContainer().getService(JPAServiceNames.getPUServiceName(puScopedName));
final PersistenceUnitServiceImpl persistenceUnitService = (PersistenceUnitServiceImpl) controller.getService();
emf = persistenceUnitService.getEntityManagerFactory();
}
private static ServiceContainer currentServiceContainer() {
return AccessController.doPrivileged(new PrivilegedAction<ServiceContainer>() {
@Override
public ServiceContainer run() {
return CurrentServiceContainer.getServiceContainer();
}
});
}
}