Skip to content

Commit 13caf24

Browse files
committed
pkg/risk/adminack: Refactor to take a version callback
Returning just the version, instead of the whole Release. Slightly more work setup-side to wrap the optr.currentVersion call, but it makes the required information more transparent.
1 parent b972175 commit 13caf24

3 files changed

Lines changed: 6 additions & 6 deletions

File tree

pkg/cvo/cvo.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -325,7 +325,7 @@ func New(
325325
risks = append(risks, source)
326326
}
327327
risks = append(risks, deletionrisk.New("ResourceDeletionInProgress", optr.currentVersion))
328-
if source, err := adminack.New("AdminAck", optr.currentVersion, cmConfigManagedInformer, cmConfigInformer, riskSourceCallback); err != nil {
328+
if source, err := adminack.New("AdminAck", func() string { return optr.currentVersion().Version }, cmConfigManagedInformer, cmConfigInformer, riskSourceCallback); err != nil {
329329
return optr, err
330330
} else {
331331
risks = append(risks, source)

pkg/risk/adminack/adminack.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,14 +35,14 @@ var adminAckGateRegexp = regexp.MustCompile(adminAckGateFmt)
3535

3636
type adminAck struct {
3737
name string
38-
currentVersion func() configv1.Release
38+
currentVersion func() string
3939
adminGatesLister listerscorev1.ConfigMapNamespaceLister
4040
adminAcksLister listerscorev1.ConfigMapNamespaceLister
4141
lastSeen []configv1.ConditionalUpdateRisk
4242
}
4343

4444
// New returns a new update-risk source, tracking administrator acknowledgements.
45-
func New(name string, currentVersion func() configv1.Release, adminGatesInformer, adminAcksInformer informerscorev1.ConfigMapInformer, changeCallback func()) (risk.Source, error) {
45+
func New(name string, currentVersion func() string, adminGatesInformer, adminAcksInformer informerscorev1.ConfigMapInformer, changeCallback func()) (risk.Source, error) {
4646
adminGatesLister := adminGatesInformer.Lister().ConfigMaps(internal.ConfigManagedNamespace)
4747
adminAcksLister := adminAcksInformer.Lister().ConfigMaps(internal.ConfigNamespace)
4848
source := &adminAck{name: name, currentVersion: currentVersion, adminGatesLister: adminGatesLister, adminAcksLister: adminAcksLister}
@@ -92,8 +92,8 @@ func (a *adminAck) Risks(_ context.Context, versions []string) ([]configv1.Condi
9292
}
9393

9494
func (a *adminAck) risks() ([]configv1.ConditionalUpdateRisk, semver.Version, error) {
95-
currentRelease := a.currentVersion()
96-
version, err := semver.Parse(currentRelease.Version)
95+
currentVersion := a.currentVersion()
96+
version, err := semver.Parse(currentVersion)
9797
if err != nil {
9898
return []configv1.ConditionalUpdateRisk{{
9999
Name: fmt.Sprintf("%sUnknown", a.name),

pkg/risk/adminack/adminack_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,7 @@ func Test_New(t *testing.T) {
285285
for _, tt := range tests {
286286
{
287287
t.Run(tt.name, func(t *testing.T) {
288-
source, err := New("AdminAck", func() configv1.Release { return configv1.Release{Version: "4.21.0"} }, cmInformer, cmInformer, nil)
288+
source, err := New("AdminAck", func() string { return "4.21.0" }, cmInformer, cmInformer, nil)
289289
if err != nil {
290290
t.Fatal(err)
291291
}

0 commit comments

Comments
 (0)