chore: Refactor message processing in SQSHandler and add waitTimePolling option

src/ext/sqs/constants.ts

@@ -1,5 +1,4 @@
 export const DEFAULT_MAX_NUMBER_OF_MESSAGES = 10;
-
 export const DEFAULT_WAIT_TIME_SECONDS = 20;
-
 export const DEFAULT_VISIBILITY_TIMEOUT = 30;
+export const DEFAULT_WAIT_TIME_POLLING = 10000;

src/ext/sqs/handlers.ts

@@ -2,6 +2,7 @@ import logging from '../../logging';
 import { AbstractMessageTranslator } from '../../message-translators';
 import { timeout } from '../../utils';
 import { BaseSQS } from './base';
+import { DEFAULT_WAIT_TIME_POLLING } from './constants';
 import { SQSProvider } from './providers';
 import { SQSClientOptions } from './types';
 
@@ -15,7 +16,8 @@ export class SQSHandler extends BaseSQS {
     super(clientOptions);
     this.provider = new SQSProvider(queueName, clientOptions);
     this.messageTranslator = clientOptions.messageTranslator;
-    this.pollingWaitTimeMs = 30000;
+    this.pollingWaitTimeMs =
+      clientOptions.waitTimePolling || DEFAULT_WAIT_TIME_POLLING;
   }
 
   async stop(): Promise<void> {

src/ext/sqs/types.ts

@@ -17,6 +17,7 @@ export type SQSClientOptions = {
   prefixBasedQueues?: boolean;
   maxNumberOfMessages?: number;
   waitTimeSeconds?: number;
+  waitTimePolling?: number;
 };
 
 export type SQSMessageAttributes = {

tests/ext/sqs/handlers.test.ts

@@ -18,6 +18,22 @@ describe('SQSHandler', () => {
       expect(sqsHandler.messageTranslator).toBeDefined();
       expect(sqsHandler.messageTranslator).toBeInstanceOf(SQSMessageTranslator);
     });
+
+    it.each([
+      [10000, 10000],
+      [20000, 20000],
+      [undefined, 10000]
+    ])(
+      `should create a new SQSHandler with waitTimePolling %s`,
+      (waitTimePolling, expected) => {
+        sqsHandler = new SQSHandler('queueName', {
+          visibilityTimeout: 10,
+          messageTranslator: new SQSMessageTranslator(),
+          waitTimePolling
+        });
+        expect(sqsHandler.pollingWaitTimeMs).toBe(expected);
+      }
+    );
   });
 
   describe('start', () => {