eloqstore-sys 1.1.1

Low-level Rust FFI bindings for EloqStore
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

#pragma once

#include <cstddef>
#include <cstdint>
#include <utility>
#include <vector>

#include "async_io_manager.h"
#include "types.h"

namespace eloqstore
{

struct WriteBufferBatch
{
    char *buffer{nullptr};
    uint16_t buffer_index{0};
    bool use_fixed{true};
    FileId file_id{0};
    uint64_t start_offset{0};
    size_t bytes{0};
    std::vector<VarPage> pages;
    std::vector<char *> release_ptrs;
    std::vector<uint16_t> release_indices;
};

class WriteBufferAggregator
{
public:
    explicit WriteBufferAggregator(size_t buffer_size);

    void Reset();
    bool HasBuffer() const;
    bool HasData() const;
    /**
     * @brief Get the file ID of the current buffer.
     *
     * Returns the file_id associated with the current write buffer.
     * Used to detect when file_id switches (indicating a file seal event).
     *
     * @return Current FileId, or invalid if no buffer/data
     */
    FileId CurrentFileId() const
    {
        return file_id_;
    }
    void SetBuffer(char *buffer,
                   uint16_t buffer_index,
                   FileId file_id,
                   uint64_t start_offset,
                   bool use_fixed);
    bool CanAppend(FileId file_id, uint64_t offset, size_t size) const;
    char *TryReserve(FileId file_id, uint64_t offset, size_t size);
    void AddPage(VarPage page, char *release_ptr, uint16_t release_index);
    bool ShouldFlush(size_t next_size) const;
    WriteBufferBatch TakeBatch();

private:
    size_t buffer_size_{0};
    char *buffer_{nullptr};
    uint16_t buffer_index_{0};
    bool use_fixed_{true};
    FileId file_id_{0};
    uint64_t start_offset_{0};
    uint64_t next_offset_{0};
    size_t used_{0};
    std::vector<VarPage> pages_;
    std::vector<char *> release_ptrs_;
    std::vector<uint16_t> release_indices_;
};

}  // namespace eloqstore