extern void reconfig()
{
read_slurmdbd_conf();
assoc_mgr_set_missing_uids();
acct_storage_g_reconfig(NULL, 0);
_update_logging(false);
}
/* _rollup_handler - Process rollup duties */
static void *_rollup_handler(void *db_conn)
{
time_t start_time = time(NULL);
time_t next_time;
/* int sigarray[] = {SIGUSR1, 0}; */
struct tm tm;
(void) pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL);
(void) pthread_setcanceltype(PTHREAD_CANCEL_ASYNCHRONOUS, NULL);
if (!localtime_r(&start_time, &tm)) {
fatal("Couldn't get localtime for rollup handler %ld",
(long)start_time);
return NULL;
}
while (1) {
if (!db_conn)
break;
/* run the roll up */
slurm_mutex_lock(&rollup_lock);
running_rollup = 1;
debug2("running rollup at %s", slurm_ctime(&start_time));
acct_storage_g_roll_usage(db_conn, 0, 0, 1);
acct_storage_g_commit(db_conn, 1);
running_rollup = 0;
slurm_mutex_unlock(&rollup_lock);
/* sleep for an hour */
tm.tm_sec = 0;
tm.tm_min = 0;
tm.tm_hour++;
tm.tm_isdst = -1;
next_time = mktime(&tm);
/* get the time now we have rolled usage */
start_time = time(NULL);
sleep((next_time-start_time));
start_time = time(NULL);
if (!localtime_r(&start_time, &tm)) {
fatal("Couldn't get localtime for rollup handler %ld",
(long)start_time);
return NULL;
}
/* Just in case some new uids were added to the system
pick them up here. */
assoc_mgr_set_missing_uids();
/* repeat ;) */
}
return NULL;
}
/* _signal_handler - Process daemon-wide signals */
static void *_signal_handler(void *no_data)
{
int rc, sig;
int sig_array[] = {SIGINT, SIGTERM, SIGHUP, SIGABRT, 0};
sigset_t set;
(void) pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL);
(void) pthread_setcanceltype(PTHREAD_CANCEL_ASYNCHRONOUS, NULL);
/* Make sure no required signals are ignored (possibly inherited) */
_default_sigaction(SIGINT);
_default_sigaction(SIGTERM);
_default_sigaction(SIGHUP);
_default_sigaction(SIGABRT);
while (1) {
xsignal_sigset_create(sig_array, &set);
rc = sigwait(&set, &sig);
if (rc == EINTR)
continue;
switch (sig) {
case SIGHUP: /* kill -1 */
info("Reconfigure signal (SIGHUP) received");
read_slurmdbd_conf();
assoc_mgr_set_missing_uids();
acct_storage_g_reconfig(NULL);
_update_logging(false);
break;
case SIGINT: /* kill -2 or <CTRL-C> */
case SIGTERM: /* kill -15 */
info("Terminate signal (SIGINT or SIGTERM) received");
shutdown_threads();
return NULL; /* Normal termination */
case SIGABRT: /* abort */
info("SIGABRT received");
abort(); /* Should terminate here */
shutdown_threads();
return NULL;
default:
error("Invalid signal (%d) received", sig);
}
}
}
/* _rollup_handler - Process rollup duties */
static void *_rollup_handler(void *db_conn)
{
time_t start_time = time(NULL);
time_t next_time;
/* int sigarray[] = {SIGUSR1, 0}; */
struct tm tm;
rollup_stats_t rollup_stats;
int i;
(void) pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL);
(void) pthread_setcanceltype(PTHREAD_CANCEL_ASYNCHRONOUS, NULL);
if (!slurm_localtime_r(&start_time, &tm)) {
fatal("Couldn't get localtime for rollup handler %ld",
(long)start_time);
return NULL;
}
while (1) {
if (!db_conn)
break;
/* run the roll up */
memset(&rollup_stats, 0, sizeof(rollup_stats_t));
slurm_mutex_lock(&rollup_lock);
running_rollup = 1;
debug2("running rollup at %s", slurm_ctime2(&start_time));
acct_storage_g_roll_usage(db_conn, 0, 0, 1, &rollup_stats);
acct_storage_g_commit(db_conn, 1);
running_rollup = 0;
slurm_mutex_unlock(&rollup_lock);
slurm_mutex_lock(&rpc_mutex);
for (i = 0; i < ROLLUP_COUNT; i++) {
if (rollup_stats.rollup_time[i] == 0)
continue;
rpc_stats.rollup_count[i]++;
rpc_stats.rollup_time[i] +=
rollup_stats.rollup_time[i];
rpc_stats.rollup_max_time[i] =
MAX(rpc_stats.rollup_max_time[i],
rollup_stats.rollup_time[i]);
}
slurm_mutex_unlock(&rpc_mutex);
/* get the time now we have rolled usage */
start_time = time(NULL);
if (!slurm_localtime_r(&start_time, &tm)) {
fatal("Couldn't get localtime for rollup handler %ld",
(long)start_time);
return NULL;
}
/* sleep until the next hour */
tm.tm_sec = 0;
tm.tm_min = 0;
tm.tm_hour++;
tm.tm_isdst = -1;
next_time = slurm_mktime(&tm);
sleep((next_time - start_time));
start_time = next_time;
/* Just in case some new uids were added to the system
pick them up here. */
assoc_mgr_set_missing_uids();
/* repeat ;) */
}
return NULL;
}
