mirror of
https://github.com/cesanta/mongoose.git
synced 2024-12-27 19:17:48 +08:00
105 lines
3.0 KiB
C
105 lines
3.0 KiB
C
/*
|
|
* Copyright (c) 2014 Cesanta Software Limited
|
|
* All rights reserved
|
|
*/
|
|
|
|
#include "db_plugin.h"
|
|
|
|
static const char *s_http_port = "8000";
|
|
static struct mg_serve_http_opts s_http_server_opts;
|
|
static int s_sig_num = 0;
|
|
static void *s_db_handle = NULL;
|
|
static const char *s_db_path = "api_server.db";
|
|
static const struct mg_str s_get_method = MG_MK_STR("GET");
|
|
static const struct mg_str s_put_method = MG_MK_STR("PUT");
|
|
static const struct mg_str s_delele_method = MG_MK_STR("DELETE");
|
|
|
|
static void signal_handler(int sig_num) {
|
|
signal(sig_num, signal_handler);
|
|
s_sig_num = sig_num;
|
|
}
|
|
|
|
static int has_prefix(const struct mg_str *uri, const struct mg_str *prefix) {
|
|
return uri->len > prefix->len && memcmp(uri->p, prefix->p, prefix->len) == 0;
|
|
}
|
|
|
|
static int is_equal(const struct mg_str *s1, const struct mg_str *s2) {
|
|
return s1->len == s2->len && memcmp(s1->p, s2->p, s2->len) == 0;
|
|
}
|
|
|
|
static void ev_handler(struct mg_connection *nc, int ev, void *ev_data) {
|
|
static const struct mg_str api_prefix = MG_MK_STR("/api/v1");
|
|
struct http_message *hm = (struct http_message *) ev_data;
|
|
struct mg_str key;
|
|
|
|
switch (ev) {
|
|
case MG_EV_HTTP_REQUEST:
|
|
if (has_prefix(&hm->uri, &api_prefix)) {
|
|
key.p = hm->uri.p + api_prefix.len;
|
|
key.len = hm->uri.len - api_prefix.len;
|
|
if (is_equal(&hm->method, &s_get_method)) {
|
|
db_op(nc, hm, &key, s_db_handle, API_OP_GET);
|
|
} else if (is_equal(&hm->method, &s_put_method)) {
|
|
db_op(nc, hm, &key, s_db_handle, API_OP_SET);
|
|
} else if (is_equal(&hm->method, &s_delele_method)) {
|
|
db_op(nc, hm, &key, s_db_handle, API_OP_DEL);
|
|
} else {
|
|
mg_printf(nc, "%s",
|
|
"HTTP/1.0 501 Not Implemented\r\n"
|
|
"Content-Length: 0\r\n\r\n");
|
|
}
|
|
} else {
|
|
mg_serve_http(nc, hm, s_http_server_opts); /* Serve static content */
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
struct mg_mgr mgr;
|
|
struct mg_connection *nc;
|
|
int i;
|
|
|
|
/* Open listening socket */
|
|
mg_mgr_init(&mgr, NULL);
|
|
nc = mg_bind(&mgr, s_http_port, ev_handler);
|
|
mg_set_protocol_http_websocket(nc);
|
|
s_http_server_opts.document_root = "web_root";
|
|
|
|
/* Parse command line arguments */
|
|
for (i = 1; i < argc; i++) {
|
|
if (strcmp(argv[i], "-D") == 0) {
|
|
mgr.hexdump_file = argv[++i];
|
|
} else if (strcmp(argv[i], "-f") == 0) {
|
|
s_db_path = argv[++i];
|
|
} else if (strcmp(argv[i], "-r") == 0) {
|
|
s_http_server_opts.document_root = argv[++i];
|
|
}
|
|
}
|
|
|
|
signal(SIGINT, signal_handler);
|
|
signal(SIGTERM, signal_handler);
|
|
|
|
/* Open database */
|
|
if ((s_db_handle = db_open(s_db_path)) == NULL) {
|
|
fprintf(stderr, "Cannot open DB [%s]\n", s_db_path);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
/* Run event loop until signal is received */
|
|
printf("Starting RESTful server on port %s\n", s_http_port);
|
|
while (s_sig_num == 0) {
|
|
mg_mgr_poll(&mgr, 1000);
|
|
}
|
|
|
|
/* Cleanup */
|
|
mg_mgr_free(&mgr);
|
|
db_close(&s_db_handle);
|
|
|
|
printf("Exiting on signal %d\n", s_sig_num);
|
|
|
|
return 0;
|
|
}
|