fix: config host path

kaito-project · Apr 4, 2024 · 5652d19 · 5652d19
1 parent 340a155
commit 5652d19
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 8 deletions.
diff --git a/pkg/tuning/preset-tuning.go b/pkg/tuning/preset-tuning.go
@@ -129,7 +129,7 @@ func handleImageDataSource(ctx context.Context, workspaceObj *kaitov1alpha1.Work
 		},
 	})
 
-	volumes, volumeMounts := utils.ConfigDataVolume()
+	volumes, volumeMounts := utils.ConfigDataVolume("")
 	return initContainers, volumes, volumeMounts
 }
 
@@ -153,13 +153,14 @@ func handleURLDataSource(ctx context.Context, workspaceObj *kaitov1alpha1.Worksp
 			},
 		},
 	})
-	volumes, volumeMounts := utils.ConfigDataVolume()
+	volumes, volumeMounts := utils.ConfigDataVolume("")
 	return initContainers, volumes, volumeMounts
 }
 
 func handleHostPathDataSource(ctx context.Context, workspaceObj *kaitov1alpha1.Workspace) ([]corev1.Container, []corev1.Volume, []corev1.VolumeMount) {
 	var initContainers []corev1.Container
-	volumes, volumeMounts := utils.ConfigDataVolume()
+	hostPath := workspaceObj.Tuning.Input.HostPath
+	volumes, volumeMounts := utils.ConfigDataVolume(hostPath)
 	return initContainers, volumes, volumeMounts
 }
 

diff --git a/pkg/utils/common-preset.go b/pkg/utils/common-preset.go
@@ -39,14 +39,24 @@ func ConfigSHMVolume(wObj *kaitov1alpha1.Workspace) (corev1.Volume, corev1.Volum
 	return volume, volumeMount
 }
 
-func ConfigDataVolume() ([]corev1.Volume, []corev1.VolumeMount) {
+func ConfigDataVolume(hostPath string) ([]corev1.Volume, []corev1.VolumeMount) {
 	var volumes []corev1.Volume
 	var volumeMounts []corev1.VolumeMount
-	volumes = append(volumes, corev1.Volume{
-		Name: "data-volume",
-		VolumeSource: corev1.VolumeSource{
+	var volumeSource corev1.VolumeSource
+	if hostPath != "" {
+		volumeSource = corev1.VolumeSource{
+			HostPath: &corev1.HostPathVolumeSource{
+				Path: hostPath,
+			},
+		}
+	} else {
+		volumeSource = corev1.VolumeSource{
 			EmptyDir: &corev1.EmptyDirVolumeSource{},
-		},
+		}
+	}
+	volumes = append(volumes, corev1.Volume{
+		Name:         "data-volume",
+		VolumeSource: volumeSource,
 	})
 
 	volumeMounts = append(volumeMounts, corev1.VolumeMount{