mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-21 14:41:12 +02:00
0081ac5fac
This extracts Docker Talos cluster provisioner as common code which might be shared between `osctl cluster` and integration-test. There should be almost no functional changes. As proof of concept, abstract cluster readiness checks were implemented based on provisioned cluster state. It implements same checks as `basic-integration.sh` in pure Go via Talos/K8s clients. `conditions` package was promoted from machined-internal to `internal/pkg` as it is used to run the checks. Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package docker
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
|
|
"github.com/docker/distribution/reference"
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/filters"
|
|
|
|
"github.com/talos-systems/talos/internal/pkg/provision"
|
|
)
|
|
|
|
func (p *provisioner) ensureImageExists(ctx context.Context, image string, options *provision.Options) error {
|
|
// In order to pull an image, the reference must be in canonical
|
|
// format (e.g. domain/repo/image:tag).
|
|
ref, err := reference.ParseNormalizedNamed(image)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
image = ref.String()
|
|
|
|
// To filter the images, we need a familiar name and a tag
|
|
// (e.g. domain/repo/image:tag => repo/image:tag).
|
|
familiarName := reference.FamiliarName(ref)
|
|
tag := ""
|
|
|
|
if tagged, isTagged := ref.(reference.Tagged); isTagged {
|
|
tag = tagged.Tag()
|
|
}
|
|
|
|
filters := filters.NewArgs()
|
|
filters.Add("reference", familiarName+":"+tag)
|
|
|
|
images, err := p.client.ImageList(ctx, types.ImageListOptions{Filters: filters})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(images) == 0 {
|
|
fmt.Fprintln(options.LogWriter, "downloading", image)
|
|
|
|
var reader io.ReadCloser
|
|
|
|
if reader, err = p.client.ImagePull(ctx, image, types.ImagePullOptions{}); err != nil {
|
|
return err
|
|
}
|
|
|
|
// nolint: errcheck
|
|
defer reader.Close()
|
|
|
|
if _, err = io.Copy(ioutil.Discard, reader); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|