allow KUBECONFIG to be set

tenjin · Apr 23, 2019 · d9cd178 · d9cd178
1 parent 0c48bca
commit d9cd178
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/internal/pkg/rotator/cluster.go b/internal/pkg/rotator/cluster.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"log"
+	"os"
 	"strings"
 	"time"
 
@@ -16,13 +17,22 @@ import (
 	"github.com/tenjin/rotate-eks-asg/internal/pkg/cmd"
 )
 
+const (
+	DefaultKubeConfigPath = "/tmp/.kube/config"
+)
+
 var (
 	DefaultNodeAwaitJoinTimeout      = 30 * time.Second
 	DefaultNodeAwaitReadinessTimeout = 10 * time.Second
 )
 
 func NewKubernetesClient() (*kubernetes.Clientset, error) {
-	config, err := clientcmd.BuildConfigFromFlags("", "/tmp/.kube/config") // TODO
+	kcfg := os.Getenv("KUBECONFIG")
+	if kcfg == "" {
+		kcfg = DefaultKubeConfigPath
+	}
+
+	config, err := clientcmd.BuildConfigFromFlags("", kcfg)
 	if err != nil {
 		return nil, err
 	}