enable users to set ec2-endpoint for nonstandard regions

kubernetes-sigs · Sep 11, 2019 · 6f8e3b4 · 6f8e3b4
1 parent ad0e4f5
commit 6f8e3b4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/cloud/cloud.go b/pkg/cloud/cloud.go
@@ -20,6 +20,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"os"
 	"time"
 
 	"github.com/aws/aws-sdk-go/aws"
@@ -224,6 +225,11 @@ func newEC2Cloud(metadata MetadataService, svc *ec2metadata.EC2Metadata) (Cloud,
 		CredentialsChainVerboseErrors: aws.Bool(true),
 	}
 
+	endpoint := os.Getenv("AWS_EC2_ENDPOINT")
+	if endpoint != "" {
+		awsConfig.Endpoint = aws.String(endpoint)
+	}
+
 	return &cloud{
 		metadata: metadata,
 		dm:       dm.NewDeviceManager(),