mirror of https://github.com/kubernetes/kops.git
122 lines
3.1 KiB
Go
122 lines
3.1 KiB
Go
/*
|
|
Copyright 2021 The Kubernetes 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 assets
|
|
|
|
import (
|
|
"fmt"
|
|
"sort"
|
|
|
|
"k8s.io/klog/v2"
|
|
"k8s.io/kops/pkg/apis/kops"
|
|
"k8s.io/kops/util/pkg/vfs"
|
|
)
|
|
|
|
type assetTask interface {
|
|
Run() error
|
|
}
|
|
|
|
func Copy(imageAssets []*ImageAsset, fileAssets []*FileAsset, vfsContext *vfs.VFSContext, cluster *kops.Cluster) error {
|
|
tasks := map[string]assetTask{}
|
|
|
|
for _, imageAsset := range imageAssets {
|
|
if imageAsset.DownloadLocation != imageAsset.CanonicalLocation {
|
|
copyImageTask := &CopyImage{
|
|
Name: imageAsset.DownloadLocation,
|
|
SourceImage: imageAsset.CanonicalLocation,
|
|
TargetImage: imageAsset.DownloadLocation,
|
|
}
|
|
|
|
if existing, ok := tasks[copyImageTask.Name]; ok {
|
|
if existing.(*CopyImage).SourceImage != copyImageTask.SourceImage {
|
|
return fmt.Errorf("different sources for same image target %s: %s vs %s", copyImageTask.Name, copyImageTask.SourceImage, existing.(*CopyImage).SourceImage)
|
|
}
|
|
}
|
|
|
|
tasks[copyImageTask.Name] = copyImageTask
|
|
}
|
|
}
|
|
|
|
for _, fileAsset := range fileAssets {
|
|
if fileAsset.DownloadURL.String() != fileAsset.CanonicalURL.String() {
|
|
copyFileTask := &CopyFile{
|
|
Name: fileAsset.CanonicalURL.String(),
|
|
TargetFile: fileAsset.DownloadURL.String(),
|
|
SourceFile: fileAsset.CanonicalURL.String(),
|
|
SHA: fileAsset.SHAValue.Hex(),
|
|
VFSContext: vfsContext,
|
|
Cluster: cluster,
|
|
}
|
|
|
|
if existing, ok := tasks[copyFileTask.Name]; ok {
|
|
e, ok := existing.(*CopyFile)
|
|
if !ok {
|
|
return fmt.Errorf("different types for copy target %s", copyFileTask.Name)
|
|
}
|
|
if e.TargetFile != copyFileTask.TargetFile {
|
|
return fmt.Errorf("different targets for same file %s: %s vs %s", copyFileTask.Name, copyFileTask.TargetFile, e.TargetFile)
|
|
}
|
|
if e.SHA != copyFileTask.SHA {
|
|
return fmt.Errorf("different sha for same file %s: %s vs %s", copyFileTask.Name, copyFileTask.SHA, e.SHA)
|
|
}
|
|
}
|
|
|
|
tasks[copyFileTask.Name] = copyFileTask
|
|
}
|
|
}
|
|
|
|
ch := make(chan error, 5)
|
|
for i := 0; i < cap(ch); i++ {
|
|
ch <- nil
|
|
}
|
|
|
|
gotError := false
|
|
names := make([]string, 0, len(tasks))
|
|
for name := range tasks {
|
|
names = append(names, name)
|
|
}
|
|
sort.Strings(names)
|
|
for _, name := range names {
|
|
task := tasks[name]
|
|
err := <-ch
|
|
if err != nil {
|
|
klog.Warning(err)
|
|
gotError = true
|
|
}
|
|
go func(n string, t assetTask) {
|
|
err := t.Run()
|
|
if err != nil {
|
|
err = fmt.Errorf("%s: %v", n, err)
|
|
}
|
|
ch <- err
|
|
}(name, task)
|
|
}
|
|
|
|
for i := 0; i < cap(ch); i++ {
|
|
err := <-ch
|
|
if err != nil {
|
|
klog.Warning(err)
|
|
gotError = true
|
|
}
|
|
}
|
|
|
|
close(ch)
|
|
if gotError {
|
|
return fmt.Errorf("not all assets copied successfully")
|
|
}
|
|
return nil
|
|
}
|