Refactored include/message.hpp

This commit is contained in:
Paul-Louis Ageneau
2020-08-21 23:56:09 +02:00
parent 3c971e05dd
commit c001bb3f9d
3 changed files with 73 additions and 40 deletions

View File

@ -1,5 +1,5 @@
/**
* Copyright (c) 2019 Paul-Louis Ageneau
* Copyright (c) 2019-2020 Paul-Louis Ageneau
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
@ -50,7 +50,7 @@ using message_ptr = std::shared_ptr<Message>;
using message_callback = std::function<void(message_ptr message)>;
using message_variant = std::variant<binary, string>;
constexpr auto message_size_func = [](const message_ptr &m) -> size_t {
inline size_t message_size_func(const message_ptr &m) {
return m->type == Message::Binary || m->type == Message::String ? m->size() : 0;
};
@ -64,47 +64,16 @@ message_ptr make_message(Iterator begin, Iterator end, Message::Type type = Mess
return message;
}
inline message_ptr make_message(size_t size, Message::Type type = Message::Binary,
unsigned int stream = 0,
std::shared_ptr<Reliability> reliability = nullptr) {
auto message = std::make_shared<Message>(size, type);
message->stream = stream;
message->reliability = reliability;
return message;
}
message_ptr make_message(size_t size, Message::Type type = Message::Binary, unsigned int stream = 0,
std::shared_ptr<Reliability> reliability = nullptr);
inline message_ptr make_message(binary &&data, Message::Type type = Message::Binary,
unsigned int stream = 0,
std::shared_ptr<Reliability> reliability = nullptr) {
auto message = std::make_shared<Message>(std::move(data), type);
message->stream = stream;
message->reliability = reliability;
return message;
}
message_ptr make_message(binary &&data, Message::Type type = Message::Binary,
unsigned int stream = 0,
std::shared_ptr<Reliability> reliability = nullptr);
inline message_ptr make_message(message_variant data) {
return std::visit( //
overloaded{
[&](binary data) { return make_message(std::move(data), Message::Binary); },
[&](string data) {
auto b = reinterpret_cast<const byte *>(data.data());
return make_message(b, b + data.size(), Message::String);
},
},
std::move(data));
}
message_ptr make_message(message_variant data);
inline std::optional<message_variant> to_variant(Message &&message) {
switch (message.type) {
case Message::String:
return std::make_optional(
string(reinterpret_cast<const char *>(message.data()), message.size()));
case Message::Binary:
return std::make_optional(std::move(message));
default:
return nullopt;
}
}
std::optional<message_variant> to_variant(Message &&message);
} // namespace rtc