From 6a0e262e967adb49aeb93c4bab64a15434a24def Mon Sep 17 00:00:00 2001
From: Joshua Boniface <joshua@boniface.me>
Date: Wed, 6 Jun 2018 20:36:21 -0400
Subject: [PATCH] Add flush component to fence

---
 fencenode.py | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/fencenode.py b/fencenode.py
index 955bcc2a..11be6961 100644
--- a/fencenode.py
+++ b/fencenode.py
@@ -28,4 +28,28 @@ import os, sys, libvirt, uuid, kazoo.client, time
 def fence(node_name, zk):
     time.sleep(3)
     print('>>> Fencing node {} via IPMI reboot signal.'.format(node_name))
-    
+
+    # DO IPMI FENCE HERE
+
+    print('>>> Moving VMs from dead hypervisor {} to new hosts.'.format(node_name))
+    dead_node_running_domains = zk.get('/nodes/{}/runningdomains'.format(node_name))[0].decode('ascii').split()
+    for dom_uuid in dead_node_running_domains:
+        most_memfree = 0
+        hypervisor_list = zk.get_children('/nodes')
+        current_hypervisor = zk.get('/dom_uuids/{}/hypervisor'.format(dom_uuid))[0].decode('ascii')
+        for hypervisor in hypervisor_list:
+            state = zk.get('/nodes/{}/state'.format(hypervisor))[0].decode('ascii')
+            if state != 'start' or hypervisor == current_hypervisor:
+                continue
+
+            memfree = int(zk.get('/nodes/{}/memfree'.format(hypervisor))[0].decode('ascii'))
+            if memfree > most_memfree:
+                most_memfree = memfree
+                target_hypervisor = hypervisor
+
+        print('>>> Moving VM "{}" to hypervisor "{}".'.format(dom_uuid, target_hypervisor))
+        transaction = zk.transaction()
+        transaction.set_data('/domains/{}/state'.format(dom_uuid), 'start'.encode('ascii'))
+        transaction.set_data('/domains/{}/hypervisor'.format(dom_uuid), target_hypervisor.encode('ascii'))
+        transaction.set_data('/domains/{}/lasthypervisor'.format(dom_uuid), current_hypervisor.encode('ascii'))
+        transaction.commit()