Fix the accounting for batch size

jianoaix · Nov 29, 2023 · 4c80cba · 4c80cba
1 parent cefcf3e
commit 4c80cba
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/core/data.go b/core/data.go
@@ -160,3 +160,14 @@ func (cb Bundles) Serialize() ([][][]byte, error) {
 	}
 	return data, nil
 }
+
+// Returns the size of the bundles in bytes.
+func (cb Bundles) Size() int {
+	size := 0
+	for _, bundle := range cb {
+		for _, chunk := range bundle {
+			size += chunk.Size()
+		}
+	}
+	return size
+}
diff --git a/node/node.go b/node/node.go
@@ -236,7 +236,7 @@ func (n *Node) ProcessBatch(ctx context.Context, header *core.BatchHeader, blobs
 	// Measure num batches received and its size in bytes
 	batchSize := 0
 	for _, blob := range blobs {
-		batchSize += blob.BlobHeader.EncodedSizeAllQuorums()
+		batchSize += blob.Bundles.Size()
 	}
 	n.Metrics.AcceptBatches("received", batchSize)