954996e231
- Move from the old github.com/golang/glog to k8s.io/klog - klog as explicit InitFlags() so we add them as necessary - we update the other repositories that we vendor that made a similar change from glog to klog * github.com/kubernetes/repo-infra * k8s.io/gengo/ * k8s.io/kube-openapi/ * github.com/google/cadvisor - Entirely remove all references to glog - Fix some tests by explicit InitFlags in their init() methods Change-Id: I92db545ff36fcec83afe98f550c9e630098b3135
148 lines
4.0 KiB
Go
148 lines
4.0 KiB
Go
/*
|
|
Copyright 2018 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 pluginwatcher
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"reflect"
|
|
"sync"
|
|
"time"
|
|
|
|
"golang.org/x/net/context"
|
|
"k8s.io/klog"
|
|
|
|
v1beta1 "k8s.io/kubernetes/pkg/kubelet/util/pluginwatcher/example_plugin_apis/v1beta1"
|
|
v1beta2 "k8s.io/kubernetes/pkg/kubelet/util/pluginwatcher/example_plugin_apis/v1beta2"
|
|
)
|
|
|
|
type exampleHandler struct {
|
|
SupportedVersions []string
|
|
ExpectedNames map[string]int
|
|
|
|
eventChans map[string]chan examplePluginEvent // map[pluginName]eventChan
|
|
|
|
m sync.Mutex
|
|
count int
|
|
}
|
|
|
|
type examplePluginEvent int
|
|
|
|
const (
|
|
exampleEventValidate examplePluginEvent = 0
|
|
exampleEventRegister examplePluginEvent = 1
|
|
exampleEventDeRegister examplePluginEvent = 2
|
|
exampleEventError examplePluginEvent = 3
|
|
)
|
|
|
|
// NewExampleHandler provide a example handler
|
|
func NewExampleHandler(supportedVersions []string) *exampleHandler {
|
|
return &exampleHandler{
|
|
SupportedVersions: supportedVersions,
|
|
ExpectedNames: make(map[string]int),
|
|
|
|
eventChans: make(map[string]chan examplePluginEvent),
|
|
}
|
|
}
|
|
|
|
func (p *exampleHandler) ValidatePlugin(pluginName string, endpoint string, versions []string) error {
|
|
p.SendEvent(pluginName, exampleEventValidate)
|
|
|
|
n, ok := p.DecreasePluginCount(pluginName)
|
|
if !ok && n > 0 {
|
|
return fmt.Errorf("pluginName('%s') wasn't expected (count is %d)", pluginName, n)
|
|
}
|
|
|
|
if !reflect.DeepEqual(versions, p.SupportedVersions) {
|
|
return fmt.Errorf("versions('%v') != supported versions('%v')", versions, p.SupportedVersions)
|
|
}
|
|
|
|
// this handler expects non-empty endpoint as an example
|
|
if len(endpoint) == 0 {
|
|
return errors.New("expecting non empty endpoint")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (p *exampleHandler) RegisterPlugin(pluginName, endpoint string) error {
|
|
p.SendEvent(pluginName, exampleEventRegister)
|
|
|
|
// Verifies the grpcServer is ready to serve services.
|
|
_, conn, err := dial(endpoint, time.Second)
|
|
if err != nil {
|
|
return fmt.Errorf("Failed dialing endpoint (%s): %v", endpoint, err)
|
|
}
|
|
defer conn.Close()
|
|
|
|
// The plugin handler should be able to use any listed service API version.
|
|
v1beta1Client := v1beta1.NewExampleClient(conn)
|
|
v1beta2Client := v1beta2.NewExampleClient(conn)
|
|
|
|
// Tests v1beta1 GetExampleInfo
|
|
_, err = v1beta1Client.GetExampleInfo(context.Background(), &v1beta1.ExampleRequest{})
|
|
if err != nil {
|
|
return fmt.Errorf("Failed GetExampleInfo for v1beta2Client(%s): %v", endpoint, err)
|
|
}
|
|
|
|
// Tests v1beta1 GetExampleInfo
|
|
_, err = v1beta2Client.GetExampleInfo(context.Background(), &v1beta2.ExampleRequest{})
|
|
if err != nil {
|
|
return fmt.Errorf("Failed GetExampleInfo for v1beta2Client(%s): %v", endpoint, err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (p *exampleHandler) DeRegisterPlugin(pluginName string) {
|
|
p.SendEvent(pluginName, exampleEventDeRegister)
|
|
}
|
|
|
|
func (p *exampleHandler) EventChan(pluginName string) chan examplePluginEvent {
|
|
return p.eventChans[pluginName]
|
|
}
|
|
|
|
func (p *exampleHandler) SendEvent(pluginName string, event examplePluginEvent) {
|
|
klog.V(2).Infof("Sending %v for plugin %s over chan %v", event, pluginName, p.eventChans[pluginName])
|
|
p.eventChans[pluginName] <- event
|
|
}
|
|
|
|
func (p *exampleHandler) AddPluginName(pluginName string) {
|
|
p.m.Lock()
|
|
defer p.m.Unlock()
|
|
|
|
v, ok := p.ExpectedNames[pluginName]
|
|
if !ok {
|
|
p.eventChans[pluginName] = make(chan examplePluginEvent)
|
|
v = 1
|
|
}
|
|
|
|
p.ExpectedNames[pluginName] = v
|
|
}
|
|
|
|
func (p *exampleHandler) DecreasePluginCount(pluginName string) (old int, ok bool) {
|
|
p.m.Lock()
|
|
defer p.m.Unlock()
|
|
|
|
v, ok := p.ExpectedNames[pluginName]
|
|
if !ok {
|
|
v = -1
|
|
}
|
|
|
|
return v, ok
|
|
}
|