static void test_handler_remove(void)
{
handler_add("http", "e");
g_assert_true(handler_remove("http"));
g_assert_false(handler_remove("http"));
}
static void
scheduler_msg_dispatch(void)
{
size_t n, sz, count;
struct evpstate evpstates[MAX_BATCH_SIZE];
uint64_t evpid, evpids[MAX_BATCH_SIZE], u64;
uint32_t msgids[MAX_BATCH_SIZE], version, msgid;
struct scheduler_info info;
int typemask, r, type, types[MAX_BATCH_SIZE];
int delay;
switch (imsg.hdr.type) {
case PROC_SCHEDULER_INIT:
log_debug("scheduler-api: PROC_SCHEDULER_INIT");
scheduler_msg_get(&version, sizeof(version));
scheduler_msg_end();
if (version != PROC_SCHEDULER_API_VERSION) {
log_warnx("warn: scheduler-api: bad API version");
fatalx("scheduler-api: exiting");
}
r = handler_init();
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r));
break;
case PROC_SCHEDULER_INSERT:
log_debug("scheduler-api: PROC_SCHEDULER_INSERT");
scheduler_msg_get(&info, sizeof(info));
scheduler_msg_end();
r = handler_insert(&info);
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r));
break;
case PROC_SCHEDULER_COMMIT:
log_debug("scheduler-api: PROC_SCHEDULER_COMMIT");
scheduler_msg_get(&msgid, sizeof(msgid));
scheduler_msg_end();
n = handler_commit(msgid);
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &n, sizeof(n));
break;
case PROC_SCHEDULER_ROLLBACK:
log_debug("scheduler-api: PROC_SCHEDULER_ROLLBACK");
scheduler_msg_get(&msgid, sizeof(msgid));
scheduler_msg_end();
n = handler_rollback(msgid);
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &n, sizeof(n));
break;
case PROC_SCHEDULER_UPDATE:
log_debug("scheduler-api: PROC_SCHEDULER_UPDATE");
scheduler_msg_get(&info, sizeof(info));
scheduler_msg_end();
r = handler_update(&info);
scheduler_msg_add(&r, sizeof(r));
if (r == 1)
scheduler_msg_add(&info, sizeof(info));
scheduler_msg_close();
break;
case PROC_SCHEDULER_DELETE:
log_debug("scheduler-api: PROC_SCHEDULER_DELETE");
scheduler_msg_get(&evpid, sizeof(evpid));
scheduler_msg_end();
r = handler_delete(evpid);
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r));
break;
case PROC_SCHEDULER_HOLD:
log_debug("scheduler-api: PROC_SCHEDULER_HOLD");
scheduler_msg_get(&evpid, sizeof(evpid));
scheduler_msg_get(&u64, sizeof(u64));
scheduler_msg_end();
r = handler_hold(evpid, u64);
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r));
break;
case PROC_SCHEDULER_RELEASE:
log_debug("scheduler-api: PROC_SCHEDULER_RELEASE");
scheduler_msg_get(&type, sizeof(type));
scheduler_msg_get(&u64, sizeof(u64));
scheduler_msg_get(&r, sizeof(r));
scheduler_msg_end();
r = handler_release(type, u64, r);
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r));
break;
case PROC_SCHEDULER_BATCH:
log_debug("scheduler-api: PROC_SCHEDULER_BATCH");
scheduler_msg_get(&typemask, sizeof(typemask));
scheduler_msg_get(&count, sizeof(count));
scheduler_msg_end();
if (count > MAX_BATCH_SIZE)
count = MAX_BATCH_SIZE;
r = handler_batch(typemask, &delay, &count, evpids, types);
scheduler_msg_add(&r, sizeof(r));
scheduler_msg_add(&delay, sizeof(delay));
scheduler_msg_add(&count, sizeof(count));
if (r > 0) {
scheduler_msg_add(evpids, sizeof(*evpids) * count);
scheduler_msg_add(types, sizeof(*types) * count);
}
scheduler_msg_close();
break;
case PROC_SCHEDULER_MESSAGES:
log_debug("scheduler-api: PROC_SCHEDULER_MESSAGES");
scheduler_msg_get(&msgid, sizeof(msgid));
scheduler_msg_get(&sz, sizeof(sz));
scheduler_msg_end();
if (sz > MAX_BATCH_SIZE)
sz = MAX_BATCH_SIZE;
n = handler_messages(msgid, msgids, sz);
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, msgids,
n * sizeof(*msgids));
break;
case PROC_SCHEDULER_ENVELOPES:
log_debug("scheduler-api: PROC_SCHEDULER_ENVELOPES");
scheduler_msg_get(&evpid, sizeof(evpid));
scheduler_msg_get(&sz, sizeof(sz));
scheduler_msg_end();
if (sz > MAX_BATCH_SIZE)
sz = MAX_BATCH_SIZE;
n = handler_envelopes(evpid, evpstates, sz);
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, evpstates,
n * sizeof(*evpstates));
break;
case PROC_SCHEDULER_SCHEDULE:
log_debug("scheduler-api: PROC_SCHEDULER_SCHEDULE");
scheduler_msg_get(&evpid, sizeof(evpid));
scheduler_msg_end();
r = handler_schedule(evpid);
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r));
break;
case PROC_SCHEDULER_REMOVE:
log_debug("scheduler-api: PROC_SCHEDULER_REMOVE");
scheduler_msg_get(&evpid, sizeof(evpid));
scheduler_msg_end();
r = handler_remove(evpid);
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r));
break;
case PROC_SCHEDULER_SUSPEND:
log_debug("scheduler-api: PROC_SCHEDULER_SUSPEND");
scheduler_msg_get(&evpid, sizeof(evpid));
scheduler_msg_end();
r = handler_suspend(evpid);
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r));
break;
case PROC_SCHEDULER_RESUME:
log_debug("scheduler-api: PROC_SCHEDULER_RESUME");
scheduler_msg_get(&evpid, sizeof(evpid));
scheduler_msg_end();
r = handler_resume(evpid);
imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r));
break;
default:
log_warnx("warn: scheduler-api: bad message %d", imsg.hdr.type);
fatalx("scheduler-api: exiting");
}
}
