From b7960dbb9a3c9385ab1f04c002b4fa50573f8ac6 Mon Sep 17 00:00:00 2001
From: Matheus Filype <67132916+Santosl2@users.noreply.github.com>
Date: Fri, 12 Dec 2025 01:50:18 -0300
Subject: [PATCH] feat: Optimize Offline Node Processing with Batching and
 Event Loop Yielding (#2138)

* fix: improve message resend logic by adding checks for message IDs

* Revert "fix: improve message resend logic by adding checks for message IDs"

This reverts commit c03f9d8e6fc6cbfbb9d1f8f67c169700e704213d.

* fix(messages-recv): improve offline node processing and error handling

* feat: improve offline node processing by yielding to event loop
---
 src/Socket/messages-recv.ts | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/src/Socket/messages-recv.ts b/src/Socket/messages-recv.ts
index cd237792..9363d2b4 100644
--- a/src/Socket/messages-recv.ts
+++ b/src/Socket/messages-recv.ts
@@ -1439,6 +1439,11 @@ export const makeMessagesRecvSocket = (config: SocketConfig) => {
 		node: BinaryNode
 	}
 
+	/** Yields control to the event loop to prevent blocking */
+	const yieldToEventLoop = (): Promise<void> => {
+		return new Promise(resolve => setImmediate(resolve))
+	}
+
 	const makeOfflineNodeProcessor = () => {
 		const nodeProcessorMap: Map<MessageType, (node: BinaryNode) => Promise<void>> = new Map([
 			['message', handleMessage],
@@ -1449,6 +1454,9 @@ export const makeMessagesRecvSocket = (config: SocketConfig) => {
 		const nodes: OfflineNode[] = []
 		let isProcessing = false
 
+		// Number of nodes to process before yielding to event loop
+		const BATCH_SIZE = 10
+
 		const enqueue = (type: MessageType, node: BinaryNode) => {
 			nodes.push({ type, node })
 
@@ -1459,6 +1467,8 @@ export const makeMessagesRecvSocket = (config: SocketConfig) => {
 			isProcessing = true
 
 			const promise = async () => {
+				let processedInBatch = 0
+
 				while (nodes.length && ws.isOpen) {
 					const { type, node } = nodes.shift()!
 
@@ -1470,6 +1480,14 @@ export const makeMessagesRecvSocket = (config: SocketConfig) => {
 					}
 
 					await nodeProcessor(node)
+					processedInBatch++
+
+					// Yield to event loop after processing a batch
+					// This prevents blocking the event loop for too long when there are many offline nodes
+					if (processedInBatch >= BATCH_SIZE) {
+						processedInBatch = 0
+						await yieldToEventLoop()
+					}
 				}
 
 				isProcessing = false