/
imagestreamimport_evaluator.go
109 lines (90 loc) · 4.13 KB
/
imagestreamimport_evaluator.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
package image
import (
"fmt"
corev1 "k8s.io/api/core/v1"
kerrors "k8s.io/apimachinery/pkg/api/errors"
"k8s.io/apimachinery/pkg/api/resource"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/runtime/schema"
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
kadmission "k8s.io/apiserver/pkg/admission"
kquota "k8s.io/kubernetes/pkg/quota/v1"
"k8s.io/kubernetes/pkg/quota/v1/generic"
"github.com/openshift/api/image"
imagev1 "github.com/openshift/api/image/v1"
imagev1lister "github.com/openshift/client-go/image/listers/image/v1"
imageapi "github.com/openshift/origin/pkg/image/apis/image"
imagev1conversions "github.com/openshift/origin/pkg/image/apis/image/v1"
)
var imageStreamImportResources = []corev1.ResourceName{
imagev1.ResourceImageStreams,
}
type imageStreamImportEvaluator struct {
store imagev1lister.ImageStreamLister
}
// NewImageStreamImportEvaluator computes resource usage for ImageStreamImport objects. This particular kind
// is a virtual resource. It depends on ImageStream usage evaluator to compute image numbers before the
// the admission can work.
func NewImageStreamImportEvaluator(store imagev1lister.ImageStreamLister) kquota.Evaluator {
return &imageStreamImportEvaluator{
store: store,
}
}
// Constraints checks that given object is an image stream import.
func (i *imageStreamImportEvaluator) Constraints(required []corev1.ResourceName, object runtime.Object) error {
_, okInt := object.(*imageapi.ImageStreamImport)
_, okExt := object.(*imagev1.ImageStreamImport)
if !okInt && !okExt {
return fmt.Errorf("unexpected input object %v", object)
}
return nil
}
func (i *imageStreamImportEvaluator) GroupResource() schema.GroupResource {
return image.Resource("imagestreamimports")
}
func (i *imageStreamImportEvaluator) Handles(a kadmission.Attributes) bool {
return a.GetOperation() == kadmission.Create
}
func (i *imageStreamImportEvaluator) Matches(resourceQuota *corev1.ResourceQuota, item runtime.Object) (bool, error) {
matchesScopeFunc := func(corev1.ScopedResourceSelectorRequirement, runtime.Object) (bool, error) { return true, nil }
return generic.Matches(resourceQuota, item, i.MatchingResources, matchesScopeFunc)
}
func (p *imageStreamImportEvaluator) MatchingScopes(item runtime.Object, scopes []corev1.ScopedResourceSelectorRequirement) ([]corev1.ScopedResourceSelectorRequirement, error) {
return []corev1.ScopedResourceSelectorRequirement{}, nil
}
func (p *imageStreamImportEvaluator) UncoveredQuotaScopes(limitedScopes []corev1.ScopedResourceSelectorRequirement, matchedQuotaScopes []corev1.ScopedResourceSelectorRequirement) ([]corev1.ScopedResourceSelectorRequirement, error) {
return []corev1.ScopedResourceSelectorRequirement{}, nil
}
func (i *imageStreamImportEvaluator) MatchingResources(input []corev1.ResourceName) []corev1.ResourceName {
return kquota.Intersection(input, imageStreamImportResources)
}
func (i *imageStreamImportEvaluator) Usage(item runtime.Object) (corev1.ResourceList, error) {
if isiInternal, ok := item.(*imageapi.ImageStreamImport); ok {
out := &imagev1.ImageStreamImport{}
if err := imagev1conversions.Convert_image_ImageStreamImport_To_v1_ImageStreamImport(isiInternal, out, nil); err != nil {
return corev1.ResourceList{}, fmt.Errorf("error converting ImageStreamImport: %v", err)
}
item = out
}
isi, ok := item.(*imagev1.ImageStreamImport)
if !ok {
return corev1.ResourceList{}, fmt.Errorf("item is not an ImageStreamImport: %T", item)
}
usage := map[corev1.ResourceName]resource.Quantity{
imagev1.ResourceImageStreams: *resource.NewQuantity(0, resource.DecimalSI),
}
if !isi.Spec.Import || (len(isi.Spec.Images) == 0 && isi.Spec.Repository == nil) {
return usage, nil
}
is, err := i.store.ImageStreams(isi.Namespace).Get(isi.Name)
if err != nil && !kerrors.IsNotFound(err) {
utilruntime.HandleError(fmt.Errorf("failed to list image streams: %v", err))
}
if is == nil || kerrors.IsNotFound(err) {
usage[imagev1.ResourceImageStreams] = *resource.NewQuantity(1, resource.DecimalSI)
}
return usage, nil
}
func (i *imageStreamImportEvaluator) UsageStats(options kquota.UsageStatsOptions) (kquota.UsageStats, error) {
return kquota.UsageStats{}, nil
}