From 6da66c15d88771c96a8f997308fc2dbc66165a15 Mon Sep 17 00:00:00 2001
From: Michael Eischer <michael.eischer@fau.de>
Date: Fri, 6 Nov 2020 23:54:20 +0100
Subject: [PATCH] check: Simplify referenced blob tracking

The result is identical as long as the context in not canceled. However,
in that case the result is incomplete anyways.
---
 internal/checker/checker.go | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/internal/checker/checker.go b/internal/checker/checker.go
index 09a71f9de..19501b4e6 100644
--- a/internal/checker/checker.go
+++ b/internal/checker/checker.go
@@ -520,6 +520,8 @@ func (c *Checker) filterTrees(ctx context.Context, backlog restic.IDs, loaderCha
 			c.blobRefs.Lock()
 			h := restic.BlobHandle{ID: nextTreeID, Type: restic.TreeBlob}
 			blobReferenced := c.blobRefs.M.Has(h)
+			// noop if already referenced
+			c.blobRefs.M.Insert(h)
 			c.blobRefs.Unlock()
 			if blobReferenced {
 				continue
@@ -540,10 +542,6 @@ func (c *Checker) filterTrees(ctx context.Context, backlog restic.IDs, loaderCha
 		case loadCh <- nextTreeID:
 			outstandingLoadTreeJobs++
 			loadCh = nil
-			c.blobRefs.Lock()
-			h := restic.BlobHandle{ID: nextTreeID, Type: restic.TreeBlob}
-			c.blobRefs.M.Insert(h)
-			c.blobRefs.Unlock()
 
 		case j, ok := <-inCh:
 			if !ok {