ff4e8d429e
While moving device_plugin_handler_test.go from pkg/kubelet/cm/ to pkg/kubelet/cm/deviceplugin/, we can no longer uses cm in its tests because that would cause a cycle dependency. To solve this problem, I moved the main cm GetResources functionality as well as part of the current device plugin handler Allocate functionality into a new device plugin handler function, GetDeviceRunContainerOptions(). This refactoring is also needed by another PR 51895 that moves device allocation into admission phase. Now device plugin handler Allocate() first checks whether there is cached device runtime state and only issues Allocate grpc call if there is no cached state available. The new GetDeviceRunContainerOptions() function simply returns device runtime config from the cached state. To support this change, extended the podDevices struct and checkpoint data structure with device runtime state.
51 lines
1.4 KiB
Go
51 lines
1.4 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 deviceplugin
|
|
|
|
import (
|
|
"k8s.io/api/core/v1"
|
|
pluginapi "k8s.io/kubernetes/pkg/kubelet/apis/deviceplugin/v1alpha"
|
|
)
|
|
|
|
// HandlerStub provides a simple stub implementation for Handler.
|
|
type HandlerStub struct{}
|
|
|
|
// NewHandlerStub creates a HandlerStub.
|
|
func NewHandlerStub() (*HandlerStub, error) {
|
|
return &HandlerStub{}, nil
|
|
}
|
|
|
|
// Start simply returns nil.
|
|
func (h *HandlerStub) Start() error {
|
|
return nil
|
|
}
|
|
|
|
// Devices returns an empty map.
|
|
func (h *HandlerStub) Devices() map[string][]pluginapi.Device {
|
|
return make(map[string][]pluginapi.Device)
|
|
}
|
|
|
|
// Allocate simply returns nil.
|
|
func (h *HandlerStub) Allocate(pod *v1.Pod, container *v1.Container, activePods []*v1.Pod) error {
|
|
return nil
|
|
}
|
|
|
|
// GetDeviceRunContainerOptions simply returns nil.
|
|
func (h *HandlerStub) GetDeviceRunContainerOptions(pod *v1.Pod, container *v1.Container) *DeviceRunContainerOptions {
|
|
return nil
|
|
}
|