chore: move setting of discoveryCommandCtor to client config

Author: trivikr

clients/client-timestream-query/TimestreamQueryClient.ts

@@ -1,5 +1,9 @@
 import { CancelQueryCommandInput, CancelQueryCommandOutput } from "./commands/CancelQueryCommand";
-import { DescribeEndpointsCommandInput, DescribeEndpointsCommandOutput } from "./commands/DescribeEndpointsCommand";
+import {
+  DescribeEndpointsCommand,
+  DescribeEndpointsCommandInput,
+  DescribeEndpointsCommandOutput,
+} from "./commands/DescribeEndpointsCommand";
 import { QueryCommandInput, QueryCommandOutput } from "./commands/QueryCommand";
 import { ClientDefaultValues as __ClientDefaultValues } from "./runtimeConfig";
 import {
@@ -215,6 +219,7 @@ export class TimestreamQueryClient extends __Client<
     let _config_7 = resolveEndpointDiscoveryClientConfig(_config_6);
     super(_config_7);
     _config_7.client = this;
+    _config_7.discoveryEndpointCommandCtor = DescribeEndpointsCommand;
     this.config = _config_7;
     this.middlewareStack.use(getRetryPlugin(this.config));
     this.middlewareStack.use(getContentLengthPlugin(this.config));

clients/client-timestream-query/commands/QueryCommand.ts

@@ -48,7 +48,6 @@ export class QueryCommand extends $Command<QueryCommandInput, QueryCommandOutput
         isDiscoveredEndpointRequired: true,
         // @ts-ignore
         getEndpointDiscoveryId: undefined,
-        discoveryEndpointCommandCtor: DescribeEndpointsCommand,
       })
     );
 

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

@@ -24,7 +24,6 @@ export const endpointDiscoveryMiddlewareOptions: FinalizeRequestHandlerOptions =
 export type EndpointDiscoveryMiddlewareConfig = {
   isDiscoveredEndpointRequired: boolean;
   getEndpointDiscoveryId: () => string | undefined;
-  discoveryEndpointCommandCtor: new (comandConfig: any) => Command<any, any, any, any, any>;
 };
 
 export const endpointDiscoveryMiddleware = (
@@ -36,7 +35,9 @@ export const endpointDiscoveryMiddleware = (
 ): FinalizeHandler<any, Output> => async (
   args: FinalizeHandlerArguments<any>
 ): Promise<FinalizeHandlerOutput<Output>> => {
-  const { isDiscoveredEndpointRequired, discoveryEndpointCommandCtor, getEndpointDiscoveryId } = middlewareConfig;
+  const { client } = config;
+  const { discoveryEndpointCommandCtor } = client?.config;
+  const { isDiscoveredEndpointRequired, getEndpointDiscoveryId } = middlewareConfig;
   const { commandName } = context;
 
   if (isDiscoveredEndpointRequired === true) {
@@ -57,7 +58,6 @@ export const endpointDiscoveryMiddleware = (
     });
   }
 
-  const { client } = config;
   const { request } = args;
   const cacheKey = await getCacheKey(commandName, client?.config, getEndpointDiscoveryId);
   if (cacheKey && HttpRequest.isInstance(request)) {

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

@@ -1,4 +1,4 @@
-import { Client } from "@aws-sdk/types";
+import { Client, Command } from "@aws-sdk/types";
 import LRUCache from "lru-cache";
 
 export interface EndpointDiscoveryClientInputConfig {}
@@ -7,6 +7,7 @@ interface PreviouslyResolved {}
 
 export interface EndpointDiscoveryClientResolvedConfig {
   client?: Client<any, any, any>;
+  discoveryEndpointCommandCtor?: new (comandConfig: any) => Command<any, any, any, any, any>;
   endpointCache: LRUCache<string, string>;
 }