Changeset 1207 for DCWoRMS/branches/coolemall/src/schedframe/scheduling/SchedulerDescription.java

Timestamp:

11/26/13 11:56:07 (11 years ago)

Author:

wojtekp

Message:

 

File:

• DCWoRMS/branches/coolemall/src/schedframe/scheduling/SchedulerDescription.java (modified) (3 diffs)

DCWoRMS/branches/coolemall/src/schedframe/scheduling/SchedulerDescription.java

@@ -2 +2 @@
 
 import java.util.ArrayList;
+import java.util.Collection;
 import java.util.HashMap;
 import java.util.Iterator;
@@ -9 +10 @@
 import schedframe.resources.StandardResourceType;
 import schedframe.resources.computing.description.ExecutingResourceDescription;
+import schedframe.resources.computing.description.ResourceDescription;
 import schedframe.resources.providers.ResourceProvider;
-import schedframe.resources.units.AbstractResourceUnit;
 import schedframe.resources.units.ResourceUnit;
 import schedframe.resources.units.ResourceUnitName;
 import schedframe.scheduling.queue.QueueDescription;
 
-public class SchedulerDescription extends ExecutingResourceDescription{
+public class SchedulerDescription extends ResourceDescription implements ExecutingResourceDescription{
 
         protected ResourceProvider provider;
         protected List<QueueDescription> accesQueues;
+        protected Map<ResourceUnitName, List<ResourceUnit>> resUnits;
 
         public SchedulerDescription(ResourceProvider provider){
@@ -62 +64 @@
                 return provider;
         }
+        
+        public void addResourceUnit(ResourceUnit unit) {
+                if (this.resUnits == null)
+                        this.resUnits = new HashMap<ResourceUnitName, List<ResourceUnit>>(1);
+                List<ResourceUnit> list = null;
+                if (this.resUnits.containsKey(unit.getName())) {
+                        list = this.resUnits.get(unit.getName());
+                } else {
+                        list = new ArrayList<ResourceUnit>(1);
+                        this.resUnits.put(unit.getName(), list);
+                }
+                list.add(unit);
+        }
+        
+        public ResourceUnit getResourceUnit(ResourceUnitName unitName) throws NoSuchFieldException {
+                return getResourceUnitList(unitName).get(0);
+        }
+
+        public List<ResourceUnit> getResourceUnitList(ResourceUnitName unitName) throws NoSuchFieldException {
+                if (resUnits.containsKey(unitName))
+                        return resUnits.get(unitName);
+                else
+                        throw new NoSuchFieldException("Resource unit " + unitName + " is not available in resource " + this.id);
+        }
+
+        public Collection<ResourceUnit> getResourceUnit() {
+                if (resUnits == null)
+                        return null;
+                List<ResourceUnit> values = new ArrayList<ResourceUnit>();
+                Collection<List<ResourceUnit>> lists = resUnits.values();
+                Iterator<List<ResourceUnit>> itr = lists.iterator();
+
+                while (itr.hasNext()) {
+                        List<ResourceUnit> list = itr.next();
+                        values.addAll(list);
+                }
+
+                return values;
+        }
+
+        public Map<ResourceUnitName, List<ResourceUnit>> getResourceUnits() {
+                return resUnits;
+        }
 }