-
Notifications
You must be signed in to change notification settings - Fork 76
/
helper.go
62 lines (51 loc) · 2.18 KB
/
helper.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
// SPDX-FileCopyrightText: 2024 SAP SE or an SAP affiliate company and Gardener contributors
//
// SPDX-License-Identifier: Apache-2.0
package worker
import (
"context"
"fmt"
kutil "github.com/gardener/gardener/pkg/utils/kubernetes"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/runtime"
"sigs.k8s.io/controller-runtime/pkg/client"
azureapi "github.com/gardener/gardener-extension-provider-azure/pkg/apis/azure"
"github.com/gardener/gardener-extension-provider-azure/pkg/apis/azure/v1alpha1"
)
func (w *workerDelegate) decodeAzureInfrastructureStatus() (*azureapi.InfrastructureStatus, error) {
infrastructureStatus := &azureapi.InfrastructureStatus{}
if _, _, err := w.lenientDecoder.Decode(w.worker.Spec.InfrastructureProviderStatus.Raw, nil, infrastructureStatus); err != nil {
return nil, err
}
return infrastructureStatus, nil
}
func (w *workerDelegate) decodeWorkerProviderStatus() (*azureapi.WorkerStatus, error) {
workerStatus := &azureapi.WorkerStatus{}
if w.worker.Status.ProviderStatus == nil {
return workerStatus, nil
}
if _, _, err := w.decoder.Decode(w.worker.Status.ProviderStatus.Raw, nil, workerStatus); err != nil {
return nil, fmt.Errorf("could not decode the worker provider status of worker '%s': %w", kutil.ObjectName(w.worker), err)
}
return workerStatus, nil
}
func (w *workerDelegate) updateWorkerProviderStatus(ctx context.Context, workerStatus *azureapi.WorkerStatus) error {
workerStatusV1alpha1 := &v1alpha1.WorkerStatus{
TypeMeta: metav1.TypeMeta{
APIVersion: v1alpha1.SchemeGroupVersion.String(),
Kind: "WorkerStatus",
},
}
if err := w.scheme.Convert(workerStatus, workerStatusV1alpha1, nil); err != nil {
return err
}
patch := client.MergeFrom(w.worker.DeepCopy())
w.worker.Status.ProviderStatus = &runtime.RawExtension{Object: workerStatusV1alpha1}
return w.client.Status().Patch(ctx, w.worker, patch)
}
func (w *workerDelegate) updateWorkerProviderStatusWithError(ctx context.Context, workerStatus *azureapi.WorkerStatus, err error) error {
if statusUpdateErr := w.updateWorkerProviderStatus(ctx, workerStatus); statusUpdateErr != nil {
return fmt.Errorf("%s: %w", err.Error(), statusUpdateErr)
}
return err
}