7939b04117
Reviewers: mislav.bradac, buda Reviewed By: buda Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D632
101 lines
2.6 KiB
C++
101 lines
2.6 KiB
C++
#pragma once
|
|
|
|
#include <atomic>
|
|
#include <cstdio>
|
|
#include <iomanip>
|
|
#include <memory>
|
|
#include <sstream>
|
|
#include <thread>
|
|
|
|
#include <glog/logging.h>
|
|
|
|
#include "io/network/network_error.hpp"
|
|
#include "io/network/stream_reader.hpp"
|
|
|
|
namespace communication {
|
|
|
|
/**
|
|
* TODO (mferencevic): document methods
|
|
*/
|
|
|
|
/**
|
|
* Communication worker.
|
|
* Listens for incomming data on connections and accepts new connections.
|
|
* Also, executes sessions on incomming data.
|
|
*
|
|
* @tparam Session the worker can handle different Sessions, each session
|
|
* represents a different protocol so the same network infrastructure
|
|
* can be used for handling different protocols
|
|
* @tparam Socket the input/output socket that should be used
|
|
* @tparam SessionData the class with objects that will be forwarded to the session
|
|
*/
|
|
template <typename Session, typename Socket, typename SessionData>
|
|
class Worker
|
|
|
|
: public io::network::StreamReader<Worker<Session, Socket, SessionData>,
|
|
Session> {
|
|
using StreamBuffer = io::network::StreamBuffer;
|
|
|
|
public:
|
|
using uptr = std::unique_ptr<Worker<Session, Socket, SessionData>>;
|
|
|
|
Worker(SessionData &session_data) : session_data_(session_data) {}
|
|
|
|
Session &OnConnect(Socket &&socket) {
|
|
DLOG(INFO) << "Accepting connection on socket " << socket.id();
|
|
|
|
// TODO fix session lifecycle handling
|
|
// dangling pointers are not cool :)
|
|
// TODO attach currently active Db
|
|
return *(new Session(std::forward<Socket>(socket), session_data_));
|
|
}
|
|
|
|
void OnError(Session &session) {
|
|
LOG(ERROR) << "Error occured in this session";
|
|
OnClose(session);
|
|
}
|
|
|
|
void OnWaitTimeout() {}
|
|
|
|
void OnRead(Session &session) {
|
|
DLOG(INFO) << "OnRead";
|
|
|
|
try {
|
|
session.Execute();
|
|
} catch (const std::exception &e) {
|
|
LOG(ERROR) << "Error occured while executing statement. " << std::endl
|
|
<< e.what();
|
|
// TODO: report to client
|
|
}
|
|
}
|
|
|
|
void OnClose(Session &session) {
|
|
std::cout << fmt::format("Client {}:{} closed the connection.",
|
|
session.socket_.endpoint().address(),
|
|
session.socket_.endpoint().port())
|
|
<< std::endl;
|
|
// TODO: remove socket from epoll object
|
|
session.Close();
|
|
delete &session;
|
|
}
|
|
|
|
template <class... Args>
|
|
void OnException(Session &, Args &&...) {
|
|
LOG(ERROR) << "Error occured in this session";
|
|
|
|
// TODO: Do something about it
|
|
}
|
|
|
|
std::thread thread_;
|
|
|
|
void Start(std::atomic<bool> &alive) {
|
|
thread_ = std::thread([&, this]() {
|
|
while (alive) this->WaitAndProcessEvents();
|
|
});
|
|
}
|
|
|
|
private:
|
|
SessionData &session_data_;
|
|
};
|
|
}
|