builder/push.go

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

package builder

import (
	"context"
	"fmt"
	"log"
	"os"
	"time"

	"github.com/Azure/acr-builder/util"
	"github.com/google/uuid"
)

const (
	maxPushRetries = 3
)

func (b *Builder) pushWithRetries(ctx context.Context, images []string) error {
	if len(images) == 0 {
		return nil
	}

	for _, img := range images {
		args := []string{
			"docker",
			"run",
			"--name", fmt.Sprintf("acb_docker_push_%s", uuid.New()),
			"--rm",

			// Mount home
			"--volume", util.DockerSocketVolumeMapping,
			"--volume", homeVol + ":" + homeWorkDir,
			"--env", homeEnv,

			dockerCLIImageName,
			"push",
			img,
		}

		attempt := 0
		for attempt < maxPushRetries {
			log.Printf("Pushing image: %s, attempt %d\n", img, attempt+1)
			if err := b.procManager.Run(ctx, args, nil, os.Stdout, os.Stderr, ""); err != nil {
				time.Sleep(util.GetExponentialBackoff(attempt))
				attempt++
			} else {
				log.Printf("Successfully pushed image: %s\n", img)
				break
			}
		}

		if attempt == maxPushRetries {
			return fmt.Errorf("failed to push images successfully")
		}
	}

	return nil
}