diff options
Diffstat (limited to 'datastore-leveldb')
| -rw-r--r-- | datastore-leveldb/Makefile | 2 | ||||
| -rw-r--r-- | datastore-leveldb/src/server.cpp | 37 |
2 files changed, 34 insertions, 5 deletions
diff --git a/datastore-leveldb/Makefile b/datastore-leveldb/Makefile index 6a6c9a3..6a55822 100644 --- a/datastore-leveldb/Makefile +++ b/datastore-leveldb/Makefile @@ -3,7 +3,7 @@ OUT = bin/server CFLAGS += -Wall -DNO_CGI -DNO_POPEN -DUSE_IPV6 -DNO_SSL CXXFLAGS += -Wall -std=c++11 -Imongoose -LDFLAGS += -lleveldb -lboost_regex +LDFLAGS += -lleveldb -lboost_regex -lboost_program_options ifdef DEBUG CFLAGS += -g -O0 diff --git a/datastore-leveldb/src/server.cpp b/datastore-leveldb/src/server.cpp index 8dd0ee2..c2c702d 100644 --- a/datastore-leveldb/src/server.cpp +++ b/datastore-leveldb/src/server.cpp @@ -5,6 +5,7 @@ #include <thread> #include <boost/regex.hpp> +#include <boost/program_options.hpp> #include "mongoose.h" #include "db.h" @@ -41,14 +42,43 @@ void sigint_handler(int s) { exit(0); } +namespace po = boost::program_options; + int main(int argc, char **argv) { - struct mg_callbacks callbacks; + // Program options + po::options_description desc("Program options"); + desc.add_options() + ("wwwroot", po::value<std::string>()->required(), "Path to www root directory") + ("port", po::value<int>()->default_value(8080), "HTTP Port") + ("help", "Print help message") + ; + + po::variables_map vm; + po::store(po::parse_command_line(argc, argv, desc), vm); + try { + po::notify(vm); + } catch(boost::program_options::required_option& e) { + std::cerr << "ERROR: " << e.what() << std::endl << std::endl; + return 1; + } + + if (vm.count("help")) { + std::cout << "Usage: " << argv[0] << std::endl << std::endl; + std::cout << desc << std::endl; + return 0; + } + const char *port = strdup(std::to_string(vm["wwwroot"].as<int>()).c_str()); + const char *wwwroot = strdup(vm["wwwroot"].as<std::string>().c_str()); + + // Mongoose options const char *options[] = { "num_threads", "8", - "listening_ports", "8080", - "document_root", "wwwroot", + "listening_ports", port, + "document_root", wwwroot, NULL}; + // Mongoose callbacks + struct mg_callbacks callbacks; memset(&callbacks, 0, sizeof(callbacks)); callbacks.begin_request = begin_request_handler; @@ -66,7 +96,6 @@ int main(int argc, char **argv) { web_handle_api_range_size_R, web_handle_api_range_size)); - // Signals: handle C-c struct sigaction action; memset(&action, 0, sizeof(action)); |