mirror of
https://github.com/EvolutionAPI/evolution-api.git
synced 2025-07-25 01:48:39 -06:00
test
This commit is contained in:
Davidson Gomes
parent
434f39de1f
commit
9c4f02634b
@ -1,34 +1,16 @@
|
||||
import { Channel, connect } from 'amqplib/callback_api';
|
||||
import * as amqp from 'amqplib/callback_api';
|
||||
|
||||
import { configService, HttpServer, Rabbitmq } from '../config/env.config';
|
||||
import { configService, Rabbitmq } from '../config/env.config';
|
||||
import { Logger } from '../config/logger.config';
|
||||
import { Events } from '../whatsapp/types/wa.types';
|
||||
|
||||
const logger = new Logger('AMQP');
|
||||
|
||||
const parseEvtName = (evt: string) => evt.replace(/_/g, '.').toLowerCase();
|
||||
|
||||
const globalQueues: { [key: string]: Events[] } = {
|
||||
contacts: [Events.CONTACTS_SET, Events.CONTACTS_UPDATE, Events.CONTACTS_UPSERT],
|
||||
messages: [
|
||||
Events.MESSAGES_DELETE,
|
||||
Events.MESSAGES_SET,
|
||||
Events.MESSAGES_UPDATE,
|
||||
Events.MESSAGES_UPSERT,
|
||||
Events.MESSAGING_HISTORY_SET,
|
||||
Events.SEND_MESSAGE,
|
||||
],
|
||||
chats: [Events.CHATS_DELETE, Events.CHATS_SET, Events.CHATS_UPDATE, Events.CHATS_UPSERT],
|
||||
groups: [Events.GROUPS_UPDATE, Events.GROUPS_UPSERT, Events.GROUP_PARTICIPANTS_UPDATE],
|
||||
others: [], // All other events not included in the above categories
|
||||
};
|
||||
|
||||
let amqpChannel: Channel | null = null;
|
||||
let amqpChannel: amqp.Channel | null = null;
|
||||
|
||||
export const initAMQP = () => {
|
||||
return new Promise<void>((resolve, reject) => {
|
||||
const rabbitConfig = configService.get<Rabbitmq>('RABBITMQ');
|
||||
connect(rabbitConfig.URI, (error, connection) => {
|
||||
const uri = configService.get<Rabbitmq>('RABBITMQ').URI;
|
||||
amqp.connect(uri, (error, connection) => {
|
||||
if (error) {
|
||||
reject(error);
|
||||
return;
|
||||
@ -40,7 +22,7 @@ export const initAMQP = () => {
|
||||
return;
|
||||
}
|
||||
|
||||
const exchangeName = rabbitConfig.EXCHANGE_NAME ?? 'evolution_exchange';
|
||||
const exchangeName = 'evolution_exchange';
|
||||
|
||||
channel.assertExchange(exchangeName, 'topic', {
|
||||
durable: true,
|
||||
@ -56,46 +38,28 @@ export const initAMQP = () => {
|
||||
});
|
||||
};
|
||||
|
||||
export const getAMQP = (): Channel | null => {
|
||||
export const getAMQP = (): amqp.Channel | null => {
|
||||
return amqpChannel;
|
||||
};
|
||||
|
||||
export const initQueues = (instanceName: string, events: string[]) => {
|
||||
if (!instanceName || !events || !events.length) return;
|
||||
const rabbitConfig = configService.get<Rabbitmq>('RABBITMQ');
|
||||
const amqp = getAMQP();
|
||||
if (!events || !events.length) return;
|
||||
|
||||
const rabbitMode = rabbitConfig.MODE || 'isolated';
|
||||
let exchangeName = rabbitConfig.EXCHANGE_NAME ?? 'evolution_exchange';
|
||||
const queues = events.map((event) => {
|
||||
return `${event.replace(/_/g, '.').toLowerCase()}`;
|
||||
});
|
||||
|
||||
const receivedEvents = events.map(parseEvtName);
|
||||
if (rabbitMode === 'isolated') {
|
||||
exchangeName = instanceName ?? 'evolution_exchange';
|
||||
queues.forEach((event) => {
|
||||
const amqp = getAMQP();
|
||||
const exchangeName = instanceName ?? 'evolution_exchange';
|
||||
|
||||
amqp.assertExchange(exchangeName, 'topic', {
|
||||
durable: true,
|
||||
autoDelete: false,
|
||||
});
|
||||
|
||||
receivedEvents.forEach((event) => {
|
||||
const queueName = `${instanceName}.${event}`;
|
||||
amqp.assertQueue(queueName, {
|
||||
durable: true,
|
||||
autoDelete: false,
|
||||
arguments: {
|
||||
'x-queue-type': 'quorum',
|
||||
},
|
||||
});
|
||||
const queueName = `${instanceName}.${event}`;
|
||||
|
||||
amqp.bindQueue(queueName, exchangeName, event);
|
||||
});
|
||||
} else if (rabbitMode === 'single') {
|
||||
amqp.assertExchange(exchangeName, 'topic', {
|
||||
durable: true,
|
||||
autoDelete: false,
|
||||
});
|
||||
|
||||
const queueName = 'evolution';
|
||||
amqp.assertQueue(queueName, {
|
||||
durable: true,
|
||||
autoDelete: false,
|
||||
@ -104,157 +68,33 @@ export const initQueues = (instanceName: string, events: string[]) => {
|
||||
},
|
||||
});
|
||||
|
||||
receivedEvents.forEach((event) => {
|
||||
amqp.bindQueue(queueName, exchangeName, event);
|
||||
});
|
||||
} else if (rabbitMode === 'global') {
|
||||
const queues = Object.keys(globalQueues);
|
||||
|
||||
const addQueues = queues.filter((evt) => {
|
||||
if (evt === 'others') {
|
||||
return receivedEvents.some(
|
||||
(e) =>
|
||||
!Object.values(globalQueues)
|
||||
.flat()
|
||||
.includes(e as Events),
|
||||
);
|
||||
}
|
||||
return globalQueues[evt].some((e) => receivedEvents.includes(e));
|
||||
});
|
||||
|
||||
addQueues.forEach((event) => {
|
||||
amqp.assertExchange(exchangeName, 'topic', {
|
||||
durable: true,
|
||||
autoDelete: false,
|
||||
});
|
||||
|
||||
const queueName = event;
|
||||
amqp.assertQueue(queueName, {
|
||||
durable: true,
|
||||
autoDelete: false,
|
||||
arguments: {
|
||||
'x-queue-type': 'quorum',
|
||||
},
|
||||
});
|
||||
|
||||
if (globalQueues[event].length === 0) {
|
||||
// Other events
|
||||
const otherEvents = Object.values(globalQueues).flat();
|
||||
for (const subEvent in Events) {
|
||||
const eventCode = Events[subEvent];
|
||||
if (otherEvents.includes(eventCode)) continue;
|
||||
if (!receivedEvents.includes(eventCode)) continue;
|
||||
amqp.bindQueue(queueName, exchangeName, eventCode);
|
||||
}
|
||||
} else {
|
||||
globalQueues[event].forEach((subEvent) => {
|
||||
amqp.bindQueue(queueName, exchangeName, subEvent);
|
||||
});
|
||||
}
|
||||
});
|
||||
} else {
|
||||
throw new Error('Invalid RabbitMQ mode');
|
||||
}
|
||||
amqp.bindQueue(queueName, exchangeName, event);
|
||||
});
|
||||
};
|
||||
|
||||
export const removeQueues = (instanceName: string, events: string[]) => {
|
||||
if (!events || !events.length) return;
|
||||
const rabbitConfig = configService.get<Rabbitmq>('RABBITMQ');
|
||||
|
||||
const rabbitMode = rabbitConfig.MODE || 'isolated';
|
||||
let exchangeName = rabbitConfig.EXCHANGE_NAME ?? 'evolution_exchange';
|
||||
const amqp = getAMQP();
|
||||
const channel = getAMQP();
|
||||
|
||||
const receivedEvents = events.map(parseEvtName);
|
||||
if (rabbitMode === 'isolated') {
|
||||
exchangeName = instanceName;
|
||||
receivedEvents.forEach((event) => {
|
||||
amqp.assertExchange(exchangeName, 'topic', {
|
||||
durable: true,
|
||||
autoDelete: false,
|
||||
});
|
||||
const queues = events.map((event) => {
|
||||
return `${event.replace(/_/g, '.').toLowerCase()}`;
|
||||
});
|
||||
|
||||
const queueName = `${instanceName}.${event}`;
|
||||
amqp.deleteQueue(queueName);
|
||||
const exchangeName = instanceName ?? 'evolution_exchange';
|
||||
|
||||
queues.forEach((event) => {
|
||||
const amqp = getAMQP();
|
||||
|
||||
amqp.assertExchange(exchangeName, 'topic', {
|
||||
durable: true,
|
||||
autoDelete: false,
|
||||
});
|
||||
amqp.deleteExchange(instanceName);
|
||||
}
|
||||
};
|
||||
|
||||
interface SendEventData {
|
||||
instanceName: string;
|
||||
wuid: string;
|
||||
event: string;
|
||||
apiKey?: string;
|
||||
data: any;
|
||||
}
|
||||
|
||||
export const sendEventData = ({ data, event, wuid, apiKey, instanceName }: SendEventData) => {
|
||||
const rabbitConfig = configService.get<Rabbitmq>('RABBITMQ');
|
||||
const rabbitMode = rabbitConfig.MODE || 'isolated';
|
||||
let exchangeName = rabbitConfig.EXCHANGE_NAME ?? 'evolution_exchange';
|
||||
if (rabbitMode === 'isolated') exchangeName = instanceName ?? 'evolution_exchange';
|
||||
|
||||
console.log('exchangeName: ', exchangeName);
|
||||
console.log('rabbitMode: ', rabbitMode);
|
||||
|
||||
amqpChannel.assertExchange(exchangeName, 'topic', {
|
||||
durable: true,
|
||||
autoDelete: false,
|
||||
});
|
||||
|
||||
let queueName = event;
|
||||
|
||||
if (rabbitMode === 'single') {
|
||||
queueName = 'evolution';
|
||||
} else if (rabbitMode === 'global') {
|
||||
let eventName = '';
|
||||
|
||||
Object.keys(globalQueues).forEach((key) => {
|
||||
if (globalQueues[key].includes(event as Events)) {
|
||||
eventName = key;
|
||||
}
|
||||
if (eventName === '' && key === 'others') {
|
||||
eventName = key;
|
||||
}
|
||||
});
|
||||
queueName = eventName;
|
||||
} else if (rabbitMode === 'isolated') {
|
||||
queueName = `${instanceName}.${event}`;
|
||||
}
|
||||
|
||||
amqpChannel.assertQueue(queueName, {
|
||||
durable: true,
|
||||
autoDelete: false,
|
||||
arguments: { 'x-queue-type': 'quorum' },
|
||||
});
|
||||
|
||||
console.log('envia na fila: ', queueName, exchangeName, event);
|
||||
|
||||
amqpChannel.bindQueue(queueName, exchangeName, event);
|
||||
|
||||
const serverUrl = configService.get<HttpServer>('SERVER').URL;
|
||||
const tzoffset = new Date().getTimezoneOffset() * 60000; //offset in milliseconds
|
||||
const localISOTime = new Date(Date.now() - tzoffset).toISOString();
|
||||
const now = localISOTime;
|
||||
|
||||
const message = {
|
||||
event,
|
||||
instance: instanceName,
|
||||
data,
|
||||
server_url: serverUrl,
|
||||
date_time: now,
|
||||
sender: wuid,
|
||||
};
|
||||
|
||||
if (apiKey) {
|
||||
message['apikey'] = apiKey;
|
||||
}
|
||||
|
||||
logger.log({
|
||||
queueName,
|
||||
exchangeName,
|
||||
event,
|
||||
});
|
||||
amqpChannel.publish(exchangeName, event, Buffer.from(JSON.stringify(message)));
|
||||
|
||||
const queueName = `${instanceName}.${event}`;
|
||||
|
||||
amqp.deleteQueue(queueName);
|
||||
});
|
||||
|
||||
channel.deleteExchange(exchangeName);
|
||||
};
|
||||
|
||||
@ -20,7 +20,7 @@ import {
|
||||
import { Logger } from '../../config/logger.config';
|
||||
import { ROOT_DIR } from '../../config/path.config';
|
||||
import { NotFoundException } from '../../exceptions';
|
||||
import { getAMQP, removeQueues, sendEventData } from '../../libs/amqp.server';
|
||||
import { getAMQP, removeQueues } from '../../libs/amqp.server';
|
||||
import { getIO } from '../../libs/socket.server';
|
||||
import { getSQS, removeQueues as removeQueuesSQS } from '../../libs/sqs.server';
|
||||
import { ChamaaiRaw, IntegrationRaw, ProxyRaw, RabbitmqRaw, SettingsRaw, SqsRaw, TypebotRaw } from '../models';
|
||||
@ -685,15 +685,40 @@ export class WAStartupService {
|
||||
|
||||
if (amqp) {
|
||||
if (Array.isArray(rabbitmqLocal) && rabbitmqLocal.includes(we)) {
|
||||
console.log('envia na fila: ', we);
|
||||
sendEventData({
|
||||
data,
|
||||
event,
|
||||
instanceName: this.instanceName,
|
||||
wuid: this.wuid,
|
||||
apiKey: expose && instanceApikey ? instanceApikey : undefined,
|
||||
const exchangeName = this.instanceName ?? 'evolution_exchange';
|
||||
|
||||
amqp.assertExchange(exchangeName, 'topic', {
|
||||
durable: true,
|
||||
autoDelete: false,
|
||||
});
|
||||
|
||||
const queueName = `${this.instanceName}.${event}`;
|
||||
|
||||
amqp.assertQueue(queueName, {
|
||||
durable: true,
|
||||
autoDelete: false,
|
||||
arguments: {
|
||||
'x-queue-type': 'quorum',
|
||||
},
|
||||
});
|
||||
|
||||
amqp.bindQueue(queueName, exchangeName, event);
|
||||
|
||||
const message = {
|
||||
event,
|
||||
instance: this.instance.name,
|
||||
data,
|
||||
server_url: serverUrl,
|
||||
date_time: now,
|
||||
sender: this.wuid,
|
||||
};
|
||||
|
||||
if (expose && instanceApikey) {
|
||||
message['apikey'] = instanceApikey;
|
||||
}
|
||||
|
||||
amqp.publish(exchangeName, event, Buffer.from(JSON.stringify(message)));
|
||||
|
||||
if (this.configService.get<Log>('LOG').LEVEL.includes('WEBHOOKS')) {
|
||||
const logData = {
|
||||
local: WAStartupService.name + '.sendData-RabbitMQ',
|
||||
|
||||
Loading…
Reference in New Issue
Block a user