From a62e4200f6f83c8017a41d637a46ba4d7b99b860 Mon Sep 17 00:00:00 2001
From: Robert Da Silva <mail@robdasilva.com>
Date: Thu, 21 Jul 2022 23:51:15 +0200
Subject: [PATCH] fix: add missing memory circuit breaker options

Signed-off-by: Robert Da Silva <mail@robdasilva.com>
---
 index.d.ts         | 6 ++----
 lib/Transport.d.ts | 7 +++++++
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/index.d.ts b/index.d.ts
index 328664ea4..8a6112c99 100644
--- 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
index 026601b12..99d34e78d 100644
--- 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;