Use ResourceId or DiscreteResourceId when specifying a resource
Change-Id: I4e29558ec649510c8d08bb5e5f8ed10c189252e5
diff --git a/core/net/src/main/java/org/onosproject/net/newresource/impl/ResourceManager.java b/core/net/src/main/java/org/onosproject/net/newresource/impl/ResourceManager.java
index 517ee10..5fc9d55 100644
--- a/core/net/src/main/java/org/onosproject/net/newresource/impl/ResourceManager.java
+++ b/core/net/src/main/java/org/onosproject/net/newresource/impl/ResourceManager.java
@@ -25,10 +25,12 @@
import org.apache.felix.scr.annotations.Service;
import org.onlab.util.GuavaCollectors;
import org.onosproject.event.AbstractListenerManager;
+import org.onosproject.net.newresource.DiscreteResourceId;
import org.onosproject.net.newresource.ResourceAdminService;
import org.onosproject.net.newresource.ResourceAllocation;
import org.onosproject.net.newresource.ResourceConsumer;
import org.onosproject.net.newresource.ResourceEvent;
+import org.onosproject.net.newresource.ResourceId;
import org.onosproject.net.newresource.ResourceListener;
import org.onosproject.net.newresource.ResourceService;
import org.onosproject.net.newresource.Resource;
@@ -115,25 +117,21 @@
}
@Override
- public List<ResourceAllocation> getResourceAllocations(Resource resource) {
- checkNotNull(resource);
+ public List<ResourceAllocation> getResourceAllocations(ResourceId id) {
+ checkNotNull(id);
- List<ResourceConsumer> consumers = store.getConsumers(resource);
- return consumers.stream()
- .map(x -> new ResourceAllocation(resource, x))
- .collect(GuavaCollectors.toImmutableList());
+ return store.getResourceAllocations(id);
}
@Override
- public <T> Collection<ResourceAllocation> getResourceAllocations(Resource parent, Class<T> cls) {
+ public <T> Collection<ResourceAllocation> getResourceAllocations(DiscreteResourceId parent, Class<T> cls) {
checkNotNull(parent);
checkNotNull(cls);
// We access store twice in this method, then the store may be updated by others
Collection<Resource> resources = store.getAllocatedResources(parent, cls);
return resources.stream()
- .flatMap(resource -> store.getConsumers(resource).stream()
- .map(consumer -> new ResourceAllocation(resource, consumer)))
+ .flatMap(resource -> store.getResourceAllocations(resource.id()).stream())
.collect(GuavaCollectors.toImmutableList());
}
@@ -148,7 +146,7 @@
}
@Override
- public Set<Resource> getAvailableResources(Resource parent) {
+ public Set<Resource> getAvailableResources(DiscreteResourceId parent) {
checkNotNull(parent);
Set<Resource> children = store.getChildResources(parent);
@@ -159,7 +157,7 @@
}
@Override
- public Set<Resource> getRegisteredResources(Resource parent) {
+ public Set<Resource> getRegisteredResources(DiscreteResourceId parent) {
checkNotNull(parent);
return store.getChildResources(parent);