Fix: [AEA-0000] - retry on ECONNABORTED

packages/serviceSearchClient/jest.debug.config.ts

@@ -1,4 +1,4 @@
-import config from "./jest.config"
+import config from "./jest.config.ts"
 import type {JestConfigWithTsJest} from "ts-jest"
 
 const debugConfig: JestConfigWithTsJest = {

packages/serviceSearchClient/src/live-serviceSearch-client.ts

@@ -1,7 +1,7 @@
 import {Logger} from "@aws-lambda-powertools/logger"
 import {getSecret} from "@aws-lambda-powertools/parameters/secrets"
 import axios, {AxiosError, AxiosInstance} from "axios"
-import axiosRetry from "axios-retry"
+import axiosRetry, {isNetworkOrIdempotentRequestError} from "axios-retry"
 import {handleUrl} from "./handleUrl"
 
 import {ServiceSearchClient} from "./serviceSearch-client"
@@ -82,7 +82,11 @@ export class LiveServiceSearchClient implements ServiceSearchClient {
         v3: process.env.ServiceSearch3ApiKey !== undefined
       })
     this.axiosInstance = axios.create()
-    axiosRetry(this.axiosInstance, {retries: 3})
+    axiosRetry(this.axiosInstance, {
+      retries: 3,
+      onRetry: this.onAxiosRetry,
+      retryCondition: this.retryCondition
+    })
 
     this.axiosInstance.interceptors.request.use((config) => {
       config.headers["request-startTime"] = Date.now()
@@ -102,7 +106,7 @@ export class LiveServiceSearchClient implements ServiceSearchClient {
       // reject with a proper Error object
       let err: Error
       if (error instanceof Error) {
-        logger.error("Error in serviceSearch request", {error})
+        this.logger.error("Error in serviceSearch request", {error})
         err = error
       } else if ((error as AxiosError).message) {
         // Only report the interesting subset of the error object.
@@ -121,10 +125,10 @@ export class LiveServiceSearchClient implements ServiceSearchClient {
           }
         }
 
-        logger.error("Axios error in serviceSearch request", {axiosErrorDetails})
+        this.logger.error("Axios error in serviceSearch request", {axiosErrorDetails})
         err = new Error("Axios error in serviceSearch request")
       } else {
-        logger.error("Unknown error in serviceSearch request", {error})
+        this.logger.error("Unknown error in serviceSearch request", {error})
         err = new Error("Unknown error in serviceSearch request")
       }
       return Promise.reject(err)
@@ -268,4 +272,13 @@ export class LiveServiceSearchClient implements ServiceSearchClient {
       }
     })
   }
+
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  onAxiosRetry = (retryCount: number, error: any) => {
+    this.logger.warn(error)
+    this.logger.warn(`Call to serviceSearch failed - retrying. Retry count ${retryCount}`, {retryCount: retryCount})
+  }
+  retryCondition(error: AxiosError): boolean {
+    return isNetworkOrIdempotentRequestError(error) || error.code === "ECONNABORTED"
+  }
 }

packages/serviceSearchClient/tests/live-serviceSearch-client.test.ts

@@ -284,15 +284,31 @@ describe("live serviceSearch client", () => {
         .onGet(serviceSearchUrl).replyOnce(500)
         .onGet(serviceSearchUrl).replyOnce(500)
         .onGet(serviceSearchUrl).reply(200, validUrlData)
+      const warnSpy = jest.spyOn(Logger.prototype, "warn")
+      client = new LiveServiceSearchClient(logger)
+
       const result = await client.searchService("z", dummyCorrelationId)
       expect(result).toEqual(new URL(validUrlData.value[0].URL))
+      expect(warnSpy).toHaveBeenCalledWith(
+        expect.stringContaining("Call to serviceSearch failed - retrying. Retry count"),
+        expect.objectContaining({retryCount: 1})
+      )
     })
 
     test("fails after exceeding retries", async () => {
       mock.onGet(serviceSearchUrl).reply(500)
       await expect(client.searchService("z", dummyCorrelationId)).rejects.toThrow("Request failed with status code 500")
     })
 
+    test("retries on timeout (ECONNABORTED) error", async () => {
+      mock.onGet(serviceSearchUrl).timeoutOnce()
+        .onGet(serviceSearchUrl).timeoutOnce()
+        .onGet(serviceSearchUrl).timeoutOnce()
+        .onGet(serviceSearchUrl).reply(200, validUrlData)
+      const result = await client.searchService("z", dummyCorrelationId)
+      expect(result).toEqual(new URL(validUrlData.value[0].URL))
+    })
+
     test("logs duration in info on success and failure", async () => {
       const infoSpy = jest.spyOn(Logger.prototype, "info")
       mock.onGet(serviceSearchUrl).networkError()