/
cleanup.go
61 lines (52 loc) · 1.67 KB
/
cleanup.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
package kube
import (
"context"
"github.com/virtengine/virtengine/manifest"
mtypes "github.com/virtengine/virtengine/x/market/types"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/labels"
"k8s.io/apimachinery/pkg/selection"
"k8s.io/client-go/kubernetes"
)
func cleanupStaleResources(ctx context.Context, kc kubernetes.Interface, lid mtypes.LeaseID, group *manifest.Group) error {
ns := lidNS(lid)
// build label selector for objects not in current manifest group
svcnames := make([]string, 0, len(group.Services))
for _, svc := range group.Services {
svcnames = append(svcnames, svc.Name)
}
req1, err := labels.NewRequirement(virtengineManifestServiceLabelName, selection.NotIn, svcnames)
if err != nil {
return err
}
req2, err := labels.NewRequirement(virtengineManagedLabelName, selection.Equals, []string{"true"})
if err != nil {
return err
}
selector := labels.NewSelector().Add(*req1).Add(*req2).String()
// delete stale deployments
if err := kc.AppsV1().Deployments(ns).DeleteCollection(ctx, metav1.DeleteOptions{}, metav1.ListOptions{
LabelSelector: selector,
}); err != nil {
return err
}
// delete stale ingresses
if err := kc.NetworkingV1().Ingresses(ns).DeleteCollection(ctx, metav1.DeleteOptions{}, metav1.ListOptions{
LabelSelector: selector,
}); err != nil {
return err
}
// delete stale services (no DeleteCollection)
services, err := kc.CoreV1().Services(ns).List(ctx, metav1.ListOptions{
LabelSelector: selector,
})
if err != nil {
return err
}
for _, svc := range services.Items {
if err := kc.CoreV1().Services(ns).Delete(ctx, svc.Name, metav1.DeleteOptions{}); err != nil {
return err
}
}
return nil
}