Add commit support

ChainSafe · Sep 2, 2024 · 4c2bcb2 · 4c2bcb2
1 parent 381018b
commit 4c2bcb2
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/pkg/mmr/mmr.go b/pkg/mmr/mmr.go
@@ -104,6 +104,10 @@ func (mmr *MMR) Root() (MMRElement, error) {
 	return mmr.bagPeaks(peaks), nil
 }
 
+func (mmr *MMR) Commit() error {
+	return mmr.batch.commit()
+}
+
 func (mmr *MMR) findElement(position uint64, values []MMRElement) (MMRElement, error) {
 	if position > mmr.size {
 		positionOffset := position - mmr.size

diff --git a/pkg/mmr/mmr_batch.go b/pkg/mmr/mmr_batch.go
@@ -48,3 +48,20 @@ func (b *MMRBatch) getElement(pos uint64) (*MMRElement, error) {
 
 	return b.storage.getElement(pos)
 }
+
+func (b *MMRBatch) commit() error {
+	for _, node := range b.drain() {
+		err := b.storage.append(node.pos, node.elements)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func (b *MMRBatch) drain() []MMRNode {
+	nodes := b.nodes
+	b.nodes = make([]MMRNode, 0)
+
+	return nodes
+}