org.apache.openejb.core.cmp.cmp2.CmrSet Maven / Gradle / Ivy
/*
* 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.openejb.core.cmp.cmp2;
import org.apache.openejb.BeanContext;
import org.apache.openejb.loader.SystemInstance;
import javax.ejb.EJBException;
import javax.ejb.EJBLocalObject;
import javax.ejb.EntityBean;
import javax.transaction.TransactionSynchronizationRegistry;
import javax.transaction.Synchronization;
import java.util.AbstractSet;
import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;
import java.util.Collections;
import java.util.ConcurrentModificationException;
public class CmrSet extends AbstractSet {
private final EntityBean source;
private final String sourceProperty;
private final BeanContext relatedInfo;
private final String relatedProperty;
private final Class relatedLocal;
private boolean mutable = true;
private Collection relatedBeans;
public CmrSet(EntityBean source, String sourceProperty, BeanContext relatedInfo, String relatedProperty, Collection relatedBeans) {
this.source = source;
this.sourceProperty = sourceProperty;
this.relatedInfo = relatedInfo;
this.relatedProperty = relatedProperty;
this.relatedBeans = relatedBeans;
relatedLocal = relatedInfo.getLocalInterface();
TransactionSynchronizationRegistry transactionRegistry = SystemInstance.get().getComponent(TransactionSynchronizationRegistry.class);
try {
transactionRegistry.registerInterposedSynchronization(new Synchronization() {
public void beforeCompletion() {
}
public void afterCompletion(int i) {
mutable = false;
}
});
} catch (IllegalStateException ignored) {
// no tx so not mutable
mutable = false;
}
}
protected void entityDeleted() {
relatedBeans = null;
mutable = false;
}
public boolean isEmpty() {
return getRelatedBeans(false, false).isEmpty();
}
public int size() {
return getRelatedBeans(false, false).size();
}
public boolean contains(Object o) {
if (relatedLocal.isInstance(o)) {
Bean entity = getEntityBean((EJBLocalObject) o);
return entity != null && getRelatedBeans(false, false).contains(entity);
}
return false;
}
public boolean addAll(Collection c) {
Set entityBeans = getEntityBeans(c, relatedLocal);
boolean changed = false;
for (Iterator iterator = entityBeans.iterator(); iterator.hasNext();) {
Bean bean = iterator.next();
changed = add(bean) || changed;
}
return changed;
}
public boolean add(Object proxy) {
if (!relatedLocal.isInstance(proxy)) {
throw new IllegalArgumentException("Object is not an instance of " + relatedLocal.getName() +
": " + (proxy == null ? "null" : proxy.getClass().getName()));
}
Bean newEntity = getEntityBean((EJBLocalObject) proxy);
if (newEntity == null) {
throw new IllegalArgumentException("Ejb has been deleted");
}
return add(newEntity);
}
private boolean add(Bean newEntity) {
boolean changed = getRelatedBeans(true, true).add(newEntity);
if (changed && relatedProperty != null) {
// set the back reference in the new related bean
if (source == null) throw new IllegalStateException("source is null");
Object oldBackRef = toCmp2Entity(newEntity).OpenEJB_addCmr(relatedProperty, source);
// if the new related beas was related to another bean, we need
// to clear the back reference in that old bean
if (oldBackRef != null) {
toCmp2Entity(oldBackRef).OpenEJB_removeCmr(sourceProperty, newEntity);
}
}
return changed;
}
public boolean remove(Object o) {
if (!relatedLocal.isInstance(o)) {
return false;
}
Bean entity = getEntityBean((EJBLocalObject) o);
boolean changed = entity != null && getRelatedBeans(false, true).remove(entity);
if (changed && relatedProperty != null) {
toCmp2Entity(entity).OpenEJB_removeCmr(relatedProperty, source);
}
return changed;
}
public boolean retainAll(Collection c) {
Set entityBeans = getEntityBeans(c, null);
boolean changed = false;
for (Iterator iterator = getRelatedBeans(false, true).iterator(); iterator.hasNext();) {
Bean entity = iterator.next();
if (!entityBeans.contains(entity)) {
iterator.remove();
if (relatedProperty != null) {
toCmp2Entity(entity).OpenEJB_removeCmr(relatedProperty, source);
}
changed = true;
}
}
return changed;
}
public Iterator iterator() {
return new Iterator() {
private Bean currentEntity;
private Iterator iterator = getRelatedBeans(true, false).iterator();
public boolean hasNext() {
return iterator.hasNext();
}
public Proxy next() {
if (relatedBeans == null) {
throw new ConcurrentModificationException("Entity has been deleted therefore this iterator can no longer be used");
}
currentEntity = iterator.next();
return getEjbProxy(currentEntity);
}
public void remove() {
if (relatedBeans == null) {
throw new ConcurrentModificationException("Entity has been deleted therefore this iterator can no longer be used");
}
if (!mutable) {
throw new ConcurrentModificationException("Transaction has completed therefore this cmr collection can no longer be modified");
}
iterator.remove();
if (relatedProperty != null) {
toCmp2Entity(currentEntity).OpenEJB_removeCmr(relatedProperty, source);
}
}
};
}
private Proxy getEjbProxy(Bean entity) throws EJBException {
if (entity == null) return null;
Proxy ejbProxy = Cmp2Util.getEjbProxy(relatedInfo, entity);
return ejbProxy;
}
private Bean getEntityBean(EJBLocalObject proxy) {
if (proxy == null) return null;
Bean bean = Cmp2Util.getEntityBean(proxy);
return bean;
}
private static Set getEntityBeans(Collection proxies, Class type) {
if (proxies == null) return null;
Set entities = new HashSet();
for (Object value : proxies) {
if (type != null && !type.isInstance(value)) {
throw new IllegalArgumentException("Object is not an instance of " + type.getName() +
": " + (value == null ? "null" : value.getClass().getName()));
}
Bean entity = Cmp2Util.getEntityBean((EJBLocalObject) value);
if (entity == null) {
throw new IllegalArgumentException("Entity has been deleted");
}
entities.add(entity);
}
return entities;
}
private Cmp2Entity toCmp2Entity(Object object) {
return (Cmp2Entity) object;
}
private Collection getRelatedBeans(boolean mustExist, boolean mutateOpation) {
if (relatedBeans == null) {
if (mustExist) {
throw new IllegalStateException("Entity has been deleted therefore this cmr collection can no longer be modified");
}
return Collections.emptySet();
}
if (mutateOpation && !mutable) {
throw new IllegalStateException("Transaction has completed therefore this cmr collection can no longer be modified");
}
return relatedBeans;
}
}