Controller: sending JSON configuration to router.

4 files changed, 288 insertions, 33 deletions

diff --git a/src/nxt_conf.c b/src/nxt_conf.c
index c091b260..f891f71e 100644
--- a/src/nxt_conf.c
+++ b/src/nxt_conf.c
@@ -137,7 +137,7 @@ nxt_conf_json_indentation(u_char *p, uint32_t level)
 }
 
 
-nxt_inline void
+void
 nxt_conf_get_string(nxt_conf_value_t *value, nxt_str_t *str)
 {
     if (value->type == NXT_CONF_VALUE_SHORT_STRING) {
@@ -151,6 +151,70 @@ nxt_conf_get_string(nxt_conf_value_t *value, nxt_str_t *str)
 
 
 nxt_uint_t
+nxt_conf_object_members_count(nxt_conf_value_t *value)
+{
+    return value->u.object->count;
+}
+
+
+nxt_conf_value_t *
+nxt_conf_create_object(nxt_mp_t *mp, nxt_uint_t count)
+{
+    size_t            size;
+    nxt_conf_value_t  *value;
+
+    size = sizeof(nxt_conf_value_t)
+           + sizeof(nxt_conf_object_t)
+           + count * sizeof(nxt_conf_object_member_t);
+
+    value = nxt_mp_get(mp, size);
+    if (nxt_slow_path(value == NULL)) {
+        return NULL;
+    }
+
+    value->u.object = nxt_pointer_to(value, sizeof(nxt_conf_value_t));
+    value->u.object->count = count;
+
+    value->type = NXT_CONF_VALUE_OBJECT;
+
+    return value;
+}
+
+
+nxt_int_t
+nxt_conf_set_object_member(nxt_mp_t *mp, nxt_conf_value_t *object,
+    nxt_str_t *name, nxt_conf_value_t *value, uint32_t index)
+{
+    nxt_conf_value_t          *name_value;
+    nxt_conf_object_member_t  *member;
+
+    member = &object->u.object->members[index];
+    name_value = &member->name;
+
+    if (name->length > NXT_CONF_MAX_SHORT_STRING) {
+        name_value->type = NXT_CONF_VALUE_STRING;
+        name_value->u.string = nxt_str_alloc(mp, 0);
+
+        if (nxt_slow_path(name_value->u.string == NULL)) {
+            return NXT_ERROR;
+        }
+
+        *name_value->u.string = *name;
+
+    } else {
+        name_value->type = NXT_CONF_VALUE_SHORT_STRING;
+        name_value->u.str[0] = name->length;
+
+        nxt_memcpy(&name_value->u.str[1], name->start, name->length);
+    }
+
+    member->value = *value;
+
+    return NXT_OK;
+}
+
+
+nxt_uint_t
 nxt_conf_type(nxt_conf_value_t *value)
 {
     switch (value->type) {
diff --git a/src/nxt_conf.h b/src/nxt_conf.h
index 64997b56..d9021732 100644
--- a/src/nxt_conf.h
+++ b/src/nxt_conf.h
@@ -77,5 +77,13 @@ u_char *nxt_conf_json_print(u_char *p, nxt_conf_value_t *value,
 
 nxt_int_t nxt_conf_validate(nxt_conf_value_t *value);
 
+void nxt_conf_get_string(nxt_conf_value_t *value, nxt_str_t *str);
+
+// FIXME reimplement and reorder functions below
+nxt_uint_t nxt_conf_object_members_count(nxt_conf_value_t *value);
+nxt_conf_value_t *nxt_conf_create_object(nxt_mp_t *mp, nxt_uint_t count);
+nxt_int_t nxt_conf_set_object_member(nxt_mp_t *mp, nxt_conf_value_t *object,
+    nxt_str_t *name, nxt_conf_value_t *value, uint32_t index);
+
 
 #endif /* _NXT_CONF_INCLUDED_ */
diff --git a/src/nxt_conf_validation.c b/src/nxt_conf_validation.c
index fd92020b..4863d163 100644
--- a/src/nxt_conf_validation.c
+++ b/src/nxt_conf_validation.c
@@ -59,6 +59,11 @@ static const nxt_conf_vldt_object_t  nxt_conf_application_members[] = {
       &nxt_conf_vldt_app_type,
       NULL },
 
+    { nxt_string("workers"),
+      NXT_CONF_INTEGER,
+      NULL,
+      NULL },
+
     { nxt_string("path"),
       NXT_CONF_STRING,
       NULL,
diff --git a/src/nxt_controller.c b/src/nxt_controller.c
index 6403c98a..170f8246 100644
--- a/src/nxt_controller.c
+++ b/src/nxt_controller.c
@@ -55,6 +55,8 @@ static nxt_int_t nxt_controller_request_content_length(void *ctx,
 
 static void nxt_controller_process_request(nxt_task_t *task,
     nxt_conn_t *c, nxt_controller_request_t *r);
+static nxt_int_t nxt_controller_conf_apply(nxt_task_t *task,
+    nxt_conf_value_t *conf);
 static nxt_int_t nxt_controller_response(nxt_task_t *task, nxt_conn_t *c,
     nxt_controller_response_t *resp);
 static nxt_buf_t *nxt_controller_response_body(nxt_controller_response_t *resp,
@@ -562,38 +564,6 @@ nxt_controller_process_request(nxt_task_t *task, nxt_conn_t *c,
 
     nxt_memzero(&resp, sizeof(nxt_controller_response_t));
 
-    if (nxt_str_eq(&req->parser.method, "POST", 4)) {
-        nxt_port_t     *port;
-        nxt_runtime_t  *rt;
-
-        rt = task->thread->runtime;
-
-        nxt_runtime_port_each(rt, port) {
-
-            if (nxt_pid == port->pid) {
-                continue;
-            }
-
-            if (port->type == NXT_PROCESS_ROUTER) {
-                nxt_buf_t *b, *src;
-
-                src = c->read;
-                b = nxt_port_mmap_get_buf(task, port,
-                                          nxt_buf_mem_used_size(&src->mem));
-
-                nxt_memcpy(b->mem.pos, src->mem.pos,
-                           nxt_buf_mem_used_size(&src->mem));
-                b->mem.free += nxt_buf_mem_used_size(&src->mem);
-
-                (void) nxt_port_socket_write(task, port, NXT_PORT_MSG_DATA,
-                                             -1, 0, 0, b);
-
-                break;
-            }
-
-        } nxt_runtime_port_loop;
-    }
-
     if (nxt_str_eq(&req->parser.method, "GET", 3)) {
 
         value = nxt_conf_get_path(nxt_controller_conf.root, &path);
@@ -654,12 +624,19 @@ nxt_controller_process_request(nxt_task_t *task, nxt_conn_t *c,
         }
 
         if (nxt_slow_path(nxt_conf_validate(value) != NXT_OK)) {
+            nxt_mp_destroy(mp);
             status = 400;
             nxt_str_set(&resp.json,
                         "{ \"error\": \"Invalid configuration.\" }");
             goto done;
         }
 
+        if (nxt_slow_path(nxt_controller_conf_apply(task, value) != NXT_OK)) {
+            nxt_mp_destroy(mp);
+            status = 500;
+            goto done;
+        }
+
         nxt_mp_destroy(nxt_controller_conf.pool);
 
         nxt_controller_conf.root = value;
@@ -715,12 +692,19 @@ nxt_controller_process_request(nxt_task_t *task, nxt_conn_t *c,
         }
 
         if (nxt_slow_path(nxt_conf_validate(value) != NXT_OK)) {
+            nxt_mp_destroy(mp);
             status = 400;
             nxt_str_set(&resp.json,
                         "{ \"error\": \"Invalid configuration.\" }");
             goto done;
         }
 
+        if (nxt_slow_path(nxt_controller_conf_apply(task, value) != NXT_OK)) {
+            nxt_mp_destroy(mp);
+            status = 500;
+            goto done;
+        }
+
         nxt_mp_destroy(nxt_controller_conf.pool);
 
         nxt_controller_conf.root = value;
@@ -769,6 +753,200 @@ done:
 
 
 static nxt_int_t
+nxt_controller_conf_apply(nxt_task_t *task, nxt_conf_value_t *conf)
+{
+    size_t            size;
+    uint32_t          next, n;
+    nxt_mp_t          *mp;
+    nxt_buf_t         *b;
+    nxt_int_t         rc;
+    nxt_str_t         ls_name, name;
+    nxt_port_t        *port;
+    nxt_uint_t        lss_count;
+    nxt_runtime_t     *rt;
+    nxt_conf_value_t  *lss, *ls, *rtr, *rtr_lss, *rtr_ls, *app_name, *apps,
+                      *app, *value;
+
+    static nxt_str_t  lss_str = nxt_string("listeners");
+    static nxt_str_t  app_str = nxt_string("application");
+    static nxt_str_t  apps_str = nxt_string("applications");
+    static nxt_str_t  type_str = nxt_string("type");
+    static nxt_str_t  app_type_str = nxt_string("_application_type");
+    static nxt_str_t  workers_str = nxt_string("workers");
+    static nxt_str_t  app_workers_str = nxt_string("_application_workers");
+    static nxt_str_t  rtr_str = nxt_string("router");
+    static nxt_str_t  http_str = nxt_string("http");
+
+    static nxt_str_t  rtr_json = nxt_string("{ \"threads\": 0 }");
+    static nxt_str_t  http_json = nxt_string(
+        "{ \"header_buffer_size\": 2096,"
+        "  \"large_header_buffer_size\": 16384, "
+        "  \"header_read_timeout\": 60000 }"
+    );
+
+    mp = nxt_mp_create(1024, 128, 256, 32);
+
+    if (nxt_slow_path(mp == NULL)) {
+        return NXT_ERROR;
+    }
+
+    lss_count = 0;
+
+    lss = nxt_conf_get_object_member(conf, &lss_str, NULL);
+
+    if (lss != NULL) {
+        lss_count = nxt_conf_object_members_count(lss);
+    }
+
+    apps = nxt_conf_get_object_member(conf, &apps_str, NULL);
+
+    if (nxt_slow_path(apps == NULL)) {
+        goto error;
+    }
+
+    rtr_lss = nxt_conf_create_object(mp, lss_count);
+
+    if (nxt_slow_path(rtr_lss == NULL)) {
+        goto error;
+    }
+
+    n = 0;
+    next = 0;
+
+    for ( ;; ) {
+        ls = nxt_conf_next_object_member(lss, &ls_name, &next);
+        if (ls == NULL) {
+            break;
+        }
+
+        rtr_ls = nxt_conf_create_object(mp, 3);
+
+        if (nxt_slow_path(rtr_ls == NULL)) {
+            goto error;
+        }
+
+        app_name = nxt_conf_get_object_member(ls, &app_str, NULL);
+
+        if (nxt_slow_path(app_name == NULL)) {
+            goto error;
+        }
+
+        rc = nxt_conf_set_object_member(mp, rtr_ls, &app_str, app_name, 0);
+
+        if (nxt_slow_path(rc != NXT_OK)) {
+            goto error;
+        }
+
+        nxt_conf_get_string(app_name, &name);
+
+        app = nxt_conf_get_object_member(apps, &name, NULL);
+
+        if (nxt_slow_path(app == NULL)) {
+            goto error;
+        }
+
+        value = nxt_conf_get_object_member(app, &type_str, NULL);
+
+        if (nxt_slow_path(value == NULL)) {
+            goto error;
+        }
+
+        rc = nxt_conf_set_object_member(mp, rtr_ls, &app_type_str, value, 1);
+
+        if (nxt_slow_path(rc != NXT_OK)) {
+            goto error;
+        }
+
+        value = nxt_conf_get_object_member(app, &workers_str, NULL);
+
+        if (nxt_slow_path(value == NULL)) {
+            goto error;
+        }
+
+        rc = nxt_conf_set_object_member(mp, rtr_ls, &app_workers_str, value, 2);
+
+        if (nxt_slow_path(rc != NXT_OK)) {
+            goto error;
+        }
+
+        rc = nxt_conf_set_object_member(mp, rtr_lss, &ls_name, rtr_ls, n);
+
+        if (nxt_slow_path(rc != NXT_OK)) {
+            goto error;
+        }
+
+        n++;
+    }
+
+    rtr = nxt_conf_create_object(mp, 3);
+
+    if (nxt_slow_path(rtr == NULL)) {
+        goto error;
+    }
+
+    rc = nxt_conf_set_object_member(mp, rtr, &lss_str, rtr_lss, 0);
+
+    if (nxt_slow_path(rc != NXT_OK)) {
+        goto error;
+    }
+
+    value = nxt_conf_json_parse_str(mp, &rtr_json);
+
+    if (nxt_slow_path(value == NULL)) {
+        goto error;
+    }
+
+    rc = nxt_conf_set_object_member(mp, rtr, &rtr_str, value, 1);
+
+    if (nxt_slow_path(rc != NXT_OK)) {
+        goto error;
+    }
+
+    value = nxt_conf_json_parse_str(mp, &http_json);
+
+    if (nxt_slow_path(value == NULL)) {
+        goto error;
+    }
+
+    rc = nxt_conf_set_object_member(mp, rtr, &http_str, value, 2);
+
+    if (nxt_slow_path(rc != NXT_OK)) {
+        goto error;
+    }
+
+    rt = task->thread->runtime;
+
+    nxt_runtime_port_each(rt, port) {
+
+        if (port->type == NXT_PROCESS_ROUTER) {
+            break;
+        }
+
+    } nxt_runtime_port_loop;
+
+    size = nxt_conf_json_length(rtr, NULL);
+
+    b = nxt_port_mmap_get_buf(task, port, size);
+
+    b->mem.free = nxt_conf_json_print(b->mem.free, rtr, NULL);
+
+    nxt_mp_destroy(mp);
+
+    nxt_debug(task, "conf for router: \"%*s\"",
+              b->mem.free - b->mem.pos, b->mem.pos);
+
+    (void) nxt_port_socket_write(task, port, NXT_PORT_MSG_DATA, -1, 0, 0, b);
+
+    return NXT_OK;
+
+error:
+
+    nxt_mp_destroy(mp);
+    return NXT_ERROR;
+}
+
+
+static nxt_int_t
 nxt_controller_response(nxt_task_t *task, nxt_conn_t *c,
     nxt_controller_response_t *resp)
 {