Code

store: Introduced sdb_store_host_tojson() to serialize a single host.
[sysdb.git] / src / core / store.c
index d195b8f0e3d029f4bf853910ba4ce05358e85d58..50b39d52f1160791cee05ecb5b3a44867380eece 100644 (file)
@@ -57,6 +57,11 @@ static sdb_type_t sdb_attribute_type;
 
 struct sdb_store_base {
        sdb_object_t super;
+
+       /* object type */
+       int type;
+
+       /* common meta information */
        sdb_time_t last_update;
        sdb_store_base_t *parent;
 };
@@ -74,9 +79,7 @@ typedef struct {
 typedef struct {
        sdb_store_base_t super;
 
-       int type;
        sdb_llist_t *children;
-
        sdb_llist_t *attributes;
 } sdb_store_obj_t;
 #define SDB_STORE_OBJ(obj) ((sdb_store_obj_t *)(obj))
@@ -100,8 +103,10 @@ static int
 store_base_init(sdb_object_t *obj, va_list ap)
 {
        sdb_store_base_t *sobj = STORE_BASE(obj);
-       sobj->last_update = va_arg(ap, sdb_time_t);
 
+       sobj->type = va_arg(ap, int);
+
+       sobj->last_update = va_arg(ap, sdb_time_t);
        sobj->parent = NULL;
        return 0;
 } /* store_base_init */
@@ -121,12 +126,11 @@ sdb_store_obj_init(sdb_object_t *obj, va_list ap)
        sdb_store_obj_t *sobj = SDB_STORE_OBJ(obj);
        int ret;
 
+       /* this will consume the first argument (type) of ap */
        ret = store_base_init(obj, ap);
        if (ret)
                return ret;
 
-       sobj->type = va_arg(ap, int);
-
        sobj->children = sdb_llist_create();
        if (! sobj->children)
                return -1;
@@ -157,6 +161,7 @@ sdb_attr_init(sdb_object_t *obj, va_list ap)
        const char *value;
        int ret;
 
+       /* this will consume the first argument (type) of ap */
        ret = store_base_init(obj, ap);
        if (ret)
                return ret;
@@ -215,14 +220,14 @@ sdb_store_lookup_in_list(sdb_llist_t *l, int type, const char *name)
                sdb_store_obj_t *sobj = SDB_STORE_OBJ(sdb_llist_iter_get_next(iter));
                assert(sobj);
 
-               if ((sobj->type == type)
+               if ((STORE_BASE(sobj)->type == type)
                                && (! strcasecmp(SDB_OBJ(sobj)->name, name))) {
                        sdb_llist_iter_destroy(iter);
                        return sobj;
                }
 
                /* don't lookups non-host types from hierarchical hosts */
-               if ((type != SDB_HOST) && (sobj->type == SDB_HOST))
+               if ((type != SDB_HOST) && (STORE_BASE(sobj)->type == SDB_HOST))
                        continue;
 
                sobj = sdb_store_lookup_in_list(sobj->children, type, name);
@@ -342,10 +347,10 @@ store_obj(int parent_type, const char *parent_name,
                if (type == SDB_ATTRIBUTE)
                        /* the value will be updated by the caller */
                        new = STORE_BASE(sdb_object_create(name, sdb_attribute_type,
-                                               last_update, NULL));
+                                               type, last_update, NULL));
                else
                        new = STORE_BASE(sdb_object_create(name, sdb_store_obj_type,
-                                               last_update, type));
+                                               type, last_update));
 
                if (! new) {
                        char errbuf[1024];
@@ -451,96 +456,114 @@ sdb_store_service(const char *hostname, const char *name,
        return status;
 } /* sdb_store_service */
 
-/* TODO: actually support hierarchical data */
 int
-sdb_store_tojson(sdb_strbuf_t *buf)
+sdb_store_host_tojson(sdb_store_base_t *h, sdb_strbuf_t *buf)
 {
-       sdb_llist_iter_t *host_iter;
+       sdb_store_obj_t *host;
 
-       if (! buf)
-               return -1;
+       sdb_llist_iter_t *svc_iter;
+       sdb_llist_iter_t *attr_iter;
 
-       pthread_rwlock_rdlock(&obj_lock);
+       char time_str[64];
 
-       host_iter = sdb_llist_get_iter(obj_list);
-       if (! host_iter) {
-               pthread_rwlock_unlock(&obj_lock);
+       if ((! h) || (h->type != SDB_HOST) || (! buf))
                return -1;
+
+       host = SDB_STORE_OBJ(h);
+
+       if (! sdb_strftime(time_str, sizeof(time_str),
+                               "%F %T %z", host->_last_update))
+               snprintf(time_str, sizeof(time_str), "<error>");
+       time_str[sizeof(time_str) - 1] = '\0';
+
+       sdb_strbuf_append(buf, "{\"name\": \"%s\", "
+                       "\"last_update\": \"%s\", "
+                       "\"attributes\": [",
+                       SDB_OBJ(host)->name, time_str);
+
+       attr_iter = sdb_llist_get_iter(host->attributes);
+       if (! attr_iter) {
+               char errbuf[1024];
+               sdb_log(SDB_LOG_ERR, "store: Failed to retrieve attributes: %s\n",
+                               sdb_strerror(errno, errbuf, sizeof(errbuf)));
+               sdb_strbuf_append(buf, "{\"error\": \"failed to retrieve "
+                               "attributes: %s\"}", errbuf);
        }
 
-       sdb_strbuf_append(buf, "{\"hosts\":[");
+       /* has_next returns false if the iterator is NULL */
+       while (sdb_llist_iter_has_next(attr_iter)) {
+               sdb_attribute_t *attr = SDB_ATTR(sdb_llist_iter_get_next(attr_iter));
+               assert(attr);
 
-       while (sdb_llist_iter_has_next(host_iter)) {
-               sdb_store_obj_t *host = SDB_STORE_OBJ(sdb_llist_iter_get_next(host_iter));
-               sdb_llist_iter_t *svc_iter;
-               sdb_llist_iter_t *attr_iter;
+               if (! sdb_strftime(time_str, sizeof(time_str),
+                                       "%F %T %z", attr->_last_update))
+                       snprintf(time_str, sizeof(time_str), "<error>");
+               time_str[sizeof(time_str) - 1] = '\0';
 
-               char time_str[64];
+               sdb_strbuf_append(buf, "{\"name\": \"%s\", "
+                               "\"value\": \"%s\", \"last_update\": \"%s\"},",
+                               SDB_OBJ(attr)->name, attr->value, time_str);
+       }
 
-               assert(host);
+       sdb_llist_iter_destroy(attr_iter);
+       sdb_strbuf_append(buf, "], \"services\": [");
+
+       svc_iter = sdb_llist_get_iter(host->children);
+       if (! svc_iter) {
+               char errbuf[1024];
+               sdb_log(SDB_LOG_ERR, "store: Failed to retrieve services: %s\n",
+                               sdb_strerror(errno, errbuf, sizeof(errbuf)));
+               sdb_strbuf_append(buf, "{\"error\": \"failed to retrieve "
+                               "services: %s\"}", errbuf);
+       }
+
+       while (sdb_llist_iter_has_next(svc_iter)) {
+               sdb_store_obj_t *svc = SDB_STORE_OBJ(sdb_llist_iter_get_next(svc_iter));
+               assert(svc);
 
                if (! sdb_strftime(time_str, sizeof(time_str),
-                                       "%F %T %z", host->_last_update))
+                                       "%F %T %z", svc->_last_update))
                        snprintf(time_str, sizeof(time_str), "<error>");
                time_str[sizeof(time_str) - 1] = '\0';
 
                sdb_strbuf_append(buf, "{\"name\": \"%s\", "
-                               "\"last_update\": \"%s\", "
-                               "\"attributes\": [",
-                               SDB_OBJ(host)->name, time_str);
-
-               attr_iter = sdb_llist_get_iter(host->attributes);
-               if (! attr_iter) {
-                       char errbuf[1024];
-                       sdb_log(SDB_LOG_ERR, "store: Failed to retrieve attributes: %s\n",
-                                       sdb_strerror(errno, errbuf, sizeof(errbuf)));
-                       sdb_strbuf_append(buf, "{\"error\": \"failed to retrieve "
-                                       "attributes: %s\"}", errbuf);
-               }
+                               "\"last_update\": \"%s\"},",
+                               SDB_OBJ(svc)->name, time_str);
+       }
 
-               /* has_next returns false if the iterator is NULL */
-               while (sdb_llist_iter_has_next(attr_iter)) {
-                       sdb_attribute_t *attr = SDB_ATTR(sdb_llist_iter_get_next(attr_iter));
-                       assert(attr);
+       sdb_llist_iter_destroy(svc_iter);
+       sdb_strbuf_append(buf, "]}");
+       return 0;
+} /* sdb_store_host_tojson */
 
-                       if (! sdb_strftime(time_str, sizeof(time_str),
-                                               "%F %T %z", attr->_last_update))
-                               snprintf(time_str, sizeof(time_str), "<error>");
-                       time_str[sizeof(time_str) - 1] = '\0';
+/* TODO: actually support hierarchical data */
+int
+sdb_store_tojson(sdb_strbuf_t *buf)
+{
+       sdb_llist_iter_t *host_iter;
 
-                       sdb_strbuf_append(buf, "{\"name\": \"%s\", "
-                                       "\"value\": \"%s\", \"last_update\": \"%s\"},",
-                                       SDB_OBJ(attr)->name, attr->value, time_str);
-               }
+       if (! buf)
+               return -1;
 
-               sdb_llist_iter_destroy(attr_iter);
-               sdb_strbuf_append(buf, "], \"services\": [");
+       pthread_rwlock_rdlock(&obj_lock);
 
-               svc_iter = sdb_llist_get_iter(host->children);
-               if (! svc_iter) {
-                       char errbuf[1024];
-                       sdb_log(SDB_LOG_ERR, "store: Failed to retrieve services: %s\n",
-                                       sdb_strerror(errno, errbuf, sizeof(errbuf)));
-                       sdb_strbuf_append(buf, "{\"error\": \"failed to retrieve "
-                                       "services: %s\"}", errbuf);
-               }
+       host_iter = sdb_llist_get_iter(obj_list);
+       if (! host_iter) {
+               pthread_rwlock_unlock(&obj_lock);
+               return -1;
+       }
 
-               while (sdb_llist_iter_has_next(svc_iter)) {
-                       sdb_store_obj_t *svc = SDB_STORE_OBJ(sdb_llist_iter_get_next(svc_iter));
-                       assert(svc);
+       sdb_strbuf_append(buf, "{\"hosts\":[");
 
-                       if (! sdb_strftime(time_str, sizeof(time_str),
-                                               "%F %T %z", svc->_last_update))
-                               snprintf(time_str, sizeof(time_str), "<error>");
-                       time_str[sizeof(time_str) - 1] = '\0';
+       while (sdb_llist_iter_has_next(host_iter)) {
+               sdb_store_base_t *host = STORE_BASE(sdb_llist_iter_get_next(host_iter));
+               assert(host);
 
-                       sdb_strbuf_append(buf, "{\"name\": \"%s\", "
-                                       "\"last_update\": \"%s\"},",
-                                       SDB_OBJ(svc)->name, time_str);
-               }
+               if (sdb_store_host_tojson(host, buf))
+                       return -1;
 
-               sdb_llist_iter_destroy(svc_iter);
-               sdb_strbuf_append(buf, "]},");
+               if (sdb_llist_iter_has_next(host_iter))
+                       sdb_strbuf_append(buf, ",");
        }
 
        sdb_strbuf_append(buf, "]}");