add the KUBECONFIG check

satheshs · Sep 24, 2018 · 17883cf · 17883cf
1 parent 33f8cd1
commit 17883cf
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/sparkctl/cmd/root.go b/sparkctl/cmd/root.go
@@ -23,7 +23,15 @@ import (
 	"github.com/spf13/cobra"
 )
 
-var defaultKubeConfig = os.Getenv("HOME") + "/.kube/config"
+func getKubeConfigPath() string {
+	var kubeConfigEnv = os.Getenv("KUBECONFIG")
+	if len(kubeConfigEnv) == 0 {
+		return os.Getenv("HOME") + "/.kube/config"
+	}
+	return kubeConfigEnv
+}
+
+var defaultKubeConfig = getKubeConfigPath()
 
 var Namespace string
 var KubeConfig string