get runtime(docker, containerd) socket file path based on toolkit container env or default
(c *corev1.Container, runtime string)
| 2637 | |
| 2638 | // get runtime(docker, containerd) socket file path based on toolkit container env or default |
| 2639 | func getRuntimeSocketFile(c *corev1.Container, runtime string) (string, error) { |
| 2640 | var runtimeSocketFile string |
| 2641 | switch runtime { |
| 2642 | case gpuv1.Docker.String(): |
| 2643 | runtimeSocketFile = DefaultDockerSocketFile |
| 2644 | if getContainerEnv(c, "DOCKER_SOCKET") != "" { |
| 2645 | runtimeSocketFile = getContainerEnv(c, "DOCKER_SOCKET") |
| 2646 | } |
| 2647 | case gpuv1.Containerd.String(): |
| 2648 | runtimeSocketFile = DefaultContainerdSocketFile |
| 2649 | if getContainerEnv(c, "CONTAINERD_SOCKET") != "" { |
| 2650 | runtimeSocketFile = getContainerEnv(c, "CONTAINERD_SOCKET") |
| 2651 | } |
| 2652 | case gpuv1.CRIO.String(): |
| 2653 | runtimeSocketFile = "" |
| 2654 | default: |
| 2655 | return "", fmt.Errorf("invalid runtime: %s", runtime) |
| 2656 | } |
| 2657 | |
| 2658 | return runtimeSocketFile, nil |
| 2659 | } |
| 2660 | |
| 2661 | func getContainerEnv(c *corev1.Container, key string) string { |
| 2662 | for _, val := range c.Env { |
no test coverage detected