ca2f1b87ad
Alter how runtime.SerializeInfo is represented to simplify negotiation and reduce the need to allocate during negotiation. Simplify the dynamic client's logic around negotiating type. Add more tests for media type handling where necessary.
108 lines
3.5 KiB
Go
Executable File
108 lines
3.5 KiB
Go
Executable File
/*
|
|
Copyright 2016 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 webhook implements a generic HTTP webhook plugin.
|
|
package webhook
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"k8s.io/kubernetes/pkg/api"
|
|
apierrors "k8s.io/kubernetes/pkg/api/errors"
|
|
"k8s.io/kubernetes/pkg/api/unversioned"
|
|
"k8s.io/kubernetes/pkg/apimachinery/registered"
|
|
"k8s.io/kubernetes/pkg/client/restclient"
|
|
"k8s.io/kubernetes/pkg/client/unversioned/clientcmd"
|
|
"k8s.io/kubernetes/pkg/runtime"
|
|
runtimeserializer "k8s.io/kubernetes/pkg/runtime/serializer"
|
|
"k8s.io/kubernetes/pkg/util/wait"
|
|
|
|
_ "k8s.io/kubernetes/pkg/apis/authorization/install"
|
|
)
|
|
|
|
type GenericWebhook struct {
|
|
RestClient *restclient.RESTClient
|
|
initialBackoff time.Duration
|
|
}
|
|
|
|
// NewGenericWebhook creates a new GenericWebhook from the provided kubeconfig file.
|
|
func NewGenericWebhook(kubeConfigFile string, groupVersions []unversioned.GroupVersion, initialBackoff time.Duration) (*GenericWebhook, error) {
|
|
for _, groupVersion := range groupVersions {
|
|
if !registered.IsEnabledVersion(groupVersion) {
|
|
return nil, fmt.Errorf("webhook plugin requires enabling extension resource: %s", groupVersion)
|
|
}
|
|
}
|
|
|
|
loadingRules := clientcmd.NewDefaultClientConfigLoadingRules()
|
|
loadingRules.ExplicitPath = kubeConfigFile
|
|
loader := clientcmd.NewNonInteractiveDeferredLoadingClientConfig(loadingRules, &clientcmd.ConfigOverrides{})
|
|
|
|
clientConfig, err := loader.ClientConfig()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
codec := api.Codecs.LegacyCodec(groupVersions...)
|
|
clientConfig.ContentConfig.NegotiatedSerializer = runtimeserializer.NegotiatedSerializerWrapper(runtime.SerializerInfo{Serializer: codec})
|
|
|
|
restClient, err := restclient.UnversionedRESTClientFor(clientConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// TODO(ericchiang): Can we ensure remote service is reachable?
|
|
|
|
return &GenericWebhook{restClient, initialBackoff}, nil
|
|
}
|
|
|
|
// WithExponentialBackoff will retry webhookFn() up to 5 times with exponentially increasing backoff when
|
|
// it returns an error for which apierrors.SuggestsClientDelay() or apierrors.IsInternalError() returns true.
|
|
func (g *GenericWebhook) WithExponentialBackoff(webhookFn func() restclient.Result) restclient.Result {
|
|
var result restclient.Result
|
|
WithExponentialBackoff(g.initialBackoff, func() error {
|
|
result = webhookFn()
|
|
return result.Error()
|
|
})
|
|
return result
|
|
}
|
|
|
|
// WithExponentialBackoff will retry webhookFn() up to 5 times with exponentially increasing backoff when
|
|
// it returns an error for which apierrors.SuggestsClientDelay() or apierrors.IsInternalError() returns true.
|
|
func WithExponentialBackoff(initialBackoff time.Duration, webhookFn func() error) error {
|
|
backoff := wait.Backoff{
|
|
Duration: initialBackoff,
|
|
Factor: 1.5,
|
|
Jitter: 0.2,
|
|
Steps: 5,
|
|
}
|
|
|
|
var err error
|
|
wait.ExponentialBackoff(backoff, func() (bool, error) {
|
|
err = webhookFn()
|
|
if _, shouldRetry := apierrors.SuggestsClientDelay(err); shouldRetry {
|
|
return false, nil
|
|
}
|
|
if apierrors.IsInternalError(err) {
|
|
return false, nil
|
|
}
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
})
|
|
return err
|
|
}
|