opengris-scaler/scaler/object_storage/tests/test_server_main.cpp at 538a85d62a97e521f0db335626c4758bcf0682ac · finos/opengris-scaler · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#include <boost/asio/awaitable.hpp>
#include <boost/asio/co_spawn.hpp>
#include <boost/asio/detached.hpp>
#include <boost/asio/io_context.hpp>
#include <boost/asio/ip/address.hpp>
#include <boost/asio/ip/tcp.hpp>
#include <boost/asio/read.hpp>
#include <boost/asio/signal_set.hpp>
#include <boost/asio/this_coro.hpp>
#include <boost/system/system_error.hpp>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>

#include "../constants.h"
#include "../io_helper.h"
#include "../object_storage_server.h"
#include "argparse/argparse.hpp"
#include "version.h"

// Helper macros to stringify the macro value
#define STRINGIFY_HELPER(x) #x
#define STRINGIFY(x)        STRINGIFY_HELPER(x)

using boost::asio::awaitable;
using boost::asio::co_spawn;
using boost::asio::detached;
using boost::asio::use_awaitable;
using boost::asio::ip::tcp;
namespace this_coro = boost::asio::this_coro;

#if defined(BOOST_ASIO_ENABLE_HANDLER_TRACKING)
#define use_awaitable boost::asio::use_awaitable_t(__FILE__, __LINE__, __PRETTY_FUNCTION__)
#endif

scaler::object_storage::ObjectStorageServer server;

awaitable<void> listener(boost::asio::ip::tcp::endpoint endpoint) {
    auto executor = co_await this_coro::executor;
    tcp::acceptor acceptor(executor, endpoint);
    for (;;) {
        tcp::socket socket = co_await acceptor.async_accept(use_awaitable);
        co_spawn(executor, server.process_request(std::move(socket)), detached);
    }
}

int main(int argc, char* argv[]) {
    std::string name = scaler::object_storage::DEFAULT_ADDR;
    std::string port = scaler::object_storage::DEFAULT_PORT;

    argparse::ArgumentParser argParser("server", STRINGIFY(VERSION));
    argParser.add_argument("-n", "--name").default_value(name).help("Name to resolve, e.g., localhost, 127.0.0.1");
    argParser.add_argument("-p", "--port").default_value(port).help("Specify port to listen on");
    try {
        argParser.parse_args(argc, argv);
    } catch (const std::exception& err) {
        std::cerr << err.what() << std::endl;
        std::cerr << argParser;
        std::exit(1);
    }

    name = argParser.get<std::string>("--name");
    port = argParser.get<std::string>("--port");

    try {
        boost::asio::io_context io_context(1);
        tcp::resolver resolver(io_context);
        auto res = resolver.resolve(name, port);

        boost::asio::signal_set signals(io_context, SIGINT, SIGTERM);
        signals.async_wait([&](auto, auto) { io_context.stop(); });

        co_spawn(io_context, listener(res.begin()->endpoint()), detached);

        io_context.run();
    } catch (std::exception& e) { std::printf("Exception: %s\n", e.what()); }
}