Add a reporter for verbs to track number of storage streams are needed

This commit is contained in:
Joseph Schorr 2018-03-23 14:39:38 -04:00
parent 110366f656
commit dd470bdc9d
7 changed files with 43 additions and 14 deletions

View file

@ -8,8 +8,8 @@ from util.registry.tarlayerformat import TarLayerFormat
class StreamLayerMerger(TarLayerFormat):
""" Class which creates a generator of the combined TAR data for a set of Docker layers. """
def __init__(self, get_tar_stream_iterator, path_prefix=None):
super(StreamLayerMerger, self).__init__(get_tar_stream_iterator, path_prefix)
def __init__(self, get_tar_stream_iterator, path_prefix=None, reporter=None):
super(StreamLayerMerger, self).__init__(get_tar_stream_iterator, path_prefix, reporter=reporter)
self.path_trie = marisa_trie.Trie()
self.path_encountered = set()