fix: convert identifiers into Map<String,String>

Author: trivikr

packages/middleware-endpoint-discovery/src/getCacheKey.ts

@@ -8,16 +8,16 @@ export const getCacheKey = async (
   commandName: string,
   config: AwsAuthResolvedConfig & RegionResolvedConfig,
   options: {
-    endpointDiscoveryId?: string;
+    identifiers?: Map<String, String>;
   }
 ) => {
   const region = await config.region();
   const { accessKeyId } = await config.credentials();
-  const { endpointDiscoveryId } = options;
+  const { identifiers } = options;
   return {
     commandName,
     ...(region && { region }),
     ...(accessKeyId && { accessKeyId }),
-    ...(endpointDiscoveryId && { endpointDiscoveryId }),
+    ...(identifiers && { identifiers }),
   }.toString();
 };

packages/middleware-endpoint-discovery/src/getEndpointDiscoveryCommandPlugin.ts

@@ -1,6 +1,5 @@
 import { HttpRequest } from "@aws-sdk/protocol-http";
 import {
-  Command,
   FinalizeHandler,
   FinalizeHandlerArguments,
   FinalizeHandlerOutput,
@@ -23,7 +22,7 @@ export const endpointDiscoveryMiddlewareOptions: FinalizeRequestHandlerOptions =
 
 export type EndpointDiscoveryMiddlewareConfig = {
   isDiscoveredEndpointRequired: boolean;
-  endpointDiscoveryId?: string;
+  identifiers?: Map<String, String>;
 };
 
 export const endpointDiscoveryMiddleware = (
@@ -37,7 +36,7 @@ export const endpointDiscoveryMiddleware = (
 ): Promise<FinalizeHandlerOutput<Output>> => {
   const { client } = config;
   const { endpointDiscoveryCommandCtor } = client?.config;
-  const { isDiscoveredEndpointRequired, endpointDiscoveryId } = middlewareConfig;
+  const { isDiscoveredEndpointRequired, identifiers } = middlewareConfig;
   const { commandName } = context;
 
   if (isDiscoveredEndpointRequired === true) {
@@ -46,20 +45,20 @@ export const endpointDiscoveryMiddleware = (
     await updateDiscoveredEndpointInCache(config, {
       commandName,
       endpointDiscoveryCommandCtor,
-      endpointDiscoveryId,
+      identifiers,
     });
   } else if (isDiscoveredEndpointRequired === false) {
     // Do not call await await on Endpoint Discovery API utility so that function
     // does not block, the command will use discovered endpoint, if available.
     updateDiscoveredEndpointInCache(config, {
       commandName,
       endpointDiscoveryCommandCtor,
-      endpointDiscoveryId,
+      identifiers,
     });
   }
 
   const { request } = args;
-  const cacheKey = await getCacheKey(commandName, client?.config, { endpointDiscoveryId });
+  const cacheKey = await getCacheKey(commandName, client?.config, { identifiers });
   if (cacheKey && HttpRequest.isInstance(request)) {
     const endpoints = client?.config.endpointCache.get(cacheKey);
     if (endpoints && endpoints.length > 0) {

packages/middleware-endpoint-discovery/src/updateDiscoveredEndpointInCache.ts

@@ -6,7 +6,7 @@ import { EndpointDiscoveryClientResolvedConfig } from "./resolveEndpointDiscover
 export type updateDiscoveredEndpointInCacheOptions = {
   commandName: string;
   endpointDiscoveryCommandCtor: new (comandConfig: any) => Command<any, any, any, any, any>;
-  endpointDiscoveryId?: string;
+  identifiers?: Map<String, String>;
 };
 
 export const updateDiscoveredEndpointInCache = async (
@@ -16,8 +16,8 @@ export const updateDiscoveredEndpointInCache = async (
   const { client } = config;
   const { endpointCache } = client?.config;
 
-  const { commandName, endpointDiscoveryId } = options;
-  const cacheKey = await getCacheKey(commandName, client?.config, { endpointDiscoveryId });
+  const { commandName, identifiers } = options;
+  const cacheKey = await getCacheKey(commandName, client?.config, { identifiers });
 
   const endpoints = endpointCache.get(cacheKey);
   if (endpoints && endpoints.length === 1 && endpoints[0].Address === "") {
@@ -40,7 +40,7 @@ export const updateDiscoveredEndpointInCache = async (
     try {
       const command = new options.endpointDiscoveryCommandCtor({
         Operation: commandName.substr(0, commandName.length - 7), // strip "Command"
-        Identifiers: endpointDiscoveryId,
+        Identifiers: identifiers,
       });
       const { Endpoints } = await client?.send(command);
       endpointCache.set(cacheKey, Endpoints);