allocator.hpp

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
#pragma once


#include <felspar/io/warden.hpp>


namespace felspar::io {


    using memory_resource = felspar::pmr::memory_resource;

Attach an allocator to the ward so that it can be used to manage IO coroutine allocations

15
16
17
18
19
20
21
22
23
24
25
    class allocator : public warden {
        warden &backing_warden;
        memory_resource &backing_allocator;

      public:
        allocator(warden &bw, memory_resource &ba)
        : backing_warden{bw}, backing_allocator{ba} {}

        void run_batch() override { backing_warden.run_batch(); }

      private:

Memory related APIs

27
28
29
30
31
32
33
34
35
36
37
38
39
40
        void *do_allocate(
                std::size_t const bytes, std::size_t const alignment) override {
            return backing_allocator.allocate(bytes, alignment);
        }
        void do_deallocate(
                void *const p,
                std::size_t const bytes,
                std::size_t const alignment) override {
            backing_allocator.deallocate(p, bytes, alignment);
        }

        bool do_is_equal(memory_resource const &other) const noexcept override {
            return backing_allocator.is_equal(other);
        }

Warden related APIs

 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
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
        void run_until(felspar::coro::coroutine_handle<> h) override {
            backing_warden.run_until(h);
        }
        iop<void> do_close(
                socket_descriptor const fd,
                felspar::source_location const &loc) override {
            return backing_warden.do_close(fd, loc);
        }
        iop<void> do_sleep(
                std::chrono::nanoseconds const time,
                felspar::source_location const &loc) override {
            return backing_warden.do_sleep(time, loc);
        }
        iop<std::size_t> do_read_some(
                socket_descriptor const fd,
                std::span<std::byte> const buffer,
                std::optional<std::chrono::nanoseconds> const timeout,
                felspar::source_location const &loc) override {
            return backing_warden.do_read_some(fd, buffer, timeout, loc);
        }
        iop<std::size_t> do_write_some(
                socket_descriptor const fd,
                std::span<std::byte const> const buffer,
                std::optional<std::chrono::nanoseconds> const timeout,
                felspar::source_location const &loc) override {
            return backing_warden.do_write_some(fd, buffer, timeout, loc);
        }
        void do_prepare_socket(
                socket_descriptor const sock,
                felspar::source_location const &loc) override {
            return backing_warden.do_prepare_socket(sock, loc);
        }
        iop<socket_descriptor> do_accept(
                socket_descriptor const fd,
                std::optional<std::chrono::nanoseconds> const timeout,
                felspar::source_location const &loc) override {
            return backing_warden.do_accept(fd, timeout, loc);
        }
        iop<void> do_connect(
                socket_descriptor const fd,
                sockaddr const *const addr,
                socklen_t const len,
                std::optional<std::chrono::nanoseconds> const timeout,
                felspar::source_location const &loc) override {
            return backing_warden.do_connect(fd, addr, len, timeout, loc);
        }
        iop<void> do_read_ready(
                socket_descriptor const fd,
                std::optional<std::chrono::nanoseconds> const timeout,
                felspar::source_location const &loc) override {
            return backing_warden.do_read_ready(fd, timeout, loc);
        }
        iop<void> do_write_ready(
                socket_descriptor const fd,
                std::optional<std::chrono::nanoseconds> const timeout,
                felspar::source_location const &loc) override {
            return backing_warden.do_write_ready(fd, timeout, loc);
        }
    };


}