franpcode/Baileys
1
0
Fork 0
mirror of https://github.com/FranP-code/Baileys.git synced 2025-10-13 00:32:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: use (participant, ID) tuple for retry counter

Browse Source
This commit is contained in:
Adhiraj Singh
2022-06-16 15:25:31 +05:30
parent 542617d79c
commit 349002857e
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/Socket/messages-recv.ts
View File

@@ -304,11 +304,17 @@ export const makeMessagesRecvSocket = (config: SocketConfig) => {
}
}
const willSendMessageAgain = (id: string) => {
const retryCount = msgRetryMap[id] || 0
const willSendMessageAgain = (id: string, participant: string) => {
const key = `${id}:${participant}`
const retryCount = msgRetryMap[key] || 0
return retryCount < 5
}
const updateSendMessageAgainCount = (id: string, participant: string) => {
const key = `${id}:${participant}`
msgRetryMap[key] = (msgRetryMap[key] || 0) + 1
}
const sendMessagesAgain = async(key: proto.IMessageKey, ids: string[]) => {
const msgs = await Promise.all(
ids.map(id => (
@@ -327,7 +333,7 @@ export const makeMessagesRecvSocket = (config: SocketConfig) => {
for(let i = 0; i < msgs.length;i++) {
if(msgs[i]) {
msgRetryMap[ids[i]] = (msgRetryMap[ids[i]] || 0) + 1
updateSendMessageAgainCount(ids[i], participant)
await relayMessage(key.remoteJid, msgs[i], {
messageId: ids[i],
participant
@@ -397,9 +403,9 @@ export const makeMessagesRecvSocket = (config: SocketConfig) => {
}
if(attrs.type === 'retry') {
if(willSendMessageAgain(ids[0])) {
// correctly set who is asking for the retry
key.participant = key.participant || attrs.from
// correctly set who is asking for the retry
key.participant = key.participant || attrs.from
if(willSendMessageAgain(ids[0], key.participant)) {
if(key.fromMe) {
try {
logger.debug({ attrs, key }, 'recv retry request')