Function setContainerProbe

controllers/object_controls.go:2763–2792

(container *corev1.Container, probe *gpuv1.ContainerProbeSpec, probeType ContainerProbe)

}

func setContainerProbe(container *corev1.Container, probe *gpuv1.ContainerProbeSpec, probeType ContainerProbe) {
	var containerProbe *corev1.Probe

	// determine probe type to update
	switch probeType {
	case Startup:
		containerProbe = container.StartupProbe
	case Liveness:
		containerProbe = container.LivenessProbe
	case Readiness:
		containerProbe = container.ReadinessProbe
	}

	// set probe parameters if specified
	if probe.InitialDelaySeconds != 0 {
		containerProbe.InitialDelaySeconds = probe.InitialDelaySeconds
	}
	if probe.TimeoutSeconds != 0 {
		containerProbe.TimeoutSeconds = probe.TimeoutSeconds
	}
	if probe.FailureThreshold != 0 {
		containerProbe.FailureThreshold = probe.FailureThreshold
	}
	if probe.SuccessThreshold != 0 {
		containerProbe.SuccessThreshold = probe.SuccessThreshold
	}
	if probe.PeriodSeconds != 0 {
		containerProbe.PeriodSeconds = probe.PeriodSeconds
	}
}

// applies MIG related configuration env to container spec
func applyMIGConfiguration(c *corev1.Container, strategy gpuv1.MIGStrategy) {