From cab2af7df1d5a91086bde6950ac57a8c475be1a5 Mon Sep 17 00:00:00 2001
From: up9-github <info@up9.com>
Date: Thu, 22 Apr 2021 13:22:30 +0300
Subject: [PATCH] Update mizu.go

---
 cli/mizu/mizu.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/cli/mizu/mizu.go b/cli/mizu/mizu.go
index 9cdeda745..45daabe7c 100644
--- a/cli/mizu/mizu.go
+++ b/cli/mizu/mizu.go
@@ -25,10 +25,11 @@ func Run(podRegex *regexp.Regexp) {
 
 	go createPodAndPortForward(ctx, kubernetesProvider, cancel, podName) //TODO convert this to job for built in pod ttl or have the running app handle this
 	go watchPodsForTapping(ctx, kubernetesProvider, cancel, podRegex)
-	waitForFinish(ctx, cancel)
+	waitForFinish(ctx, cancel) //block until exit signal or error
 
 	// TODO handle incoming traffic from tapper using a channel
 
+	//cleanup
 	fmt.Printf("\nremoving pod %s\n", podName)
 	removalCtx, _ := context.WithTimeout(context.Background(), 2 * time.Second)
 	kubernetesProvider.RemovePod(removalCtx, podName)