@@ -797,11 +797,9 @@ def process_message(
797797 # ... but either way update the job ID in the job proxy (it only
798798 # comes in via the submission message).
799799 if itask .run_mode != RunMode .SIMULATION :
800- job_tokens = itask .tokens .duplicate (
801- job = str (itask .submit_num )
802- )
803800 self .data_store_mgr .delta_job_attr (
804- job_tokens , 'job_id' , itask .summary ['submit_method_id' ])
801+ itask , 'job_id' , itask .summary ['submit_method_id' ]
802+ )
805803 else :
806804 # In simulation mode submitted implies started:
807805 self .spawn_children (itask , TASK_OUTPUT_STARTED , forced )
@@ -1354,9 +1352,8 @@ def _process_message_failed(
13541352 if event_time is None :
13551353 event_time = get_current_time_string ()
13561354 itask .set_summary_time ('finished' , event_time )
1357- job_tokens = itask .tokens .duplicate (job = str (itask .submit_num ))
1358- self .data_store_mgr .delta_job_time (job_tokens , 'finished' , event_time )
1359- self .data_store_mgr .delta_job_state (job_tokens , TASK_STATUS_FAILED )
1355+ self .data_store_mgr .delta_job_time (itask , 'finished' , event_time )
1356+ self .data_store_mgr .delta_job_state (itask , TASK_STATUS_FAILED )
13601357 self .workflow_db_mgr .put_update_task_jobs (itask , {
13611358 "run_status" : 1 ,
13621359 "time_run_exit" : event_time ,
@@ -1394,12 +1391,11 @@ def _process_message_failed(
13941391
13951392 def _process_message_started (self , itask , event_time , forced ):
13961393 """Helper for process_message, handle a started message."""
1394+ self .data_store_mgr .delta_job_time (itask , 'started' , event_time )
1395+ self .data_store_mgr .delta_job_state (itask , TASK_STATUS_RUNNING )
13971396 if itask .job_vacated :
13981397 itask .job_vacated = False
13991398 LOG .info (f"[{ itask } ] Vacated job restarted" )
1400- job_tokens = itask .tokens .duplicate (job = str (itask .submit_num ))
1401- self .data_store_mgr .delta_job_time (job_tokens , 'started' , event_time )
1402- self .data_store_mgr .delta_job_state (job_tokens , TASK_STATUS_RUNNING )
14031399 itask .set_summary_time ('started' , event_time )
14041400 self .workflow_db_mgr .put_update_task_jobs (itask , {
14051401 "time_run" : itask .summary ['started_time_string' ]})
@@ -1430,9 +1426,8 @@ def _process_message_succeeded(self, itask, event_time, forced):
14301426 Ignore forced.
14311427 """
14321428
1433- job_tokens = itask .tokens .duplicate (job = str (itask .submit_num ))
1434- self .data_store_mgr .delta_job_time (job_tokens , 'finished' , event_time )
1435- self .data_store_mgr .delta_job_state (job_tokens , TASK_STATUS_SUCCEEDED )
1429+ self .data_store_mgr .delta_job_time (itask , 'finished' , event_time )
1430+ self .data_store_mgr .delta_job_state (itask , TASK_STATUS_SUCCEEDED )
14361431 itask .set_summary_time ('finished' , event_time )
14371432 self .workflow_db_mgr .put_update_task_jobs (itask , {
14381433 "run_status" : 0 ,
@@ -1505,13 +1500,9 @@ def _process_message_submit_failed(
15051500 self .setup_event_handlers (itask , self .EVENT_SUBMIT_RETRY , msg )
15061501
15071502 # Register newly submit-failed job with the database and datastore.
1508- job_tokens = itask .tokens .duplicate (job = str (itask .submit_num ))
15091503 self ._insert_task_job (
15101504 itask , event_time , self .JOB_SUBMIT_FAIL_FLAG , forced = forced )
1511- self .data_store_mgr .delta_job_state (
1512- job_tokens ,
1513- TASK_STATUS_SUBMIT_FAILED
1514- )
1505+ self .data_store_mgr .delta_job_state (itask , TASK_STATUS_SUBMIT_FAILED )
15151506 self ._reset_job_timers (itask )
15161507
15171508 return no_retries
@@ -1560,24 +1551,12 @@ def _process_message_submitted(
15601551 # Do after itask has changed state
15611552 self ._insert_task_job (
15621553 itask , event_time , self .JOB_SUBMIT_SUCCESS_FLAG , forced = forced )
1563- job_tokens = itask .tokens .duplicate (job = str (itask .submit_num ))
1564- self .data_store_mgr .delta_job_time (
1565- job_tokens ,
1566- 'submitted' ,
1567- event_time ,
1568- )
1554+ self .data_store_mgr .delta_job_time (itask , 'submitted' , event_time )
15691555 if itask .run_mode == RunMode .SIMULATION :
15701556 # Simulate job started as well.
1571- self .data_store_mgr .delta_job_time (
1572- job_tokens ,
1573- 'started' ,
1574- event_time ,
1575- )
1557+ self .data_store_mgr .delta_job_time (itask , 'started' , event_time )
15761558 else :
1577- self .data_store_mgr .delta_job_state (
1578- job_tokens ,
1579- TASK_STATUS_SUBMITTED ,
1580- )
1559+ self .data_store_mgr .delta_job_state (itask , TASK_STATUS_SUBMITTED )
15811560
15821561 def _insert_task_job (
15831562 self ,
0 commit comments