summaryrefslogtreecommitdiff
path: root/datastore-leveldb
diff options
context:
space:
mode:
Diffstat (limited to 'datastore-leveldb')
-rw-r--r--datastore-leveldb/Makefile4
-rw-r--r--datastore-leveldb/src/main.cpp103
2 files changed, 39 insertions, 68 deletions
diff --git a/datastore-leveldb/Makefile b/datastore-leveldb/Makefile
index 33f61bd..3117123 100644
--- a/datastore-leveldb/Makefile
+++ b/datastore-leveldb/Makefile
@@ -1,10 +1,10 @@
CC = gcc
CPP = foo
-OBJ= src/http_parser.o src/server_eh.o src/main.o mongoose/mongoose.o
+OBJ= src/main.o mongoose/mongoose.o
OUT = main
CFLAGS =
CXXFLAGS = -std=c++11 -Imongoose
-LDFLAGS = -lleveldb -lev -lboost_regex -lmagic
+LDFLAGS = -lleveldb -lboost_regex
all: build
diff --git a/datastore-leveldb/src/main.cpp b/datastore-leveldb/src/main.cpp
index dab88cd..f7f2f5d 100644
--- a/datastore-leveldb/src/main.cpp
+++ b/datastore-leveldb/src/main.cpp
@@ -2,7 +2,6 @@ extern "C" {
#include <unistd.h>
#include <signal.h>
#include <string.h>
-#include <magic.h>
#include "mongoose.h"
}
@@ -25,8 +24,6 @@ std::forward_list<std::pair<boost::regex,std::function<void(const boost::cmatch&
static std::map<std::string,leveldb::DB*> dbs;
-static magic_t magic_cookie;
-
struct mg_context *ctx;
bool sensor_name_is_sane(std::string& name) {
@@ -67,26 +64,6 @@ leveldb::DB *getDB(std::string& name) {
}
}
-// see http_parser.h
-// char DELETE = 0;
-// char GET = 1;
-// char HEAD = 2;
-// char POST = 3;
-// char PUT = 4;
-
-
-static inline void http_ok(int fd, const char *content_type, const char *extra_headers) {
-#define write_const(fd, text) \
- write(fd, text, strlen(text));
-
- write_const(fd, "HTTP/1.1 200 OK\r\nContent-Type: ");
- write_const(fd, content_type);
- write_const(fd, "\r\n");
- if (extra_headers) write_const(fd, extra_headers);
- write_const(fd, "\r\n");
-#undef write_const
-}
-
std::string make_key(uint64_t timestamp) {
std::stringstream key;
key << "ts-";
@@ -129,11 +106,11 @@ void web_handle_api_range(const boost::cmatch &match, struct mg_connection *conn
mg_printf(conn, "HTTP/1.1 500 Internal Error\r\n\r\n");
return;
}
-
+
mg_printf(conn,
- "HTTP/1.1 200 Value received\r\n"
- "Content-Type: application/json; encoding=UTF-8\r\n"
- "\r\n");
+ "HTTP/1.1 200 Value received\r\n"
+ "Content-Type: application/json; encoding=UTF-8\r\n"
+ "\r\n");
std::cout << "sensor=" << sensor << " start=" << start << " end=" << end << std::endl;
@@ -145,33 +122,33 @@ void web_handle_api_range(const boost::cmatch &match, struct mg_connection *conn
bool first = true;
std::ostringstream outbuf;
for (it->Seek(key_start);
- it->Valid() && cmp->Compare(it->key(), key_end) < 0;
- it->Next()) {
- const char *key = it->key().data();
- size_t key_size = it->key().size();
- if (key_size != 20+3) {
- std::cerr << "invalid key" << std::endl;
- return;
- }
-
- int offset = 3; //ts-
- // skip zeros in timestamp
- while (offset < key_size-1 and *(key+offset) == '0')
- offset++;
-
- if (first)
- first = false;
- else
- outbuf << ',';
-
- outbuf << '[';
- outbuf.write(key+offset, key_size-offset);
+ it->Valid() && cmp->Compare(it->key(), key_end) < 0;
+ it->Next()) {
+ const char *key = it->key().data();
+ size_t key_size = it->key().size();
+ if (key_size != 20+3) {
+ std::cerr << "invalid key" << std::endl;
+ return;
+ }
+
+ int offset = 3; //ts-
+ // skip zeros in timestamp
+ while (offset < key_size-1 and *(key+offset) == '0')
+ offset++;
+
+ if (first)
+ first = false;
+ else
outbuf << ',';
- outbuf.write(it->value().data(), it->value().size());
- outbuf << ']';
- mg_write(conn, outbuf.str().c_str(), outbuf.tellp());
- outbuf.seekp(0);
+ outbuf << '[';
+ outbuf.write(key+offset, key_size-offset);
+ outbuf << ',';
+ outbuf.write(it->value().data(), it->value().size());
+ outbuf << ']';
+
+ mg_write(conn, outbuf.str().c_str(), outbuf.tellp());
+ outbuf.seekp(0);
}
mg_printf(conn, "]}\r\n");
delete it;
@@ -188,11 +165,13 @@ int begin_request_handler(struct mg_connection *conn) {
return 1;
}
}
-
+
return 0;
}
+struct sigaction old_action;
+
void sigint_handler(int s) {
// Stop the server.
mg_stop(ctx);
@@ -203,7 +182,6 @@ void sigint_handler(int s) {
delete (*it).second;
dbs.erase(it);
}
-
exit(0);
}
@@ -227,22 +205,15 @@ int main(int argc, char **argv) {
boost::regex("/api/range/([a-zA-Z0-9]+)/([0-9]+)/([0-9]+)"),
web_handle_api_range));
- magic_cookie = magic_open(MAGIC_MIME_TYPE);
- magic_load(magic_cookie, NULL);
-
// Signals: handle C-c
- struct sigaction on_sigint;
- on_sigint.sa_handler = sigint_handler;
- sigemptyset(&on_sigint.sa_mask);
- on_sigint.sa_flags = 0;
-
- sigaction(SIGINT, &on_sigint, NULL);
-
+ struct sigaction action;
+ memset(&action, 0, sizeof(action));
+ action.sa_handler = &sigint_handler;
+ sigaction(SIGINT, &action, &old_action);
// Start the web server.
ctx = mg_start(&callbacks, NULL, options);
-
- while (1) sleep(1);
+ pause();
return 1;
}