/
machine_images.go
69 lines (55 loc) · 2.89 KB
/
machine_images.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
// SPDX-FileCopyrightText: 2024 SAP SE or an SAP affiliate company and Gardener contributors
//
// SPDX-License-Identifier: Apache-2.0
package worker
import (
"context"
"fmt"
"github.com/gardener/gardener/extensions/pkg/controller/worker"
kutil "github.com/gardener/gardener/pkg/utils/kubernetes"
api "github.com/gardener/gardener-extension-provider-azure/pkg/apis/azure"
"github.com/gardener/gardener-extension-provider-azure/pkg/apis/azure/helper"
)
// UpdateMachineImagesStatus stores the used machine images for the `Worker` resource in the worker-provider-status.
func (w *workerDelegate) UpdateMachineImagesStatus(ctx context.Context) error {
if w.machineImages == nil {
if err := w.generateMachineConfig(ctx); err != nil {
return fmt.Errorf("unable to generate the machine config: %w", err)
}
}
// Decode the current worker provider status.
workerStatus, err := w.decodeWorkerProviderStatus()
if err != nil {
return fmt.Errorf("unable to decode the worker provider status: %w", err)
}
workerStatus.MachineImages = w.machineImages
if err := w.updateWorkerProviderStatus(ctx, workerStatus); err != nil {
return fmt.Errorf("unable to update worker provider status: %w", err)
}
return nil
}
func (w *workerDelegate) findMachineImage(name, version string, architecture *string) (urn, id, communityGalleryImageID *string, sharedGalleryImageID *string, acceleratedNetworking *bool, err error) {
machineImage, err := helper.FindImageFromCloudProfile(w.cloudProfileConfig, name, version, architecture)
if err == nil {
return machineImage.URN, machineImage.ID, machineImage.CommunityGalleryImageID, machineImage.SharedGalleryImageID, machineImage.AcceleratedNetworking, nil
}
// Try to look up machine image in worker provider status as it was not found in componentconfig.
if providerStatus := w.worker.Status.ProviderStatus; providerStatus != nil {
workerStatus := &api.WorkerStatus{}
if _, _, err := w.decoder.Decode(providerStatus.Raw, nil, workerStatus); err != nil {
return nil, nil, nil, nil, nil, fmt.Errorf("could not decode worker status of worker '%s': %w", kutil.ObjectName(w.worker), err)
}
machineImage, err := helper.FindMachineImage(workerStatus.MachineImages, name, version, architecture)
if err != nil {
return nil, nil, nil, nil, nil, worker.ErrorMachineImageNotFound(name, version, *architecture)
}
return machineImage.URN, machineImage.ID, machineImage.CommunityGalleryImageID, machineImage.SharedGalleryImageID, machineImage.AcceleratedNetworking, nil
}
return nil, nil, nil, nil, nil, worker.ErrorMachineImageNotFound(name, version, *architecture)
}
func appendMachineImage(machineImages []api.MachineImage, machineImage api.MachineImage) []api.MachineImage {
if _, err := helper.FindMachineImage(machineImages, machineImage.Name, machineImage.Version, machineImage.Architecture); err != nil {
return append(machineImages, machineImage)
}
return machineImages
}