fix: export types (#542)

Author: karenc-bq

common/lib/authentication/aws_secrets_manager_plugin.ts

@@ -30,8 +30,9 @@ import { WrapperProperties } from "../wrapper_property";
 import { logAndThrowError } from "../utils/utils";
 import { ClientWrapper } from "../client_wrapper";
 import { TelemetryTraceLevel } from "../utils/telemetry/telemetry_trace_level";
+import { CanReleaseResources } from "../can_release_resources";
 
-export class AwsSecretsManagerPlugin extends AbstractConnectionPlugin {
+export class AwsSecretsManagerPlugin extends AbstractConnectionPlugin implements CanReleaseResources {
   private static readonly TELEMETRY_UPDATE_SECRETS = "fetch credentials";
   private static readonly TELEMETRY_FETCH_CREDENTIALS_COUNTER = "secretsManager.fetchCredentials.count";
   private static SUBSCRIBED_METHODS: Set<string> = new Set<string>(["connect", "forceConnect"]);
@@ -169,6 +170,11 @@ export class AwsSecretsManagerPlugin extends AbstractConnectionPlugin {
     }
     throw new AwsWrapperError(Messages.get("AwsSecretsManagerConnectionPlugin.failedToFetchDbCredentials"));
   }
+
+  releaseResources(): Promise<void> {
+    AwsSecretsManagerPlugin.secretsCache.clear();
+    return;
+  }
 }
 
 export class SecretCacheKey {

common/lib/authentication/iam_authentication_plugin.ts

@@ -25,8 +25,9 @@ import { WrapperProperties } from "../wrapper_property";
 import { IamAuthUtils, TokenInfo } from "../utils/iam_auth_utils";
 import { ClientWrapper } from "../client_wrapper";
 import { RegionUtils } from "../utils/region_utils";
+import { CanReleaseResources } from "../can_release_resources";
 
-export class IamAuthenticationPlugin extends AbstractConnectionPlugin {
+export class IamAuthenticationPlugin extends AbstractConnectionPlugin implements CanReleaseResources {
   private static readonly SUBSCRIBED_METHODS = new Set<string>(["connect", "forceConnect"]);
   protected static readonly tokenCache = new Map<string, TokenInfo>();
   private readonly telemetryFactory;
@@ -133,7 +134,8 @@ export class IamAuthenticationPlugin extends AbstractConnectionPlugin {
     }
   }
 
-  static clearCache(): void {
+  releaseResources(): Promise<void> {
     IamAuthenticationPlugin.tokenCache.clear();
+    return;
   }
 }

common/lib/aws_client.ts

@@ -30,12 +30,10 @@ import { DriverDialect } from "./driver_dialect/driver_dialect";
 import { WrapperProperties } from "./wrapper_property";
 import { DriverConfigurationProfiles } from "./profile/driver_configuration_profiles";
 import { ConfigurationProfile } from "./profile/configuration_profile";
-import { AwsWrapperError } from "./utils/errors";
+import { AwsWrapperError, TransactionIsolationLevel, ConnectionProvider } from "./";
 import { Messages } from "./utils/messages";
-import { TransactionIsolationLevel } from "./utils/transaction_isolation_level";
 import { HostListProviderService } from "./host_list_provider_service";
 import { SessionStateClient } from "./session_state_client";
-import { ConnectionProvider } from "./connection_provider";
 import { DriverConnectionProvider } from "./driver_connection_provider";
 
 const { EventEmitter } = pkgStream;

common/lib/driver_connection_provider.ts

@@ -19,7 +19,7 @@ import { HostRole } from "./host_role";
 import { HostInfo } from "./host_info";
 import { HostSelector } from "./host_selector";
 import { RandomHostSelector } from "./random_host_selector";
-import { UnsupportedStrategyError } from "./utils/errors";
+import { UnsupportedStrategyError } from "./index";
 import { WrapperProperties } from "./wrapper_property";
 import { Messages } from "./utils/messages";
 import { RdsUtils } from "./utils/rds_utils";

common/lib/host_list_provider/monitoring/monitoring_host_list_provider.ts

@@ -15,13 +15,12 @@
 */
 
 import { RdsHostListProvider } from "../rds_host_list_provider";
-import { HostInfo } from "../../host_info";
-import { ClusterTopologyMonitor, ClusterTopologyMonitorImpl } from "./cluster_topology_monitor";
+import { HostInfo, AwsWrapperError } from "../../";
 import { PluginService } from "../../plugin_service";
+import { ClusterTopologyMonitor, ClusterTopologyMonitorImpl } from "./cluster_topology_monitor";
 import { HostListProviderService } from "../../host_list_provider_service";
 import { ClientWrapper } from "../../client_wrapper";
 import { DatabaseDialect } from "../../database_dialect/database_dialect";
-import { AwsWrapperError } from "../../utils/errors";
 import { Messages } from "../../utils/messages";
 import { WrapperProperties } from "../../wrapper_property";
 import { BlockingHostListProvider } from "../host_list_provider";

common/lib/host_list_provider/rds_host_list_provider.ts

@@ -21,7 +21,7 @@ import { RdsUrlType } from "../utils/rds_url_type";
 import { RdsUtils } from "../utils/rds_utils";
 import { HostListProviderService } from "../host_list_provider_service";
 import { ConnectionUrlParser } from "../utils/connection_url_parser";
-import { AwsWrapperError } from "../utils/errors";
+import { AwsWrapperError } from "../";
 import { Messages } from "../utils/messages";
 import { WrapperProperties } from "../wrapper_property";
 import { logger } from "../../logutils";

common/lib/index.ts

@@ -16,3 +16,17 @@
 
 export * from "./connection_plugin";
 export * from "./plugin_manager";
+export * from "./utils/errors";
+
+export { AwsPoolConfig } from "./aws_pool_config";
+export { HostInfo } from "./host_info";
+export { HostRole } from "./host_role";
+export { HostInfoBuilder } from "./host_info_builder";
+export type { ConnectionProvider } from "./connection_provider";
+export { InternalPooledConnectionProvider } from "./internal_pooled_connection_provider";
+export type { InternalPoolMapping } from "./utils/internal_pool_mapping";
+
+export { TransactionIsolationLevel } from "./utils/transaction_isolation_level";
+
+export { HostAvailability } from "./host_availability/host_availability";
+export type { HostAvailabilityStrategy } from "./host_availability/host_availability_strategy";

common/lib/internal_pooled_connection_provider.ts

@@ -27,7 +27,7 @@ import { lookup, LookupAddress } from "dns";
 import { promisify } from "util";
 import { HostInfoBuilder } from "./host_info_builder";
 import { RdsUrlType } from "./utils/rds_url_type";
-import { AwsWrapperError } from "./utils/errors";
+import { AwsWrapperError } from "./index";
 import { Messages } from "./utils/messages";
 import { HostSelector } from "./host_selector";
 import { RandomHostSelector } from "./random_host_selector";

common/lib/plugin_service.ts

@@ -23,7 +23,7 @@ import { BlockingHostListProvider, HostListProvider } from "./host_list_provider
 import { ConnectionUrlParser } from "./utils/connection_url_parser";
 import { DatabaseDialect, DatabaseType } from "./database_dialect/database_dialect";
 import { HostInfoBuilder } from "./host_info_builder";
-import { AwsWrapperError } from "./utils/errors";
+import { AwsWrapperError } from "./";
 import { HostAvailability } from "./host_availability/host_availability";
 import { CacheMap } from "./utils/cache_map";
 import { HostChangeOptions } from "./host_change_options";

common/lib/plugins/efm/host_monitoring_connection_plugin.ts

@@ -14,9 +14,14 @@
   limitations under the License.
 */
 
+import {
+  HostInfo,
+  AwsWrapperError,
+  UnavailableHostError,
+  HostAvailability
+} from "../../";
 import { PluginService } from "../../plugin_service";
 import { HostChangeOptions } from "../../host_change_options";
-import { HostInfo } from "../../host_info";
 import { OldConnectionSuggestionAction } from "../../old_connection_suggestion_action";
 import { RdsUtils } from "../../utils/rds_utils";
 import { AbstractConnectionPlugin } from "../../abstract_connection_plugin";
@@ -26,9 +31,7 @@ import { MonitorConnectionContext } from "./monitor_connection_context";
 import { logger, uniqueId } from "../../../logutils";
 import { Messages } from "../../utils/messages";
 import { MonitorService, MonitorServiceImpl } from "./monitor_service";
-import { AwsWrapperError, UnavailableHostError } from "../../utils/errors";
 import { HostListProvider } from "../../host_list_provider/host_list_provider";
-import { HostAvailability } from "../../host_availability/host_availability";
 import { CanReleaseResources } from "../../can_release_resources";
 import { SubscribedMethodHelper } from "../../utils/subscribed_method_helper";
 import { ClientWrapper } from "../../client_wrapper";