mirror of
https://github.com/cesanta/mongoose.git
synced 2026-10-11 17:04:33 +07:00
Revert "Stop publish examples to mongoose repo"
This reverts commit 1a17e17c462bdd4e1d26d8742f8b7087273e04c2. PUBLISHED_FROM=80028de308c9a021955d1425d2bfee8feb85f193
This commit is contained in:
Alexander Alashkin
committed by
Cesanta Bot
Cesanta Bot
parent
7cc7df8c0e
commit
eaef5bd133
@@ -0,0 +1,28 @@
|
||||
PROG = api_server
|
||||
SOURCES = $(PROG).c sqlite3.c db_plugin_sqlite.c ../../mongoose.c
|
||||
CFLAGS = -W -Wall $(CFLAGS_EXTRA)
|
||||
|
||||
|
||||
ifeq ($(OS), Windows_NT)
|
||||
CFLAGS += -lws2_32
|
||||
CC = gcc
|
||||
else
|
||||
UNAME_S := $(shell uname -s)
|
||||
ifeq ($(UNAME_S), Linux)
|
||||
CFLAGS += -ldl -lm -pthread
|
||||
endif
|
||||
endif
|
||||
|
||||
all: $(PROG)
|
||||
|
||||
$(PROG): $(SOURCES)
|
||||
$(CC) $(SOURCES) -o $@ $(CFLAGS)
|
||||
|
||||
$(PROG).exe: $(SOURCES)
|
||||
cl $(SOURCES) /I.. /MD /Fe$@
|
||||
|
||||
test: $(PROG)
|
||||
sh unit_test.sh $$(pwd)/$(PROG)
|
||||
|
||||
clean:
|
||||
rm -rf *.gc* *.dSYM *.exe *.obj *.o a.out $(PROG)
|
||||
@@ -0,0 +1,104 @@
|
||||
/*
|
||||
* 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;
|
||||
}
|
||||
@@ -0,0 +1,19 @@
|
||||
/*
|
||||
* Copyright (c) 2014 Cesanta Software Limited
|
||||
* All rights reserved
|
||||
*/
|
||||
|
||||
#ifndef CS_MONGOOSE_EXAMPLES_API_SERVER_DB_PLUGIN_H_
|
||||
#define CS_MONGOOSE_EXAMPLES_API_SERVER_DB_PLUGIN_H_
|
||||
|
||||
#include "../../mongoose.h"
|
||||
|
||||
void *db_open(const char *db_path);
|
||||
void db_close(void **db_handle);
|
||||
|
||||
enum { API_OP_GET, API_OP_SET, API_OP_DEL };
|
||||
|
||||
void db_op(struct mg_connection *nc, const struct http_message *hm,
|
||||
const struct mg_str *key, void *db, int op);
|
||||
|
||||
#endif /* CS_MONGOOSE_EXAMPLES_API_SERVER_DB_PLUGIN_H_ */
|
||||
@@ -0,0 +1,119 @@
|
||||
/*
|
||||
* Copyright (c) 2014 Cesanta Software Limited
|
||||
* All rights reserved
|
||||
*/
|
||||
|
||||
#include "db_plugin.h"
|
||||
#include "sqlite3.h"
|
||||
|
||||
void *db_open(const char *db_path) {
|
||||
sqlite3 *db = NULL;
|
||||
if (sqlite3_open_v2(db_path, &db, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE |
|
||||
SQLITE_OPEN_FULLMUTEX,
|
||||
NULL) == SQLITE_OK) {
|
||||
sqlite3_exec(db, "CREATE TABLE IF NOT EXISTS kv(key PRIMARY KEY, val);", 0,
|
||||
0, 0);
|
||||
}
|
||||
return db;
|
||||
}
|
||||
|
||||
void db_close(void **db_handle) {
|
||||
if (db_handle != NULL && *db_handle != NULL) {
|
||||
sqlite3_close(*db_handle);
|
||||
*db_handle = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
static void op_set(struct mg_connection *nc, const struct http_message *hm,
|
||||
const struct mg_str *key, void *db) {
|
||||
sqlite3_stmt *stmt = NULL;
|
||||
char value[200];
|
||||
const struct mg_str *body =
|
||||
hm->query_string.len > 0 ? &hm->query_string : &hm->body;
|
||||
|
||||
mg_get_http_var(body, "value", value, sizeof(value));
|
||||
if (sqlite3_prepare_v2(db, "INSERT OR REPLACE INTO kv VALUES (?, ?);", -1,
|
||||
&stmt, NULL) == SQLITE_OK) {
|
||||
sqlite3_bind_text(stmt, 1, key->p, key->len, SQLITE_STATIC);
|
||||
sqlite3_bind_text(stmt, 2, value, strlen(value), SQLITE_STATIC);
|
||||
sqlite3_step(stmt);
|
||||
sqlite3_finalize(stmt);
|
||||
}
|
||||
mg_printf(nc, "%s", "HTTP/1.1 200 OK\r\nContent-Length: 0\r\n\r\n");
|
||||
}
|
||||
|
||||
static void op_get(struct mg_connection *nc, const struct http_message *hm,
|
||||
const struct mg_str *key, void *db) {
|
||||
sqlite3_stmt *stmt = NULL;
|
||||
const char *data = NULL;
|
||||
int result;
|
||||
(void) hm;
|
||||
|
||||
if (sqlite3_prepare_v2(db, "SELECT val FROM kv WHERE key = ?;", -1, &stmt,
|
||||
NULL) == SQLITE_OK) {
|
||||
sqlite3_bind_text(stmt, 1, key->p, key->len, SQLITE_STATIC);
|
||||
result = sqlite3_step(stmt);
|
||||
data = (char *) sqlite3_column_text(stmt, 0);
|
||||
if ((result == SQLITE_OK || result == SQLITE_ROW) && data != NULL) {
|
||||
mg_printf(nc,
|
||||
"HTTP/1.1 200 OK\r\n"
|
||||
"Content-Type: text/plain\r\n"
|
||||
"Content-Length: %d\r\n\r\n%s",
|
||||
(int) strlen(data), data);
|
||||
} else {
|
||||
mg_printf(nc, "%s",
|
||||
"HTTP/1.1 404 Not Found\r\n"
|
||||
"Content-Length: 0\r\n\r\n");
|
||||
}
|
||||
sqlite3_finalize(stmt);
|
||||
} else {
|
||||
mg_printf(nc, "%s",
|
||||
"HTTP/1.1 500 Server Error\r\n"
|
||||
"Content-Length: 0\r\n\r\n");
|
||||
}
|
||||
}
|
||||
|
||||
static void op_del(struct mg_connection *nc, const struct http_message *hm,
|
||||
const struct mg_str *key, void *db) {
|
||||
sqlite3_stmt *stmt = NULL;
|
||||
int result;
|
||||
(void) hm;
|
||||
|
||||
if (sqlite3_prepare_v2(db, "DELETE FROM kv WHERE key = ?;", -1, &stmt,
|
||||
NULL) == SQLITE_OK) {
|
||||
sqlite3_bind_text(stmt, 1, key->p, key->len, SQLITE_STATIC);
|
||||
result = sqlite3_step(stmt);
|
||||
if (result == SQLITE_OK || result == SQLITE_ROW) {
|
||||
mg_printf(nc, "%s", "HTTP/1.1 200 OK\r\nContent-Length: 0\r\n\r\n");
|
||||
} else {
|
||||
mg_printf(nc, "%s",
|
||||
"HTTP/1.1 404 Not Found\r\n"
|
||||
"Content-Length: 0\r\n\r\n");
|
||||
}
|
||||
sqlite3_finalize(stmt);
|
||||
} else {
|
||||
mg_printf(nc, "%s",
|
||||
"HTTP/1.1 500 Server Error\r\n"
|
||||
"Content-Length: 0\r\n\r\n");
|
||||
}
|
||||
}
|
||||
|
||||
void db_op(struct mg_connection *nc, const struct http_message *hm,
|
||||
const struct mg_str *key, void *db, int op) {
|
||||
switch (op) {
|
||||
case API_OP_GET:
|
||||
op_get(nc, hm, key, db);
|
||||
break;
|
||||
case API_OP_SET:
|
||||
op_set(nc, hm, key, db);
|
||||
break;
|
||||
case API_OP_DEL:
|
||||
op_del(nc, hm, key, db);
|
||||
break;
|
||||
default:
|
||||
mg_printf(nc, "%s",
|
||||
"HTTP/1.0 501 Not Implemented\r\n"
|
||||
"Content-Length: 0\r\n\r\n");
|
||||
break;
|
||||
}
|
||||
}
|
||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,35 @@
|
||||
#!/bin/sh
|
||||
|
||||
PROG=$1
|
||||
PORT=${2:-8000} # If second param is given, this is load balancer port
|
||||
DB_FILE=/tmp/_$$.db
|
||||
URL=http://127.0.0.1:$PORT/api/v1
|
||||
|
||||
cleanup() {
|
||||
rm -rf $DB_FILE
|
||||
kill -9 $PID >/dev/null 2>&1
|
||||
}
|
||||
|
||||
#set -x
|
||||
trap cleanup EXIT
|
||||
|
||||
cleanup
|
||||
$PROG -f $DB_FILE &
|
||||
PID=$!
|
||||
|
||||
#sleep 1
|
||||
curl -s -X PUT -d 'value=123' $URL/foo
|
||||
curl -s -X PUT -d 'value=success' $URL/bar/baz
|
||||
|
||||
# Fetch existing key
|
||||
RESULT=$(curl -s $URL/bar/baz)
|
||||
test "$RESULT" = "success" || exit 1
|
||||
|
||||
# Delete it
|
||||
curl -s -X DELETE $URL/bar/baz
|
||||
|
||||
# Make sure it's deleted - GET must result in 404
|
||||
RESULT=$(curl -s -i $URL/bar/baz | head -1 | tr -d '\r')
|
||||
test "$RESULT" = "HTTP/1.1 404 Not Found" || exit 1
|
||||
|
||||
exit 0
|
||||
@@ -0,0 +1,38 @@
|
||||
<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="utf-8" />
|
||||
<title>RESTful API demo</title>
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1" />
|
||||
<style type="text/css">
|
||||
* { outline: none; }
|
||||
body {
|
||||
background-color: #789; margin: 0;
|
||||
padding: 0; font: 16px/1.4 Helvetica, Arial, sans-serif;
|
||||
font: 16px/1.4 Helvetica, Arial, sans-serif;
|
||||
}
|
||||
div.content {
|
||||
width: 800px; margin: 2em auto; padding: 20px 50px;
|
||||
background-color: #fff; border-radius: 1em;
|
||||
}
|
||||
label { display: inline-block; min-width: 7em; }
|
||||
input { border: 1px solid #ccc; padding: 0.2em; }
|
||||
a:link, a:visited { color: #69c; text-decoration: none; }
|
||||
@media (max-width: 700px) {
|
||||
body { background-color: #fff; }
|
||||
div.content { width: auto; margin: 0 auto; padding: 1em; }
|
||||
}
|
||||
</style>
|
||||
|
||||
<script src="http://code.jquery.com/jquery-1.11.0.min.js"></script>
|
||||
<script language="javascript" type="text/javascript">
|
||||
jQuery(function() {
|
||||
});
|
||||
</script>
|
||||
</head>
|
||||
<body>
|
||||
<div class="content">
|
||||
<h1>API server</h1>
|
||||
</div>
|
||||
</body>
|
||||
</html>
|
||||
Reference in New Issue
Block a user