/* * get info from the storage * IN: slurmdb_qos_cond_t * * RET: List of slurmdb_qos_rec_t * * note List needs to be freed with slurm_list_destroy() when called */ extern List slurmdb_qos_get(void *db_conn, slurmdb_qos_cond_t *qos_cond) { if (db_api_uid == -1) db_api_uid = getuid(); return acct_storage_g_get_qos(db_conn, db_api_uid, qos_cond); }
extern int sacctmgr_list_cluster(int argc, char *argv[]) { int rc = SLURM_SUCCESS; slurmdb_cluster_cond_t *cluster_cond = xmalloc(sizeof(slurmdb_cluster_cond_t)); List cluster_list; int i=0; ListIterator itr = NULL; ListIterator itr2 = NULL; slurmdb_cluster_rec_t *cluster = NULL; char *tmp_char = NULL; int field_count = 0; print_field_t *field = NULL; List format_list = list_create(slurm_destroy_char); List print_fields_list; /* types are of print_field_t */ slurmdb_init_cluster_cond(cluster_cond, 0); cluster_cond->cluster_list = list_create(slurm_destroy_char); for (i=0; i<argc; i++) { int command_len = strlen(argv[i]); if (!strncasecmp(argv[i], "Where", MAX(command_len, 5)) || !strncasecmp(argv[i], "Set", MAX(command_len, 3))) i++; _set_cond(&i, argc, argv, cluster_cond, format_list); } if(exit_code) { slurmdb_destroy_cluster_cond(cluster_cond); list_destroy(format_list); return SLURM_ERROR; } if(!list_count(format_list)) { slurm_addto_char_list(format_list, "Cl,Controlh,Controlp,RPC"); if(!without_limits) slurm_addto_char_list(format_list, "Fa,GrpJ,GrpN,GrpS,MaxJ,MaxN," "MaxS,MaxW,QOS,DefaultQOS"); } cluster_cond->with_deleted = with_deleted; print_fields_list = sacctmgr_process_format_list(format_list); list_destroy(format_list); if(exit_code) { slurmdb_destroy_cluster_cond(cluster_cond); list_destroy(print_fields_list); return SLURM_ERROR; } cluster_list = acct_storage_g_get_clusters(db_conn, my_uid, cluster_cond); slurmdb_destroy_cluster_cond(cluster_cond); if(!cluster_list) { exit_code=1; fprintf(stderr, " Problem with query.\n"); list_destroy(print_fields_list); return SLURM_ERROR; } itr = list_iterator_create(cluster_list); itr2 = list_iterator_create(print_fields_list); print_fields_header(print_fields_list); field_count = list_count(print_fields_list); while((cluster = list_next(itr))) { int curr_inx = 1; slurmdb_association_rec_t *assoc = cluster->root_assoc; /* set up the working cluster rec so nodecnt's and node names * are handled correctly */ working_cluster_rec = cluster; while((field = list_next(itr2))) { switch(field->type) { case PRINT_CLUSTER: field->print_routine(field, cluster->name, (curr_inx == field_count)); break; case PRINT_CHOST: field->print_routine(field, cluster->control_host, (curr_inx == field_count)); break; case PRINT_CPORT: field->print_routine(field, cluster->control_port, (curr_inx == field_count)); break; case PRINT_CLASS: field->print_routine(field, get_classification_str( cluster-> classification), (curr_inx == field_count)); break; case PRINT_CPUS: { char tmp_char[9]; convert_num_unit((float)cluster->cpu_count, tmp_char, sizeof(tmp_char), UNIT_NONE); field->print_routine(field, tmp_char, (curr_inx == field_count)); break; } case PRINT_DQOS: if(!g_qos_list) { g_qos_list = acct_storage_g_get_qos( db_conn, my_uid, NULL); } tmp_char = slurmdb_qos_str(g_qos_list, assoc->def_qos_id); field->print_routine( field, tmp_char, (curr_inx == field_count)); break; case PRINT_FAIRSHARE: field->print_routine( field, assoc->shares_raw, (curr_inx == field_count)); break; case PRINT_FLAGS: { char *tmp_char = slurmdb_cluster_flags_2_str( cluster->flags); field->print_routine( field, tmp_char, (curr_inx == field_count)); xfree(tmp_char); break; } case PRINT_GRPC: field->print_routine(field, assoc->grp_cpus, (curr_inx == field_count)); break; case PRINT_GRPJ: field->print_routine(field, assoc->grp_jobs, (curr_inx == field_count)); break; case PRINT_GRPN: field->print_routine(field, assoc->grp_nodes, (curr_inx == field_count)); break; case PRINT_GRPS: field->print_routine(field, assoc->grp_submit_jobs, (curr_inx == field_count)); break; case PRINT_MAXCM: field->print_routine( field, assoc->max_cpu_mins_pj, (curr_inx == field_count)); break; case PRINT_MAXC: field->print_routine(field, assoc->max_cpus_pj, (curr_inx == field_count)); break; case PRINT_MAXJ: field->print_routine(field, assoc->max_jobs, (curr_inx == field_count)); break; case PRINT_MAXN: field->print_routine(field, assoc->max_nodes_pj, (curr_inx == field_count)); break; case PRINT_MAXS: field->print_routine(field, assoc->max_submit_jobs, (curr_inx == field_count)); break; case PRINT_MAXW: field->print_routine( field, assoc->max_wall_pj, (curr_inx == field_count)); break; case PRINT_NODECNT: { hostlist_t hl = hostlist_create(cluster->nodes); int cnt = 0; if(hl) { cnt = hostlist_count(hl); hostlist_destroy(hl); } field->print_routine( field, cnt, (curr_inx == field_count)); break; } case PRINT_CLUSTER_NODES: field->print_routine( field, cluster->nodes, (curr_inx == field_count)); break; case PRINT_QOS: if(!g_qos_list) g_qos_list = acct_storage_g_get_qos( db_conn, my_uid, NULL); field->print_routine(field, g_qos_list, assoc->qos_list, (curr_inx == field_count)); break; case PRINT_QOS_RAW: field->print_routine(field, assoc->qos_list, (curr_inx == field_count)); break; case PRINT_RPC_VERSION: field->print_routine( field, cluster->rpc_version, (curr_inx == field_count)); break; case PRINT_SELECT: field->print_routine( field, cluster->plugin_id_select, (curr_inx == field_count)); break; default: field->print_routine( field, NULL, (curr_inx == field_count)); break; } curr_inx++; } list_iterator_reset(itr2); printf("\n"); } /* clear the working cluster rec */ working_cluster_rec = NULL; list_iterator_destroy(itr2); list_iterator_destroy(itr); list_destroy(cluster_list); list_destroy(print_fields_list); return rc; }
extern int sacctmgr_list_qos(int argc, char *argv[]) { int rc = SLURM_SUCCESS; slurmdb_qos_cond_t *qos_cond = xmalloc(sizeof(slurmdb_qos_cond_t)); int i=0; ListIterator itr = NULL; ListIterator itr2 = NULL; slurmdb_qos_rec_t *qos = NULL; List qos_list = NULL; int field_count = 0; print_field_t *field = NULL; List format_list = list_create(slurm_destroy_char); List print_fields_list; /* types are of print_field_t */ for (i=0; i<argc; i++) { int command_len = strlen(argv[i]); if (!strncasecmp (argv[i], "Where", MAX(command_len, 5)) || !strncasecmp (argv[i], "Set", MAX(command_len, 3))) i++; _set_cond(&i, argc, argv, qos_cond, format_list); } if(exit_code) { slurmdb_destroy_qos_cond(qos_cond); list_destroy(format_list); return SLURM_ERROR; } else if(!list_count(format_list)) { slurm_addto_char_list(format_list, "Name,Prio,GraceT,Preempt,PreemptM," "Flags%40,UsageThres,UsageFactor," "GrpCPUs,GrpCPUMins,GrpCPURunMins," "GrpJ,GrpMEM,GrpN,GrpS,GrpW," "MaxCPUs,MaxCPUMins,MaxN,MaxW," "MaxCPUsPerUser," "MaxJobsPerUser,MaxNodesPerUser," "MaxSubmitJobsPerUser"); } print_fields_list = sacctmgr_process_format_list(format_list); list_destroy(format_list); if(exit_code) { list_destroy(print_fields_list); return SLURM_ERROR; } qos_list = acct_storage_g_get_qos(db_conn, my_uid, qos_cond); slurmdb_destroy_qos_cond(qos_cond); if(!qos_list) { exit_code=1; fprintf(stderr, " Problem with query.\n"); list_destroy(print_fields_list); return SLURM_ERROR; } itr = list_iterator_create(qos_list); itr2 = list_iterator_create(print_fields_list); print_fields_header(print_fields_list); field_count = list_count(print_fields_list); while((qos = list_next(itr))) { int curr_inx = 1; while((field = list_next(itr2))) { switch(field->type) { case PRINT_DESC: field->print_routine( field, qos->description, (curr_inx == field_count)); break; case PRINT_FLAGS: { char *tmp_char = slurmdb_qos_flags_str( qos->flags); field->print_routine( field, tmp_char, (curr_inx == field_count)); xfree(tmp_char); break; } case PRINT_UT: field->print_routine( field, qos->usage_thres, (curr_inx == field_count)); break; case PRINT_GRACE: field->print_routine( field, (uint64_t)qos->grace_time, (curr_inx == field_count)); break; case PRINT_GRPCM: field->print_routine( field, qos->grp_cpu_mins, (curr_inx == field_count)); break; case PRINT_GRPCRM: field->print_routine( field, qos->grp_cpu_run_mins, (curr_inx == field_count)); break; case PRINT_GRPC: field->print_routine(field, qos->grp_cpus, (curr_inx == field_count)); break; case PRINT_GRPJ: field->print_routine(field, qos->grp_jobs, (curr_inx == field_count)); break; case PRINT_GRPMEM: field->print_routine(field, qos->grp_mem, (curr_inx == field_count)); break; case PRINT_GRPN: field->print_routine(field, qos->grp_nodes, (curr_inx == field_count)); break; case PRINT_GRPS: field->print_routine(field, qos->grp_submit_jobs, (curr_inx == field_count)); break; case PRINT_GRPW: field->print_routine( field, qos->grp_wall, (curr_inx == field_count)); break; case PRINT_ID: field->print_routine( field, qos->id, (curr_inx == field_count)); break; case PRINT_MAXCM: field->print_routine( field, qos->max_cpu_mins_pj, (curr_inx == field_count)); break; case PRINT_MAXCRM: field->print_routine( field, qos->max_cpu_run_mins_pu, (curr_inx == field_count)); break; case PRINT_MAXC: field->print_routine(field, qos->max_cpus_pj, (curr_inx == field_count)); break; case PRINT_MAXCU: field->print_routine(field, qos->max_cpus_pu, (curr_inx == field_count)); break; case PRINT_MAXJ: field->print_routine(field, qos->max_jobs_pu, (curr_inx == field_count)); break; case PRINT_MAXN: field->print_routine(field, qos->max_nodes_pj, (curr_inx == field_count)); break; case PRINT_MAXNU: field->print_routine(field, qos->max_nodes_pu, (curr_inx == field_count)); break; case PRINT_MAXS: field->print_routine(field, qos->max_submit_jobs_pu, (curr_inx == field_count)); break; case PRINT_MAXW: field->print_routine( field, qos->max_wall_pj, (curr_inx == field_count)); break; case PRINT_NAME: field->print_routine( field, qos->name, (curr_inx == field_count)); break; case PRINT_PREE: if(!g_qos_list) g_qos_list = acct_storage_g_get_qos( db_conn, my_uid, NULL); field->print_routine( field, g_qos_list, qos->preempt_bitstr, (curr_inx == field_count)); break; case PRINT_PREEM: { char *tmp_char = "cluster"; if(qos->preempt_mode) tmp_char = xstrtolower( preempt_mode_string( qos->preempt_mode)); field->print_routine( field, tmp_char, (curr_inx == field_count)); break; } case PRINT_PRIO: field->print_routine( field, qos->priority, (curr_inx == field_count)); break; case PRINT_UF: field->print_routine( field, qos->usage_factor, (curr_inx == field_count)); break; default: field->print_routine( field, NULL, (curr_inx == field_count)); break; } curr_inx++; } list_iterator_reset(itr2); printf("\n"); } list_iterator_destroy(itr2); list_iterator_destroy(itr); list_destroy(qos_list); list_destroy(print_fields_list); return rc; }
extern int sacctmgr_add_qos(int argc, char *argv[]) { int rc = SLURM_SUCCESS; int i=0, limit_set=0; ListIterator itr = NULL; slurmdb_qos_rec_t *qos = NULL; slurmdb_qos_rec_t *start_qos = xmalloc(sizeof(slurmdb_qos_rec_t)); List name_list = list_create(slurm_destroy_char); char *description = NULL; char *name = NULL; List qos_list = NULL; char *qos_str = NULL; slurmdb_init_qos_rec(start_qos, 0); for (i=0; i<argc; i++) { int command_len = strlen(argv[i]); if (!strncasecmp (argv[i], "Where", MAX(command_len, 5)) || !strncasecmp (argv[i], "Set", MAX(command_len, 3))) i++; limit_set += _set_rec(&i, argc, argv, name_list, start_qos); } if(exit_code) { list_destroy(name_list); xfree(description); return SLURM_ERROR; } else if(!list_count(name_list)) { list_destroy(name_list); slurmdb_destroy_qos_rec(start_qos); exit_code=1; fprintf(stderr, " Need name of qos to add.\n"); return SLURM_SUCCESS; } if(!g_qos_list) { g_qos_list = acct_storage_g_get_qos(db_conn, my_uid, NULL); if(!g_qos_list) { exit_code=1; fprintf(stderr, " Problem getting qos's " "from database. " "Contact your admin.\n"); list_destroy(name_list); xfree(description); return SLURM_ERROR; } } qos_list = list_create(slurmdb_destroy_qos_rec); itr = list_iterator_create(name_list); while((name = list_next(itr))) { qos = NULL; if(!sacctmgr_find_qos_from_list(g_qos_list, name)) { qos = xmalloc(sizeof(slurmdb_qos_rec_t)); slurmdb_init_qos_rec(qos, 0); qos->name = xstrdup(name); if(start_qos->description) qos->description = xstrdup(start_qos->description); else qos->description = xstrdup(name); qos->grace_time = start_qos->grace_time; qos->grp_cpu_mins = start_qos->grp_cpu_mins; qos->grp_cpus = start_qos->grp_cpus; qos->grp_jobs = start_qos->grp_jobs; qos->grp_mem = start_qos->grp_mem; qos->grp_nodes = start_qos->grp_nodes; qos->grp_submit_jobs = start_qos->grp_submit_jobs; qos->grp_wall = start_qos->grp_wall; qos->max_cpu_mins_pj = start_qos->max_cpu_mins_pj; qos->max_cpu_run_mins_pu = start_qos->max_cpu_run_mins_pu; qos->max_cpus_pj = start_qos->max_cpus_pj; qos->max_cpus_pu = start_qos->max_cpus_pu; qos->max_jobs_pu = start_qos->max_jobs_pu; qos->max_nodes_pj = start_qos->max_nodes_pj; qos->max_nodes_pu = start_qos->max_nodes_pu; qos->max_submit_jobs_pu = start_qos->max_submit_jobs_pu; qos->max_wall_pj = start_qos->max_wall_pj; qos->preempt_list = copy_char_list(start_qos->preempt_list); qos->preempt_mode = start_qos->preempt_mode; qos->priority = start_qos->priority; qos->usage_factor = start_qos->usage_factor; qos->usage_thres = start_qos->usage_thres; xstrfmtcat(qos_str, " %s\n", name); list_append(qos_list, qos); } } list_iterator_destroy(itr); list_destroy(name_list); if (g_qos_list) { list_destroy(g_qos_list); g_qos_list = NULL; } if(!list_count(qos_list)) { printf(" Nothing new added.\n"); goto end_it; } if(qos_str) { printf(" Adding QOS(s)\n%s", qos_str); printf(" Settings\n"); if(description) printf(" Description = %s\n", description); else printf(" Description = %s\n", "QOS Name"); sacctmgr_print_qos_limits(start_qos); xfree(qos_str); } notice_thread_init(); if(list_count(qos_list)) rc = acct_storage_g_add_qos(db_conn, my_uid, qos_list); else goto end_it; notice_thread_fini(); if(rc == SLURM_SUCCESS) { if(commit_check("Would you like to commit changes?")) { acct_storage_g_commit(db_conn, 1); } else { printf(" Changes Discarded\n"); acct_storage_g_commit(db_conn, 0); } } else { exit_code=1; fprintf(stderr, " Problem adding QOS: %s\n", slurm_strerror(rc)); rc = SLURM_ERROR; } end_it: list_destroy(qos_list); xfree(description); return rc; }
static int _set_rec(int *start, int argc, char *argv[], List name_list, slurmdb_qos_rec_t *qos) { int i, mins; int set = 0; int end = 0; int command_len = 0; int option = 0; for (i=(*start); i<argc; i++) { end = parse_option_end(argv[i]); if(!end) command_len=strlen(argv[i]); else { command_len=end-1; if(argv[i][end] == '=') { option = (int)argv[i][end-1]; end++; } } if (!strncasecmp (argv[i], "Where", MAX(command_len, 5))) { i--; break; } else if(!end && !strncasecmp(argv[i], "set", MAX(command_len, 3))) { continue; } else if(!end || !strncasecmp (argv[i], "Name", MAX(command_len, 1))) { if(name_list) slurm_addto_char_list(name_list, argv[i]+end); } else if (!strncasecmp (argv[i], "Description", MAX(command_len, 1))) { if(!qos->description) qos->description = strip_quotes(argv[i]+end, NULL, 1); set = 1; } else if (!strncasecmp (argv[i], "Flags", MAX(command_len, 2))) { if(!qos) continue; qos->flags = str_2_qos_flags(argv[i]+end, option); if (qos->flags == QOS_FLAG_NOTSET) { char *tmp_char = NULL; qos->flags = INFINITE; qos->flags &= (~QOS_FLAG_NOTSET & ~QOS_FLAG_ADD & ~QOS_FLAG_REMOVE); tmp_char = slurmdb_qos_flags_str(qos->flags); printf(" Unknown QOS flag used in:\n '%s'\n" " Valid QOS flags are\n '%s'\n", argv[i]+end, tmp_char); xfree(tmp_char); exit_code = 1; } else set = 1; } else if (!strncasecmp (argv[i], "GraceTime", MAX(command_len, 3))) { if (!qos) continue; if (get_uint(argv[i]+end, &qos->grace_time, "GraceTime") == SLURM_SUCCESS) { set = 1; } } else if (!strncasecmp (argv[i], "GrpCPUMins", MAX(command_len, 7))) { if(!qos) continue; if (get_uint64(argv[i]+end, &qos->grp_cpu_mins, "GrpCPUMins") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "GrpCPURunMins", MAX(command_len, 7))) { if(!qos) continue; if (get_uint64(argv[i]+end, &qos->grp_cpu_run_mins, "GrpCPURunMins") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "GrpCPUs", MAX(command_len, 7))) { if(!qos) continue; if (get_uint(argv[i]+end, &qos->grp_cpus, "GrpCPUs") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "GrpJobs", MAX(command_len, 4))) { if(!qos) continue; if (get_uint(argv[i]+end, &qos->grp_jobs, "GrpJobs") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "GrpMemory", MAX(command_len, 4))) { if(!qos) continue; if (get_uint(argv[i]+end, &qos->grp_mem, "GrpMemory") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "GrpNodes", MAX(command_len, 4))) { if(!qos) continue; if (get_uint(argv[i]+end, &qos->grp_nodes, "GrpNodes") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "GrpSubmitJobs", MAX(command_len, 4))) { if(!qos) continue; if (get_uint(argv[i]+end, &qos->grp_submit_jobs, "GrpSubmitJobs") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "GrpWall", MAX(command_len, 4))) { if(!qos) continue; mins = time_str2mins(argv[i]+end); if (mins != NO_VAL) { qos->grp_wall = (uint32_t) mins; set = 1; } else { exit_code=1; fprintf(stderr, " Bad GrpWall time format: %s\n", argv[i]); } } else if (!strncasecmp (argv[i], "MaxCPUMinsPerJob", MAX(command_len, 7))) { if(!qos) continue; if (get_uint64(argv[i]+end, &qos->max_cpu_mins_pj, "MaxCPUMins") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "MaxCPUsPerJob", MAX(command_len, 7))) { if(!qos) continue; if (get_uint(argv[i]+end, &qos->max_cpus_pj, "MaxCPUs") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "MaxCPUsPerUser", MAX(command_len, 11))) { if(!qos) continue; if (get_uint(argv[i]+end, &qos->max_cpus_pu, "MaxCPUsPerUser") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "MaxJobsPerUser", MAX(command_len, 4))) { if(!qos) continue; if (get_uint(argv[i]+end, &qos->max_jobs_pu, "MaxJobs") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "MaxNodesPerJob", MAX(command_len, 4))) { if(!qos) continue; if (get_uint(argv[i]+end, &qos->max_nodes_pj, "MaxNodes") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "MaxNodesPerUser", MAX(command_len, 8))) { if(!qos) continue; if (get_uint(argv[i]+end, &qos->max_nodes_pu, "MaxNodesPerUser") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "MaxSubmitJobsPerUser", MAX(command_len, 4))) { if(!qos) continue; if (get_uint(argv[i]+end, &qos->max_submit_jobs_pu, "MaxSubmitJobs") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "MaxWallDurationPerJob", MAX(command_len, 4))) { if(!qos) continue; mins = time_str2mins(argv[i]+end); if (mins != NO_VAL) { qos->max_wall_pj = (uint32_t) mins; set = 1; } else { exit_code=1; fprintf(stderr, " Bad MaxWall time format: %s\n", argv[i]); } } else if (!strncasecmp (argv[i], "PreemptMode", MAX(command_len, 8))) { if(!qos) continue; qos->preempt_mode = preempt_mode_num(argv[i]+end); if(qos->preempt_mode == (uint16_t)NO_VAL) { fprintf(stderr, " Bad Preempt Mode given: %s\n", argv[i]); exit_code = 1; } else if (qos->preempt_mode == PREEMPT_MODE_SUSPEND) { printf("PreemptType and PreemptMode " "values incompatible\n"); exit_code = 1; } else set = 1; /* Preempt needs to follow PreemptMode */ } else if (!strncasecmp (argv[i], "Preempt", MAX(command_len, 7))) { if(!qos) continue; if(!qos->preempt_list) qos->preempt_list = list_create(slurm_destroy_char); if(!g_qos_list) g_qos_list = acct_storage_g_get_qos( db_conn, my_uid, NULL); if(slurmdb_addto_qos_char_list(qos->preempt_list, g_qos_list, argv[i]+end, option)) set = 1; else exit_code = 1; } else if (!strncasecmp (argv[i], "Priority", MAX(command_len, 3))) { if(!qos) continue; if (get_uint(argv[i]+end, &qos->priority, "Priority") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "UsageFactor", MAX(command_len, 6))) { if(!qos) continue; if (get_double(argv[i]+end, &qos->usage_factor, "UsageFactor") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp (argv[i], "UsageThreshold", MAX(command_len, 6))) { if(!qos) continue; if (get_double(argv[i]+end, &qos->usage_thres, "UsageThreshold") == SLURM_SUCCESS) set = 1; } else { exit_code = 1; printf(" Unknown option: %s\n" " Use keyword 'where' to modify condition\n", argv[i]); } } (*start) = i; return set; }
extern int sacctmgr_delete_qos(int argc, char *argv[]) { int rc = SLURM_SUCCESS; slurmdb_qos_cond_t *qos_cond = xmalloc(sizeof(slurmdb_qos_cond_t)); int i=0; List ret_list = NULL; int set = 0; for (i=0; i<argc; i++) { int command_len = strlen(argv[i]); if (!strncasecmp (argv[i], "Where", MAX(command_len, 5)) || !strncasecmp (argv[i], "Set", MAX(command_len, 3))) i++; set += _set_cond(&i, argc, argv, qos_cond, NULL); } if(!set) { exit_code=1; fprintf(stderr, " No conditions given to remove, not executing.\n"); slurmdb_destroy_qos_cond(qos_cond); return SLURM_ERROR; } else if(set == -1) { slurmdb_destroy_qos_cond(qos_cond); return SLURM_ERROR; } if (!g_qos_list) g_qos_list = acct_storage_g_get_qos( db_conn, my_uid, NULL); notice_thread_init(); ret_list = acct_storage_g_remove_qos(db_conn, my_uid, qos_cond); notice_thread_fini(); slurmdb_destroy_qos_cond(qos_cond); if(ret_list && list_count(ret_list)) { char *object = NULL; ListIterator itr = NULL; /* Check to see if person is trying to remove a default * qos of an association. _isdefault only works with the * output from acct_storage_g_remove_qos, and * with a previously got g_qos_list. */ if (_isdefault(ret_list)) { exit_code=1; fprintf(stderr, " Please either remove the qos' listed " "above from list and resubmit,\n" " or change the default qos to " "remove the qos.\n" " Changes Discarded\n"); acct_storage_g_commit(db_conn, 0); goto end_it; } itr = list_iterator_create(ret_list); printf(" Deleting QOS(s)...\n"); while((object = list_next(itr))) { printf(" %s\n", object); } list_iterator_destroy(itr); if(commit_check("Would you like to commit changes?")) { acct_storage_g_commit(db_conn, 1); } else { printf(" Changes Discarded\n"); acct_storage_g_commit(db_conn, 0); } } else if(ret_list) { printf(" Nothing deleted\n"); } else { exit_code=1; fprintf(stderr, " Error with request: %s\n", slurm_strerror(errno)); rc = SLURM_ERROR; } end_it: if(ret_list) list_destroy(ret_list); return rc; }
/* * get info from the storage * IN: slurmdb_qos_cond_t * * RET: List of slurmdb_qos_rec_t * * note List needs to be freed with slurm_list_destroy() when called */ extern List slurmdb_qos_get(void *db_conn, slurmdb_qos_cond_t *qos_cond) { return acct_storage_g_get_qos(db_conn, getuid(), qos_cond); }
extern void sacctmgr_print_association_rec(slurmdb_association_rec_t *assoc, print_field_t *field, List tree_list, bool last) { char *print_acct = NULL; char *tmp_char = NULL; xassert(field); if(!assoc) { field->print_routine(field, NULL, last); return; } switch(field->type) { case PRINT_ACCT: if (tree_display) { char *local_acct = NULL; char *parent_acct = NULL; if (assoc->user) { local_acct = xstrdup_printf("|%s", assoc->acct); parent_acct = assoc->acct; } else { local_acct = xstrdup(assoc->acct); parent_acct = assoc->parent_acct; } print_acct = slurmdb_tree_name_get( local_acct, parent_acct, tree_list); xfree(local_acct); } else { print_acct = assoc->acct; } field->print_routine(field, print_acct, last); break; case PRINT_CLUSTER: field->print_routine(field, assoc->cluster, last); break; case PRINT_DQOS: if (!g_qos_list) g_qos_list = acct_storage_g_get_qos( db_conn, my_uid, NULL); tmp_char = slurmdb_qos_str(g_qos_list, assoc->def_qos_id); field->print_routine(field, tmp_char, last); break; case PRINT_FAIRSHARE: if (assoc->shares_raw == SLURMDB_FS_USE_PARENT) print_fields_str(field, "parent", last); else field->print_routine(field, assoc->shares_raw, last); break; case PRINT_GRPCM: field->print_routine(field, assoc->grp_cpu_mins, last); break; case PRINT_GRPCRM: field->print_routine(field, assoc->grp_cpu_run_mins, last); break; case PRINT_GRPC: field->print_routine(field, assoc->grp_cpus, last); break; case PRINT_GRPJ: field->print_routine(field, assoc->grp_jobs, last); break; case PRINT_GRPN: field->print_routine(field, assoc->grp_nodes, last); break; case PRINT_GRPS: field->print_routine(field, assoc->grp_submit_jobs, last); break; case PRINT_GRPW: field->print_routine(field, assoc->grp_wall, last); break; case PRINT_ID: field->print_routine(field, assoc->id, last); break; case PRINT_LFT: field->print_routine(field, assoc->lft, last); break; case PRINT_MAXCM: field->print_routine(field, assoc->max_cpu_mins_pj, last); break; case PRINT_MAXCRM: field->print_routine(field, assoc->max_cpu_run_mins, last); break; case PRINT_MAXC: field->print_routine(field, assoc->max_cpus_pj, last); break; case PRINT_MAXJ: field->print_routine(field, assoc->max_jobs, last); break; case PRINT_MAXN: field->print_routine(field, assoc->max_nodes_pj, last); break; case PRINT_MAXS: field->print_routine(field, assoc->max_submit_jobs, last); break; case PRINT_MAXW: field->print_routine(field, assoc->max_wall_pj, last); break; case PRINT_PID: field->print_routine(field, assoc->parent_id, last); break; case PRINT_PNAME: field->print_routine(field, assoc->parent_acct, last); break; case PRINT_PART: field->print_routine(field, assoc->partition, last); break; case PRINT_QOS: if (!g_qos_list) g_qos_list = acct_storage_g_get_qos( db_conn, my_uid, NULL); field->print_routine(field, g_qos_list, assoc->qos_list, last); break; case PRINT_QOS_RAW: field->print_routine(field, assoc->qos_list, last); break; case PRINT_RGT: field->print_routine(field, assoc->rgt, last); break; case PRINT_USER: field->print_routine(field, assoc->user, last); break; default: field->print_routine(field, NULL, last); break; } }
extern int sacctmgr_set_association_rec(slurmdb_association_rec_t *assoc, char *type, char *value, int command_len, int option) { int set = 0; uint32_t mins = NO_VAL; if (!assoc) return set; if (!strncasecmp (type, "DefaultQOS", MAX(command_len, 8))) { if(!g_qos_list) g_qos_list = acct_storage_g_get_qos( db_conn, my_uid, NULL); if(atoi(value) == -1) assoc->def_qos_id = -1; else assoc->def_qos_id = str_2_slurmdb_qos( g_qos_list, value); if(assoc->def_qos_id == NO_VAL) { fprintf(stderr, "You gave a bad default qos '%s'. " "Use 'list qos' to get " "complete list.\n", value); exit_code = 1; } set = 1; } else if (!strncasecmp(type, "FairShare", MAX(command_len, 1)) || !strncasecmp(type, "Shares", MAX(command_len, 1))) { if (!strncasecmp(value, "parent", 6)) { assoc->shares_raw = SLURMDB_FS_USE_PARENT; set = 1; } else if (get_uint(value, &assoc->shares_raw, "FairShare") == SLURM_SUCCESS) { set = 1; } } else if (!strncasecmp(type, "GrpCPUMins", MAX(command_len, 7))) { if (get_uint64(value, &assoc->grp_cpu_mins, "GrpCPUMins") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp(type, "GrpCPURunMins", MAX(command_len, 7))) { if (get_uint64(value, &assoc->grp_cpu_run_mins, "GrpCPURunMins") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp(type, "GrpCpus", MAX(command_len, 7))) { if (get_uint(value, &assoc->grp_cpus, "GrpCpus") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp(type, "GrpJobs", MAX(command_len, 4))) { if (get_uint(value, &assoc->grp_jobs, "GrpJobs") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp(type, "GrpNodes", MAX(command_len, 4))) { if (get_uint(value, &assoc->grp_nodes, "GrpNodes") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp(type, "GrpSubmitJobs", MAX(command_len, 4))) { if (get_uint(value, &assoc->grp_submit_jobs, "GrpSubmitJobs") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp(type, "GrpWall", MAX(command_len, 4))) { mins = time_str2mins(value); if (mins != NO_VAL) { assoc->grp_wall = mins; set = 1; } else { exit_code=1; fprintf(stderr, " Bad GrpWall time format: %s\n", type); } } else if (!strncasecmp(type, "MaxCPUMinsPerJob", MAX(command_len, 7))) { if (get_uint64(value, &assoc->max_cpu_mins_pj, "MaxCPUMins") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp(type, "MaxCPURunMins", MAX(command_len, 7))) { if (get_uint64(value, &assoc->max_cpu_run_mins, "MaxCPURunMins") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp(type, "MaxCpusPerJob", MAX(command_len, 7))) { if (get_uint(value, &assoc->max_cpus_pj, "MaxCpus") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp(type, "MaxJobs", MAX(command_len, 4))) { if (get_uint(value, &assoc->max_jobs, "MaxJobs") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp(type, "MaxNodesPerJob", MAX(command_len, 4))) { if (get_uint(value, &assoc->max_nodes_pj, "MaxNodes") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp(type, "MaxSubmitJobs", MAX(command_len, 4))) { if (get_uint(value, &assoc->max_submit_jobs, "MaxSubmitJobs") == SLURM_SUCCESS) set = 1; } else if (!strncasecmp(type, "MaxWallDurationPerJob", MAX(command_len, 4))) { mins = time_str2mins(value); if (mins != NO_VAL) { assoc->max_wall_pj = mins; set = 1; } else { exit_code=1; fprintf(stderr, " Bad MaxWall time format: %s\n", type); } } else if (!strncasecmp(type, "Parent", MAX(command_len, 1))) { assoc->parent_acct = strip_quotes(value, NULL, 1); set = 1; } else if (!strncasecmp(type, "QosLevel", MAX(command_len, 1))) { if (!assoc->qos_list) assoc->qos_list = list_create(slurm_destroy_char); if (!g_qos_list) g_qos_list = acct_storage_g_get_qos( db_conn, my_uid, NULL); if (slurmdb_addto_qos_char_list(assoc->qos_list, g_qos_list, value, option)) set = 1; } return set; }
extern int sacctmgr_set_association_cond(slurmdb_association_cond_t *assoc_cond, char *type, char *value, int command_len, int option) { int set =0; xassert(assoc_cond); xassert(type); if (!strncasecmp (type, "Accounts", MAX(command_len, 2)) || !strncasecmp (type, "Acct", MAX(command_len, 4))) { if (!assoc_cond->acct_list) assoc_cond->acct_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->acct_list, value)) set = 1; } else if (!strncasecmp(type, "Ids", MAX(command_len, 1)) || !strncasecmp(type, "Associations", MAX(command_len, 2))) { ListIterator itr = NULL; char *temp = NULL; uint32_t id = 0; if (!assoc_cond->id_list) assoc_cond->id_list =list_create(slurm_destroy_char); slurm_addto_char_list(assoc_cond->id_list, value); /* check to make sure user gave ints here */ itr = list_iterator_create(assoc_cond->id_list); while ((temp = list_next(itr))) { if (get_uint(temp, &id, "AssocId") != SLURM_SUCCESS) { exit_code = 1; list_delete_item(itr); } } list_iterator_destroy(itr); set = 1; } else if (!strncasecmp (type, "Clusters", MAX(command_len, 1))) { if (!assoc_cond->cluster_list) assoc_cond->cluster_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->cluster_list, value)) set = 1; } else if (!strncasecmp (type, "DefaultQOS", MAX(command_len, 8))) { if (!assoc_cond->def_qos_id_list) assoc_cond->def_qos_id_list = list_create(slurm_destroy_char); if (!g_qos_list) g_qos_list = acct_storage_g_get_qos( db_conn, my_uid, NULL); if (slurmdb_addto_qos_char_list(assoc_cond->def_qos_id_list, g_qos_list, value, 0)) set = 1; else exit_code = 1; } else if (!strncasecmp (type, "FairShare", MAX(command_len, 1)) || !strncasecmp (type, "Shares", MAX(command_len, 1))) { if (!assoc_cond->fairshare_list) assoc_cond->fairshare_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->fairshare_list, value)) set = 1; } else if (!strncasecmp (type, "GrpCPUMins", MAX(command_len, 7))) { if (!assoc_cond->grp_cpu_mins_list) assoc_cond->grp_cpu_mins_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->grp_cpu_mins_list, value)) set = 1; } else if (!strncasecmp (type, "GrpCPURunMins", MAX(command_len, 7))) { if (!assoc_cond->grp_cpu_run_mins_list) assoc_cond->grp_cpu_run_mins_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->grp_cpu_run_mins_list, value)) set = 1; } else if (!strncasecmp (type, "GrpCpus", MAX(command_len, 7))) { if (!assoc_cond->grp_cpus_list) assoc_cond->grp_cpus_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->grp_cpus_list, value)) set = 1; } else if (!strncasecmp (type, "GrpJobs", MAX(command_len, 4))) { if (!assoc_cond->grp_jobs_list) assoc_cond->grp_jobs_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->grp_jobs_list, value)) set = 1; } else if (!strncasecmp (type, "GrpNodes", MAX(command_len, 4))) { if (!assoc_cond->grp_nodes_list) assoc_cond->grp_nodes_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->grp_nodes_list, value)) set = 1; } else if (!strncasecmp (type, "GrpSubmitJobs", MAX(command_len, 4))) { if (!assoc_cond->grp_submit_jobs_list) assoc_cond->grp_submit_jobs_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->grp_submit_jobs_list, value)) set = 1; } else if (!strncasecmp (type, "GrpWall", MAX(command_len, 4))) { if (!assoc_cond->grp_wall_list) assoc_cond->grp_wall_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->grp_wall_list, value)) set = 1; } else if (!strncasecmp (type, "MaxCPUMinsPerJob", MAX(command_len, 7))) { if (!assoc_cond->max_cpu_mins_pj_list) assoc_cond->max_cpu_mins_pj_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->max_cpu_mins_pj_list, value)) set = 1; } else if (!strncasecmp (type, "MaxCPURunMins", MAX(command_len, 7))) { if (!assoc_cond->max_cpu_run_mins_list) assoc_cond->max_cpu_run_mins_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->max_cpu_run_mins_list, value)) set = 1; } else if (!strncasecmp (type, "MaxCpusPerJob", MAX(command_len, 7))) { if (!assoc_cond->max_cpus_pj_list) assoc_cond->max_cpus_pj_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->max_cpus_pj_list, value)) set = 1; } else if (!strncasecmp (type, "MaxJobs", MAX(command_len, 4))) { if (!assoc_cond->max_jobs_list) assoc_cond->max_jobs_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->max_jobs_list, value)) set = 1; } else if (!strncasecmp (type, "MaxNodesPerJob", MAX(command_len, 4))) { if (!assoc_cond->max_nodes_pj_list) assoc_cond->max_nodes_pj_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->max_nodes_pj_list, value)) set = 1; } else if (!strncasecmp (type, "MaxSubmitJobs", MAX(command_len, 4))) { if (!assoc_cond->max_submit_jobs_list) assoc_cond->max_submit_jobs_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->max_submit_jobs_list, value)) set = 1; } else if (!strncasecmp (type, "MaxWallDurationPerJob", MAX(command_len, 4))) { if (!assoc_cond->max_wall_pj_list) assoc_cond->max_wall_pj_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->max_wall_pj_list, value)) set = 1; } else if (!strncasecmp (type, "Partitions", MAX(command_len, 3))) { if (!assoc_cond->partition_list) assoc_cond->partition_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->partition_list, value)) set = 1; } else if (!strncasecmp(type, "Parents", MAX(command_len, 4))) { if (!assoc_cond->parent_acct_list) assoc_cond->parent_acct_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->parent_acct_list, value)) set = 1; } else if (!strncasecmp (type, "QosLevel", MAX(command_len, 1))) { if(!assoc_cond->qos_list) assoc_cond->qos_list = list_create(slurm_destroy_char); if(!g_qos_list) g_qos_list = acct_storage_g_get_qos( db_conn, my_uid, NULL); if(slurmdb_addto_qos_char_list(assoc_cond->qos_list, g_qos_list, value, option)) set = 1; } else if (!strncasecmp (type, "Users", MAX(command_len, 1))) { if (!assoc_cond->user_list) assoc_cond->user_list = list_create(slurm_destroy_char); if (slurm_addto_char_list(assoc_cond->user_list, value)) set = 1; } return set; }
extern bool sacctmgr_check_default_qos(uint32_t qos_id, slurmdb_association_cond_t *assoc_cond) { char *object = NULL; ListIterator itr; slurmdb_association_rec_t *assoc; List no_access_list = NULL; List assoc_list = NULL; if (qos_id == NO_VAL) return true; assoc_list = acct_storage_g_get_associations( db_conn, my_uid, assoc_cond); if (!assoc_list) { fprintf(stderr, "Couldn't get a list back for checking qos.\n"); return false; } if (!g_qos_list) g_qos_list = acct_storage_g_get_qos(db_conn, my_uid, NULL); itr = list_iterator_create(assoc_list); while ((assoc = list_next(itr))) { char *qos = NULL; if (assoc->qos_list) { int check_qos = qos_id; if (check_qos == -1) check_qos = assoc->def_qos_id; if ((int)check_qos > 0) { ListIterator qos_itr = list_iterator_create(assoc->qos_list); while ((qos = list_next(qos_itr))) { if (qos[0] == '-') continue; else if (qos[0] == '+') qos++; /* info("looking for %u ?= %u", */ /* check_qos, slurm_atoul(qos)); */ if (check_qos == slurm_atoul(qos)) break; } list_iterator_destroy(qos_itr); } else qos = ""; } if (!qos) { char *name = slurmdb_qos_str(g_qos_list, assoc->def_qos_id); if (!assoc->user) { // see if this isn't a user object = xstrdup_printf( " DefQOS = %-10s C = %-10s A = %-20s ", name, assoc->cluster, assoc->acct); } else if (assoc->partition) { // see if there is a partition name object = xstrdup_printf( " DefQOS = %-10s C = %-10s A = %-20s " "U = %-9s P = %s", name, assoc->cluster, assoc->acct, assoc->user, assoc->partition); } else { object = xstrdup_printf( " DefQOS = %-10s C = %-10s A = %-20s " "U = %-9s", name, assoc->cluster, assoc->acct, assoc->user); } if (!no_access_list) no_access_list = list_create(slurm_destroy_char); list_append(no_access_list, object); } } list_iterator_destroy(itr); list_destroy(assoc_list); if (!no_access_list) return true; fprintf(stderr, " These associations don't have access to their " "default qos.\n"); fprintf(stderr, " Please give them access before they the default " "can be set to this.\n"); itr = list_iterator_create(no_access_list); while ((object = list_next(itr))) fprintf(stderr, "%s\n", object); list_iterator_destroy(itr); list_destroy(no_access_list); return 0; }