traits.hpp

#pragma once
 
#include <ydb-cpp-sdk/client/params/fwd.h>
#include <ydb-cpp-sdk/client/value/fwd.h>
 
#include <string_view>
#include <type_traits>
 
USERVER_NAMESPACE_BEGIN
 
namespace ydb {
 
/// Used for error reporting in ydb::ValueTraits::Parse.
struct ParseContext final {
    std::string_view column_name;
};
 
template <typename T, typename Enable = void>
struct ValueTraits {
    /// Parses an element of type `T` from `parser`. `context` may be used
    /// for diagnostic messages.
    static T Parse([[maybe_unused]] NYdb::TValueParser& parser, [[maybe_unused]] const ParseContext& context) {
        static_assert(
            !sizeof(T),
            "Cannot convert from YDB value. No ValueTraits defined for "
            "this type. userver/ydb/io/supported_types.hpp "
            "contains all known definitions"
        );
        return T{};
    }
 
    /// Writes an element of type `T` to `builder`.
    template <typename Builder>
    static void Write([[maybe_unused]] NYdb::TValueBuilderBase<Builder>& builder, [[maybe_unused]] const T& value) {
        static_assert(
            !sizeof(T),
            "Cannot convert to YDB value. No ValueTraits defined for "
            "this type. userver/ydb/io/supported_types.hpp "
            "contains all known definitions"
        );
    }
 
    /// Uses NYdb::TTypeBuilder to create a representation of `T` type.
    static NYdb::TType MakeType() = delete;
};
 
/// A shorthand for calling ydb::ValueTraits<T>::Parse.
template <typename T>
inline constexpr auto Parse = [](NYdb::TValueParser& parser, const ParseContext& context) -> T {
    // Note: using a CPO instead of a global function to avoid ADL.
    // Customization should be performed using ValueTraits.
    return ValueTraits<T>::Parse(parser, context);
};
 
/// A shorthand for calling ydb::ValueTraits<T>::Write.
inline constexpr auto Write = [](auto& builder, auto&& value) {
    // Note: using a CPO instead of a global function to avoid ADL.
    // Customization should be performed using ValueTraits.
    using RawValueType = std::remove_const_t<std::remove_reference_t<decltype(value)>>;
    using ValueType =
        std::conditional_t<std::is_convertible_v<RawValueType, std::string_view>, std::string, RawValueType>;
    ValueTraits<ValueType>::Write(builder, value);
};
 
}  // namespace ydb
 
USERVER_NAMESPACE_END