fix(receive-one): add missing options

Sagacify · Oct 10, 2024 · c003d16 · c003d16
1 parent 4c356af
commit c003d16
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -22,7 +22,7 @@
   "license": "MIT",
   "repository": {
     "type": "git",
-    "url": "https://github.com/Sagacify/sqs-handler.git"
+    "url": "git+https://github.com/Sagacify/sqs-handler.git"
   },
   "bugs": {
     "url": "https://github.com/Sagacify/sqs-handler/issues"

diff --git a/src/SQSHandler.ts b/src/SQSHandler.ts
@@ -27,6 +27,7 @@ export type ReceiveOptions = Omit<
   ReceiveMessageCommandInput,
   'QueueUrl' | 'VisibilityTimeout' | 'WaitTimeSeconds'
 >;
+export type ReceiveOneOptions = Omit<ReceiveOptions, 'MaxNumberOfMessages'>;
 export type SendOptions = Omit<
   SendMessageCommandInput,
   'QueueUrl' | 'MessageAttributes' | 'MessageBody'
@@ -226,8 +227,9 @@ export class SQSHandler<ParsedBody> {
     return parsedMessages;
   }
 
-  async receiveOne(): Promise<ParsedMessage<ParsedBody> | null> {
+  async receiveOne(options: ReceiveOneOptions = {}): Promise<ParsedMessage<ParsedBody> | null> {
     const command = new ReceiveMessageCommand({
+      ...options,
       QueueUrl: this.queueUrl,
       MaxNumberOfMessages: 1,
       VisibilityTimeout: this.visibilityTimeout,