Merge pull request #1 from benvan/master

feat: improve perf of groupMessagesForPartition
terebentina · Jun 9, 2023 · 6e68af6 · 6e68af6
2 parents ff3b111 + fa724d8
commit 6e68af6
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/producer/groupMessagesPerPartition.js b/src/producer/groupMessagesPerPartition.js
@@ -3,9 +3,14 @@ module.exports = ({ topic, partitionMetadata, messages, partitioner }) => {
     return {}
   }
 
-  return messages.reduce((result, message) => {
+  const grouping = {}
+  for (const message of messages) {
     const partition = partitioner({ topic, partitionMetadata, message })
-    const current = result[partition] || []
-    return Object.assign(result, { [partition]: [...current, message] })
-  }, {})
+    if (grouping[partition] === undefined) {
+      grouping[partition] = []
+    }
+    grouping[partition].push(message)
+  }
+
+  return grouping
 }