fix: add missing memory circuit breaker options

Signed-off-by: Robert Da Silva <mail@robdasilva.com>
opensearch-project · Jul 21, 2022 · a62e420 · a62e420
1 parent ca12492
commit a62e420
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/index.d.ts b/index.d.ts
@@ -45,6 +45,7 @@ import Transport, {
   RequestBody,
   RequestNDBody,
   Context,
+  MemoryCircuitBreakerOptions,
 } from './lib/Transport';
 import { URL } from 'url';
 import Connection, { AgentOptions, agentFn } from './lib/Connection';
@@ -137,10 +138,7 @@ interface ClientOptions {
     password?: string;
   };
   disablePrototypePoisoningProtection?: boolean | 'proto' | 'constructor';
-  memoryCircuitBreaker?: {
-    enabled: boolean;
-    maxPercentage: number;
-  };
+  memoryCircuitBreaker?: MemoryCircuitBreakerOptions
 }
 
 declare class Client {

diff --git a/lib/Transport.d.ts b/lib/Transport.d.ts
@@ -59,6 +59,11 @@ export interface generateRequestIdFn {
   (params: TransportRequestParams, options: TransportRequestOptions): any;
 }
 
+export interface MemoryCircuitBreakerOptions {
+  enabled: boolean;
+  maxPercentage: number;
+}
+
 interface TransportOptions {
   emit: (event: string | symbol, ...args: any[]) => boolean;
   connectionPool: ConnectionPool | CloudConnectionPool;
@@ -77,6 +82,7 @@ interface TransportOptions {
   generateRequestId?: generateRequestIdFn;
   name?: string;
   opaqueIdPrefix?: string;
+  memoryCircuitBreaker: MemoryCircuitBreakerOptions;
 }
 
 export interface RequestEvent<TResponse = Record<string, any>, TContext = Context> {
@@ -172,6 +178,7 @@ export default class Transport {
   sniffInterval: number;
   sniffOnConnectionFault: boolean;
   opaqueIdPrefix: string | null;
+  memoryCircuitBreaker: MemoryCircuitBreakerOptions;
   sniffEndpoint: string;
   _sniffEnabled: boolean;
   _nextSniff: number;