From 48e79d5dd4a3a8bedc7086a4e8c175a15451c689 Mon Sep 17 00:00:00 2001
From: stefanprodan <stefan.prodan@gmail.com>
Date: Sat, 16 Mar 2019 14:12:23 +0200
Subject: [PATCH] Add mesh provider flag

---
 cmd/flagger/main.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/cmd/flagger/main.go b/cmd/flagger/main.go
index b12c745d..542182ca 100644
--- a/cmd/flagger/main.go
+++ b/cmd/flagger/main.go
@@ -34,6 +34,7 @@ var (
 	zapReplaceGlobals   bool
 	zapEncoding         string
 	namespace           string
+	meshProvider        string
 )
 
 func init() {
@@ -50,6 +51,7 @@ func init() {
 	flag.BoolVar(&zapReplaceGlobals, "zap-replace-globals", false, "Whether to change the logging level of the global zap logger.")
 	flag.StringVar(&zapEncoding, "zap-encoding", "json", "Zap logger encoding.")
 	flag.StringVar(&namespace, "namespace", "", "Namespace that flagger would watch canary object")
+	flag.StringVar(&meshProvider, "mesh-provider", "istio", "Service mesh provider, can be istio or appmesh")
 }
 
 func main() {
@@ -77,7 +79,7 @@ func main() {
 		logger.Fatalf("Error building kubernetes clientset: %v", err)
 	}
 
-	istioClient, err := clientset.NewForConfig(cfg)
+	meshClient, err := clientset.NewForConfig(cfg)
 	if err != nil {
 		logger.Fatalf("Error building istio clientset: %v", err)
 	}
@@ -125,7 +127,7 @@ func main() {
 
 	c := controller.NewController(
 		kubeClient,
-		istioClient,
+		meshClient,
 		flaggerClient,
 		canaryInformer,
 		controlLoopInterval,