[mirror] Feature: Push parallelism support

Signed-off-by: Maxim Vasilenko <[email protected]>
deckhouse · Sep 23, 2024 · 908b82d · 908b82d
1 parent e35bc43
commit 908b82d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/internal/mirror/cmd/push/push.go b/internal/mirror/cmd/push/push.go
@@ -149,8 +149,8 @@ func buildPushContext() *contexts.PushContext {
 		},
 
 		Parallelism: contexts.ParallelismConfig{
-			Blobs:  4,
-			Images: 10,
+			Blobs:  2,
+			Images: 5,
 		},
 	}
 	return mirrorCtx

diff --git a/pkg/libmirror/layouts/push.go b/pkg/libmirror/layouts/push.go
@@ -69,7 +69,7 @@ func PushLayoutToRepo(
 	batchesCount, imagesCount := 1, 1
 
 	for _, manifestSet := range batches {
-		if len(manifestSet) == 1 {
+		if parallelismConfig.Images == 1 {
 			tag := manifestSet[0].Annotations["io.deckhouse.image.short_tag"]
 			imageRef := registryRepo + ":" + tag
 			logger.InfoF("[%d / %d] Pushing image %s", imagesCount, len(indexManifest.Manifests), imageRef)
@@ -81,9 +81,7 @@ func PushLayoutToRepo(
 		err = logger.Process(fmt.Sprintf("Pushing batch %d / %d", batchesCount, len(batches)), func() error {
 			logger.InfoLn("Images in batch:")
 			for _, manifest := range manifestSet {
-				tag := manifest.Annotations["io.deckhouse.image.short_tag"]
-				imageRef := registryRepo + ":" + tag
-				logger.InfoF("- %s", imageRef)
+				logger.InfoF("- %s", registryRepo+":"+manifest.Annotations["io.deckhouse.image.short_tag"])
 			}
 
 			parallel.ForEach(manifestSet, pushImage(logger, registryRepo, index, imagesCount, refOpts, remoteOpts))