forked from openshift/origin
/
imagestreamtag_evaluator.go
93 lines (76 loc) · 3.23 KB
/
imagestreamtag_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
package image
import (
"fmt"
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"
kapi "k8s.io/kubernetes/pkg/apis/core"
kquota "k8s.io/kubernetes/pkg/quota"
"k8s.io/kubernetes/pkg/quota/generic"
imageapi "github.com/openshift/origin/pkg/image/apis/image"
imageclient "github.com/openshift/origin/pkg/image/generated/internalclientset/typed/image/internalversion"
imageinternalversion "github.com/openshift/origin/pkg/image/generated/listers/image/internalversion"
)
var imageStreamTagResources = []kapi.ResourceName{
imageapi.ResourceImageStreams,
}
type imageStreamTagEvaluator struct {
store imageinternalversion.ImageStreamLister
istGetter imageclient.ImageStreamTagsGetter
}
// NewImageStreamTagEvaluator computes resource usage of ImageStreamsTags. Its sole purpose is to handle
// UPDATE admission operations on imageStreamTags resource.
func NewImageStreamTagEvaluator(store imageinternalversion.ImageStreamLister, istGetter imageclient.ImageStreamTagsGetter) kquota.Evaluator {
return &imageStreamTagEvaluator{
store: store,
istGetter: istGetter,
}
}
// Constraints checks that given object is an image stream tag
func (i *imageStreamTagEvaluator) Constraints(required []kapi.ResourceName, object runtime.Object) error {
if _, ok := object.(*imageapi.ImageStreamTag); !ok {
return fmt.Errorf("unexpected input object %v", object)
}
return nil
}
func (i *imageStreamTagEvaluator) GroupResource() schema.GroupResource {
return imageapi.Resource("imagestreamtags")
}
func (i *imageStreamTagEvaluator) Handles(a kadmission.Attributes) bool {
operation := a.GetOperation()
return operation == kadmission.Create || operation == kadmission.Update
}
func (i *imageStreamTagEvaluator) Matches(resourceQuota *kapi.ResourceQuota, item runtime.Object) (bool, error) {
matchesScopeFunc := func(kapi.ResourceQuotaScope, runtime.Object) (bool, error) { return true, nil }
return generic.Matches(resourceQuota, item, i.MatchingResources, matchesScopeFunc)
}
func (i *imageStreamTagEvaluator) MatchingResources(input []kapi.ResourceName) []kapi.ResourceName {
return kquota.Intersection(input, imageStreamTagResources)
}
func (i *imageStreamTagEvaluator) Usage(item runtime.Object) (kapi.ResourceList, error) {
ist, ok := item.(*imageapi.ImageStreamTag)
if !ok {
return kapi.ResourceList{}, nil
}
res := map[kapi.ResourceName]resource.Quantity{
imageapi.ResourceImageStreams: *resource.NewQuantity(0, resource.BinarySI),
}
isName, _, err := imageapi.ParseImageStreamTagName(ist.Name)
if err != nil {
return kapi.ResourceList{}, err
}
is, err := i.store.ImageStreams(ist.Namespace).Get(isName)
if err != nil && !kerrors.IsNotFound(err) {
utilruntime.HandleError(fmt.Errorf("failed to get image stream %s/%s: %v", ist.Namespace, isName, err))
}
if is == nil || kerrors.IsNotFound(err) {
res[imageapi.ResourceImageStreams] = *resource.NewQuantity(1, resource.BinarySI)
}
return res, nil
}
func (i *imageStreamTagEvaluator) UsageStats(options kquota.UsageStatsOptions) (kquota.UsageStats, error) {
return kquota.UsageStats{}, nil
}