feat: close connection in correct way (#10)

Author: Sebastian-Iwanczyszyn

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nestjstools/messaging-rabbitmq-extension",
-  "version": "2.12.0",
+  "version": "2.13.0",
   "description": "Extension to handle messages and dispatch them over AMQP protocol",
   "author": "Sebastian Iwanczyszyn",
   "private": false,
@@ -66,7 +66,7 @@
     "@nestjs/core": "^11.0.0",
     "@nestjs/schematics": "^11.0.0",
     "@nestjs/testing": "^11.0.0",
-    "@nestjstools/messaging": "^2.15.0",
+    "@nestjstools/messaging": "^2.21.0",
     "@semantic-release/github": "^11.0.1",
     "@types/express": "^5.0.0",
     "@types/jest": "^29.5.2",

src/channel/amqp.channel.ts

@@ -1,5 +1,4 @@
-import { Channel } from '@nestjstools/messaging';
-import { AmqpChannelConfig } from '@nestjstools/messaging';
+import { AmqpChannelConfig, Channel } from '@nestjstools/messaging';
 import { RmqChannelConfig as ExtensionAmqpChannelConfig } from './rmq-channel.config';
 import { Connection } from 'rabbitmq-client';
 
@@ -13,4 +12,9 @@ export class AmqpChannel extends Channel<
     super(config);
     this.connection = new Connection(config.connectionUri);
   }
+
+  async onChannelDestroy(): Promise<void> {
+    await this.connection.close();
+    return Promise.resolve();
+  }
 }

src/channel/rmq-channel-factory.service.ts renamed to src/channel/rmq-channel-factory.ts

@@ -12,7 +12,7 @@ import { Buffer } from 'buffer';
 @Injectable()
 @MessageConsumer(AmqpChannel)
 export class RabbitmqMessagingConsumer
-  implements IMessagingConsumer<AmqpChannel>, OnApplicationShutdown
+  implements IMessagingConsumer<AmqpChannel>
 {
   private channel?: AmqpChannel = undefined;
 
@@ -70,12 +70,6 @@ export class RabbitmqMessagingConsumer
 
     return Promise.resolve();
   }
-
-  async onApplicationShutdown(signal?: string): Promise<any> {
-    if (this.channel) {
-      await this.channel.connection.close();
-    }
-  }
 }
 
 interface RabbitMQMessage {

src/consumer/rabbitmq-messaging.consumer.ts

@@ -3,7 +3,7 @@ import { AmqpMessageBusFactory } from './message-bus/amqp-message-bus-factory';
 import { AmqpChannelFactory } from './channel/amqp-channel.factory';
 import { RabbitmqMessagingConsumer } from './consumer/rabbitmq-messaging.consumer';
 import { RabbitmqMigrator } from './migrator/rabbitmq.migrator';
-import { RmqChannelFactory } from './channel/rmq-channel-factory.service';
+import { RmqChannelFactory } from './channel/rmq-channel-factory';
 
 @Global()
 @Module({