feat(application): add Environment field and inject into Slurm job submission

2026-04-21 10:23:31 +08:00
parent d9ca9233b3
commit 4fd331ebd8
4 changed files with 29 additions and 0 deletions
--- a/internal/service/job_service.go
+++ b/internal/service/job_service.go
@@ -49,6 +49,9 @@ func (s *JobService) SubmitJob(ctx context.Context, req *model.SubmitJobRequest)
 		"PATH=/usr/local/bin:/usr/bin:/bin",
 		"HOME=/root",
 	}
+	for k, v := range req.Environment {
+		jobDesc.Environment = append(jobDesc.Environment, k+"="+v)
+	}

 	if req.MemoryPerNode != nil {
 		jobDesc.MemoryPerNode = &slurm.Uint64NoVal{Set: slurm.Ptr(true), Number: req.MemoryPerNode}