feat: add circuit breaker support

r2don · May 14, 2023 · 74c581f · 74c581f
1 parent c327db8
commit 74c581f
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 2 deletions.
diff --git a/lib/builders/circuit-breaker.builder.ts b/lib/builders/circuit-breaker.builder.ts
@@ -0,0 +1,13 @@
+import CircuitBreaker from 'opossum';
+
+export class CircuitBreakerBuilder {
+  constructor(private readonly options?: CircuitBreaker.Options) {}
+
+  async build(executor: () => Promise<Response>): Promise<Response> {
+    if (this.options == null) {
+      return await executor();
+    }
+
+    return await new CircuitBreaker(executor, this.options).fire();
+  }
+}
diff --git a/lib/decorators/constants.ts b/lib/decorators/constants.ts
@@ -6,3 +6,4 @@ export const REQUEST_BODY_METADATA = Symbol('REQUEST_BODY_METADATA');
 export const REQUEST_FORM_METADATA = Symbol('REQUEST_FORM_METADATA');
 export const REQUEST_HEADER_METADATA = Symbol('REQUEST_HEADER_METADATA');
 export const RESPONSE_BODY_METADATA = Symbol('RESPONSE_BODY_METADATA');
+export const CIRCUIT_BREAKER_METADATA = Symbol('CIRCUIT_BREAKER_METADATA');
diff --git a/lib/supports/node-fetch.injector.ts b/lib/supports/node-fetch.injector.ts
@@ -1,9 +1,11 @@
 import { Injectable, type OnModuleInit } from '@nestjs/common';
 import { DiscoveryService, MetadataScanner } from '@nestjs/core';
 import { type InstanceWrapper } from '@nestjs/core/injector/instance-wrapper';
+import { CircuitBreakerBuilder } from '../builders/circuit-breaker.builder';
 import { type HttpRequestBuilder } from '../builders/http-request.builder';
 import { type ResponseBodyBuilder } from '../builders/response-body.builder';
 import {
+  CIRCUIT_BREAKER_METADATA,
   HTTP_EXCHANGE_METADATA,
   HTTP_INTERFACE_METADATA,
   RESPONSE_BODY_METADATA,
@@ -43,14 +45,25 @@ export class NodeFetchInjector implements OnModuleInit {
         return;
       }
 
+      const circuitBreaker: CircuitBreakerBuilder =
+        Reflect.getMetadata(CIRCUIT_BREAKER_METADATA, prototype, methodName) ??
+        new CircuitBreakerBuilder(
+          this.httpInterfaceConfig?.circuitBreakerOption,
+        );
       const responseBodyBuilder: ResponseBodyBuilder<unknown> | undefined =
         Reflect.getMetadata(RESPONSE_BODY_METADATA, prototype, methodName);
 
       httpRequestBuilder.setBaseUrl(baseUrl);
 
       wrapper.instance[methodName] = async (...args: never[]) =>
-        await this.httpClient
-          .request(httpRequestBuilder.build(args), httpRequestBuilder.options)
+        await circuitBreaker
+          .build(
+            async () =>
+              await this.httpClient.request(
+                httpRequestBuilder.build(args),
+                httpRequestBuilder.options,
+              ),
+          )
           .then(async (response) => {
             if (responseBodyBuilder != null) {
               const res = await response.json();

diff --git a/lib/types/http-interface-config.ts b/lib/types/http-interface-config.ts
@@ -1,8 +1,10 @@
 import { type ClassTransformOptions } from 'class-transformer';
+import type CircuitBreaker from 'opossum';
 import { type HttpClient } from './http-client.interface';
 
 export interface HttpInterfaceConfig {
   timeout?: number;
   httpClient?: HttpClient;
   transformOption?: ClassTransformOptions;
+  circuitBreakerOption?: CircuitBreaker.Options;
 }