Controller: trivial abilities to save and request configuration.

Now you can get current configuration with:

   $ curl 127.0.0.1:8443

and put new configuration with:

   $ curl -X PUT -d @conf.json 127.0.0.1:8443
This commit is contained in:
Valentin Bartenev
2017-05-15 22:39:53 +03:00
parent 952ec2e0ff
commit 0ca2c2fce2
3 changed files with 115 additions and 54 deletions

View File

@@ -12,7 +12,7 @@
typedef struct nxt_conf_json_value_s nxt_conf_json_value_t; typedef struct nxt_conf_json_value_s nxt_conf_json_value_t;
nxt_conf_json_value_t *nxt_conf_json_parse(nxt_buf_mem_t *b, nxt_conf_json_value_t *nxt_conf_json_parse(u_char *pos, size_t length,
nxt_mem_pool_t *pool); nxt_mem_pool_t *pool);
nxt_buf_t *nxt_conf_json_print(nxt_conf_json_value_t *value, nxt_buf_t *nxt_conf_json_print(nxt_conf_json_value_t *value,
nxt_mem_pool_t *pool); nxt_mem_pool_t *pool);

View File

@@ -169,9 +169,9 @@ nxt_conf_json_object_member_get(nxt_lvlhsh_t *lvlhsh, u_char *name,
nxt_conf_json_value_t * nxt_conf_json_value_t *
nxt_conf_json_parse(nxt_buf_mem_t *b, nxt_mem_pool_t *pool) nxt_conf_json_parse(u_char *pos, size_t length, nxt_mem_pool_t *pool)
{ {
u_char *pos, *end; u_char *end;
nxt_conf_json_value_t *value; nxt_conf_json_value_t *value;
value = nxt_mem_alloc(pool, sizeof(nxt_conf_json_value_t)); value = nxt_mem_alloc(pool, sizeof(nxt_conf_json_value_t));
@@ -179,8 +179,7 @@ nxt_conf_json_parse(nxt_buf_mem_t *b, nxt_mem_pool_t *pool)
return NULL; return NULL;
} }
pos = b->pos; end = pos + length;
end = b->free;
pos = nxt_conf_json_skip_space(pos, end); pos = nxt_conf_json_skip_space(pos, end);

View File

@@ -11,14 +11,27 @@
#include <nxt_conf.h> #include <nxt_conf.h>
typedef struct {
nxt_conf_json_value_t *root;
nxt_mem_pool_t *pool;
} nxt_controller_conf_t;
typedef struct { typedef struct {
nxt_http_request_parse_t parser; nxt_http_request_parse_t parser;
size_t length; size_t length;
nxt_conf_json_value_t *conf; nxt_controller_conf_t conf;
} nxt_controller_request_t; } nxt_controller_request_t;
typedef struct {
nxt_str_t status_line;
nxt_conf_json_value_t *json_value;
nxt_str_t json_string;
} nxt_controller_response_t;
static void nxt_controller_conn_init(nxt_task_t *task, void *obj, void *data); static void nxt_controller_conn_init(nxt_task_t *task, void *obj, void *data);
static void nxt_controller_conn_read(nxt_task_t *task, void *obj, void *data); static void nxt_controller_conn_read(nxt_task_t *task, void *obj, void *data);
static nxt_msec_t nxt_controller_conn_timeout_value(nxt_event_conn_t *c, static nxt_msec_t nxt_controller_conn_timeout_value(nxt_event_conn_t *c,
@@ -44,8 +57,8 @@ static void nxt_controller_process_request(nxt_task_t *task,
nxt_event_conn_t *c, nxt_controller_request_t *r); nxt_event_conn_t *c, nxt_controller_request_t *r);
static nxt_int_t nxt_controller_request_body_parse(nxt_task_t *task, static nxt_int_t nxt_controller_request_body_parse(nxt_task_t *task,
nxt_event_conn_t *c, nxt_controller_request_t *r); nxt_event_conn_t *c, nxt_controller_request_t *r);
static void nxt_controller_conf_output(nxt_task_t *task, nxt_event_conn_t *c, static nxt_int_t nxt_controller_response(nxt_task_t *task, nxt_event_conn_t *c,
nxt_controller_request_t *r); nxt_controller_response_t *resp);
static nxt_http_fields_t nxt_controller_request_fields[] = { static nxt_http_fields_t nxt_controller_request_fields[] = {
@@ -55,10 +68,12 @@ static nxt_http_fields_t nxt_controller_request_fields[] = {
{ nxt_null_string, NULL, 0 } { nxt_null_string, NULL, 0 }
}; };
static nxt_http_fields_hash_t *nxt_controller_request_fields_hash; static nxt_http_fields_hash_t *nxt_controller_request_fields_hash;
static nxt_controller_conf_t nxt_controller_conf;
static const nxt_event_conn_state_t nxt_controller_conn_read_state; static const nxt_event_conn_state_t nxt_controller_conn_read_state;
static const nxt_event_conn_state_t nxt_controller_conn_body_read_state; static const nxt_event_conn_state_t nxt_controller_conn_body_read_state;
static const nxt_event_conn_state_t nxt_controller_conn_write_state; static const nxt_event_conn_state_t nxt_controller_conn_write_state;
@@ -68,8 +83,13 @@ static const nxt_event_conn_state_t nxt_controller_conn_close_state;
nxt_int_t nxt_int_t
nxt_controller_start(nxt_task_t *task, nxt_runtime_t *rt) nxt_controller_start(nxt_task_t *task, nxt_runtime_t *rt)
{ {
nxt_mem_pool_t *mp;
nxt_conf_json_value_t *conf;
nxt_http_fields_hash_t *hash; nxt_http_fields_hash_t *hash;
static const nxt_str_t json
= nxt_string("{ \"sockets\": {}, \"applications\": {} }");
hash = nxt_http_fields_hash(nxt_controller_request_fields, rt->mem_pool); hash = nxt_http_fields_hash(nxt_controller_request_fields, rt->mem_pool);
if (nxt_slow_path(hash == NULL)) { if (nxt_slow_path(hash == NULL)) {
@@ -82,6 +102,21 @@ nxt_controller_start(nxt_task_t *task, nxt_runtime_t *rt)
return NXT_ERROR; return NXT_ERROR;
} }
mp = nxt_mem_pool_create(256);
if (nxt_slow_path(mp == NULL)) {
return NXT_ERROR;
}
conf = nxt_conf_json_parse(json.start, json.length, mp);
if (conf == NULL) {
return NXT_ERROR;
}
nxt_controller_conf.root = conf;
nxt_controller_conf.pool = mp;
return NXT_OK; return NXT_OK;
} }
@@ -515,35 +550,41 @@ nxt_controller_request_content_length(void *ctx, nxt_str_t *name,
static void static void
nxt_controller_process_request(nxt_task_t *task, nxt_event_conn_t *c, nxt_controller_process_request(nxt_task_t *task, nxt_event_conn_t *c,
nxt_controller_request_t *r) nxt_controller_request_t *req)
{ {
nxt_buf_t *b; nxt_controller_response_t resp;
static const nxt_str_t resp418 nxt_memzero(&resp, sizeof(nxt_controller_response_t));
= nxt_string("HTTP/1.0 418 I'm a teapot\r\n\r\nerror\r\n");
if (nxt_controller_request_body_parse(task, c, r) != NXT_OK) { if (nxt_str_eq(&req->parser.method, "GET", 3)) {
goto error; nxt_str_set(&resp.status_line, "200 OK");
resp.json_value = nxt_controller_conf.root;
} else if (nxt_str_eq(&req->parser.method, "PUT", 3)) {
if (nxt_controller_request_body_parse(task, c, req) == NXT_OK) {
nxt_mem_pool_destroy(nxt_controller_conf.pool);
nxt_controller_conf = req->conf;
nxt_str_set(&resp.status_line, "201 Created");
nxt_str_set(&resp.json_string,
"{ \"success\": \"Configuration updated\" }");
} else {
nxt_str_set(&resp.status_line, "400 Bad Request");
nxt_str_set(&resp.json_string,
"{ \"error\": \"Invalid JSON\" }");
} }
nxt_controller_conf_output(task, c, r); } else {
nxt_str_set(&resp.status_line, "405 Method Not Allowed");
return; nxt_str_set(&resp.json_string, "{ \"error\": \"Invalid method\" }");
error:
b = nxt_buf_mem_alloc(c->mem_pool, resp418.length, 0);
if (nxt_slow_path(b == NULL)) {
nxt_controller_conn_close(task, c, r);
return;
} }
b->mem.free = nxt_cpymem(b->mem.free, resp418.start, resp418.length); if (nxt_controller_response(task, c, &resp) != NXT_OK) {
nxt_controller_conn_close(task, c, req);
c->write = b; }
c->write_state = &nxt_controller_conn_write_state;
nxt_event_conn_write(task->thread->engine, c);
} }
@@ -551,51 +592,72 @@ static nxt_int_t
nxt_controller_request_body_parse(nxt_task_t *task, nxt_event_conn_t *c, nxt_controller_request_body_parse(nxt_task_t *task, nxt_event_conn_t *c,
nxt_controller_request_t *r) nxt_controller_request_t *r)
{ {
nxt_buf_t *b; nxt_buf_mem_t *mbuf;
nxt_mem_pool_t *mp;
nxt_conf_json_value_t *value; nxt_conf_json_value_t *value;
b = c->read; mp = nxt_mem_pool_create(512);
value = nxt_conf_json_parse(&b->mem, c->mem_pool); if (nxt_slow_path(mp == NULL)) {
return NXT_ERROR;
}
mbuf = &c->read->mem;
value = nxt_conf_json_parse(mbuf->pos, mbuf->free - mbuf->pos, mp);
if (value == NULL) { if (value == NULL) {
return NXT_ERROR; return NXT_ERROR;
} }
r->conf = value; r->conf.root = value;
r->conf.pool = mp;
return NXT_OK; return NXT_OK;
} }
static void static nxt_int_t
nxt_controller_conf_output(nxt_task_t *task, nxt_event_conn_t *c, nxt_controller_response(nxt_task_t *task, nxt_event_conn_t *c,
nxt_controller_request_t *r) nxt_controller_response_t *resp)
{ {
size_t size;
nxt_buf_t *b; nxt_buf_t *b;
static const nxt_str_t head = nxt_string("HTTP/1.0 200 OK\r\n\r\n"); size = sizeof("HTTP/1.0 " "\r\n\r\n") - 1
+ resp->status_line.length
+ resp->json_string.length;
b = nxt_buf_mem_alloc(c->mem_pool, head.length, 0); b = nxt_buf_mem_alloc(c->mem_pool, size, 0);
if (nxt_slow_path(b == NULL)) { if (nxt_slow_path(b == NULL)) {
nxt_controller_conn_close(task, c, r); return NXT_ERROR;
return;
} }
b->mem.free = nxt_cpymem(b->mem.free, head.start, head.length); b->mem.free = nxt_cpymem(b->mem.free, "HTTP/1.0 ", sizeof("HTTP/1.0 ") - 1);
b->mem.free = nxt_cpymem(b->mem.free, resp->status_line.start,
resp->status_line.length);
b->mem.free = nxt_cpymem(b->mem.free, "\r\n\r\n", sizeof("\r\n\r\n") - 1);
if (resp->json_string.length > 0) {
b->mem.free = nxt_cpymem(b->mem.free, resp->json_string.start,
resp->json_string.length);
}
c->write = b; c->write = b;
c->write_state = &nxt_controller_conn_write_state;
b = nxt_conf_json_print(r->conf, c->mem_pool); if (resp->json_value != NULL) {
b = nxt_conf_json_print(resp->json_value, c->mem_pool);
if (b == NULL) { if (b == NULL) {
nxt_controller_conn_close(task, c, r); return NXT_ERROR;
return;
} }
c->write->next = b; c->write->next = b;
c->write_state = &nxt_controller_conn_write_state; }
nxt_event_conn_write(task->thread->engine, c); nxt_event_conn_write(task->thread->engine, c);
return;
return NXT_OK;
} }