feat: optimize batch processing in incrementBatch and setBatch methods

Author: lohanidamodar

src/Usage/Adapter/Database.php

@@ -99,7 +99,7 @@ protected function getColumnDefinition(string $id): string
 
     public function incrementBatch(array $metrics, int $batchSize = 1000): bool
     {
-        $this->db->getAuthorization()->skip(function () use ($metrics) {
+        $this->db->getAuthorization()->skip(function () use ($metrics, $batchSize) {
             $documentsById = [];
             foreach ($metrics as $metric) {
                 $period = $metric['period'] ?? '1h';
@@ -133,13 +133,13 @@ public function incrementBatch(array $metrics, int $batchSize = 1000): bool
                 }
             }
 
-            $documents = [];
-            foreach ($documentsById as $doc) {
-                $documents[] = new Document($doc);
-            }
+            $documents = array_values(array_map(
+                static fn (array $doc) => new Document($doc),
+                $documentsById
+            ));
 
-            if (!empty($documents)) {
-                $this->db->upsertDocumentsWithIncrease($this->collection, 'value', $documents);
+            foreach (array_chunk($documents, max(1, $batchSize)) as $chunk) {
+                $this->db->upsertDocumentsWithIncrease($this->collection, 'value', $chunk);
             }
         });
 
@@ -157,7 +157,7 @@ public function incrementBatch(array $metrics, int $batchSize = 1000): bool
      */
     public function setBatch(array $metrics, int $batchSize = 1000): bool
     {
-        $this->db->getAuthorization()->skip(function () use ($metrics) {
+        $this->db->getAuthorization()->skip(function () use ($metrics, $batchSize) {
             $documentsById = [];
             foreach ($metrics as $metric) {
                 $period = $metric['period'] ?? '1h';
@@ -176,7 +176,7 @@ public function setBatch(array $metrics, int $batchSize = 1000): bool
 
                 $id = $this->buildDeterministicId($metric['metric'], $period, $time);
 
-                // Last one wins for the same ID (counter behavior, not aggregating)
+                // Last one wins for the same ID (replace behavior, not aggregating)
                 $documentsById[$id] = [
                     '$id' => $id,
                     '$permissions' => [],
@@ -188,13 +188,13 @@ public function setBatch(array $metrics, int $batchSize = 1000): bool
                 ];
             }
 
-            $documents = [];
-            foreach ($documentsById as $doc) {
-                $documents[] = new Document($doc);
-            }
+            $documents = array_values(array_map(
+                static fn (array $doc) => new Document($doc),
+                $documentsById
+            ));
 
-            if (!empty($documents)) {
-                $this->db->upsertDocuments($this->collection, $documents);
+            foreach (array_chunk($documents, max(1, $batchSize)) as $chunk) {
+                $this->db->upsertDocuments($this->collection, $chunk);
             }
         });