mirrors/tgfs
1
0
Fork 0
mirror of https://github.com/TheodoreKrypton/tgfs synced 2024-11-22 07:04:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

shorten wait time for grouping requests

Browse Source
This commit is contained in:
Wheat Carrier 2023-11-12 09:00:59 +08:00
parent 2a8fbb8b74
commit 4f82c9141f
1 changed files with 24 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/api/client/message-api.ts
View File

@ -23,13 +23,14 @@ class MessageBroker {
protected buffer: Array<{
ids: number[];
resolve: (result: unknown) => void;
reject: (error: unknown) => void;
}> = [],
protected timeout: NodeJS.Timeout = null,
) {}
async getMessagesByIds(ids: number[]) {
return new Promise((resolve, reject) => {
this.buffer.push({ ids, resolve });
this.buffer.push({ ids, resolve, reject });
if (this.timeout) {
clearTimeout(this.timeout);
}
@ -37,18 +38,28 @@ class MessageBroker {
let buffer = [];
[buffer, this.buffer] = [[...this.buffer], []];
const ids = [...new Set(buffer.map((item) => item.ids).flat())];
const messages = await this.client.getMessages(this.privateChannelId, {
ids,
});
const messageMap = new Map();
messages.forEach((message) => {
messageMap.set(message.id, message);
});
buffer.forEach((item) => {
const result = item.ids.map((id: number) => messageMap.get(id));
item.resolve(result);
});
}, 500);
try {
const messages = await this.client.getMessages(
this.privateChannelId,
{
ids,
},
);
const messageMap = new Map();
messages.forEach((message) => {
messageMap.set(message.id, message);
});
buffer.forEach((item) => {
const result = item.ids.map((id: number) => messageMap.get(id));
item.resolve(result);
});
} catch (err) {
buffer.forEach((item) => {
item.reject(err);
});
}
}, 100);
});
}
}