extern int select_p_job_init(List job_list)
{
static bool run_already = false;
/* Execute only on initial startup. We don't support bgblock
* creation on demand today, so there is no need to re-sync data. */
if (run_already)
return other_job_init(job_list);
run_already = true;
if (!(slurmctld_conf.select_type_param & CR_NHC_NO)
&& job_list && list_count(job_list)) {
ListIterator itr = list_iterator_create(job_list);
struct job_record *job_ptr;
if (debug_flags & DEBUG_FLAG_SELECT_TYPE)
info("select_p_job_init: syncing jobs");
while ((job_ptr = list_next(itr))) {
select_jobinfo_t *jobinfo =
job_ptr->select_jobinfo->data;
if (!(slurmctld_conf.select_type_param & CR_NHC_STEP_NO)
&& job_ptr->step_list
&& list_count(job_ptr->step_list)) {
ListIterator itr_step = list_iterator_create(
job_ptr->step_list);
struct step_record *step_ptr;
while ((step_ptr = list_next(itr_step))) {
jobinfo =
step_ptr->select_jobinfo->data;
if (jobinfo && jobinfo->cleaning)
_spawn_cleanup_thread(
step_ptr, _step_fini);
}
list_iterator_destroy(itr_step);
}
jobinfo = job_ptr->select_jobinfo->data;
if (jobinfo && jobinfo->cleaning)
_spawn_cleanup_thread(job_ptr, _job_fini);
}
list_iterator_destroy(itr);
}
return other_job_init(job_list);
}
extern int select_p_step_finish(struct step_record *step_ptr)
{
select_jobinfo_t *jobinfo = step_ptr->select_jobinfo->data;
if (slurmctld_conf.select_type_param & CR_NHC_STEP_NO) {
debug3("NHC_No_Steps set not running NHC on steps.");
other_step_finish(step_ptr);
/* free resources on the job */
post_job_step(step_ptr);
return SLURM_SUCCESS;
}
/* The NHC needs to be ran after each step even if the job is
about to run the NHC for the allocation. The NHC
developers feel this is needed. If it ever changes just
remove the below commented code.
*/
/* else if (IS_JOB_COMPLETING(step_ptr->job_ptr)) { */
/* debug3("step completion %u.%u was received after job " */
/* "allocation is already completing, no extra NHC needed.", */
/* step_ptr->job_ptr->job_id, step_ptr->step_id); */
/* other_step_finish(step_ptr); */
/* /\* free resources on the job *\/ */
/* post_job_step(step_ptr); */
/* return SLURM_SUCCESS; */
/* } */
jobinfo->cleaning = 1;
_spawn_cleanup_thread(step_ptr, _step_fini);
return SLURM_SUCCESS;
}
extern int select_p_job_fini(struct job_record *job_ptr)
{
select_jobinfo_t *jobinfo = job_ptr->select_jobinfo->data;
jobinfo->cleaning = 1;
_spawn_cleanup_thread(job_ptr, _job_fini);
return SLURM_SUCCESS;
}
extern int select_p_job_init(List job_list)
{
if (job_list && list_count(job_list)) {
ListIterator itr = list_iterator_create(job_list);
struct job_record *job_ptr;
if (debug_flags & DEBUG_FLAG_SELECT_TYPE)
info("select_p_job_init: syncing jobs");
while ((job_ptr = list_next(itr))) {
select_jobinfo_t *jobinfo =
job_ptr->select_jobinfo->data;
if (!jobinfo->cleaning && job_ptr->step_list
&& list_count(job_ptr->step_list)) {
ListIterator itr_step = list_iterator_create(
job_ptr->step_list);
struct step_record *step_ptr;
while ((step_ptr = list_next(itr_step))) {
jobinfo =
step_ptr->select_jobinfo->data;
if (!jobinfo->cleaning)
continue;
_spawn_cleanup_thread(step_ptr,
_step_fini);
}
list_iterator_destroy(itr_step);
continue;
}
_spawn_cleanup_thread(job_ptr, _job_fini);
}
list_iterator_destroy(itr);
}
return other_job_init(job_list);
}
extern int select_p_job_fini(struct job_record *job_ptr)
{
select_jobinfo_t *jobinfo = job_ptr->select_jobinfo->data;
if (slurmctld_conf.select_type_param & CR_NHC_NO) {
debug3("NHC_No set, not running NHC after allocations");
other_job_fini(job_ptr);
return SLURM_SUCCESS;
}
jobinfo->cleaning = 1;
_spawn_cleanup_thread(job_ptr, _job_fini);
return SLURM_SUCCESS;
}
extern int select_p_step_finish(struct step_record *step_ptr)
{
select_jobinfo_t *jobinfo = step_ptr->select_jobinfo->data;
if (IS_JOB_COMPLETING(step_ptr->job_ptr)) {
debug3("step completion %u.%u was received after job "
"allocation is already completing, no extra NHC needed.",
step_ptr->job_ptr->job_id, step_ptr->step_id);
other_step_finish(step_ptr);
/* free resources on the job */
post_job_step(step_ptr);
return SLURM_SUCCESS;
}
jobinfo->cleaning = 1;
_spawn_cleanup_thread(step_ptr, _step_fini);
return SLURM_SUCCESS;
}
/*
* init() is called when the plugin is loaded, before any other functions
* are called. Put global initialization here.
*/
extern int init ( void )
{
/* We must call the api here since we call this from other
* things other than the slurmctld.
*/
uint16_t select_type_param = slurm_get_select_type_param();
if (select_type_param & CR_OTHER_CONS_RES)
plugin_id = 108;
debug_flags = slurm_get_debug_flags();
#ifdef HAVE_NATIVE_CRAY
// Spawn the aeld thread, only in slurmctld.
if (run_in_daemon("slurmctld")) {
_spawn_cleanup_thread(NULL, _aeld_event_loop);
}
#endif
verbose("%s loaded", plugin_name);
return SLURM_SUCCESS;
}
