update controller code

src/controller/myconf.c

@@ -0,0 +1,173 @@
+#include <errno.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <tlog.h>
+
+#include "controller.capnp.h"
+#include "myconf.h"
+#include "objmgr.h"
+#include "utils.h"
+#include "xce/cluster_def.h"
+#include "xce/types.h"
+
+int add_config(void* c,
+               char* name,
+               char* value)
+{
+    cluster_t*   cls = (cluster_t*)c;
+    xce_config_t var = {
+        .name  = name,
+        .value = value
+    };
+    xce_config_t* cv = NULL;
+
+    cv = create_obj_by_typename(&(cls->objmgr), "xconfig",
+                                &var);
+    if (cv == NULL) {
+        return -errno;
+    }
+
+    return 0;
+}
+
+struct cluster_configuration {
+    cluster_config_t* cluster;
+};
+
+static int config_fini(int     type,
+                       int     situation,
+                       void*   data,
+                       va_list ap)
+{
+    struct cluster_configuration* cf = (struct cluster_configuration*)data;
+
+    switch (situation) {
+    case HOOK_AFTER_EVENT:
+        if (cf->cluster != NULL) {
+            free_ClusterConfig(cf->cluster);
+            FREEUP(cf->cluster);
+        }
+        break;
+    default:
+        break;
+    }
+    return 0;
+}
+
+static int cluster_hook_fun(int     type,
+                            int     situation,
+                            int     detail,
+                            void*   data,
+                            va_list ap)
+{
+    if (detail == HOOK_DETAIL_CLUSTER_SHUTDOWN) {
+        return config_fini(type, situation, data, ap);
+    }
+
+    return 0;
+}
+
+#define SET_DEFAULT_IF_UNSET(var, unset, val) \
+    do {                                      \
+        if ((var) == (unset)) {               \
+            (var) = (val);                    \
+        }                                     \
+    } while (0)
+
+static void fill_default_cluster_config(cluster_config_t* c)
+{
+    SET_DEFAULT_IF_UNSET(c->port, 0, 9990);
+    SET_DEFAULT_IF_UNSET(c->heartbeat_interval, 0, 5);
+    SET_DEFAULT_IF_UNSET(c->vcpus, 0, 8);
+    SET_DEFAULT_IF_UNSET(c->fetch_job_interval, 0, 10);
+}
+
+static int load_cluster_config_from_file(char*              file,
+                                         cluster_config_t** config)
+{
+    struct capn       c;
+    ClusterConfig_ptr p;
+    FILE*             fp;
+    int               rc = 0;
+
+    fp = fopen(file, "rb");
+    if (fp == NULL) {
+        rc = -errno;
+        perror("fopen");
+        return rc;
+    }
+
+    capn_init_fp(&c, fp, 0);
+    p.p = capn_getp(capn_root(&c), 0, 1);
+
+    decode_ClusterConfig_ptr(config, p);
+
+    capn_free(&c);
+
+    fclose(fp);
+
+    return 0;
+}
+
+int load_cluster_config(cluster_t*                     c,
+                        struct cluster_configuration** cluster_config)
+{
+    cluster_config_t* cfg;
+    int               rc;
+
+    *cluster_config = (struct cluster_configuration*)calloc(1, sizeof(*cluster_config));
+    if ((*cluster_config) == NULL) {
+        return -ENOMEM;
+    }
+
+    rc = load_cluster_config_from_file(c->myconf->conffile, &((*cluster_config)->cluster));
+
+    if (rc < 0) {
+        return rc;
+    }
+
+    if ((*cluster_config)->cluster == NULL) {
+        return -errno;
+    }
+
+    cfg = (*cluster_config)->cluster;
+
+    fill_default_cluster_config(cfg);
+
+    xce_add_hook(&(c->hooksys),
+                 XCE_CONTROLLER_HOOK_CLUSTER,
+                 (hook_fn_t)cluster_hook_fun,
+                 (*cluster_config));
+
+    return 0;
+}
+
+static struct cluster_configuration* cf = NULL;
+
+cluster_config_t* get_default_conf(void* cl)
+{
+    cluster_config_t* c;
+    cluster_t*        cls = (cluster_t*)cl;
+
+    load_cluster_config(cls, &cf);
+    if (cf == NULL) {
+        PERROR("load_cluster_config");
+        return NULL;
+    }
+
+    c = cf->cluster;
+
+    return c;
+}
+
+int load_initial_config(char*              file,
+                        initial_config_t** conf,
+                        int                verbose)
+{
+    return 0;
+}
+
+void free_initial_config(initial_config_t* conf)
+{
+}