mirror of
https://github.com/docker/compose.git
synced 2025-05-25 08:50:12 +02:00
371 lines
9.0 KiB
Go
371 lines
9.0 KiB
Go
/*
|
|
Copyright 2020 Docker Compose CLI authors
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package compose
|
|
|
|
import (
|
|
"context"
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"strings"
|
|
|
|
"github.com/compose-spec/compose-go/types"
|
|
"github.com/distribution/distribution/v3/reference"
|
|
"github.com/docker/buildx/driver"
|
|
moby "github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/pkg/jsonmessage"
|
|
"github.com/docker/docker/registry"
|
|
"github.com/hashicorp/go-multierror"
|
|
"golang.org/x/sync/errgroup"
|
|
|
|
"github.com/docker/compose/v2/pkg/api"
|
|
"github.com/docker/compose/v2/pkg/progress"
|
|
)
|
|
|
|
func (s *composeService) Pull(ctx context.Context, project *types.Project, options api.PullOptions) error {
|
|
if options.Quiet {
|
|
return s.pull(ctx, project, options)
|
|
}
|
|
return progress.Run(ctx, func(ctx context.Context) error {
|
|
return s.pull(ctx, project, options)
|
|
})
|
|
}
|
|
|
|
func (s *composeService) pull(ctx context.Context, project *types.Project, opts api.PullOptions) error { //nolint:gocyclo
|
|
info, err := s.apiClient().Info(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if info.IndexServerAddress == "" {
|
|
info.IndexServerAddress = registry.IndexServer
|
|
}
|
|
|
|
images, err := s.getLocalImagesDigests(ctx, project)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
w := progress.ContextWriter(ctx)
|
|
eg, ctx := errgroup.WithContext(ctx)
|
|
eg.SetLimit(s.maxConcurrency)
|
|
|
|
var (
|
|
mustBuild []string
|
|
pullErrors = make([]error, len(project.Services))
|
|
imagesBeingPulled = map[string]string{}
|
|
)
|
|
|
|
for i, service := range project.Services {
|
|
i, service := i, service
|
|
if service.Image == "" {
|
|
w.Event(progress.Event{
|
|
ID: service.Name,
|
|
Status: progress.Done,
|
|
Text: "Skipped - No image to be pulled",
|
|
})
|
|
continue
|
|
}
|
|
|
|
switch service.PullPolicy {
|
|
case types.PullPolicyNever, types.PullPolicyBuild:
|
|
w.Event(progress.Event{
|
|
ID: service.Name,
|
|
Status: progress.Done,
|
|
Text: "Skipped",
|
|
})
|
|
continue
|
|
case types.PullPolicyMissing, types.PullPolicyIfNotPresent:
|
|
if imageAlreadyPresent(service.Image, images) {
|
|
w.Event(progress.Event{
|
|
ID: service.Name,
|
|
Status: progress.Done,
|
|
Text: "Skipped - Image is already present locally",
|
|
})
|
|
continue
|
|
}
|
|
}
|
|
|
|
if service.Build != nil && opts.IgnoreBuildable {
|
|
w.Event(progress.Event{
|
|
ID: service.Name,
|
|
Status: progress.Done,
|
|
Text: "Skipped - Image can be built",
|
|
})
|
|
continue
|
|
}
|
|
|
|
if s, ok := imagesBeingPulled[service.Image]; ok {
|
|
w.Event(progress.Event{
|
|
ID: service.Name,
|
|
Status: progress.Done,
|
|
Text: fmt.Sprintf("Skipped - Image is already being pulled by %v", s),
|
|
})
|
|
continue
|
|
}
|
|
|
|
imagesBeingPulled[service.Image] = service.Name
|
|
|
|
eg.Go(func() error {
|
|
_, err := s.pullServiceImage(ctx, service, info, s.configFile(), w, false, project.Environment["DOCKER_DEFAULT_PLATFORM"])
|
|
if err != nil {
|
|
pullErrors[i] = err
|
|
if service.Build != nil {
|
|
mustBuild = append(mustBuild, service.Name)
|
|
}
|
|
if !opts.IgnoreFailures && service.Build == nil {
|
|
// fail fast if image can't be pulled nor built
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
|
|
err = eg.Wait()
|
|
|
|
if len(mustBuild) > 0 {
|
|
w.TailMsgf("WARNING: Some service image(s) must be built from source by running:\n docker compose build %s", strings.Join(mustBuild, " "))
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if opts.IgnoreFailures {
|
|
return nil
|
|
}
|
|
return multierror.Append(nil, pullErrors...).ErrorOrNil()
|
|
}
|
|
|
|
func imageAlreadyPresent(serviceImage string, localImages map[string]string) bool {
|
|
normalizedImage, err := reference.ParseDockerRef(serviceImage)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
tagged, ok := normalizedImage.(reference.NamedTagged)
|
|
if !ok {
|
|
return false
|
|
}
|
|
_, ok = localImages[serviceImage]
|
|
return ok && tagged.Tag() != "latest"
|
|
}
|
|
|
|
func (s *composeService) pullServiceImage(ctx context.Context, service types.ServiceConfig, info moby.Info,
|
|
configFile driver.Auth, w progress.Writer, quietPull bool, defaultPlatform string) (string, error) {
|
|
w.Event(progress.Event{
|
|
ID: service.Name,
|
|
Status: progress.Working,
|
|
Text: "Pulling",
|
|
})
|
|
ref, err := reference.ParseNormalizedNamed(service.Image)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
encodedAuth, err := encodedAuth(ref, info, configFile)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
platform := service.Platform
|
|
if platform == "" {
|
|
platform = defaultPlatform
|
|
}
|
|
|
|
stream, err := s.apiClient().ImagePull(ctx, service.Image, moby.ImagePullOptions{
|
|
RegistryAuth: encodedAuth,
|
|
Platform: platform,
|
|
})
|
|
|
|
// check if has error and the service has a build section
|
|
// then the status should be warning instead of error
|
|
if err != nil && service.Build != nil {
|
|
w.Event(progress.Event{
|
|
ID: service.Name,
|
|
Status: progress.Warning,
|
|
Text: "Warning",
|
|
})
|
|
return "", WrapCategorisedComposeError(err, PullFailure)
|
|
}
|
|
|
|
if err != nil {
|
|
w.Event(progress.Event{
|
|
ID: service.Name,
|
|
Status: progress.Error,
|
|
Text: "Error",
|
|
})
|
|
return "", WrapCategorisedComposeError(err, PullFailure)
|
|
}
|
|
|
|
dec := json.NewDecoder(stream)
|
|
for {
|
|
var jm jsonmessage.JSONMessage
|
|
if err := dec.Decode(&jm); err != nil {
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
return "", WrapCategorisedComposeError(err, PullFailure)
|
|
}
|
|
if jm.Error != nil {
|
|
return "", WrapCategorisedComposeError(errors.New(jm.Error.Message), PullFailure)
|
|
}
|
|
if !quietPull {
|
|
toPullProgressEvent(service.Name, jm, w)
|
|
}
|
|
}
|
|
w.Event(progress.Event{
|
|
ID: service.Name,
|
|
Status: progress.Done,
|
|
Text: "Pulled",
|
|
})
|
|
|
|
inspected, _, err := s.dockerCli.Client().ImageInspectWithRaw(ctx, service.Image)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return inspected.ID, nil
|
|
}
|
|
|
|
func encodedAuth(ref reference.Named, info moby.Info, configFile driver.Auth) (string, error) {
|
|
repoInfo, err := registry.ParseRepositoryInfo(ref)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
key := repoInfo.Index.Name
|
|
if repoInfo.Index.Official {
|
|
key = info.IndexServerAddress
|
|
}
|
|
|
|
authConfig, err := configFile.GetAuthConfig(key)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
buf, err := json.Marshal(authConfig)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return base64.URLEncoding.EncodeToString(buf), nil
|
|
}
|
|
|
|
func (s *composeService) pullRequiredImages(ctx context.Context, project *types.Project, images map[string]string, quietPull bool) error {
|
|
info, err := s.apiClient().Info(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if info.IndexServerAddress == "" {
|
|
info.IndexServerAddress = registry.IndexServer
|
|
}
|
|
|
|
var needPull []types.ServiceConfig
|
|
for _, service := range project.Services {
|
|
if service.Image == "" {
|
|
continue
|
|
}
|
|
switch service.PullPolicy {
|
|
case "", types.PullPolicyMissing, types.PullPolicyIfNotPresent:
|
|
if _, ok := images[service.Image]; ok {
|
|
continue
|
|
}
|
|
case types.PullPolicyNever, types.PullPolicyBuild:
|
|
continue
|
|
case types.PullPolicyAlways:
|
|
// force pull
|
|
}
|
|
needPull = append(needPull, service)
|
|
}
|
|
if len(needPull) == 0 {
|
|
return nil
|
|
}
|
|
|
|
return progress.Run(ctx, func(ctx context.Context) error {
|
|
w := progress.ContextWriter(ctx)
|
|
eg, ctx := errgroup.WithContext(ctx)
|
|
eg.SetLimit(s.maxConcurrency)
|
|
pulledImages := make([]string, len(needPull))
|
|
for i, service := range needPull {
|
|
i, service := i, service
|
|
eg.Go(func() error {
|
|
id, err := s.pullServiceImage(ctx, service, info, s.configFile(), w, quietPull, project.Environment["DOCKER_DEFAULT_PLATFORM"])
|
|
pulledImages[i] = id
|
|
if err != nil && isServiceImageToBuild(service, project.Services) {
|
|
// image can be built, so we can ignore pull failure
|
|
return nil
|
|
}
|
|
return err
|
|
})
|
|
}
|
|
err := eg.Wait()
|
|
for i, service := range needPull {
|
|
if pulledImages[i] != "" {
|
|
images[service.Image] = pulledImages[i]
|
|
}
|
|
}
|
|
return err
|
|
})
|
|
}
|
|
|
|
func isServiceImageToBuild(service types.ServiceConfig, services []types.ServiceConfig) bool {
|
|
if service.Build != nil {
|
|
return true
|
|
}
|
|
|
|
for _, depService := range services {
|
|
if depService.Image == service.Image && depService.Build != nil {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func toPullProgressEvent(parent string, jm jsonmessage.JSONMessage, w progress.Writer) {
|
|
if jm.ID == "" || jm.Progress == nil {
|
|
return
|
|
}
|
|
|
|
var (
|
|
text string
|
|
status = progress.Working
|
|
)
|
|
|
|
text = jm.Progress.String()
|
|
|
|
if jm.Status == "Pull complete" ||
|
|
jm.Status == "Already exists" ||
|
|
strings.Contains(jm.Status, "Image is up to date") ||
|
|
strings.Contains(jm.Status, "Downloaded newer image") {
|
|
status = progress.Done
|
|
}
|
|
|
|
if jm.Error != nil {
|
|
status = progress.Error
|
|
text = jm.Error.Message
|
|
}
|
|
|
|
w.Event(progress.Event{
|
|
ID: jm.ID,
|
|
ParentID: parent,
|
|
Text: jm.Status,
|
|
Status: status,
|
|
StatusText: text,
|
|
})
|
|
}
|