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
|
#ifndef RFL_AVRO_WRITE_HPP_
#define RFL_AVRO_WRITE_HPP_
#include <avro.h>
#include <bit>
#include <cstdint>
#include <ostream>
#include <sstream>
#include <string>
#include <type_traits>
#include <utility>
#include "../parsing/Parent.hpp"
#include "Parser.hpp"
#include "Schema.hpp"
#include "Writer.hpp"
#include "to_schema.hpp"
namespace rfl::avro {
/// Returns AVRO bytes.
template <class... Ps>
std::vector<char> write(const auto& _obj, const auto& _schema) noexcept {
using T = std::remove_cvref_t<decltype(_obj)>;
using U = typename std::remove_cvref_t<decltype(_schema)>::Type;
using ParentType = parsing::Parent<Writer>;
static_assert(std::is_same<T, U>(),
"The schema must be compatible with the type to write.");
avro_value_t root;
avro_generic_value_new(_schema.iface(), &root);
const auto writer = Writer(&root);
Parser<T, Processors<Ps...>>::write(writer, _obj,
typename ParentType::Root{});
size_t size = 0;
avro_value_sizeof(&root, &size);
std::vector<char> buffer(size);
avro_writer_t avro_writer = avro_writer_memory(buffer.data(), buffer.size());
avro_value_write(avro_writer, &root);
avro_value_decref(&root);
avro_writer_free(avro_writer);
return buffer;
}
/// Returns AVRO bytes.
template <class... Ps>
std::vector<char> write(const auto& _obj) {
using T = std::remove_cvref_t<decltype(_obj)>;
const auto schema = to_schema<T, Ps...>();
return write<Ps...>(_obj, schema);
}
/// Writes a AVRO into an ostream.
template <class... Ps>
std::ostream& write(const auto& _obj, std::ostream& _stream) {
auto buffer = write<Ps...>(_obj);
_stream.write(buffer.data(), buffer.size());
return _stream;
}
} // namespace rfl::avro
#endif
|
