/
obliviate_alpha_snapshot_crd.go
121 lines (100 loc) · 3.02 KB
/
obliviate_alpha_snapshot_crd.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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
// Copyright 2020 NetApp, Inc. All Rights Reserved.
package cmd
import (
"strings"
log "github.com/sirupsen/logrus"
"github.com/spf13/cobra"
)
func init() {
obliviateCmd.AddCommand(obliviateAlphaSnapshotCRDCmd)
obliviateAlphaSnapshotCRDCmd.Flags().StringVar(&configPath, "k8s-config-path", kubeConfigPath(),
"Path to KubeConfig file.")
}
var obliviateAlphaSnapshotCRDCmd = &cobra.Command{
Use: "alpha-snapshot-crd",
Short: "Delete kubernetes snapshot CRDs that are v1alpha1",
PersistentPreRun: func(cmd *cobra.Command, args []string) {},
RunE: func(cmd *cobra.Command, args []string) error {
var err error
initLogging()
if OperatingMode == ModeTunnel {
command := []string{"obliviate", "alpha-snapshot-crd"}
TunnelCommand(command)
return nil
} else {
forceObliviate = true // Don't require confirmation for this command
if err = initClients(); err != nil {
return err
}
return deleteAlphaSnapshotCRDs()
}
},
}
func deleteAlphaSnapshotCRDs() error {
crdNames := []string{
"volumesnapshotclasses.snapshot.storage.k8s.io",
"volumesnapshotcontents.snapshot.storage.k8s.io",
"volumesnapshots.snapshot.storage.k8s.io",
}
for _, crdName := range crdNames {
logFields := log.Fields{"CRD": crdName}
// See if CRD exists
exists, err := kubeClient.CheckCRDExists(crdName)
if err != nil {
return err
}
if !exists {
log.WithFields(logFields).Info("CRD not present.")
continue
}
// Get the CRD and check version
crd, err := kubeClient.GetCRD(crdName)
if err != nil {
return err
}
alpha := false
if strings.ToLower(crd.Spec.Version) == "v1alpha1" {
alpha = true
}
for _, version := range crd.Spec.Versions {
if strings.ToLower(version.Name) == "v1alpha1" {
alpha = true
}
}
if alpha == false {
log.WithFields(logFields).Info("CRD is not alpha")
continue
}
log.WithFields(logFields).Debug("Deleting CRD.")
// Try deleting CRD
if err = kubeClient.DeleteCRD(crdName); err != nil {
log.WithFields(logFields).Errorf("Could not delete CRD; %v", err)
return err
}
// Check if CRD still exists (mostly likely pinned by finalizer)
if exists, err = kubeClient.CheckCRDExists(crdName); err != nil {
return err
} else if !exists {
log.WithFields(logFields).Info("CRD deleted.")
continue
}
log.WithFields(logFields).Debug("CRD still present, must remove finalizers.")
// Remove finalizer
if err = kubeClient.RemoveFinalizerFromCRD(crdName); err != nil {
log.WithFields(logFields).Errorf("Could not remove finalizer from CRD; %v", err)
return err
} else {
log.WithFields(logFields).Debug("Removed finalizer from CRD.")
}
// Check if removing the finalizer allowed the CRD to be deleted
if exists, err = kubeClient.CheckCRDExists(crdName); err != nil {
return err
} else if !exists {
log.WithFields(logFields).Info("CRD deleted after removing finalizers.")
continue
} else {
log.WithFields(logFields).Error("CRD still not deleted after removing finalizers.")
}
}
return nil
}