Merge pull request #267 from jpwhite4/job_uniq2_withfixes

Cherry pick #265 into v2 branch
ubccr · Jun 22, 2023 · 325645e · 325645e
2 parents 7c81feb + 0793600
commit 325645e
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/src/supremm/xdmodaccount.py b/src/supremm/xdmodaccount.py
@@ -24,7 +24,14 @@ def __init__(self, resource_id, config):
                 SELECT
                     jf.`job_id` AS `job_id`,
                     jf.`resource_id` AS `resource_id`,
-                    COALESCE(jf.`local_job_id_raw`, jf.`local_jobid`) AS `local_job_id`,
+                    CASE
+                      WHEN jf.`local_job_id_raw` IS NULL THEN IF(jf.`local_job_array_index` = -1, jf.`local_jobid`, CONCAT(jf.`local_jobid`, '_', jf.`local_job_array_index`))
+                      WHEN sj.`source_format` = 'slurm' THEN jf.`local_job_id_raw`
+                      ELSE IF(jf.`local_job_array_index` = -1, jf.`local_jobid`, CONCAT(jf.`local_jobid`, '_', jf.`local_job_array_index`))
+                    END AS `job_uniq_id`,
+                    jf.`local_jobid` AS `local_job_id`,
+                    jf.`local_job_array_index` AS `local_job_array_index`,
+                    jf.`local_job_id_raw` AS `local_job_id_raw`,
                     jf.`start_time_ts` AS `start_time`,
                     jf.`end_time_ts` AS `end_time`,
                     jf.`submit_time_ts` AS `submit`,
@@ -70,7 +77,14 @@ def __init__(self, resource_id, config):
                 SELECT
                     jf.`job_id` as `job_id`,
                     jf.`resource_id` as `resource_id`,
-                    COALESCE(jf.`local_job_id_raw`, jf.`local_jobid`) as `local_job_id`,
+                    CASE
+                      WHEN jf.`local_job_id_raw` IS NULL THEN IF(jf.`local_job_array_index` = -1, jf.`local_jobid`, CONCAT(jf.`local_jobid`, '_', jf.`local_job_array_index`))
+                      WHEN sj.`source_format` = 'slurm' THEN jf.`local_job_id_raw`
+                      ELSE IF(jf.`local_job_array_index` = -1, jf.`local_jobid`, CONCAT(jf.`local_jobid`, '_', jf.`local_job_array_index`))
+                    END AS `job_uniq_id`,
+                    jf.`local_jobid` AS `local_job_id`,
+                    jf.`local_job_array_index` AS `local_job_array_index`,
+                    jf.`local_job_id_raw` AS `local_job_id_raw`,
                     jf.`start_time_ts` as `start_time`,
                     jf.`end_time_ts` as `end_time`,
                     jf.`submit_time_ts` as `submit`,
@@ -280,7 +294,7 @@ def executequery(self, query, data):
             jobpk = record['job_id']
             del record['job_id']
             record['host_list'] = hostlist
-            job = Job(jobpk, str(record['local_job_id']), record)
+            job = Job(jobpk, str(record['job_uniq_id']), record)
             job.set_nodes(hostlist)
             job.set_rawarchives(hostarchives)