5a5ffface3
Summary: Removed old encoder. Changed namespace from bolt to communication::bolt. Removed old include from new encoder. Added an empty message success to encoder. Changed order in communication::Server. Changed bolt session to use new encoder. Merge remote-tracking branch 'origin/dev' into mg_hardcoded_queries Fixed PrintRecordStream. Reviewers: buda, dgleich Reviewed By: buda Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D158
116 lines
3.0 KiB
C++
116 lines
3.0 KiB
C++
#include <signal.h>
|
|
#include <iostream>
|
|
|
|
#include "dbms/dbms.hpp"
|
|
#include "query/engine.hpp"
|
|
|
|
#include "communication/bolt/v1/session.hpp"
|
|
#include "communication/server.hpp"
|
|
|
|
#include "io/network/network_endpoint.hpp"
|
|
#include "io/network/network_error.hpp"
|
|
#include "io/network/socket.hpp"
|
|
|
|
#include "logging/default.hpp"
|
|
#include "logging/streams/stdout.hpp"
|
|
|
|
#include "utils/config/config.hpp"
|
|
#include "utils/signals/handler.hpp"
|
|
#include "utils/stacktrace/log.hpp"
|
|
#include "utils/terminate_handler.hpp"
|
|
|
|
using endpoint_t = io::network::NetworkEndpoint;
|
|
using socket_t = io::network::Socket;
|
|
using session_t = communication::bolt::Session<socket_t>;
|
|
using result_stream_t = communication::bolt::ResultStream<socket_t>;
|
|
using bolt_server_t = communication::Server<session_t, result_stream_t, socket_t>;
|
|
|
|
static bolt_server_t *serverptr;
|
|
|
|
Logger logger;
|
|
|
|
// TODO: load from config
|
|
static constexpr const char *interface = "0.0.0.0";
|
|
static constexpr const char *port = "7687";
|
|
|
|
void throw_and_stacktace(std::string message) {
|
|
Stacktrace stacktrace;
|
|
logger.info(stacktrace.dump());
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
// logging init
|
|
#ifdef SYNC_LOGGER
|
|
logging::init_sync();
|
|
#else
|
|
logging::init_async();
|
|
#endif
|
|
logging::log->pipe(std::make_unique<Stdout>());
|
|
|
|
// get logger
|
|
logger = logging::log->logger("Main");
|
|
logger.info("{}", logging::log->type());
|
|
|
|
// unhandled exception handler init
|
|
std::set_terminate(&terminate_handler);
|
|
|
|
// signal handling init
|
|
SignalHandler::register_handler(Signal::SegmentationFault, []() {
|
|
log_stacktrace("SegmentationFault signal raised");
|
|
std::exit(EXIT_FAILURE);
|
|
});
|
|
SignalHandler::register_handler(Signal::Terminate, []() {
|
|
log_stacktrace("Terminate signal raised");
|
|
std::exit(EXIT_FAILURE);
|
|
});
|
|
SignalHandler::register_handler(Signal::Abort, []() {
|
|
log_stacktrace("Abort signal raised");
|
|
std::exit(EXIT_FAILURE);
|
|
});
|
|
|
|
// register args
|
|
CONFIG_REGISTER_ARGS(argc, argv);
|
|
|
|
// initialize endpoint
|
|
endpoint_t endpoint;
|
|
try {
|
|
endpoint = endpoint_t(interface, port);
|
|
} catch (io::network::NetworkEndpointException &e) {
|
|
logger.error("{}", e.what());
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
// initialize socket
|
|
socket_t socket;
|
|
if (!socket.Bind(endpoint)) {
|
|
logger.error("Cannot bind to socket on {} at {}", interface, port);
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
if (!socket.SetNonBlocking()) {
|
|
logger.error("Cannot set socket to non blocking!");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
if (!socket.Listen(1024)) {
|
|
logger.error("Cannot listen on socket!");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
logger.info("Listening on {} at {}", interface, port);
|
|
|
|
Dbms dbms;
|
|
QueryEngine<result_stream_t> query_engine;
|
|
|
|
// initialize server
|
|
bolt_server_t server(std::move(socket), dbms, query_engine);
|
|
serverptr = &server;
|
|
|
|
// server start with N threads
|
|
// TODO: N should be configurable
|
|
auto N = std::thread::hardware_concurrency();
|
|
logger.info("Starting {} workers", N);
|
|
server.Start(N);
|
|
|
|
logger.info("Shutting down...");
|
|
return EXIT_SUCCESS;
|
|
}
|