-
Notifications
You must be signed in to change notification settings - Fork 99
/
packageinstall_version_handler.go
119 lines (99 loc) · 4.17 KB
/
packageinstall_version_handler.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
// Copyright 2021 VMware, Inc.
// SPDX-License-Identifier: Apache-2.0
package packageinstall
import (
"context"
"github.com/go-logr/logr"
pkgingv1alpha1 "github.com/vmware-tanzu/carvel-kapp-controller/pkg/apis/packaging/v1alpha1"
datapkgingv1alpha1 "github.com/vmware-tanzu/carvel-kapp-controller/pkg/apiserver/apis/datapackaging/v1alpha1"
kcclient "github.com/vmware-tanzu/carvel-kapp-controller/pkg/client/clientset/versioned"
"github.com/vmware-tanzu/carvel-vendir/pkg/vendir/versions"
verv1alpha1 "github.com/vmware-tanzu/carvel-vendir/pkg/vendir/versions/v1alpha1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/types"
"k8s.io/client-go/util/workqueue"
"sigs.k8s.io/controller-runtime/pkg/event"
"sigs.k8s.io/controller-runtime/pkg/handler"
"sigs.k8s.io/controller-runtime/pkg/reconcile"
)
// TODO For this PoC, we are simply going to add all packages to
// the workqueue, but in the future, we will only enqueue the
// packages that are eligible for upgrade based on the new
// packages
type PackageInstallVersionHandler struct {
client kcclient.Interface
globalNS string
log logr.Logger
}
var _ handler.EventHandler = &PackageInstallVersionHandler{}
func NewPackageInstallVersionHandler(c kcclient.Interface, globalNS string, log logr.Logger) *PackageInstallVersionHandler {
return &PackageInstallVersionHandler{c, globalNS, log}
}
// Create is called in response to an create event
func (ipvh *PackageInstallVersionHandler) Create(_ context.Context, evt event.CreateEvent, q workqueue.RateLimitingInterface) {
ipvh.log.Info("enqueueing PackageInstallList")
err := ipvh.enqueueEligiblePackageInstalls(q, evt.Object)
if err != nil {
ipvh.log.Error(err, "enqueueing all PackageInstalls")
}
}
// Update is called in response to an update event
func (ipvh *PackageInstallVersionHandler) Update(_ context.Context, evt event.UpdateEvent, q workqueue.RateLimitingInterface) {
ipvh.log.Info("enqueueing PackageInstallList")
err := ipvh.enqueueEligiblePackageInstalls(q, evt.ObjectNew)
if err != nil {
ipvh.log.Error(err, "enqueueing all PackageInstalls")
}
}
// Delete is called in response to a delete event
func (ipvh *PackageInstallVersionHandler) Delete(_ context.Context, evt event.DeleteEvent, q workqueue.RateLimitingInterface) {
ipvh.log.Info("enqueueing PackageInstallList")
err := ipvh.enqueueEligiblePackageInstalls(q, evt.Object)
if err != nil {
ipvh.log.Error(err, "enqueueing all PackageInstalls")
}
}
// Generic is called in response to an event of an unknown type or a synthetic event triggered as a cron or
// external trigger request - e.g. reconcile Autoscaling, or a Webhook.
func (ipvh *PackageInstallVersionHandler) Generic(_ context.Context, evt event.GenericEvent, q workqueue.RateLimitingInterface) {
ipvh.log.Info("enqueueing installedPkgList")
err := ipvh.enqueueEligiblePackageInstalls(q, evt.Object)
if err != nil {
ipvh.log.Error(err, "enqueueing all PackageInstalls")
}
}
func (ipvh *PackageInstallVersionHandler) enqueueEligiblePackageInstalls(q workqueue.RateLimitingInterface, obj runtime.Object) error {
pv := obj.(*datapkgingv1alpha1.Package)
namespace := ""
if pv.Namespace != ipvh.globalNS {
namespace = pv.Namespace
}
installedPkgList, err := ipvh.client.PackagingV1alpha1().PackageInstalls(namespace).List(context.Background(), metav1.ListOptions{})
if err != nil {
return err
}
for _, ip := range installedPkgList.Items {
if ip.Spec.PackageRef == nil {
continue
}
if ip.Spec.PackageRef.RefName == pv.Spec.RefName && ipvh.isEligibleForVersionUpgrade(pv.Spec.Version, ip) {
q.Add(reconcile.Request{NamespacedName: types.NamespacedName{
Name: ip.Name,
Namespace: ip.Namespace,
}})
}
}
return nil
}
func (ipvh *PackageInstallVersionHandler) isEligibleForVersionUpgrade(version string, installedPkg pkgingv1alpha1.PackageInstall) bool {
if installedPkg.Spec.PackageRef == nil {
return false
}
semverConfig := installedPkg.Spec.PackageRef.VersionSelection
selectedVersion, err := versions.HighestConstrainedVersion([]string{version}, verv1alpha1.VersionSelection{Semver: semverConfig})
if selectedVersion == "" || err != nil {
return false
}
return true
}