From 1d41e1544a3f909c697341cb3339da30f92fa095 Mon Sep 17 00:00:00 2001
From: vyzo <vyzo@hackzen.org>
Date: Fri, 2 Jul 2021 14:46:02 +0300
Subject: [PATCH] optimize transitive write tracking a bit

---
 blockstore/splitstore/splitstore.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/blockstore/splitstore/splitstore.go b/blockstore/splitstore/splitstore.go
index 10cf22e0e..82e98e60f 100644
--- a/blockstore/splitstore/splitstore.go
+++ b/blockstore/splitstore/splitstore.go
@@ -592,8 +592,15 @@ func (s *SplitStore) flushPendingWrites(locked bool) {
 	}
 
 	cids := make([]cid.Cid, 0, len(s.pendingWrites))
+	seen := make(map[cid.Cid]struct{})
 	for c := range s.pendingWrites {
+		_, ok := seen[c]
+		if ok {
+			continue
+		}
+
 		cids = append(cids, c)
+		seen[c] = struct{}{}
 
 		// recursively walk dags to propagate dependent references
 		if c.Prefix().Codec != cid.DagCBOR {
@@ -601,9 +608,10 @@ func (s *SplitStore) flushPendingWrites(locked bool) {
 		}
 
 		err := s.walkLinks(c, cid.NewSet(), func(c cid.Cid) error {
-			_, has := s.pendingWrites[c]
-			if !has {
+			_, ok := seen[c]
+			if !ok {
 				cids = append(cids, c)
+				seen[c] = struct{}{}
 			}
 
 			return nil