1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586 |
- /*
- 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 checkpoint
- import (
- "encoding/json"
- "k8s.io/kubernetes/pkg/kubelet/checkpointmanager"
- "k8s.io/kubernetes/pkg/kubelet/checkpointmanager/checksum"
- )
- // DeviceManagerCheckpoint defines the operations to retrieve pod devices
- type DeviceManagerCheckpoint interface {
- checkpointmanager.Checkpoint
- GetData() ([]PodDevicesEntry, map[string][]string)
- }
- // PodDevicesEntry connects pod information to devices
- type PodDevicesEntry struct {
- PodUID string
- ContainerName string
- ResourceName string
- DeviceIDs []string
- AllocResp []byte
- }
- // checkpointData struct is used to store pod to device allocation information
- // in a checkpoint file.
- // TODO: add version control when we need to change checkpoint format.
- type checkpointData struct {
- PodDeviceEntries []PodDevicesEntry
- RegisteredDevices map[string][]string
- }
- // Data holds checkpoint data and its checksum
- type Data struct {
- Data checkpointData
- Checksum checksum.Checksum
- }
- // New returns an instance of Checkpoint
- func New(devEntries []PodDevicesEntry,
- devices map[string][]string) DeviceManagerCheckpoint {
- return &Data{
- Data: checkpointData{
- PodDeviceEntries: devEntries,
- RegisteredDevices: devices,
- },
- }
- }
- // MarshalCheckpoint returns marshalled data
- func (cp *Data) MarshalCheckpoint() ([]byte, error) {
- cp.Checksum = checksum.New(cp.Data)
- return json.Marshal(*cp)
- }
- // UnmarshalCheckpoint returns unmarshalled data
- func (cp *Data) UnmarshalCheckpoint(blob []byte) error {
- return json.Unmarshal(blob, cp)
- }
- // VerifyChecksum verifies that passed checksum is same as calculated checksum
- func (cp *Data) VerifyChecksum() error {
- return cp.Checksum.Verify(cp.Data)
- }
- // GetData returns device entries and registered devices
- func (cp *Data) GetData() ([]PodDevicesEntry, map[string][]string) {
- return cp.Data.PodDeviceEntries, cp.Data.RegisteredDevices
- }
|