zio
/
bsky-app
2
0
Fork 0
Code Pull Requests Packages Releases Activity

Aggregate quickly-sent messages into batches (#4061)

zio/stable
Eric Bailey 2024-05-16 16:39:05 -05:00 committed by GitHub
parent a84a14d084
commit d424945eed
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 23 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/state/messages/convo/agent.ts
View File

@ -766,12 +766,12 @@ export class Convo {
logger.DebugContext.convo, logger.DebugContext.convo,
) )
const pendingMessage = Array.from(this.pendingMessages.values()).shift() const pendingMessages = Array.from(this.pendingMessages.values())
/* /*
* If there are no pending messages, we're done. * If there are no pending messages, we're done.
*/ */
if (!pendingMessage) { if (pendingMessages.length === 0) {
this.isProcessingPendingMessages = false this.isProcessingPendingMessages = false
return return
} }
@ -779,28 +779,34 @@ export class Convo {
try { try {
this.isProcessingPendingMessages = true this.isProcessingPendingMessages = true
const {id, message} = pendingMessage
const response = await networkRetry(2, () => { const response = await networkRetry(2, () => {
return this.agent.api.chat.bsky.convo.sendMessage( return this.agent.api.chat.bsky.convo.sendMessageBatch(
{ {
items: pendingMessages.map(({message}) => ({
convoId: this.convoId, convoId: this.convoId,
message, message,
})),
}, },
{encoding: 'application/json', headers: DM_SERVICE_HEADERS}, {encoding: 'application/json', headers: DM_SERVICE_HEADERS},
) )
}) })
const res = response.data
const {items} = response.data
for (let i = 0; i < items.length; i++) {
const msg = items[i]
const tempId = pendingMessages[i].id
/* /*
* Insert into `newMessages` as soon as we have a real ID. That way, when * Insert into `newMessages` as soon as we have a real ID. That way, when
* we get an event log back, we can replace in situ. * we get an event log back, we can replace in situ.
*/ */
this.newMessages.set(res.id, { this.newMessages.set(msg.id, {
...res, ...msg,
$type: 'chat.bsky.convo.defs#messageView', $type: 'chat.bsky.convo.defs#messageView',
}) })
this.pendingMessages.delete(id) this.pendingMessages.delete(tempId)
}
await this.processPendingMessages() await this.processPendingMessages()