/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.geronimo.persistence;
import java.util.Map;
import javax.persistence.EntityManager;
import javax.persistence.FlushModeType;
import javax.persistence.LockModeType;
import javax.persistence.Query;
import javax.persistence.EntityTransaction;
import javax.persistence.EntityManagerFactory;
import javax.persistence.TransactionRequiredException;
import javax.transaction.Status;
import org.apache.geronimo.transaction.manager.TransactionManagerImpl;
import org.apache.geronimo.transaction.manager.TransactionImpl;
/**
* @version $Rev: 476049 $ $Date: 2006-11-16 23:35:17 -0500 (Thu, 16 Nov 2006) $
*/
public class CMPEntityManagerTxScoped implements EntityManager {
private final TransactionManagerImpl transactionManager;
private final String persistenceUnit;
private final EntityManagerFactory entityManagerFactory;
private final Map entityManagerProperties;
public CMPEntityManagerTxScoped(TransactionManagerImpl transactionManager, String persistenceUnit, EntityManagerFactory entityManagerFactory, Map entityManagerProperties) {
this.transactionManager = transactionManager;
this.persistenceUnit = persistenceUnit;
this.entityManagerFactory = entityManagerFactory;
this.entityManagerProperties = entityManagerProperties;
}
private EntityManager getEntityManager(boolean activeRequired) {
TransactionImpl transaction = (TransactionImpl) transactionManager.getTransaction();
if (activeRequired && (transaction == null || transaction.getStatus() != Status.STATUS_ACTIVE)) {
throw new TransactionRequiredException("No active transaction");
}
if (transaction == null) {
return null;
}
EntityManagerWrapper entityManagerWrapper = (EntityManagerWrapper) transaction.getEntityManager(persistenceUnit);
if (entityManagerWrapper == null) {
EntityManager entityManager = createEntityManager();
entityManagerWrapper = new EntityManagerWrapperTxScoped(entityManager);
transaction.setEntityManager(persistenceUnit, entityManagerWrapper);
}
return entityManagerWrapper.getEntityManager();
}
private EntityManager createEntityManager() {
EntityManager entityManager;
if (entityManagerProperties == null) {
entityManager = entityManagerFactory.createEntityManager();
} else {
entityManager = entityManagerFactory.createEntityManager(entityManagerProperties);
}
return entityManager;
}
public void persist(Object o) {
EntityManager entityManager = getEntityManager(true);
if (entityManager != null) {
entityManager.persist(o);
} else {
entityManager = createEntityManager();
try {
entityManager.persist(o);
} finally {
entityManager.close();
}
}
}
public <T>T merge(T t) {
EntityManager entityManager = getEntityManager(true);
if (entityManager != null) {
return entityManager.merge(t);
} else {
entityManager = createEntityManager();
try {
return entityManager.merge(t);
} finally {
entityManager.close();
}
}
}
public void remove(Object o) {
EntityManager entityManager = getEntityManager(true);
if (entityManager != null) {
entityManager.remove(o);
} else {
entityManager = createEntityManager();
try {
entityManager.remove(o);
} finally {
entityManager.close();
}
}
}
public <T>T find(Class<T> aClass, Object o) {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
return entityManager.find(aClass, o);
} else {
entityManager = createEntityManager();
try {
return entityManager.find(aClass, o);
} finally {
entityManager.close();
}
}
}
public <T>T getReference(Class<T> aClass, Object o) {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
return entityManager.getReference(aClass, o);
} else {
entityManager = createEntityManager();
try {
return entityManager.getReference(aClass, o);
} finally {
entityManager.close();
}
}
}
public void flush() {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
entityManager.flush();
} else {
entityManager = createEntityManager();
try {
entityManager.flush();
} finally {
entityManager.close();
}
}
}
public void setFlushMode(FlushModeType flushModeType) {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
entityManager.setFlushMode(flushModeType);
} else {
entityManager = createEntityManager();
try {
entityManager.setFlushMode(flushModeType);
} finally {
entityManager.close();
}
}
}
public FlushModeType getFlushMode() {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
return entityManager.getFlushMode();
} else {
entityManager = createEntityManager();
try {
return entityManager.getFlushMode();
} finally {
entityManager.close();
}
}
}
public void lock(Object o, LockModeType lockModeType) {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
entityManager.lock(o, lockModeType);
} else {
entityManager = createEntityManager();
try {
entityManager.lock(o, lockModeType);
} finally {
entityManager.close();
}
}
}
public void refresh(Object o) {
EntityManager entityManager = getEntityManager(true);
if (entityManager != null) {
entityManager.refresh(o);
} else {
entityManager = createEntityManager();
try {
entityManager.refresh(o);
} finally {
entityManager.close();
}
}
}
public void clear() {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
entityManager.clear();
} else {
entityManager = createEntityManager();
try {
entityManager.clear();
} finally {
entityManager.close();
}
}
}
public boolean contains(Object o) {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
return entityManager.contains(o);
} else {
entityManager = createEntityManager();
try {
return entityManager.contains(o);
} finally {
entityManager.close();
}
}
}
public Query createQuery(String s) {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
return entityManager.createQuery(s);
} else {
entityManager = createEntityManager();
try {
return entityManager.createQuery(s);
} finally {
entityManager.close();
}
}
}
public Query createNamedQuery(String s) {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
return entityManager.createNamedQuery(s);
} else {
entityManager = createEntityManager();
try {
return entityManager.createNamedQuery(s);
} finally {
entityManager.close();
}
}
}
public Query createNativeQuery(String s) {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
return entityManager.createNativeQuery(s);
} else {
entityManager = createEntityManager();
try {
return entityManager.createNativeQuery(s);
} finally {
entityManager.close();
}
}
}
public Query createNativeQuery(String s, Class aClass) {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
return entityManager.createNativeQuery(s, aClass);
} else {
entityManager = createEntityManager();
try {
return entityManager.createNativeQuery(s, aClass);
} finally {
entityManager.close();
}
}
}
public Query createNativeQuery(String s, String s1) {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
return entityManager.createNativeQuery(s, s1);
} else {
entityManager = createEntityManager();
try {
return entityManager.createNativeQuery(s, s1);
} finally {
entityManager.close();
}
}
}
public void close() {
throw new IllegalStateException("You cannot call close on a Container Managed Entity Manager");
}
public boolean isOpen() {
return true;
}
public EntityTransaction getTransaction() {
throw new IllegalStateException("You cannot call getTransaction on a container managed EntityManager");
}
public void joinTransaction() {
throw new IllegalStateException("You cannot call joinTransaction on a container managed EntityManager");
}
public Object getDelegate() {
EntityManager entityManager = getEntityManager(false);
if (entityManager != null) {
return entityManager.getDelegate();
} else {
entityManager = createEntityManager();
try {
return entityManager.getDelegate();
} finally {
entityManager.close();
}
}
}
private static class EntityManagerWrapperTxScoped implements EntityManagerWrapper {
private final EntityManager entityManager;
public EntityManagerWrapperTxScoped(EntityManager entityManager) {
if (entityManager == null) {
throw new IllegalArgumentException("Need a non-null entity manager");
}
this.entityManager = entityManager;
}
public void close() {
entityManager.close();
}
public EntityManager getEntityManager() {
return entityManager;
}
}
}