utils

Documentation

Index

• Constants
• Variables
• func GetCheckpointData(devicePluginPath string) (*types.Checkpoint, error)
• func GetContainerIndexByName(pod *v1.Pod, containerName string) (int, error)
• func GetGPUData(annotations map[string]string) (gpuUtil int64, gpuMemory int64, deviceNames []string)
• func GetGPUMinorID(path string) (int, error)
• func GetGPUResourceOfContainer(container *v1.Container, resourceName v1.ResourceName) uint
• func GetGPUResourceOfPod(pod *v1.Pod, resourceName v1.ResourceName) uint
• func GetPredicateTimeOfPod(pod *v1.Pod) (predicateTime uint64)
• func IsGPUAssignedPod(pod *v1.Pod) bool
• func IsGPUPredicatedPod(pod *v1.Pod) (predicated bool)
• func IsGPURequiredContainer(c *v1.Container) bool
• func IsGPURequiredPod(pod *v1.Pod) bool
• func IsStringSliceEqual(a, b []string) bool
• func IsValidGPUPath(path string) bool
• func MakeContainerNamePrefix(containerName string) string
• func NewFSWatcher(files ...string) (*fsnotify.Watcher, error)
• func ShouldDelete(pod *v1.Pod) bool
• func ShouldRetry(err error) bool
• func UnixDial(addr string, timeout time.Duration) (net.Conn, error)
• func WaitForServer(socket string) error

Constants

const (
	TruncateLen = 31
)

constants used in this package

Variables

var (
	//DefaultDialOptions contains default dial options used in grpc dial
	DefaultDialOptions = []grpc.DialOption{grpc.WithInsecure(), grpc.WithDialer(UnixDial), grpc.WithBlock()}
)

Functions

func GetCheckpointData

func GetCheckpointData(devicePluginPath string) (*types.Checkpoint, error)

func GetContainerIndexByName

func GetContainerIndexByName(pod *v1.Pod, containerName string) (int, error)

func GetGPUData

func GetGPUData(annotations map[string]string) (gpuUtil int64, gpuMemory int64, deviceNames []string)

GetGPUData get cores, memory and device names from annotations

func GetGPUMinorID

func GetGPUMinorID(path string) (int, error)

GetGPUMinorID returns id in Nvidia GPU device path

func GetGPUResourceOfContainer

func GetGPUResourceOfContainer(container *v1.Container, resourceName v1.ResourceName) uint

func GetGPUResourceOfPod

func GetGPUResourceOfPod(pod *v1.Pod, resourceName v1.ResourceName) uint

func GetPredicateTimeOfPod

func GetPredicateTimeOfPod(pod *v1.Pod) (predicateTime uint64)

func IsGPUAssignedPod

func IsGPUAssignedPod(pod *v1.Pod) bool

Check if pod has already been assigned

func IsGPUPredicatedPod

func IsGPUPredicatedPod(pod *v1.Pod) (predicated bool)

func IsGPURequiredContainer

func IsGPURequiredContainer(c *v1.Container) bool

func IsGPURequiredPod

func IsGPURequiredPod(pod *v1.Pod) bool

func IsStringSliceEqual

func IsStringSliceEqual(a, b []string) bool

func IsValidGPUPath

func IsValidGPUPath(path string) bool

IsValidGPUPath checks if path is valid Nvidia GPU device path

func MakeContainerNamePrefix

func MakeContainerNamePrefix(containerName string) string

func NewFSWatcher

func NewFSWatcher(files ...string) (*fsnotify.Watcher, error)

NewFSWatcher returns a file watcher created by fsnotify.NewWatcher

func ShouldDelete

func ShouldDelete(pod *v1.Pod) bool

func ShouldRetry

func ShouldRetry(err error) bool

func UnixDial

func UnixDial(addr string, timeout time.Duration) (net.Conn, error)

UnixDial dials to a unix socket using net.DialTimeout

func WaitForServer

func WaitForServer(socket string) error

WaitForServer checks if grpc server is alive by making grpc blocking connection to the server socket