From e4b866bda50bd4e660b8ac5df866a57e3d0c70a2 Mon Sep 17 00:00:00 2001
From: Joshua Boniface <joshua@boniface.me>
Date: Mon, 4 Jun 2018 03:09:51 -0400
Subject: [PATCH] Add unflush function

---
 NodeInstance.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/NodeInstance.py b/NodeInstance.py
index 5d40aed9..93ee7487 100644
--- a/NodeInstance.py
+++ b/NodeInstance.py
@@ -90,6 +90,22 @@ class NodeInstance(threading.Thread):
             # Wait 1s between migrations
             time.sleep(1)
 
+    def unflush(self):
+        print('>>> Restoring node %s to active service' % self.name)
+        for domain in self.s_domain:
+            former_hypervisor = self.zk.get("/domains/" + domain + '/formerhypervisor').decode('ascii')
+            if former_hypervisor = self.name:
+                print(">>> Setting unmigration for %s" % domain)
+                transaction = self.zk.transaction()
+                transaction.set_data('/domains/' + domain + '/state', 'migrate'.encode('ascii'))
+                transaction.set_data('/domains/' + domain + '/hypervisor', self.name.encode('ascii'))
+                result = transaction.commit()
+
+                # Wait 1s between migrations
+                time.sleep(1)
+
+        self.zk.set("/nodes/" + self.name + "/state", 'start'.encode('ascii'))
+
     def run(self):
         if self.name == socket.gethostname():
             self.setup_local_node()