From d5482089bfd0348ed886be9f89c4d6b177cc6dce Mon Sep 17 00:00:00 2001
From: Matt Moore <mattmoor@google.com>
Date: Tue, 31 Jan 2017 16:36:17 -0800
Subject: [PATCH] Add tracking to elided layer pulls.

Signed-off-by: Matt Moore <mattmoor@google.com>
---
 distribution/xfer/download.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/distribution/xfer/download.go b/distribution/xfer/download.go
index 4370e72e13..8bd48646d4 100644
--- a/distribution/xfer/download.go
+++ b/distribution/xfer/download.go
@@ -126,6 +126,11 @@ func (ldm *LayerDownloadManager) Download(ctx context.Context, initialRootFS ima
 					topLayer = l
 					missingLayer = false
 					rootFS.Append(diffID)
+					// Register this repository as a source of this layer.
+					withRegistered, hasRegistered := descriptor.(DownloadDescriptorWithRegistered)
+					if hasRegistered {
+						withRegistered.Registered(diffID)
+					}
 					continue
 				}
 			}