org.gradle.api.internal.CompositeDomainObjectSet Maven / Gradle / Ivy
/*
* Copyright 2009 the original author or authors.
*
* Licensed 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.gradle.api.internal;
import com.google.common.collect.Lists;
import com.google.common.collect.Sets;
import org.gradle.api.Action;
import org.gradle.api.DomainObjectCollection;
import org.gradle.api.internal.collections.ElementSource;
import org.gradle.api.internal.provider.CollectionProviderInternal;
import org.gradle.api.internal.provider.ProviderInternal;
import org.gradle.api.specs.Spec;
import org.gradle.internal.Actions;
import java.util.Collection;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
/**
* A domain object collection that presents a combined view of one or more collections.
*
* @param The type of domain objects in the component collections of this collection.
*/
public class CompositeDomainObjectSet extends DelegatingDomainObjectSet implements WithEstimatedSize {
private final Spec uniqueSpec = new ItemIsUniqueInCompositeSpec();
private final Spec notInSpec = new ItemNotInCompositeSpec();
private final DefaultDomainObjectSet backingSet;
private final CollectionCallbackActionDecorator callbackActionDecorator;
@SafeVarargs
@SuppressWarnings("varargs")
public static CompositeDomainObjectSet create(Class type, DomainObjectCollection... collections) {
return create(type, CollectionCallbackActionDecorator.NOOP, collections);
}
@SafeVarargs
public static CompositeDomainObjectSet create(Class type, CollectionCallbackActionDecorator callbackActionDecorator, DomainObjectCollection... collections) {
DefaultDomainObjectSet backingSet = new DefaultDomainObjectSet(type, new DomainObjectCompositeCollection(), callbackActionDecorator);
CompositeDomainObjectSet out = new CompositeDomainObjectSet(backingSet, callbackActionDecorator);
for (DomainObjectCollection c : collections) {
out.addCollection(c);
}
return out;
}
private CompositeDomainObjectSet(DefaultDomainObjectSet backingSet, CollectionCallbackActionDecorator callbackActionDecorator) {
super(backingSet);
this.backingSet = backingSet;
this.callbackActionDecorator = callbackActionDecorator;
}
public class ItemIsUniqueInCompositeSpec implements Spec {
@Override
public boolean isSatisfiedBy(T element) {
int matches = 0;
for (DomainObjectCollection collection : getStore().store) {
if (collection.contains(element)) {
if (++matches > 1) {
return false;
}
}
}
return true;
}
}
public class ItemNotInCompositeSpec implements Spec {
@Override
public boolean isSatisfiedBy(T element) {
return !getStore().contains(element);
}
}
@SuppressWarnings("unchecked")
protected DomainObjectCompositeCollection getStore() {
return (DomainObjectCompositeCollection) this.backingSet.getStore();
}
@Override
public Action whenObjectAdded(Action action) {
return super.whenObjectAdded(Actions.filter(action, uniqueSpec));
}
@Override
public Action whenObjectRemoved(Action action) {
return super.whenObjectRemoved(Actions.filter(action, notInSpec));
}
public void addCollection(DomainObjectCollection collection) {
if (!getStore().containsCollection(collection)) {
getStore().addComposited(collection);
collection.all(new InternalAction() {
@Override
public void execute(T t) {
backingSet.getEventRegister().fireObjectAdded(t);
}
});
collection.whenObjectRemoved(new Action() {
@Override
public void execute(T t) {
backingSet.getEventRegister().fireObjectRemoved(t);
}
});
}
}
public void removeCollection(DomainObjectCollection collection) {
getStore().removeComposited(collection);
for (T item : collection) {
backingSet.getEventRegister().fireObjectRemoved(item);
}
}
@SuppressWarnings({"NullableProblems", "unchecked"})
@Override
public Iterator iterator() {
return getStore().iterator();
}
@Override
@SuppressWarnings("unchecked")
/**
* This method is expensive. Avoid calling it if possible. If all you need is a rough
* estimate, call {@link #estimatedSize()} instead.
*/
public int size() {
return getStore().size();
}
@Override
public int estimatedSize() {
return getStore().estimatedSize();
}
@Override
public void all(Action action) {
//calling overloaded method with extra behavior:
whenObjectAdded(action);
for (T t : this) {
callbackActionDecorator.decorate(action).execute(t);
}
}
// TODO Make this work with pending elements
private final static class DomainObjectCompositeCollection implements ElementSource {
private final List> store = Lists.newLinkedList();
public boolean containsCollection(DomainObjectCollection collection) {
for (DomainObjectCollection ts : store) {
if (ts == collection) {
return true;
}
}
return false;
}
Set collect() {
if (store.isEmpty()) {
return Collections.emptySet();
}
Set tmp = Sets.newLinkedHashSetWithExpectedSize(estimatedSize());
for (DomainObjectCollection collection : store) {
tmp.addAll(collection);
}
return tmp;
}
@Override
public int size() {
return collect().size();
}
@Override
public boolean isEmpty() {
for (DomainObjectCollection ts : store) {
if (!ts.isEmpty()) {
return false;
}
}
return true;
}
@Override
public boolean contains(Object o) {
for (DomainObjectCollection ts : store) {
if (ts.contains(o)) {
return true;
}
}
return false;
}
@Override
@SuppressWarnings("unchecked")
public Iterator iterator() {
if (store.isEmpty()) {
return Collections.emptyIterator();
}
if (store.size() == 1) {
return (Iterator) store.get(0).iterator();
}
return collect().iterator();
}
@Override
public boolean add(T t) {
throw new UnsupportedOperationException();
}
@Override
public boolean addRealized(T element) {
throw new UnsupportedOperationException();
}
@Override
public boolean remove(Object o) {
throw new UnsupportedOperationException();
}
@Override
public boolean containsAll(Collection c) {
throw new UnsupportedOperationException();
}
@Override
public void clear() {
throw new UnsupportedOperationException();
}
public void addComposited(DomainObjectCollection collection) {
this.store.add(collection);
}
public void removeComposited(DomainObjectCollection collection) {
Iterator> iterator = store.iterator();
while (iterator.hasNext()) {
DomainObjectCollection next = iterator.next();
if (next == collection) {
iterator.remove();
break;
}
}
}
@Override
public boolean constantTimeIsEmpty() {
return store.isEmpty();
}
@Override
public int estimatedSize() {
int size = 0;
for (DomainObjectCollection ts : store) {
size += Estimates.estimateSizeOf(ts);
}
return size;
}
@Override
public Iterator iteratorNoFlush() {
return iterator();
}
@Override
public void realizePending() {
}
@Override
public void realizePending(Class type) {
}
@Override
public boolean addPending(ProviderInternal provider) {
throw new UnsupportedOperationException();
}
@Override
public boolean removePending(ProviderInternal provider) {
throw new UnsupportedOperationException();
}
@Override
public boolean addPendingCollection(CollectionProviderInternal> provider) {
throw new UnsupportedOperationException();
}
@Override
public boolean removePendingCollection(CollectionProviderInternal> provider) {
throw new UnsupportedOperationException();
}
@Override
public void onRealize(Action action) {
}
@Override
public void realizeExternal(ProviderInternal provider) {
}
@Override
public MutationGuard getMutationGuard() {
return MutationGuards.identity();
}
}
}