Add WorkloadGeneratorStateCleaner that resets zookeeper

theodolite-quarkus/src/main/kotlin/theodolite/Main.kt

@@ -6,15 +6,6 @@ import io.quarkus.runtime.annotations.QuarkusMain
 object Main {
     @JvmStatic
     fun main(args: Array<String>) {
-        println("Running main method")
-
-        val run = RunUc()
-
-        val testtopic = mapOf<String,Int>("test" to 1)
-
-        run.createTopics(testtopic, 1.toShort())
-        run.deleteTopics(listOf("test"))
         //Quarkus.run()
-        run.resetZookeeper()
     }
 }

theodolite-quarkus/src/main/kotlin/theodolite/WorkloadGeneratorStateCleaner.kt

+package theodolite
+
+import org.apache.zookeeper.KeeperException
+import org.apache.zookeeper.WatchedEvent
+import org.apache.zookeeper.Watcher
+import org.apache.zookeeper.ZooKeeper
+
+class WorkloadGeneratorStateCleaner(ip: String) {
+    val path = "/workload-generation"
+    val sessionTimeout = 60
+    val retryTime = 3000L
+    lateinit var zookeeperClient: ZooKeeper
+
+    init {
+        try {
+            val watcher: Watcher = ZookeperWatcher()  // defined below
+            zookeeperClient = ZooKeeper(ip, sessionTimeout, watcher)
+        } catch (e:Exception) {
+            System.out.println(e.toString())
+        }
+    }
+
+    fun deleteAll() {
+        var deleted = false
+        while (!deleted) {
+
+            //
+            try {
+                zookeeperClient.delete(path, -1)
+            } catch (ex: Exception) {
+                System.out.println(ex.toString())
+            }
+
+            try {
+                val clients = zookeeperClient.getChildren(path, true)
+                /*if (clients.isEmpty()){
+                    break;
+                }*/
+            } catch (ex: Exception) {
+                when (ex) {
+                    is KeeperException -> { deleted = true }
+                    is InterruptedException -> {
+                        System.out.println(ex.toString())
+                    }
+                }
+            }
+            Thread.sleep(retryTime)
+            System.out.println("ZooKeeper reset was not successful. Retrying in 5s")
+        }
+
+        System.out.println("ZooKeeper reset was successful")
+    }
+}
+
+private class ZookeperWatcher : Watcher {
+
+    override fun process(event: WatchedEvent) {}
+}