Changeset 866 for branches/1.0/jobmond

Timestamp:

05/15/13 22:34:08 (10 years ago)

Author:

ramonb

Message:

jobmond.py:

• added down/offline node detection for SLURM
• state: down = down, drain = offline
• see #162

File:

• branches/1.0/jobmond/jobmond.py (modified) (2 diffs)

branches/1.0/jobmond/jobmond.py

@@ -854 +854 @@
         # Report down/offline nodes in batch (PBS only ATM)
         #
-        if BATCH_API == 'pbs':
+        if BATCH_API in [ 'pbs', 'slurm' ]:
 
             domain        = fqdn_parts( socket.getfqdn() )[1]
@@ -1259 +1259 @@
     def getNodeData( self ):
 
-        slurm_type = pyslurm.node()
-
-        nodedict = slurm_type.get()
+        slurm_type  = pyslurm.node()
+
+        slurm_nodes = slurm_type.get()
+
+        nodedict    = { }
+
+        for node, attrs in slurm_nodes.items():
+
+            ( num_state, name_state ) = attrs['node_state'] 
+
+            if name_state == 'DOWN':
+
+                nodedict[ node ] = { 'state' : 'down' }
+
+            elif name_state == 'DRAIN':
+
+                nodedict[ node ] = { 'state' : 'offline' }
 
         return nodedict