prime-sdk 1.5.1 → 2.0.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prime-sdk",
-  "version": "1.5.1",
+  "version": "2.0.0",
   "description": "swiss army knife",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -28,18 +28,18 @@
     "@aws-sdk/client-sns": "^3.572.0",
     "@aws-sdk/client-sqs": "^3.569.0",
     "@aws-sdk/s3-request-presigner": "^3.572.0",
-    "@nestjs/common": "^10.4.15",
-    "@nestjs/config": "^3.2.2",
-    "@nestjs/core": "^10.4.15",
-    "@nestjs/platform-express": "^10.4.15",
+    "@nestjs/common": "^11.0.20",
+    "@nestjs/config": "^4.0.2",
+    "@nestjs/core": "^11.0.12",
+    "@nestjs/platform-express": "^11.0.12",
     "ioredis": "^5.4.1",
     "reflect-metadata": "^0.2.0",
     "rxjs": "^7.8.1"
   },
   "devDependencies": {
-    "@nestjs/cli": "^10.4.8",
-    "@nestjs/schematics": "^10.2.3",
-    "@nestjs/testing": "^10.4.15",
+    "@nestjs/cli": "^11.0.5",
+    "@nestjs/schematics": "^11.0.2",
+    "@nestjs/testing": "^11.0.12",
     "@types/express": "^4.17.17",
     "@types/jest": "^29.5.2",
     "@types/node": "^20.3.1",

package/src/app.module.ts

@@ -15,40 +15,40 @@ import { Cache } from './lib/cache/cache';
   imports: [
     QueueModule.forRoot({
       credentials: {
-        accessKeyId: '000000000',
-        secretAccessKey: '000000000',
+        accessKeyId: '',
+        secretAccessKey: '',
       },
       endpoint: 'http://localhost:4566/',
       region: 'us-east-1',
     }),
     StorageModule.forRoot({
       credentials: {
-        accessKeyId: '000000000',
-        secretAccessKey: '000000000',
+        accessKeyId: '',
+        secretAccessKey: '',
       },
       endpoint: 'http://localhost:4566/sample-bucket',
       region: 'us-east-1',
     }),
     EventModule.forRoot({
       credentials: {
-        accessKeyId: '000000000',
-        secretAccessKey: '000000000',
+        accessKeyId: '',
+        secretAccessKey: '',
       },
       endpoint: 'http://localhost:4566/',
       region: 'us-east-1',
     }),
     CryptoModule.forRoot({
       credentials: {
-        accessKeyId: '000000000',
-        secretAccessKey: '000000000',
+        accessKeyId: '',
+        secretAccessKey: '',
       },
       endpoint: 'http://localhost:4566/',
       region: 'us-east-1',
     }),
     EmailModule.forRoot({
       credentials: {
-        accessKeyId: '000000000',
-        secretAccessKey: '000000000',
+        accessKeyId: '',
+        secretAccessKey: '',
       },
       endpoint: 'http://localhost:4566/',
       region: 'us-east-1',

package/src/external/aws/sqs/sqs.service.ts

@@ -1,5 +1,6 @@
 import {
   DeleteMessageCommand,
+  QueueAttributeName,
   ReceiveMessageCommand,
   ReceiveMessageResult,
   SQSClient,
@@ -12,13 +13,15 @@ export class Sqs implements IQueue {
   private queueUrl: string;
   private bulk: number;
   private waitTimeSeconds: number;
+  private attributeNames: Array<QueueAttributeName> = [QueueAttributeName.All];
 
   constructor(@Inject('CONNECTION_SQS') private readonly client: SQSClient) {}
 
-  setup({ waitTimeSeconds, bulk, queueUrl }) {
+  setup({ waitTimeSeconds, bulk, queueUrl, attributeNames }) {
     this.waitTimeSeconds = waitTimeSeconds;
     this.bulk = bulk;
     this.queueUrl = queueUrl;
+    this.attributeNames = attributeNames;
   }
 
   async sendCommand(
@@ -39,6 +42,19 @@ export class Sqs implements IQueue {
     return (await this.sendCommand(command)) as ReceiveMessageResult;
   }
 
+  async receiveMessagesV2(): Promise<ReceiveMessageResult> {
+    const input = {
+      QueueUrl: this.queueUrl,
+      AttributeNames: this.attributeNames,
+      MaxNumberOfMessages: this.bulk,
+      WaitTimeSeconds: this.waitTimeSeconds,
+    };
+
+    const command = new ReceiveMessageCommand(input);
+
+    return (await this.sendCommand(command)) as ReceiveMessageResult;
+  }
+
   async deleteMessage(receiptHandle): Promise<unknown> {
     const deleteCommand = new DeleteMessageCommand({
       QueueUrl: this.queueUrl,

package/src/lib/queue/interfaces/queue.interface.ts

@@ -3,7 +3,8 @@ import { Message, ReceiveMessageResult } from '@aws-sdk/client-sqs';
 export interface IQueue {
   deleteMessage(message: Array<Message> | Message): Promise<unknown>;
   receiveMessages(): Promise<ReceiveMessageResult>;
-  setup({ waitTimeSeconds, bulk, queueUrl }): void;
+  receiveMessagesV2(): Promise<ReceiveMessageResult>;
+  setup({ waitTimeSeconds, bulk, queueUrl, attributeNames }): void;
 }
 
 export const IQueue = Symbol('IQueue');

package/src/lib/queue/queue.ts

@@ -2,7 +2,11 @@ import { Inject, Injectable } from '@nestjs/common';
 import { QueueConfig } from './types/queue.type';
 import { IConsumer } from './interfaces/consumer.interface';
 import { IQueue } from './interfaces/queue.interface';
-import { Message, ReceiveMessageResult } from '@aws-sdk/client-sqs';
+import {
+  Message,
+  QueueAttributeName,
+  ReceiveMessageResult,
+} from '@aws-sdk/client-sqs';
 
 @Injectable()
 export class Queue {
@@ -16,6 +20,10 @@ export class Queue {
 
   private waitTimeSeconds: number = 20;
 
+  private attributeNames: Array<QueueAttributeName> = [QueueAttributeName.All];
+
+  private version: number = 1;
+
   constructor(@Inject(IQueue) private readonly broker: IQueue) {}
 
   async handler({
@@ -23,26 +31,32 @@ export class Queue {
     context = null,
     bulk = 1,
     waitTimeSeconds = 20,
+    attributeNames = [QueueAttributeName.All],
+    version = 1,
   }: QueueConfig): Promise<void> {
     this.queueUrl = QueueUrl;
     this.bulk = bulk;
     this.context = context;
     this.waitTimeSeconds = waitTimeSeconds;
+    this.attributeNames = attributeNames;
+    this.version = version;
 
     this.broker.setup({
       waitTimeSeconds: this.waitTimeSeconds,
       bulk: this.bulk,
       queueUrl: this.queueUrl,
+      attributeNames: this.attributeNames,
     });
 
     while (!this.inWork) {
-      await this.run();
+      await this.run(version);
     }
   }
 
-  async run() {
+  async run(version: number) {
     this.inWork = true;
-    const response = await this.receiveMessages();
+
+    const response = await this.receiveMessages(version);
 
     try {
       if (this.bulk === 1) {
@@ -70,8 +84,12 @@ export class Queue {
     this.inWork = false;
   }
 
-  async receiveMessages(): Promise<ReceiveMessageResult> {
-    return await this.broker.receiveMessages();
+  async receiveMessages(version: number): Promise<ReceiveMessageResult> {
+    if (version === 1) {
+      return await this.broker.receiveMessages();
+    } else if (version === 2) {
+      return await this.broker.receiveMessagesV2();
+    }
   }
 
   async deleteMessage(receiptHandle): Promise<unknown> {

package/src/lib/queue/types/queue.type.ts

@@ -1,3 +1,4 @@
+import { QueueAttributeName } from '@aws-sdk/client-sqs';
 import { IConsumer } from '../interfaces/consumer.interface';
 
 export type QueueConfig = {
@@ -5,4 +6,6 @@ export type QueueConfig = {
   context: IConsumer;
   bulk: number;
   waitTimeSeconds?: number;
+  attributeNames?: Array<QueueAttributeName>;
+  version?: number;
 };