98 lines
3.3 KiB
Go
98 lines
3.3 KiB
Go
/*
|
|
Copyright 2017 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package configuration
|
|
|
|
import (
|
|
"fmt"
|
|
"sort"
|
|
"sync/atomic"
|
|
|
|
"k8s.io/api/admissionregistration/v1beta1"
|
|
"k8s.io/apimachinery/pkg/labels"
|
|
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
|
|
"k8s.io/apiserver/pkg/admission/plugin/webhook/generic"
|
|
"k8s.io/client-go/informers"
|
|
admissionregistrationlisters "k8s.io/client-go/listers/admissionregistration/v1beta1"
|
|
"k8s.io/client-go/tools/cache"
|
|
)
|
|
|
|
// validatingWebhookConfigurationManager collects the validating webhook objects so that they can be called.
|
|
type validatingWebhookConfigurationManager struct {
|
|
configuration *atomic.Value
|
|
lister admissionregistrationlisters.ValidatingWebhookConfigurationLister
|
|
hasSynced func() bool
|
|
}
|
|
|
|
var _ generic.Source = &validatingWebhookConfigurationManager{}
|
|
|
|
func NewValidatingWebhookConfigurationManager(f informers.SharedInformerFactory) generic.Source {
|
|
informer := f.Admissionregistration().V1beta1().ValidatingWebhookConfigurations()
|
|
manager := &validatingWebhookConfigurationManager{
|
|
configuration: &atomic.Value{},
|
|
lister: informer.Lister(),
|
|
hasSynced: informer.Informer().HasSynced,
|
|
}
|
|
|
|
// Start with an empty list
|
|
manager.configuration.Store(&v1beta1.ValidatingWebhookConfiguration{})
|
|
|
|
// On any change, rebuild the config
|
|
informer.Informer().AddEventHandler(cache.ResourceEventHandlerFuncs{
|
|
AddFunc: func(_ interface{}) { manager.updateConfiguration() },
|
|
UpdateFunc: func(_, _ interface{}) { manager.updateConfiguration() },
|
|
DeleteFunc: func(_ interface{}) { manager.updateConfiguration() },
|
|
})
|
|
|
|
return manager
|
|
}
|
|
|
|
// Webhooks returns the merged ValidatingWebhookConfiguration.
|
|
func (v *validatingWebhookConfigurationManager) Webhooks() []v1beta1.Webhook {
|
|
return v.configuration.Load().(*v1beta1.ValidatingWebhookConfiguration).Webhooks
|
|
}
|
|
|
|
// HasSynced returns true if the shared informers have synced.
|
|
func (v *validatingWebhookConfigurationManager) HasSynced() bool {
|
|
return v.hasSynced()
|
|
}
|
|
|
|
func (v *validatingWebhookConfigurationManager) updateConfiguration() {
|
|
configurations, err := v.lister.List(labels.Everything())
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("error updating configuration: %v", err))
|
|
return
|
|
}
|
|
v.configuration.Store(mergeValidatingWebhookConfigurations(configurations))
|
|
}
|
|
|
|
func mergeValidatingWebhookConfigurations(
|
|
configurations []*v1beta1.ValidatingWebhookConfiguration,
|
|
) *v1beta1.ValidatingWebhookConfiguration {
|
|
sort.SliceStable(configurations, ValidatingWebhookConfigurationSorter(configurations).ByName)
|
|
var ret v1beta1.ValidatingWebhookConfiguration
|
|
for _, c := range configurations {
|
|
ret.Webhooks = append(ret.Webhooks, c.Webhooks...)
|
|
}
|
|
return &ret
|
|
}
|
|
|
|
type ValidatingWebhookConfigurationSorter []*v1beta1.ValidatingWebhookConfiguration
|
|
|
|
func (a ValidatingWebhookConfigurationSorter) ByName(i, j int) bool {
|
|
return a[i].Name < a[j].Name
|
|
}
|