-
Notifications
You must be signed in to change notification settings - Fork 113
/
preupgrade.go
83 lines (71 loc) · 2.39 KB
/
preupgrade.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
/*
Copyright 2019 The OpenEBS 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 v040_041
import (
"context"
apis "github.com/openebs/node-disk-manager/pkg/apis/openebs/v1alpha1"
"github.com/openebs/node-disk-manager/pkg/util"
"sigs.k8s.io/controller-runtime/pkg/client"
)
const (
// oldBDCFinalizer is the old string from which BDC should be updated
oldBDCFinalizer = "blockdeviceclaim.finalizer"
// newBDCFinalizer is the new string to which BDC to be updated
newBDCFinalizer = "openebs.io/bdc-protection"
)
// UpgradeTask is the struct which implements the Task interface
// which can be used to perform the upgrade
type UpgradeTask struct {
from string
to string
client client.Client
err error
}
// NewUpgradeTask creates a new pre-upgrade with given client
// and specified `from` and `to` version
func NewUpgradeTask(from, to string, c client.Client) *UpgradeTask {
return &UpgradeTask{from: from, to: to, client: c}
}
// PreUpgrade runs the pre-upgrade tasks and returns whether it succeeded or not
func (p *UpgradeTask) PreUpgrade() bool {
var err error
bdcList := &apis.BlockDeviceClaimList{}
opts := &client.ListOptions{}
err = p.client.List(context.TODO(), opts, bdcList)
if err != nil {
p.err = err
return false
}
for _, bdc := range bdcList.Items {
err = p.renameFinalizer(&bdc)
if err != nil {
p.err = err
return false
}
}
return true
}
// IsSuccess returns error if the upgrade failed, at any step. Else nil will
// be returned
func (p *UpgradeTask) IsSuccess() error {
return p.err
}
// renameFinalizer renames the finalizer from old to new in BDC
func (p *UpgradeTask) renameFinalizer(claim *apis.BlockDeviceClaim) error {
if util.Contains(claim.Finalizers, oldBDCFinalizer) {
claim.Finalizers = util.RemoveString(claim.Finalizers, oldBDCFinalizer)
claim.Finalizers = append(claim.Finalizers, newBDCFinalizer)
return p.client.Update(context.TODO(), claim)
}
return nil
}