chore: updating types

bbc · Nov 24, 2023 · 6e9b93d · 6e9b93d
1 parent a12cada
commit 6e9b93d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/consumer.ts b/src/consumer.ts
@@ -44,7 +44,7 @@ export class Consumer extends TypedEventEmitter {
   private postReceiveMessageCallback?: () => Promise<void>;
   private sqs: SQSClient;
   private handleMessageTimeout: number;
-  private attributeNames: string[];
+  private attributeNames: QueueAttributeName[];
   private messageAttributeNames: string[];
   private shouldDeleteMessages: boolean;
   private batchSize: number;
@@ -199,7 +199,7 @@ export class Consumer extends TypedEventEmitter {
     let currentPollingTimeout = this.pollingWaitTimeMs;
     this.receiveMessage({
       QueueUrl: this.queueUrl,
-      AttributeNames: this.attributeNames as QueueAttributeName[],
+      AttributeNames: this.attributeNames,
       MessageAttributeNames: this.messageAttributeNames,
       MaxNumberOfMessages: this.batchSize,
       WaitTimeSeconds: this.waitTimeSeconds,

diff --git a/src/types.ts b/src/types.ts
@@ -1,4 +1,4 @@
-import { SQSClient, Message } from '@aws-sdk/client-sqs';
+import { SQSClient, Message, QueueAttributeName } from '@aws-sdk/client-sqs';
 
 export interface ConsumerOptions {
   /**
@@ -10,7 +10,7 @@ export interface ConsumerOptions {
    * `['All', 'ApproximateFirstReceiveTimestamp', 'ApproximateReceiveCount']`).
    * @defaultvalue `[]`
    */
-  attributeNames?: string[];
+  attributeNames?: QueueAttributeName[];
   /**
    * List of message attributes to retrieve (i.e. `['name', 'address']`).
    * @defaultvalue `[]`