rmq.service.ts

import {Injectable} from "@nestjs/common";
import {AmqpConnection, RequestOptions} from "@golevelup/nestjs-rabbitmq";
import {DEFAULT_EXCHANGE_NAME, DEFAULT_TIMEOUT} from "./rmq.constant";

type MessageOptions = Pick<RequestOptions, | 'payload' | 'routingKey'> & {
    exchange?: string,
    timeout?: number,
    headers?: any
};

@Injectable()
export class RmqService {
    constructor(private readonly rmq: AmqpConnection) {
    }

    async publish(options: MessageOptions) {
        const {exchange, routingKey, payload, ...otherOptions} = options;
        await this.rmq.publish(
            exchange || DEFAULT_EXCHANGE_NAME,
            routingKey,
            payload,
            otherOptions
        );
    }

    async request<T>(options: MessageOptions) {
        const {exchange, timeout} = options;
        return await this.rmq.request<T>({
            ...options,
            exchange: exchange || DEFAULT_EXCHANGE_NAME,
            timeout: timeout || DEFAULT_TIMEOUT,
        });
    }
}