From 9a9e5fda0ae183a34e7b5d567d2d4d98c122abb9 Mon Sep 17 00:00:00 2001
From: RamiBerm <54766858+RamiBerm@users.noreply.github.com>
Date: Thu, 4 Nov 2021 14:47:48 +0200
Subject: [PATCH] Fix clean again (#431)

* Revert "Fix mizu clean (#430)"

This reverts commit 57cd7a365be0f22d08d7fe9ec19422576ecfcbf3.

* Update cleanRunner.go and tapRunner.go
---
 cli/cmd/cleanRunner.go | 4 +++-
 cli/cmd/tapRunner.go   | 9 +++++----
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/cli/cmd/cleanRunner.go b/cli/cmd/cleanRunner.go
index 471ef3fee..5cc3ca6b1 100644
--- a/cli/cmd/cleanRunner.go
+++ b/cli/cmd/cleanRunner.go
@@ -1,10 +1,12 @@
 package cmd
 
+import "github.com/up9inc/mizu/cli/apiserver"
+
 func performCleanCommand() {
 	kubernetesProvider, err := getKubernetesProviderForCli()
 	if err != nil {
 		return
 	}
 
-	finishMizuExecution(kubernetesProvider)
+	finishMizuExecution(kubernetesProvider, apiserver.NewProvider(GetApiServerUrl(), apiserver.DefaultRetries, apiserver.DefaultTimeout))
 }
diff --git a/cli/cmd/tapRunner.go b/cli/cmd/tapRunner.go
index 6c84b0bb1..a3f3bd2af 100644
--- a/cli/cmd/tapRunner.go
+++ b/cli/cmd/tapRunner.go
@@ -41,10 +41,11 @@ type tapState struct {
 }
 
 var state tapState
-var apiProvider = apiserver.NewProvider(GetApiServerUrl(), apiserver.DefaultRetries, apiserver.DefaultTimeout)
+var apiProvider *apiserver.Provider
 
 func RunMizuTap() {
 	mizuApiFilteringOptions, err := getMizuApiFilteringOptions()
+	apiProvider = apiserver.NewProvider(GetApiServerUrl(), apiserver.DefaultRetries, apiserver.DefaultTimeout)
 	if err != nil {
 		logger.Log.Errorf(uiUtils.Error, fmt.Sprintf("Error parsing regex-masking: %v", errormessage.FormatError(err)))
 		return
@@ -133,13 +134,13 @@ func RunMizuTap() {
 	}
 	if config.Config.Tap.DaemonMode {
 		if err := handleDaemonModePostCreation(cancel, kubernetesProvider); err != nil {
-			defer finishMizuExecution(kubernetesProvider)
+			defer finishMizuExecution(kubernetesProvider, apiProvider)
 			cancel()
 		} else {
 			logger.Log.Infof(uiUtils.Magenta, "Mizu is now running in daemon mode, run `mizu view` to connect to the mizu daemon instance")
 		}
 	} else {
-		defer finishMizuExecution(kubernetesProvider)
+		defer finishMizuExecution(kubernetesProvider, apiProvider)
 
 		if err = startTapperSyncer(ctx, cancel, kubernetesProvider, targetNamespaces, *mizuApiFilteringOptions); err != nil {
 			logger.Log.Errorf(uiUtils.Error, fmt.Sprintf("Error starting mizu tapper syncer: %v", err))
@@ -416,7 +417,7 @@ func getSyncEntriesConfig() *shared.SyncEntriesConfig {
 	}
 }
 
-func finishMizuExecution(kubernetesProvider *kubernetes.Provider) {
+func finishMizuExecution(kubernetesProvider *kubernetes.Provider, apiProvider *apiserver.Provider) {
 	telemetry.ReportAPICalls(apiProvider)
 	removalCtx, cancel := context.WithTimeout(context.Background(), cleanupTimeout)
 	defer cancel()