cpp_backend/main.cpp

114 lines
3.1 KiB
C++
Raw Normal View History

#include "src/sdk.h"
//
#include <boost/asio/signal_set.hpp>
#include <iostream>
#include <mutex>
#include <thread>
#include <vector>
#include "src/http_server.h"
namespace
{
namespace net = boost::asio;
using namespace std::literals;
namespace sys = boost::system;
namespace http = boost::beast::http;
2023-09-13 14:57:22 +02:00
using namespace std;
using net::ip::tcp;
// Запрос, тело которого представлено в виде строки
using StringRequest = http::request<http::string_body>;
// Ответ, тело которого представлено в виде строки
using StringResponse = http::response<http::string_body>;
2024-01-20 10:24:00 +01:00
struct content_type
{
2024-01-20 10:24:00 +01:00
content_type() = delete;
constexpr static std::string_view k_TextHTML = "text/html"sv;
2024-03-20 14:55:57 +01:00
constexpr static std::string_view k_JSON = "application/json"sv;
2024-01-20 10:24:00 +01:00
// При необходимости внутрь content_type можно добавить и другие типы контента
};
StringResponse HandleRequest(StringRequest&& req)
{
2024-01-20 10:46:22 +01:00
auto route = req.target();
2024-03-20 14:55:57 +01:00
switch (true)
{
default:
{
StringResponse res(http::status::bad_request, 1);
res.set(http::field::content_type, content_type::k_JSON);
string body = "{\n"
" \"code\": \"badRequest\",\n"
" \"message\": \"Bad request\"\n"
"} "s;
res.body() = body;
res.content_length(body.size());
res.keep_alive(true);
return res;
}
}
}
// Запускает функцию fn на n потоках, включая текущий
template<typename Fn>
void RunWorkers(unsigned n, const Fn& fn)
{
n = std::max(1u, n);
std::vector<std::jthread> workers;
workers.reserve(n - 1);
// Запускаем n-1 рабочих потоков, выполняющих функцию fn
while (--n)
{
workers.emplace_back(fn);
}
fn();
}
} // namespace
int main()
{
const unsigned num_threads = std::thread::hardware_concurrency();
net::io_context ioc(num_threads);
2024-02-21 03:57:28 +01:00
tcp::acceptor acceptor(net::make_strand(ioc));
// Подписываемся на сигналы и при их получении завершаем работу сервера
net::signal_set signals(ioc, SIGINT, SIGTERM);
2023-11-04 00:05:40 +01:00
signals.async_wait([&ioc](const sys::error_code& ec,
[[maybe_unused]] int signal_number)
{
if (!ec)
{
2023-11-04 00:05:40 +01:00
cout << "Signal "sv << signal_number << " received"sv
<< endl;
ioc.stop();
}
});
2024-02-21 03:57:28 +01:00
const auto address = net::ip::make_address("0.0.0.0");
constexpr net::ip::port_type port = 8080;
http_server::ServeHttp(ioc, {address, port}, [](auto&& req, auto&& sender)
{
sender(HandleRequest(std::forward<decltype(req)>(req)));
});
2023-09-13 14:57:22 +02:00
net::steady_timer t {ioc, 30s};
t.async_wait([](sys::error_code ec)
{
cout << "Timer expired"sv << endl;
});
std::cout << "Server has started..."sv << std::endl;
RunWorkers(num_threads, [&ioc]
{
ioc.run();
});
}