From dd4a0eb64c70abec8748b065ee940735d50cf953 Mon Sep 17 00:00:00 2001
From: Claudio-Daniel Freire <claudio-daniel.freire@inria.fr>
Date: Tue, 27 Sep 2011 04:01:25 +0200
Subject: [PATCH] Filter blacklisted nodes in util.getNodes

---
 src/nepi/testbeds/planetlab/util.py | 24 ++++++++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/src/nepi/testbeds/planetlab/util.py b/src/nepi/testbeds/planetlab/util.py
index dbe80183..133f3af7 100644
--- a/src/nepi/testbeds/planetlab/util.py
+++ b/src/nepi/testbeds/planetlab/util.py
@@ -31,6 +31,24 @@ def getAPI(user, pass_):
     import plcapi
     return plcapi.PLCAPI(username=user, password=pass_)
 
+def filterBlacklist(candidates):
+    blpath = environ.homepath('plblacklist')
+    
+    try:
+        bl = open(blpath, "r")
+    except:
+        return candidates
+        
+    try:
+        blacklist = set(
+            map(int,
+                map(str.strip, bl.readlines())
+            )
+        )
+        return [ x for x in candidates if x not in blacklist ]
+    finally:
+        bl.close()
+
 def getNodes(api, num, **constraints):
     # Now do the backtracking search for a suitable solution
     # First with existing slice nodes
@@ -45,7 +63,7 @@ def getNodes(api, num, **constraints):
         nodes.append(node)
     
     node = nodes[0]
-    candidates = node.find_candidates()
+    candidates = filterBlacklist(node.find_candidates())
     reqs = [candidates] * num
 
     def pickbest(fullset, nreq, node=nodes[0]):
@@ -60,8 +78,10 @@ def getNodes(api, num, **constraints):
     solution = resourcealloc.alloc(reqs, sample=pickbest)
     
     # Do assign nodes
+    runner = ParallelRun(maxthreads=4)
     for node, node_id in zip(nodes, solution):
-        node.assign_node_id(node_id)
+        runner.put(node.assign_node_id, node_id)
+    runner.join()
     
     return nodes
 
-- 
2.43.0