diff options
author | neodarz <neodarz@neodarz.net> | 2019-08-07 11:03:58 +0200 |
---|---|---|
committer | neodarz <neodarz@neodarz.net> | 2019-08-07 11:03:58 +0200 |
commit | 4c65baa7b6e0a85635f8028a5ff9979b870d7be0 (patch) | |
tree | 01a285bb270ada1363ca56c00538ef8b56c5e821 /umosapi.cpp | |
parent | 7dc4e5bfc85474700b3731fc06e48a0e5c1d387b (diff) | |
download | umosapicpp-4c65baa7b6e0a85635f8028a5ff9979b870d7be0.tar.xz umosapicpp-4c65baa7b6e0a85635f8028a5ff9979b870d7be0.zip |
Refactoring umosapi service
Diffstat (limited to 'umosapi.cpp')
-rw-r--r-- | umosapi.cpp | 244 |
1 files changed, 0 insertions, 244 deletions
diff --git a/umosapi.cpp b/umosapi.cpp deleted file mode 100644 index 72c4728..0000000 --- a/umosapi.cpp +++ /dev/null @@ -1,244 +0,0 @@ -#include "umosapi.h" - -#include <string.h> -#include <unistd.h> -#include <sys/types.h> -#include <pwd.h> - -#include <bsoncxx/builder/stream/document.hpp> -#include <bsoncxx/json.hpp> - -#include <mongocxx/client.hpp> -#include <mongocxx/instance.hpp> - -#include <mongocxx/client.hpp> -#include <mongocxx/stdx.hpp> -#include <mongocxx/uri.hpp> - -#include <json-c/json.h> - -#include <nlohmann/json.hpp> - -#include "clara.hpp" -#include "shared.h" -#include "config.h" - -std::map<std::string, std::string> config; - -using namespace std; -using namespace Pistache; - -using bsoncxx::builder::stream::close_array; -using bsoncxx::builder::stream::close_document; -using bsoncxx::builder::stream::document; -using bsoncxx::builder::stream::finalize; -using bsoncxx::builder::stream::open_array; -using bsoncxx::builder::stream::open_document; - -using json = nlohmann::json; - -using namespace clara; - -namespace Generic { - void handleReady(const Rest::Request&, Http::ResponseWriter response) { - response.send(Http::Code::Ok, "1"); - } -} - -UmosapiService::UmosapiService(Address addr) - : httpEndpoint(std::make_shared<Http::Endpoint>(addr)) - , desc("Unity Mongo Save API", "0.1") -{ } - -void UmosapiService::init(size_t thr = 2) { - auto opts = Http::Endpoint::options() - .threads(thr); - httpEndpoint->init(opts); - createDescription(); -} - -void UmosapiService::start(std::string swaggerui) { - router.initFromDescription(desc); - - Rest::Swagger swagger(desc); - swagger - .uiPath("/doc") - .uiDirectory(swaggerui) - .apiPath("/api") - .serializer(&Rest::Serializer::rapidJson) - .install(router); - - httpEndpoint->setHandler(router.handler()); - httpEndpoint->serve(); -} - -void UmosapiService::createDescription() { - desc - .info() - .license("Apache", "https://neodarz.net"); - - auto backendErrorResponse = - desc.response(Http::Code::Internal_Server_Error, "Backend is dead, it's the end of the world"); - - desc - .schemes(Rest::Scheme::Http) - .basePath("/v1") - .produces(MIME(Application, Json)) - .consumes(MIME(Application, Json)); - - desc - .route(desc.get("/ready")) - .bind(&Generic::handleReady) - .response(Http::Code::Ok, "Api started") - .response(backendErrorResponse); - - auto versionPath = desc.path("/v1"); - - versionPath - .route(desc.get("/:mcollection")) - .bind(&UmosapiService::UmosapiService::retrieveAll, this) - .produces(MIME(Application, Json)) - .parameter<Rest::Type::String>("mcollection", "Name of the collection where the uobjects are located") - .response(Http::Code::Ok, "List of uobjects") - .response(backendErrorResponse); - - versionPath - .route(desc.post("/:mcollection")) - .bind(&UmosapiService::UmosapiService::addUObject, this) - .produces(MIME(Application, Json)) - .parameter<Rest::Type::String>("mcollection", "Name of the collection where the uobjects are located") - .response(Http::Code::Ok, "Uobject created") - .response(backendErrorResponse); -} - -void UmosapiService::retrieveAll(const Rest::Request& request, Http::ResponseWriter response) { - mongocxx::client conn{mongocxx::uri{config["mongoURI"]}}; - - auto collection = conn[config["mongo_db"]][request.param(":mcollection").as<string>()]; - - auto cursor = collection.find({}); - - auto jsonObjects = json_object_new_array(); - - for (auto&& doc : cursor) { - json_object_array_add(jsonObjects, json_tokener_parse(bsoncxx::to_json(doc).c_str())); - } - - response.send(Http::Code::Ok, json_object_to_json_string_ext(jsonObjects, JSON_C_TO_STRING_SPACED | JSON_C_TO_STRING_PRETTY), MIME(Application, Json)); - json_object_put(jsonObjects); -} - -void UmosapiService::addUObject(const Rest::Request& request, Http::ResponseWriter response) { - mongocxx::client conn{mongocxx::uri{config["mongoURI"]}}; - - auto collection = conn[config["mongo_db"]][request.param(":mcollection").as<string>()]; - - auto document = bsoncxx::from_json(request.body().c_str()); - - auto result = collection.insert_one(document.view()); - - std::string oid = result->inserted_id().get_oid().value.to_string(); - - auto jsonObject = json_object_new_object(); - - int stringlen = 0; - stringlen = strlen( request.body().c_str()); - struct json_tokener *tok = json_tokener_new(); - enum json_tokener_error jerr; - - json_object *json_datas = NULL; - - do { - json_datas = json_tokener_parse_ex(tok, request.body().c_str(), stringlen); - } while ((jerr = json_tokener_get_error(tok)) == json_tokener_continue); - - if (jerr != json_tokener_success) - { - fprintf(stderr, "Error: %s\n", json_tokener_error_desc(jerr)); - } - - if (json_object_get_type(json_datas) == json_type_object) { - /* Create an object with the following template: - * { - * "_id": { - * "$oid": "5d484d371ec4865f767d8424" - * }, - * "datas": { - * [...] - * } - * } - */ - - auto json_id = json_object_new_object(); - auto json_oid = json_object_new_string(oid.c_str()); - - json_object_object_add(json_id, "$oid", json_oid); - json_object_object_add(jsonObject, "_id", json_id); - json_object_object_add(jsonObject, "datas", json_datas); - } else { - cout << typeid(json_datas).name() << endl; - cout << "json_datas type: " << json_type_to_name(json_object_get_type(json_datas)) << endl; - } - json_tokener_reset(tok); - - response.send(Http::Code::Ok, json_object_to_json_string_ext(jsonObject, JSON_C_TO_STRING_SPACED | JSON_C_TO_STRING_PRETTY), MIME(Application, Json)); - json_object_put(jsonObject); -} - - -int main(int argc, char *argv[]) { - - string config_path = ""; - const char *homedir; - - if ((homedir = getenv("XDG_CONFIG_HOME")) == NULL || (homedir = getenv("HOME")) == NULL) { - homedir = getpwuid(getuid())->pw_dir; - } - - config_path.append(homedir); - config_path.append("/.config/umosapi/config.txt"); - - bool showHelp = false; - int config_port = 9080; - int thr = 2; - auto cli = clara::detail::Help(showHelp) - | clara::detail::Opt( config_path, "config" )["-c"]["--config"]("Config file path. Default `~/.config/umosapi/config.txt`.") - | clara::detail::Opt( config_port, "port" )["-p"]["--port"]("Port to listen. Default: `9080`.") - | clara::detail::Opt( thr, "treads" )["-t"]["--threads"]("Number of threads. Default: `2`."); - auto result = cli.parse( clara::detail::Args( argc, argv ) ); - if( !result ) - { - std::cerr << "Error in command line: " << result.errorMessage() << std::endl; - std::cerr << cli << std::endl; - exit(1); - } - - if ( showHelp ) { - std::cerr << cli << std::endl; - exit(1); - } - - Address addr(Ipv4::any(), Port(config_port)); - - cout << "Using " << hardware_concurrency() << " cores"; - cout << " - " << thr << " threads" << endl; - cout << "Listen on 0.0.0.0:" << config_port << endl; - - cout << "Using config file '" << config_path << "'" << endl; - - if (!std::filesystem::exists(config_path)) { - cout << "Error fatal : config file '" << config_path << "' not found" << endl; - cout << "config.txt is search here: ~/.config/umosapi/config.txt" << endl; - exit (EXIT_FAILURE); - } - - load_config(config_path); - - cout << "Using swaggerui " << config["swaggerui"] << " path" << endl; - cout << "Using mongoURI " << config["mongoURI"] << endl; - - UmosapiService umosapi(addr); - - umosapi.init(thr); - umosapi.start(config["swaggerui"]); -} |