手把手教你使用 Nest.js 集成 RabbitMQ

2024-02-24 10:32:30

介绍

在构建现代分布式系统时, 常常需要在不同的服务之间进行通信. RabbitMQ 是一个流行的消息队列系统, 可以帮助您在服务之间可靠地传递消息, 并提供诸如负载均衡、消息持久化、消息重试等功能, 可以帮助您提高系统的可靠性和性能.

集成步骤

安装依赖

首先, 在 Nest.js 项目中安装 RabbitMQ 的相关依赖包:

npm install @nestjs/microservices @nestjs/rabbitmq

创建模块

接下来, 创建一个模块来配置 RabbitMQ 的连接:

import { Module } from '@nestjs/common';
import { ClientsModule, Transport } from '@nestjs/microservices';

@Module({
  imports: [
    ClientsModule.register([
      {
        name: 'RABBITMQ_SERVICE',
        transport: Transport.RMQ,
        options: {
          urls: ['amqp://localhost:5672'],
          queue: 'my_queue',
        },
      },
    ]),
  ],
})
export class RabbitMQModule {}

创建服务

然后, 创建一个服务来消费 RabbitMQ 消息:

import { Injectable, OnModuleInit } from '@nestjs/common';
import { ClientProxy, ClientProxyFactory, MessagePattern } from '@nestjs/microservices';

@Injectable()
export class AppService implements OnModuleInit {
  private client: ClientProxy;

  onModuleInit() {
    this.client = ClientProxyFactory.create({
      transport: Transport.RMQ,
      options: {
        urls: ['amqp://localhost:5672'],
        queue: 'my_queue',
      },
    });
  }

  @MessagePattern('my_pattern')
  async handleMessage(data: any) {
    console.log('Received message:', data);
  }
}

使用服务

最后, 在需要使用 RabbitMQ 消息队列的地方, 直接使用前面创建的服务即可:

import { Injectable } from '@nestjs/common';
import { AppService } from './app.service';

@Injectable()
export class MyController {
  constructor(private readonly appService: AppService) {}

  async sendMessage() {
    this.appService.client.send('my_pattern', { message: 'Hello world!' });
  }
}