NeoAnd/evolution-api
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/evolution-api.git synced 2025-07-24 17:38:40 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: Recover lost messages from cache in WhatsApp Baileys service

Browse Source
This commit is contained in:
Davidson Gomes 2024-06-26 16:57:55 -03:00
parent a737fed5cb
commit 0ded76d366
1 changed files with 33 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
src/api/services/channels/whatsapp.baileys.service.ts
View File

@ -162,22 +162,45 @@ export class BaileysStartupService extends ChannelStartupService {
public phoneNumber: string; public phoneNumber: string;
private async recoveringMessages() { private async recoveringMessages() {
this.logger.info('Recovering messages lost');
const cacheConf = this.configService.get<CacheConf>('CACHE'); const cacheConf = this.configService.get<CacheConf>('CACHE');
if ((cacheConf?.REDIS?.ENABLED && cacheConf?.REDIS?.URI !== '') || cacheConf?.LOCAL?.ENABLED) { if ((cacheConf?.REDIS?.ENABLED && cacheConf?.REDIS?.URI !== '') || cacheConf?.LOCAL?.ENABLED) {
this.logger.info('Recovering messages lost from cache');
setInterval(async () => { setInterval(async () => {
this.baileysCache.keys().then((keys) => { this.baileysCache.keys().then((keys) => {
keys.forEach(async (key) => { keys.forEach(async (key) => {
const message = await this.baileysCache.get(key.split(':')[2]); const data = await this.baileysCache.get(key.split(':')[2]);
let message: any;
let retry: number;
if (!data?.message) {
message = data;
retry = 0;
} else {
message = data.message;
retry = data.retry;
}
if (message.messageStubParameters && message.messageStubParameters[0] === 'Message absent from node') { if (message.messageStubParameters && message.messageStubParameters[0] === 'Message absent from node') {
this.logger.info('Message absent from node, retrying to send, key: ' + key.split(':')[2]); retry = retry + 1;
await this.client.sendMessageAck(JSON.parse(message.messageStubParameters[1], BufferJSON.reviver)); this.logger.info(`Message absent from node, retrying to send, key: ${key.split(':')[2]} retry: ${retry}`);
if (message.messageStubParameters[1]) {
await this.client.sendMessageAck(JSON.parse(message.messageStubParameters[1], BufferJSON.reviver));
}
this.baileysCache.set(key.split(':')[2], { message, retry });
if (retry >= 100) {
this.logger.warn(`Message absent from node, retry limit reached, key: ${key.split(':')[2]}`);
this.baileysCache.delete(key.split(':')[2]);
return;
}
} }
}); });
}); });
}, 30000); // 15 minutes
}, 15 * 60 * 1000);
} }
} }
@ -1112,9 +1135,12 @@ export class BaileysStartupService extends ChannelStartupService {
} }
if (received.messageStubParameters && received.messageStubParameters[0] === 'Message absent from node') { if (received.messageStubParameters && received.messageStubParameters[0] === 'Message absent from node') {
this.logger.info('Recovering message lost'); this.logger.info(`Recovering message lost messageId: ${received.key.id}`);
await this.baileysCache.set(received.key.id, received); await this.baileysCache.set(received.key.id, {
message: received,
retry: 0,
});
continue; continue;
} }