78 lines
2.7 KiB
C++
78 lines
2.7 KiB
C++
// Copyright 2023 Memgraph Ltd.
|
|
//
|
|
// Use of this software is governed by the Business Source License
|
|
// included in the file licenses/BSL.txt; by using this file, you agree to be bound by the terms of the Business Source
|
|
// License, and you may not use this file except in compliance with the Business Source License.
|
|
//
|
|
// As of the Change Date specified in that file, in accordance with
|
|
// the Business Source License, use of this software will be governed
|
|
// by the Apache License, Version 2.0, included in the file
|
|
// licenses/APL.txt.
|
|
|
|
#pragma once
|
|
|
|
#include <thread>
|
|
|
|
#include <boost/asio/io_context.hpp>
|
|
#include <boost/asio/ip/tcp.hpp>
|
|
|
|
#include "communication/http/listener.hpp"
|
|
#include "io/network/endpoint.hpp"
|
|
|
|
namespace memgraph::communication::http {
|
|
|
|
template <class TRequestHandler, typename TSessionContext>
|
|
class Server final {
|
|
using tcp = boost::asio::ip::tcp;
|
|
|
|
public:
|
|
explicit Server(io::network::Endpoint endpoint, TSessionContext *session_context, ServerContext *context);
|
|
|
|
Server(const Server &) = delete;
|
|
Server(Server &&) = delete;
|
|
Server &operator=(const Server &) = delete;
|
|
Server &operator=(Server &&) = delete;
|
|
|
|
~Server() {
|
|
MG_ASSERT(!background_thread_ || (ioc_.stopped() && !background_thread_->joinable()),
|
|
"Server wasn't shutdown properly");
|
|
}
|
|
|
|
void Start();
|
|
|
|
void Shutdown() { ioc_.stop(); }
|
|
|
|
void AwaitShutdown() {
|
|
if (background_thread_ && background_thread_->joinable()) {
|
|
background_thread_->join();
|
|
}
|
|
}
|
|
bool IsRunning() const { return background_thread_ && !ioc_.stopped(); }
|
|
tcp::endpoint GetEndpoint() const;
|
|
|
|
private:
|
|
boost::asio::io_context ioc_;
|
|
|
|
std::shared_ptr<Listener<TRequestHandler, TSessionContext>> listener_;
|
|
std::optional<std::thread> background_thread_;
|
|
};
|
|
template <class TRequestHandler, typename TSessionContext>
|
|
Server<TRequestHandler, TSessionContext>::Server(io::network::Endpoint endpoint, TSessionContext *session_context,
|
|
ServerContext *context)
|
|
: listener_{Listener<TRequestHandler, TSessionContext>::Create(
|
|
ioc_, session_context, context,
|
|
tcp::endpoint{boost::asio::ip::make_address(endpoint.address), endpoint.port})} {}
|
|
|
|
template <class TRequestHandler, typename TSessionContext>
|
|
void Server<TRequestHandler, TSessionContext>::Start() {
|
|
MG_ASSERT(!background_thread_, "The server was already started!");
|
|
listener_->Run();
|
|
background_thread_.emplace([this] { ioc_.run(); });
|
|
}
|
|
|
|
template <class TRequestHandler, typename TSessionContext>
|
|
boost::asio::ip::tcp::endpoint Server<TRequestHandler, TSessionContext>::GetEndpoint() const {
|
|
return listener_->GetEndpoint();
|
|
}
|
|
} // namespace memgraph::communication::http
|