update controller code

src/controller/opget.c

@@ -0,0 +1,225 @@
+#include "config.h"
+
+#include <errno.h>
+#include <string.h>
+#include <unistd.h>
+
+#include "cluster.h"
+#include "controller.capnp.h"
+#include "utils.h"
+#include "xce/cluster_def.h"
+
+static int
+do_reply_reslist(cluster_t*     c,
+                 endpoint_t*    ep,
+                 xce_request_t* request)
+{
+    xce_reply_t         reply;
+    xce_resource_data_t resdata;
+    xce_data_t          getdata = {
+                 .table = request->body.getmsg->table,
+                 .data  = {
+                      .resources = &resdata }
+    };
+    int rc = 0;
+
+    init_reply_by_req(&reply, request);
+    collect_resources(c, NULL, &(resdata.n_total),
+                      &(resdata.resources));
+
+    reply.status       = 0;
+    reply.data_which   = RequestReply_data_getdata;
+    reply.data.getdata = &getdata;
+
+    rc = xce_reply_get(ep, c->myhost, &reply, NULL, NULL, -1);
+
+    deref_objs(resdata.n_total, (void**)resdata.resources);
+    FREEUP(resdata.resources);
+
+    return rc;
+}
+
+static int
+do_reply_resvals(cluster_t*     c,
+                 endpoint_t*    ep,
+                 xce_request_t* request)
+{
+    xce_reply_t       reply;
+    xce_resval_data_t resvaldata;
+    xce_get_t*        getmsg  = request->body.getmsg;
+    xce_data_t        getdata = {
+               .data = {
+                   .resvals = &resvaldata }
+    };
+    int           rc = 0;
+    xce_val_t     val;
+    xce_resval_t  resval;
+    xce_resval_t* vals[] = {
+        &resval
+    };
+
+    init_reply_by_req(&reply, request);
+
+    resvaldata.n_total = 0;
+    resvaldata.vals    = NULL;
+
+    if (getmsg != NULL) {
+        xce_match_t* condition = getmsg->condition;
+
+        if (condition != NULL) {
+            getdata.table = getmsg->table;
+            rc            = get_resval(c, condition->condition.reskey->resname,
+                                       condition->condition.reskey->hostname,
+                                       &val);
+            if (rc >= 0) {
+                resval.val         = &val;
+                resval.name        = condition->condition.reskey->resname;
+                resval.host        = condition->condition.reskey->hostname;
+                resvaldata.n_total = 1;
+                resvaldata.vals    = vals;
+            }
+        } else {
+            rc = -EINVAL;
+        }
+    } else {
+        rc = -EINVAL;
+    }
+
+    reply.status       = rc;
+    reply.data_which   = RequestReply_data_getdata;
+    reply.data.getdata = &getdata;
+
+    rc = xce_reply_get(ep, c->myhost, &reply, NULL, NULL, -1);
+
+    return rc;
+}
+
+int check_worker_(xce_worker_t* w)
+{
+    if (w == NULL) {
+        return 0;
+    }
+
+    return 1;
+}
+
+static int
+do_reply_workerlist(cluster_t*     c,
+                    endpoint_t*    ep,
+                    xce_request_t* request)
+{
+    xce_reply_t        reply;
+    xce_workers_data_t workerdata;
+    /* clang-format off */
+    xce_data_t         getdata = {
+                .table = request->body.getmsg->table,
+                .data  = {
+                     .workers = &workerdata }
+    };
+    /* clang-format on */
+    int rc = 0;
+    int i;
+
+    init_reply_by_req(&reply, request);
+    collect_workers(c, (collect_check_fn_t)check_worker_,
+                    &(workerdata.n_total), &(workerdata.workers));
+
+    reply.status       = 0;
+    reply.data_which   = RequestReply_data_getdata;
+    reply.data.getdata = &getdata;
+
+    rc = xce_reply_get(ep, c->myhost, &reply,
+                       check_controller, c, 100);
+
+    for (i = 0; i < workerdata.n_total; i++) {
+        xce_worker_t* w = workerdata.workers[i];
+
+        free_resval_with_idx_list(&(w->resvals_to_sched),
+                                  w->nresvals_to_sched);
+    }
+
+    deref_objs(workerdata.n_total, (void**)workerdata.workers);
+    FREEUP(workerdata.workers);
+
+    return rc;
+}
+
+static int check_job_(xce_job_t* j)
+{
+    if (j == NULL) {
+        return 0;
+    }
+
+    return 1;
+}
+
+static int
+do_reply_joblist(cluster_t*     c,
+                 endpoint_t*    ep,
+                 xce_request_t* request)
+{
+    xce_reply_t     reply;
+    xce_jobs_data_t jobdata;
+    xce_data_t      getdata = {
+             .table = request->body.getmsg->table,
+             .data  = {
+                  .jobs = &jobdata }
+    };
+    int rc = 0;
+
+    init_reply_by_req(&reply, request);
+    collect_jobs(c, (collect_check_fn_t)check_job_,
+                 &(jobdata.n_total), &(jobdata.jobs));
+
+    reply.status       = 0;
+    reply.data_which   = RequestReply_data_getdata;
+    reply.data.getdata = &getdata;
+
+    rc = xce_reply_get(ep, c->myhost, &reply,
+                       check_controller, c, 100);
+
+    deref_objs(jobdata.n_total, (void**)jobdata.jobs);
+    FREEUP(jobdata.jobs);
+
+    return rc;
+}
+
+int do_get_op(cluster_t*     c,
+              endpoint_t*    ep,
+              xce_request_t* request)
+{
+    int         rc = 0;
+    xce_reply_t reply;
+
+    TRACE("%s table %d",
+          __func__, request->body.getmsg->table);
+
+    init_reply_by_req(&reply, request);
+
+    switch (request->body.getmsg->table) {
+    case Data_data_resources:
+        rc = do_reply_reslist(c, ep, request);
+        break;
+    case Data_data_workers:
+    case Data_data_schedworkers:
+        rc = do_reply_workerlist(c, ep, request);
+        break;
+    case Data_data_schedjobs:
+    case Data_data_jobs:
+        rc = do_reply_joblist(c, ep, request);
+        break;
+    case Data_data_resvals:
+        rc = do_reply_resvals(c, ep, request);
+        break;
+    default:
+        rc = -ENOENT;
+
+        reply.status     = rc;
+        reply.data_which = RequestReply_data_nodata;
+
+        xce_reply(ep, c->myhost, &reply);
+        break;
+    }
+
+    return rc;
+}