runmq 1.4.2 → 1.4.3

package/dist/index.js

@@ -27,8 +27,8 @@ var RunMQException = class extends Error {
   }
 };
 
-// src/core/clients/AmqplibClientAdapter.ts
-import amqplib from "amqplib";
+// src/core/clients/RabbitMQClientAdapter.ts
+import { Connection } from "rabbitmq-client";
 
 // src/core/exceptions/Exceptions.ts
 var Exceptions = class {
@@ -40,8 +40,8 @@ Exceptions.INVALID_MESSAGE_FORMAT = "MESSAGE_SHOULD_BE_VALID_RECORD";
 Exceptions.UNSUPPORTED_SCHEMA = "UNSUPPORTED_SCHEMA";
 Exceptions.FAILURE_TO_DEFINE_TTL_POLICY = "FAILURE_TO_DEFINE_TTL_POLICY";
 
-// src/core/clients/AmqplibChannel.ts
-var AmqplibChannel = class {
+// src/core/clients/RabbitMQClientChannel.ts
+var RabbitMQClientChannel = class {
   constructor(channel) {
     this.channel = channel;
   }
@@ -51,7 +51,8 @@ var AmqplibChannel = class {
     if (options == null ? void 0 : options.deadLetterRoutingKey) args["x-dead-letter-routing-key"] = options.deadLetterRoutingKey;
     if (options == null ? void 0 : options.messageTtl) args["x-message-ttl"] = options.messageTtl;
     if (options == null ? void 0 : options.arguments) Object.assign(args, options.arguments);
-    const result = await this.channel.assertQueue(queue, {
+    const result = await this.channel.queueDeclare({
+      queue,
       durable: options == null ? void 0 : options.durable,
       exclusive: options == null ? void 0 : options.exclusive,
       autoDelete: options == null ? void 0 : options.autoDelete,
@@ -64,7 +65,10 @@ var AmqplibChannel = class {
     };
   }
   async checkQueue(queue) {
-    const result = await this.channel.checkQueue(queue);
+    const result = await this.channel.queueDeclare({
+      queue,
+      passive: true
+    });
     return {
       queue: result.queue,
       messageCount: result.messageCount,
@@ -72,7 +76,8 @@ var AmqplibChannel = class {
     };
   }
   async deleteQueue(queue, options) {
-    const result = await this.channel.deleteQueue(queue, {
+    const result = await this.channel.queueDelete({
+      queue,
       ifUnused: options == null ? void 0 : options.ifUnused,
       ifEmpty: options == null ? void 0 : options.ifEmpty
     });
@@ -84,7 +89,9 @@ var AmqplibChannel = class {
     const args = {};
     if (options == null ? void 0 : options.alternateExchange) args["alternate-exchange"] = options.alternateExchange;
     if (options == null ? void 0 : options.arguments) Object.assign(args, options.arguments);
-    await this.channel.assertExchange(exchange, type, {
+    await this.channel.exchangeDeclare({
+      exchange,
+      type,
       durable: options == null ? void 0 : options.durable,
       internal: options == null ? void 0 : options.internal,
       autoDelete: options == null ? void 0 : options.autoDelete,
@@ -95,26 +102,37 @@ var AmqplibChannel = class {
     };
   }
   async checkExchange(exchange) {
-    await this.channel.checkExchange(exchange);
+    await this.channel.exchangeDeclare({
+      exchange,
+      passive: true
+    });
     return {
       exchange
     };
   }
   async deleteExchange(exchange, options) {
-    await this.channel.deleteExchange(exchange, {
+    await this.channel.exchangeDelete({
+      exchange,
       ifUnused: options == null ? void 0 : options.ifUnused
     });
   }
   async bindQueue(queue, source, pattern, args) {
-    await this.channel.bindQueue(queue, source, pattern, args);
+    await this.channel.queueBind({
+      queue,
+      exchange: source,
+      routingKey: pattern,
+      arguments: args
+    });
   }
   publish(exchange, routingKey, content, options) {
     var _a2;
-    return this.channel.publish(exchange, routingKey, content, {
+    this.channel.basicPublish({
+      exchange,
+      routingKey,
       correlationId: options == null ? void 0 : options.correlationId,
       messageId: options == null ? void 0 : options.messageId,
       headers: options == null ? void 0 : options.headers,
-      persistent: options == null ? void 0 : options.persistent,
+      durable: options == null ? void 0 : options.persistent,
       expiration: (_a2 = options == null ? void 0 : options.expiration) == null ? void 0 : _a2.toString(),
       contentType: options == null ? void 0 : options.contentType,
       contentEncoding: options == null ? void 0 : options.contentEncoding,
@@ -124,67 +142,69 @@ var AmqplibChannel = class {
       type: options == null ? void 0 : options.type,
       userId: options == null ? void 0 : options.userId,
       appId: options == null ? void 0 : options.appId
-    });
+    }, content);
+    return true;
   }
   async consume(queue, onMessage, options) {
-    const result = await this.channel.consume(queue, (msg) => {
-      if (msg === null) {
-        onMessage(null);
-        return;
-      }
+    const result = await this.channel.basicConsume({
+      queue,
+      consumerTag: options == null ? void 0 : options.consumerTag,
+      noLocal: options == null ? void 0 : options.noLocal,
+      noAck: options == null ? void 0 : options.noAck,
+      exclusive: options == null ? void 0 : options.exclusive,
+      arguments: options == null ? void 0 : options.arguments
+    }, (msg) => {
+      const body = msg.body;
+      const content = Buffer.isBuffer(body) ? body : typeof body === "string" ? Buffer.from(body) : Buffer.from(JSON.stringify(body));
       const consumeMessage = {
-        content: msg.content,
+        content,
         fields: {
-          consumerTag: msg.fields.consumerTag,
-          deliveryTag: msg.fields.deliveryTag,
-          redelivered: msg.fields.redelivered,
-          exchange: msg.fields.exchange,
-          routingKey: msg.fields.routingKey
+          consumerTag: msg.consumerTag,
+          deliveryTag: msg.deliveryTag,
+          redelivered: msg.redelivered,
+          exchange: msg.exchange,
+          routingKey: msg.routingKey
         },
         properties: {
-          contentType: msg.properties.contentType || void 0,
-          contentEncoding: msg.properties.contentEncoding || void 0,
-          headers: msg.properties.headers || {},
-          deliveryMode: msg.properties.deliveryMode,
-          priority: msg.properties.priority,
-          correlationId: msg.properties.correlationId || void 0,
-          replyTo: msg.properties.replyTo || void 0,
-          expiration: msg.properties.expiration || void 0,
-          messageId: msg.properties.messageId || void 0,
-          timestamp: msg.properties.timestamp,
-          type: msg.properties.type || void 0,
-          userId: msg.properties.userId || void 0,
-          appId: msg.properties.appId || void 0
+          contentType: msg.contentType,
+          contentEncoding: msg.contentEncoding,
+          headers: msg.headers || {},
+          deliveryMode: msg.durable ? 2 : 1,
+          priority: msg.priority,
+          correlationId: msg.correlationId,
+          replyTo: msg.replyTo,
+          expiration: msg.expiration,
+          messageId: msg.messageId,
+          timestamp: msg.timestamp,
+          type: msg.type,
+          userId: msg.userId,
+          appId: msg.appId
         }
       };
       onMessage(consumeMessage);
-    }, {
-      consumerTag: options == null ? void 0 : options.consumerTag,
-      noLocal: options == null ? void 0 : options.noLocal,
-      noAck: options == null ? void 0 : options.noAck,
-      exclusive: options == null ? void 0 : options.exclusive,
-      priority: options == null ? void 0 : options.priority,
-      arguments: options == null ? void 0 : options.arguments
     });
     return {
       consumerTag: result.consumerTag
     };
   }
   ack(message, allUpTo) {
-    this.channel.ack(
-      { fields: message.fields, content: message.content, properties: message.properties },
-      allUpTo
-    );
+    this.channel.basicAck({
+      deliveryTag: message.fields.deliveryTag,
+      multiple: allUpTo
+    });
   }
   nack(message, allUpTo, requeue) {
-    this.channel.nack(
-      { fields: message.fields, content: message.content, properties: message.properties },
-      allUpTo,
+    this.channel.basicNack({
+      deliveryTag: message.fields.deliveryTag,
+      multiple: allUpTo,
       requeue
-    );
+    });
   }
   async prefetch(count, global) {
-    await this.channel.prefetch(count, global);
+    await this.channel.basicQos({
+      prefetchCount: count,
+      global
+    });
   }
   async close() {
     await this.channel.close();
@@ -219,8 +239,8 @@ var RunMQConsoleLogger = class {
   }
 };
 
-// src/core/clients/AmqplibClientAdapter.ts
-var AmqplibClientAdapter = class {
+// src/core/clients/RabbitMQClientAdapter.ts
+var RabbitMQClientAdapter = class {
   constructor(config, logger = new RunMQConsoleLogger()) {
     this.config = config;
     this.logger = logger;
@@ -239,20 +259,38 @@ var AmqplibClientAdapter = class {
         }
         this.connection = void 0;
       }
-      const connection = await amqplib.connect(this.config.url);
-      this.connection = connection;
+      this.connection = new Connection({
+        url: this.config.url,
+        // Disable automatic retries - we handle retries at RunMQ level
+        retryLow: 100,
+        retryHigh: 200,
+        connectionTimeout: 5e3
+      });
       this.connection.on("error", (err) => {
         this.logger.error("RabbitMQ connection error:", { error: err });
         this.isConnected = false;
       });
-      this.connection.on("close", () => {
-        this.isConnected = false;
+      this.connection.on("connection", () => {
+        this.isConnected = true;
+      });
+      this.connection.on("connection.blocked", (reason) => {
+        this.logger.warn("RabbitMQ connection blocked:", { reason });
       });
+      this.connection.on("connection.unblocked", () => {
+        this.logger.info("RabbitMQ connection unblocked");
+      });
+      await this.connection.onConnect(5e3, true);
       this.isConnected = true;
       return this.connection;
     } catch (error) {
       this.isConnected = false;
-      this.connection = void 0;
+      if (this.connection) {
+        try {
+          this.connection.close();
+        } catch (e) {
+        }
+        this.connection = void 0;
+      }
       throw new RunMQException(
         Exceptions.CONNECTION_NOT_ESTABLISHED,
         {
@@ -263,9 +301,9 @@ var AmqplibClientAdapter = class {
   }
   async getChannel() {
     const connection = await this.connect();
-    const rawChannel = await connection.createChannel();
+    const rawChannel = await connection.acquire();
     this.acquiredChannels.push(rawChannel);
-    return new AmqplibChannel(rawChannel);
+    return new RabbitMQClientChannel(rawChannel);
   }
   async getDefaultChannel() {
     if (!this.defaultChannel) {
@@ -282,7 +320,9 @@ var AmqplibClientAdapter = class {
     this.acquiredChannels = [];
     for (const channel of channels) {
       try {
-        await channel.close();
+        if (channel.active) {
+          await channel.close();
+        }
       } catch (e) {
       }
     }
@@ -1291,7 +1331,7 @@ var RunMQ = class _RunMQ {
       reconnectDelay: (_a2 = config.reconnectDelay) != null ? _a2 : DEFAULTS.RECONNECT_DELAY,
       maxReconnectAttempts: (_b = config.maxReconnectAttempts) != null ? _b : DEFAULTS.MAX_RECONNECT_ATTEMPTS
     });
-    this.client = new AmqplibClientAdapter(this.config, this.logger);
+    this.client = new RabbitMQClientAdapter(this.config, this.logger);
     this.consumer = new RunMQConsumerCreator(this.client, this.logger, this.config.management);
   }
   /**