diff options
Diffstat (limited to 'forstio/codec/message.h')
-rw-r--r-- | forstio/codec/message.h | 564 |
1 files changed, 0 insertions, 564 deletions
diff --git a/forstio/codec/message.h b/forstio/codec/message.h deleted file mode 100644 index f0cadfd..0000000 --- a/forstio/codec/message.h +++ /dev/null @@ -1,564 +0,0 @@ -#pragma once - -#include <cstdint> -#include <tuple> -#include <type_traits> -#include <variant> -#include <vector> - -#include <cassert> - -#include <forstio/common.h> -#include <forstio/string_literal.h> - -#include "message_container.h" -#include "schema.h" - -namespace saw { -class message_base { -protected: - bool set_explicitly_ = false; - -public: - template <class T> T &as() { - static_assert(std::is_base_of<message_base, T>()); - return dynamic_cast<T &>(*this); - } - - template <class T> const T &as() const { - static_assert(std::is_base_of<message_base, T>()); - return dynamic_cast<const T &>(*this); - } -}; -/* - * Representing all message types - * Description which type to use happens through the use of the schema classes - * in schema.h The message classes are wrapper classes which store the data - * according to the specified container class. - * - * The reader and builder classe exist to create some clarity while implementing - * parsers. Also minor guarantess are provided if the message is used as a - * template parameter. - */ - -/* - * Struct Message - */ -template <class... V, string_literal... Keys, class Container> -class message<schema::Struct<schema::NamedMember<V, Keys>...>, Container> final - : public message_base { -private: - using SchemaType = schema::Struct<schema::NamedMember<V, Keys>...>; - using MessageType = message<SchemaType, Container>; - Container container_; - - static_assert(std::is_same_v<SchemaType, typename Container::SchemaType>, - "Container should have same the schema as Message"); - - friend class builder; - friend class reader; - -public: - class reader; - class builder { - private: - MessageType &message_; - - public: - builder(MessageType &msg) : message_{msg} {} - - reader as_reader() { return reader{message_}; } - - /* - * Initialize a member by index - */ - template <size_t i> - typename std::enable_if< - !schema_is_array< - typename message_parameter_pack_type<i, V...>::Type>::Value, - typename Container::template ElementType<i>::builder>::type - init() { - return typename Container::template ElementType<i>::builder{ - message_.container_.template get<i>()}; - } - - /* - * Initialize a member by their name - * This is the preferred method for schema::Struct messages - */ - template <string_literal Literal> - typename std::enable_if< - !schema_is_array<typename message_parameter_pack_type< - message_parameter_key_pack_index<Literal, Keys...>::Value, - V...>::Type>::Value, - typename Container::template ElementType< - message_parameter_key_pack_index<Literal, Keys...>::Value>:: - builder>::type - init() { - constexpr size_t i = - message_parameter_key_pack_index<Literal, Keys...>::Value; - - return init<i>(); - } - - template <size_t i> - typename std::enable_if< - schema_is_array< - typename message_parameter_pack_type<i, V...>::Type>::Value, - typename Container::template ElementType<i>::builder>::type - init(size_t size = 0) { - auto array_builder = - typename Container::template ElementType<i>::builder{ - message_.container_.template get<i>(), size}; - return array_builder; - } - - /* - * Version for array schema type - */ - template <string_literal Literal> - typename std::enable_if< - schema_is_array<typename message_parameter_pack_type< - message_parameter_key_pack_index<Literal, Keys...>::Value, - V...>::Type>::Value, - typename Container::template ElementType< - message_parameter_key_pack_index<Literal, Keys...>::Value>:: - builder>::type - init(size_t size) { - constexpr size_t i = - message_parameter_key_pack_index<Literal, Keys...>::Value; - - return init<i>(size); - } - }; - - class reader { - private: - MessageType &message_; - - public: - reader(MessageType &msg) : message_{msg} {} - - builder as_builder() { return builder{message_}; } - - /* - * Get member by index - */ - template <size_t i> - typename Container::template ElementType<i>::reader get() { - return typename Container::template ElementType<i>::reader{ - message_.container_.template get<i>()}; - } - - /* - * Get member by name - * This is the preferred method for schema::Struct messages - */ - template <string_literal Literal> - typename Container::template ElementType< - message_parameter_key_pack_index<Literal, Keys...>::Value>::reader - get() { - // The index of the first match - constexpr size_t i = - message_parameter_key_pack_index<Literal, Keys...>::Value; - - return get<i>(); - } - }; - - builder build() { return builder{*this}; } - - reader read() { return reader{*this}; } -}; - -/* - * Union message class. Wrapper object - */ -template <class... V, string_literal... Keys, class Container> -class message<schema::Union<schema::NamedMember<V, Keys>...>, Container> final - : public message_base { -private: - using SchemaType = schema::Union<schema::NamedMember<V, Keys>...>; - using MessageType = message<SchemaType, Container>; - - Container container_; - - static_assert(std::is_same_v<SchemaType, typename Container::SchemaType>, - "Container should have same the schema as Message"); - - friend class builder; - friend class reader; - -public: - class reader; - class builder { - private: - MessageType &message_; - - public: - builder(MessageType &msg) : message_{msg} {} - - reader as_reader() { return reader{message_}; } - - template <size_t i> - typename std::enable_if< - !schema_is_array< - typename message_parameter_pack_type<i, V...>::Type>::Value, - typename Container::template ElementType<i>::builder>::type - - init() { - return typename Container::template ElementType<i>::builder{ - message_.container_.template get<i>()}; - } - - template <string_literal Literal> - typename std::enable_if< - !schema_is_array<typename message_parameter_pack_type< - message_parameter_key_pack_index<Literal, Keys...>::Value, - V...>::Type>::Value, - typename Container::template ElementType< - message_parameter_key_pack_index<Literal, Keys...>::Value>:: - builder>::type - init() { - constexpr size_t i = - message_parameter_key_pack_index<Literal, Keys...>::Value; - - return init<i>(); - } - - /* - * If Schema is Array - */ - template <size_t i> - typename std::enable_if< - schema_is_array< - typename message_parameter_pack_type<i, V...>::Type>::Value, - typename Container::template ElementType<i>::builder>::type - init(size_t size = 0) { - return typename Container::template ElementType<i>::builder{ - message_.container_.template get<i>(), size}; - } - - template <string_literal Literal> - typename std::enable_if< - schema_is_array<typename message_parameter_pack_type< - message_parameter_key_pack_index<Literal, Keys...>::Value, - V...>::Type>::Value, - typename Container::template ElementType< - message_parameter_key_pack_index<Literal, Keys...>::Value>:: - builder>::type - init(size_t size) { - constexpr size_t i = - message_parameter_key_pack_index<Literal, Keys...>::Value; - - return init<i>(size); - } - }; - - class reader { - private: - MessageType &message_; - - public: - reader(MessageType &msg) : message_{msg} {} - - builder as_builder() { return builder{message_}; } - - template <size_t i> - typename Container::template ElementType<i>::reader get() { - return typename Container::template ElementType<i>::reader{ - message_.container_.template get<i>()}; - } - - template <string_literal Literal> - typename Container::template ElementType< - message_parameter_key_pack_index<Literal, Keys...>::Value>::reader - get() { - // The index of the first match - constexpr size_t i = - message_parameter_key_pack_index<Literal, Keys...>::Value; - - return get<i>(); - } - - template <string_literal Literal> - static constexpr size_t to_index() noexcept { - return message_parameter_key_pack_index<Literal, Keys...>::Value; - } - - size_t index() const noexcept { return message_.container_.index(); } - - template <string_literal Literal> bool has_alternative() const { - return index() == to_index<Literal>(); - } - }; -}; - -/* - * Array message class. Wrapper around an array schema element - */ -template <class T, class Container> -class message<schema::Array<T>, Container> final : public message_base { -private: - using SchemaType = schema::Array<T>; - using MessageType = message<SchemaType, Container>; - - Container container_; - - static_assert(std::is_same_v<SchemaType, typename Container::SchemaType>, - "Container should have same Schema as Message"); - - friend class builder; - friend class reader; - -public: - class reader; - class builder { - private: - MessageType &message_; - - public: - builder(MessageType &msg, size_t size) : message_{msg} { - if (size > 0) { - message_.container_.resize(size); - } - } - - reader as_reader() { return reader{message_}; } - - typename Container::ElementType::builder init(size_t i) { - return typename Container::ElementType::builder{ - message_.container_.get(i)}; - } - - size_t size() const { return message_.container_.size(); } - - void resize(size_t size) { message_.container_.resize(size); } - }; - - class reader { - private: - MessageType &message_; - - public: - reader(MessageType &msg) : message_{msg} {} - - builder as_builder() { return builder{message_, 0}; } - - typename Container::ElementType::reader get(size_t i) { - return typename Container::ElementType::reader{ - message_.container_.get(i)}; - } - - size_t size() const { return message_.container_.size(); } - }; -}; - -/* - * Tuple message class. Wrapper around a tuple schema - */ -template <class... T, class Container> -class message<schema::Tuple<T...>, Container> final : public message_base { -private: - using SchemaType = schema::Tuple<T...>; - using MessageType = message<SchemaType, Container>; - - Container container_; - - static_assert(std::is_same_v<SchemaType, typename Container::SchemaType>, - "Container should have same the schema as Message"); - - friend class builder; - friend class reader; - -public: - class reader; - class builder { - MessageType &message_; - - public: - builder(MessageType &msg) : message_{msg} {} - - reader as_reader() { return reader{message_}; } - - template <size_t i> - typename std::enable_if< - !schema_is_array< - typename message_parameter_pack_type<i, T...>::Type>::Value, - typename Container::template ElementType<i>::builder>::type - init() { - return typename Container::template ElementType<i>::builder{ - message_.container_.template get<i>()}; - } - - template <size_t i> - typename std::enable_if< - schema_is_array< - typename message_parameter_pack_type<i, T...>::Type>::Value, - typename Container::template ElementType<i>::builder>::type - init(size_t size = 0) { - return typename Container::template ElementType<i>::builder{ - message_.container_.template get<i>(), size}; - } - }; - class reader { - private: - MessageType &message_; - - public: - reader(MessageType &msg) : message_{msg} {} - - builder as_builder() { return builder{message_}; } - - template <size_t i> - typename Container::template ElementType<i>::reader get() { - return typename Container::template ElementType<i>::reader{ - message_.container_.template get<i>()}; - } - }; -}; - -/* - * Primitive type (float, double, uint8_t, uint16_t, uint32_t, uint64_t, int8_t, - * int16_t, int32_t, int64_t) message class - */ -template <class T, size_t N, class Container> -class message<schema::Primitive<T, N>, Container> final : public message_base { -private: - using SchemaType = schema::Primitive<T, N>; - using MessageType = message<SchemaType, Container>; - - Container container_; - - static_assert(std::is_same_v<SchemaType, typename Container::SchemaType>, - "Container should have same the schema as Message"); - - friend class builder; - friend class reader; - -public: - class reader; - class builder { - private: - MessageType &message_; - - public: - builder(MessageType &msg) : message_{msg} {} - - reader as_reader() { return reader{message_}; } - - void set(const typename Container::ValueType &value) { - message_.container_.set(value); - } - }; - - class reader { - private: - MessageType &message_; - - public: - reader(message &msg) : message_{msg} {} - - builder as_builder() { return builder{message_}; } - - const typename Container::ValueType &get() const { - return message_.container_.get(); - } - }; -}; - -template <class Container> -class message<schema::String, Container> final : public message_base { -private: - using SchemaType = schema::String; - using MessageType = message<SchemaType, Container>; - - Container container_; - - static_assert(std::is_same_v<SchemaType, typename Container::SchemaType>, - "Container should have same the schema as Message"); - - friend class builder; - friend class reader; - -public: - class reader; - class builder { - private: - MessageType &message_; - - public: - builder(MessageType &msg) : message_{msg} {} - - reader as_reader() { return reader{message_}; } - - void set(std::string &&str) { message_.container_.set(std::move(str)); } - void set(const std::string_view str) { message_.container_.set(str); } - void set(const char *str) { set(std::string_view{str}); } - }; - - class reader { - private: - MessageType &message_; - - public: - reader(MessageType &msg) : message_{msg} {} - - builder as_builder() { return builder{message_}; } - - const std::string_view get() const { return message_.container_.get(); } - }; -}; - -template <class Schema, class Container = message_container<Schema>> -class heap_message_root { -private: - own<message<Schema, Container>> root_; - -public: - heap_message_root(own<message<Schema, Container>> r) - : root_{std::move(r)} {} - - typename message<Schema, Container>::builder build() { - assert(root_); - return typename message<Schema, Container>::builder{*root_}; - } - - typename message<Schema, Container>::reader read() { - assert(root_); - return typename message<Schema, Container>::reader{*root_}; - } -}; - -template <class T, class Container> -class heap_message_root<schema::Array<T>, Container> { -public: - using Schema = schema::Array<T>; - -private: - own<message<Schema, Container>> root_; - -public: - heap_message_root(own<message<Schema, Container>> r) - : root_{std::move(r)} {} - - typename message<Schema, Container>::builder build(size_t size) { - assert(root_); - return typename message<Schema, Container>::builder{*root_, size}; - } - - typename message<Schema, Container>::reader read() { - assert(root_); - return typename message<Schema, Container>::reader{*root_}; - } -}; - -/* - * Minor helper for creating a message root - */ -template <class Schema, class Container = message_container<Schema>> -inline heap_message_root<Schema, Container> new_heap_message_root() { - own<message<Schema, Container>> root = heap<message<Schema, Container>>(); - return heap_message_root<Schema, Container>{std::move(root)}; -} -} // namespace saw |