Function BatchContainers

pkg/workloads/k8s.go:281–316

(api spec.API, job *spec.BatchJob)

}

func BatchContainers(api spec.API, job *spec.BatchJob) ([]kcore.Container, []kcore.Volume) {
	userContainers, userVolumes := userPodContainers(api)
	dequeuerContainer, dequeuerVolume := batchDequeuerProxyContainer(api, job.ID, job.SQSUrl)

	// make sure the dequeuer starts first to allow it to start watching the graveyard before user containers begin
	containers := append([]kcore.Container{dequeuerContainer}, userContainers...)
	volumes := append([]kcore.Volume{dequeuerVolume}, userVolumes...)

	k8sName := job.K8sName()

	volumes = append(volumes,
		KubexitVolume(),
		APIConfigVolume(k8sName),
	)

	containerNames := userconfig.GetContainerNames(api.Pod.Containers)
	containerNames.Add(dequeuerContainer.Name)

	for i, c := range containers {
		containers[i].VolumeMounts = append(containers[i].VolumeMounts,
			KubexitMount(),
			APIConfigMount(k8sName),
		)

		containerDeathDependencies := containerNames.Copy()
		containerDeathDependencies.Remove(c.Name)
		containerDeathEnvVars := getKubexitEnvVars(c.Name, containerDeathDependencies.SliceSorted(), nil)
		containers[i].Env = append(containers[i].Env, containerDeathEnvVars...)

		if c.Command[0] != "/cortex/kubexit" {
			containers[i].Command = append([]string{"/cortex/kubexit"}, c.Command...)
		}
	}

	return containers, volumes
}

func userPodContainers(api spec.API) ([]kcore.Container, []kcore.Volume) {
	volumes := []kcore.Volume{