forked from kyma-project/kyma
/
status-manager.go
102 lines (83 loc) · 3.13 KB
/
status-manager.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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package statusmanager
import (
installationv1alpha1 "github.com/kyma-project/kyma/components/installer/pkg/apis/installer/v1alpha1"
installationClientset "github.com/kyma-project/kyma/components/installer/pkg/client/clientset/versioned"
listers "github.com/kyma-project/kyma/components/installer/pkg/client/listers/installer/v1alpha1"
"github.com/kyma-project/kyma/components/installer/pkg/consts"
"k8s.io/client-go/util/retry"
)
// StatusManager .
type StatusManager interface {
InProgress(description string) error
InstallDone(url, kymaVersion string) error
UpdateDone(url, kymaVersion string) error
UninstallDone() error
Error(description string) error
}
type statusManager struct {
lister listers.InstallationLister
client installationClientset.Interface
}
// NewKymaStatusManager .
func NewKymaStatusManager(installationClient installationClientset.Interface, installationLister listers.InstallationLister) StatusManager {
sm := &statusManager{
lister: installationLister,
client: installationClient,
}
return sm
}
//InProgress .
func (sm *statusManager) InProgress(description string) error {
instObj, getErr := sm.lister.Installations(consts.InstNamespace).Get(consts.InstResource)
if getErr != nil {
return getErr
}
installationCopy := instObj.DeepCopy()
instStatus := getStatus(installationv1alpha1.StateInProgress, description, installationCopy.Status.URL, installationCopy.Status.KymaVersion)
return sm.update(instStatus)
}
//InstallDone .
func (sm *statusManager) InstallDone(url, kymaVersion string) error {
instStatus := getStatus(installationv1alpha1.StateInstalled, "Kyma installed", url, kymaVersion)
return sm.update(instStatus)
}
//UpdateDone .
func (sm *statusManager) UpdateDone(url, kymaVersion string) error {
instStatus := getStatus(installationv1alpha1.StateUpdated, "Kyma updated", url, kymaVersion)
return sm.update(instStatus)
}
//UninstallDone .
func (sm *statusManager) UninstallDone() error {
instStatus := getStatus(installationv1alpha1.StateUninstalled, "Kyma uninstalled", "", "")
return sm.update(instStatus)
}
//Error .
func (sm *statusManager) Error(description string) error {
instStatus := getStatus(installationv1alpha1.StateError, description, "", "")
return sm.update(instStatus)
}
func (sm *statusManager) update(installationStatus *installationv1alpha1.InstallationStatus) error {
retryErr := retry.RetryOnConflict(retry.DefaultRetry, func() error {
instObj, getErr := sm.lister.Installations(consts.InstNamespace).Get(consts.InstResource)
if getErr != nil {
return getErr
}
installationCopy := instObj.DeepCopy()
installationCopy.Status = *installationStatus
installationCopy.Status.Conditions = instObj.Status.Conditions
_, updateErr := sm.client.InstallerV1alpha1().Installations(consts.InstNamespace).Update(installationCopy)
return updateErr
})
if retryErr != nil {
return retryErr
}
return nil
}
func getStatus(state installationv1alpha1.StateEnum, description, url, version string) *installationv1alpha1.InstallationStatus {
return &installationv1alpha1.InstallationStatus{
State: state,
Description: description,
URL: url,
KymaVersion: version,
}
}