add more helper functions more example code

This commit is contained in:
Rongsong Shen 2026-01-21 16:34:04 +08:00
parent 70970434ca
commit 00d161aa11
6 changed files with 403 additions and 55 deletions

251
src/lib/helper.c Normal file
View file

@ -0,0 +1,251 @@
#include <errno.h>
#include <stdlib.h>
#include "controller.capnp.h"
#include "helper.h"
#include "typemap.h"
#include "xce/types.h"
void* xce_request_body(xce_request_t* req)
{
switch (req->body_which) {
case Request_body_nomsg:
break;
case Request_body_replymsg:
return req->body.replymsg;
case Request_body_hellomsg:
return req->body.hellomsg;
case Request_body_joinmsg:
return req->body.joinmsg;
case Request_body_configpushmsg:
return req->body.configpushmsg;
case Request_body_pushmsg:
return req->body.pushmsg;
case Request_body_getmsg:
return req->body.getmsg;
case Request_body_datamsg:
return req->body.datamsg;
case Request_body_dispatchmsg:
return req->body.dispatchmsg;
case Request_body_jobstartmsg:
return req->body.jobstartmsg;
case Request_body_notifymsg:
return req->body.notifymsg;
case Request_body_jobsubmsg:
return req->body.jobsubmsg;
case Request_body_updatemsg:
return req->body.updatemsg;
case Request_body_jobctrlmsg:
return req->body.jobctrlmsg;
case Request_body_queuectrlmsg:
return req->body.queuectrlmsg;
case Request_body_runtaskmsg:
return req->body.runtaskmsg;
case Request_body_preasonmsg:
return req->body.preasonmsg;
case Request_body_clrpreasonmsg:
return req->body.clrpreasonmsg;
default:
break;
}
return NULL;
}
char* xce_request_op_name(xce_request_t* req)
{
switch (req->opcode) {
case OPCode_nop:
return "nop";
case OPCode_reply:
return "reply";
case OPCode_hello:
return "hello";
case OPCode_join:
return "join";
case OPCode_shutdown:
return "shutdown";
case OPCode_get:
return "get";
case OPCode_push:
return "push";
case OPCode_update:
return "update";
case OPCode_jobdispatch:
return "jobdispatch";
case OPCode_startjob:
return "startjob";
case OPCode_notify:
return "notify";
case OPCode_jobsub:
return "jobsub";
case OPCode_jobstatus:
return "jobstatus";
case OPCode_jobrusage:
return "jobrusage";
case OPCode_hostcontrol:
return "hostcontrol";
case OPCode_jobcontrol:
return "jobcontrol";
case OPCode_clusterconf:
return "clusterconf";
case OPCode_queuecontrol:
return "queuecontrol";
case OPCode_runtasks:
return "runtasks";
case OPCode_setpreason:
return "setpreason";
case OPCode_clrpreason:
return "clrpreason";
default:
break;
}
return "unknown";
}
void *xce_val_ptr(xce_val_t *v) {
switch(v->which) {
case Value_u_ivalue32:
return &(v->u.ivalue32);
case Value_u_uvalue32:
return &(v->u.uvalue32);
case Value_u_ivalue64:
return &(v->u.ivalue64);
case Value_u_uvalue64:
return &(v->u.uvalue64);
case Value_u_svalue:
return v->u.svalue;
case Value_u_fvalue:
return &(v->u.fvalue);
case Value_u_bvalue:
return &(v->u.bvalue);
default:
break;
}
return NULL;
}
void* xce_configdata_ptr(xce_configdata_t *d) {
switch(d->data_which) {
case ConfigData_data_schedconfig:
return d->data.conf4sched;
case ConfigData_data_workerconfig:
return d->data.conf4worker;
case ConfigData_data_clusterconfig:
return d->data.clusterconfig;
default:
break;
}
return NULL;
}
void* xce_data_ptr(xce_data_t *d) {
switch(d->table) {
case Data_data_schedjobs:
return d->data.jobs;
case Data_data_schedworkers:
return d->data.workers;
case Data_data_jobs:
return d->data.jobs;
case Data_data_workers:
return d->data.workers;
case Data_data_resources:
return d->data.resources;
case Data_data_resvals:
return d->data.resvals;
default:
break;
}
return NULL;
}
void* xce_update_data_ptr(xce_update_data_t *d) {
switch(d->table) {
case Update_data_nodata:
break;
case Update_data_resvals:
return d->data.resvals;
case Update_data_jobrusages:
return d->data.jobrusages;
default:
break;
}
return NULL;
}
void* xce_notify_data_ptr(xce_notify_data_t *d) {
switch(d->event) {
case NotifyData_data_jobPidInfo:
return d->data.pidinfo;
case NotifyData_data_jobFailInfo:
return d->data.failinfo;
case NotifyData_data_jobDispatchFail:
return d->data.dispatchfail;
case NotifyData_data_jobFinishInfo:
return d->data.finishinfo;
case NotifyData_data_schedStartJob:
return (void *)d->data.sched_start_job;
case NotifyData_data_schedExitJob:
return (void *)d->data.sched_exit_job;
case NotifyData_data_schedFailJob:
return (void *)d->data.sched_fail_job;
case NotifyData_data_schedUpdateJob:
return d->data.sched_update_job;
case NotifyData_data_schedUpdateResvals:
return d->data.sched_update_resvals;
case NotifyData_data_stoppedJob:
return (void *)d->data.stopped_job;
case NotifyData_data_resumedJob:
return (void *)d->data.resumed_job;
case NotifyData_data_ckptinfo:
return d->data.ckptinfo;
case NotifyData_data_schedUpdateQueue:
return d->data.sched_update_queue;
case NotifyData_data_taskPidInfo:
return d->data.taskinfo;
case NotifyData_data_taskFailInfo:
return d->data.taskfailinfo;
case NotifyData_data_taskFinish:
return d->data.taskfinish;
case NotifyData_data_postDoneJob:
return (void *)d->data.pdone_job;
case NotifyData_data_schedJobRusages:
return d->data.sched_job_rusages;
default:
break;
}
return NULL;
}
void* xce_jobctrl_data_ptr(xce_jobctrl_t *d) {
switch(d->action) {
case JobControl_u_siginfo:
return d->u.siginfo;
case JobControl_u_ckptoption:
return d->u.ckptoption;
default:
break;
}
return NULL;
}
void* xce_reply_data_ptr(xce_reply_t *reply) {
switch(reply->data_which) {
case RequestReply_data_nodata:
break;
case RequestReply_data_debugmsg:
return reply->data.debugmsg;
case RequestReply_data_getdata:
return reply->data.getdata;
case RequestReply_data_jobid:
return (void *)reply->data.jobid;
case RequestReply_data_clusterconf:
return reply->data.clusterconf;
default:
break;
}
return NULL;
}

View file

@ -36,6 +36,7 @@ libxce_src = [ gen_srcs[0],
'group.c',
'xcelimit.c',
'xvec.c',
'helper.c',
]
incdir = include_directories('../include')