From 76b4c8494897f4646cbb2103ecbd1f2b0584d040 Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Tue, 22 Nov 2022 16:02:03 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/grampus/grampus.go | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/modules/grampus/grampus.go b/modules/grampus/grampus.go
index 26e85a1cf..899091a94 100755
--- a/modules/grampus/grampus.go
+++ b/modules/grampus/grampus.go
@@ -121,6 +121,21 @@ func getDatasetGrampus(datasetInfos map[string]models.DatasetInfo) []models.Gram
 	}
 	return datasetGrampus
 }
+func getDatasetGPUGrampus(datasetInfos map[string]models.DatasetInfo) []models.GrampusDataset {
+	var datasetGrampus []models.GrampusDataset
+	for _, datasetInfo := range datasetInfos {
+		datasetGrampus = append(datasetGrampus, models.GrampusDataset{
+			Name:          datasetInfo.FullName,
+			Bucket:        setting.Attachment.Minio.Bucket,
+			EndPoint:      setting.Attachment.Minio.Endpoint,
+			ObjectKey:     datasetInfo.DataLocalPath + "/" + datasetInfo.FullName,
+			ReadOnly:      true,
+			ContainerPath: "/dataset/" + datasetInfo.Name,
+		})
+
+	}
+	return datasetGrampus
+}
 
 func GenerateNotebookJob(ctx *context.Context, req *GenerateNotebookJobReq) (jobId string, err error) {
 	createTime := timeutil.TimeStampNow()
@@ -136,12 +151,14 @@ func GenerateNotebookJob(ctx *context.Context, req *GenerateNotebookJobReq) (job
 			ObjectKey: req.CodeStoragePath + cloudbrain.DefaultBranchName + ".zip",
 		}
 	} else {
-
+		datasetGrampus = getDatasetGPUGrampus(req.DatasetInfos)
 		codeGrampus = models.GrampusDataset{
-			Name:      req.CodeName,
-			Bucket:    setting.Bucket,
-			EndPoint:  getEndPoint(),
-			ObjectKey: req.CodeStoragePath + cloudbrain.DefaultBranchName + ".zip",
+			Name:          req.CodeName,
+			Bucket:        setting.Attachment.Minio.Bucket,
+			EndPoint:      setting.Attachment.Minio.Endpoint,
+			ObjectKey:     req.CodeStoragePath + cloudbrain.DefaultBranchName + ".zip",
+			ReadOnly:      false,
+			ContainerPath: "/code",
 		}
 
 	}