get service endpoint in wrappers

amplify-education · Jul 11, 2024 · 3ff604a · 3ff604a
1 parent 8662f8f
commit 3ff604a
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 8 deletions.
diff --git a/src/aws/acm-wrapper.ts b/src/aws/acm-wrapper.ts
@@ -26,7 +26,8 @@ class ACMWrapper {
       credentials,
       region: isEdge ? Globals.defaultRegion : Globals.getRegion(),
       retryStrategy: Globals.getRetryStrategy(),
-      requestHandler: Globals.getRequestHandler()
+      requestHandler: Globals.getRequestHandler(),
+      endpoint: Globals.getServiceEndpoint("acm")
     });
   }
 

diff --git a/src/aws/api-gateway-v1-wrapper.ts b/src/aws/api-gateway-v1-wrapper.ts
@@ -27,13 +27,15 @@ import { getAWSPagedResults } from "../utils";
 class APIGatewayV1Wrapper extends APIGatewayBase {
   public readonly apiGateway: APIGatewayClient;
 
+
   constructor (credentials?: any) {
     super();
     this.apiGateway = new APIGatewayClient({
       credentials,
       region: Globals.getRegion(),
       retryStrategy: Globals.getRetryStrategy(),
-      requestHandler: Globals.getRequestHandler()
+      requestHandler: Globals.getRequestHandler(),
+      endpoint: Globals.getServiceEndpoint("apigateway")
     });
   }
 

diff --git a/src/aws/api-gateway-v2-wrapper.ts b/src/aws/api-gateway-v2-wrapper.ts
@@ -33,7 +33,8 @@ class APIGatewayV2Wrapper extends APIGatewayBase {
       credentials,
       region: Globals.getRegion(),
       retryStrategy: Globals.getRetryStrategy(),
-      requestHandler: Globals.getRequestHandler()
+      requestHandler: Globals.getRequestHandler(),
+      endpoint: Globals.getServiceEndpoint("apigatewayv2")
     });
   }
 

diff --git a/src/aws/cloud-formation-wrapper.ts b/src/aws/cloud-formation-wrapper.ts
@@ -31,7 +31,8 @@ class CloudFormationWrapper {
       credentials,
       region: Globals.getRegion(),
       retryStrategy: Globals.getRetryStrategy(),
-      requestHandler: Globals.getRequestHandler()
+      requestHandler: Globals.getRequestHandler(),
+      endpoint: Globals.getServiceEndpoint("cloudformation")
     });
   }
 

diff --git a/src/aws/route53-wrapper.ts b/src/aws/route53-wrapper.ts
@@ -17,20 +17,23 @@ class Route53Wrapper {
 
   constructor (credentials?: any, region?: string) {
     // not null and not undefined
+    const serviceEndpoint = Globals.getServiceEndpoint("route53");
     if (credentials) {
       this.region = region || Globals.getRegion();
       this.route53 = new Route53Client({
         credentials,
         region: this.region,
         retryStrategy: Globals.getRetryStrategy(),
-        requestHandler: Globals.getRequestHandler()
+        requestHandler: Globals.getRequestHandler(),
+        endpoint: serviceEndpoint
       });
     } else {
       this.region = Globals.getRegion();
       this.route53 = new Route53Client({
         region: this.region,
         retryStrategy: Globals.getRetryStrategy(),
-        requestHandler: Globals.getRequestHandler()
+        requestHandler: Globals.getRequestHandler(),
+        endpoint: serviceEndpoint
       });
     }
   }
@@ -59,7 +62,7 @@ class Route53Wrapper {
         "NextMarker",
         new ListHostedZonesCommand({})
       );
-      Logging.logInfo(`Founded hosted zones list: ${hostedZones.map((zone) => zone.Name)}.`);
+      Logging.logInfo(`Found hosted zones list: ${hostedZones.map((zone) => zone.Name)}.`);
     } catch (err) {
       throw new Error(`Unable to list hosted zones in Route53.\n${err.message}`);
     }

diff --git a/src/aws/s3-wrapper.ts b/src/aws/s3-wrapper.ts
@@ -10,7 +10,8 @@ class S3Wrapper {
     this.s3 = new S3Client({
       credentials,
       region: Globals.getRegion(),
-      requestHandler: Globals.getRequestHandler()
+      requestHandler: Globals.getRequestHandler(),
+      endpoint: Globals.getServiceEndpoint("s3")
     });
   }