#ifndef QUERY_COLUMN_GENERATOR_HPP #define QUERY_COLUMN_GENERATOR_HPP #include "matador/sql/column.hpp" #include "matador/sql/types.hpp" #include "matador/utils/access.hpp" #include "matador/utils/field_attributes.hpp" #include "matador/utils/identifiable.hpp" #include "matador/utils/identifier.hpp" #include namespace matador::utils { enum class cascade_type; } namespace matador::sql { namespace detail { class null_identifiable : public utils::identifiable { public: void reset(const utils::identifier &) override {}; [[nodiscard]] bool has_primary_key() const override { return false; } [[nodiscard]] const utils::identifier& primary_key() const override { return id_; } utils::identifier& primary_key() override { return id_; } [[nodiscard]] utils::identifier create_identifier() const override { return utils::identifier(id_); } private: utils::identifier id_; }; static null_identifiable null_id; } class fk_column_generator : public utils::identifier_serializer { public: column generate(const char *id, utils::identifiable &x); void serialize(short &i, const utils::field_attributes &attributes) override; void serialize(int &i, const utils::field_attributes &attributes) override; void serialize(long &i, const utils::field_attributes &attributes) override; void serialize(long long int &i, const utils::field_attributes &attributes) override; void serialize(unsigned short &i, const utils::field_attributes &attributes) override; void serialize(unsigned int &i, const utils::field_attributes &attributes) override; void serialize(unsigned long &i, const utils::field_attributes &attributes) override; void serialize(unsigned long long int &i, const utils::field_attributes &attributes) override; void serialize(std::string &string, const utils::field_attributes &attributes) override; void serialize(utils::null_type_t &type, const utils::field_attributes &attributes) override; private: utils::identifiable &identifiable_{detail::null_id}; const char *id_{}; column column_; }; class column_generator { private: explicit column_generator(std::vector &columns); public: ~column_generator() = default; template < class Type > static std::vector generate() { std::vector columns; column_generator gen(columns); Type obj; matador::utils::access::process(gen, obj); return std::move(columns); } template < class V > void on_primary_key(const char *, V &x, typename std::enable_if::value && !std::is_same::value>::type* = 0); void on_primary_key(const char *id, std::string &pk, size_t size); void on_revision(const char *id, unsigned long long &rev); template void on_attribute(const char *id, Type &x, const utils::field_attributes &attr = utils::null_attributes); void on_belongs_to(const char *id, utils::identifiable &x, utils::cascade_type); void on_has_one(const char *id, utils::identifiable &x, utils::cascade_type); template void on_has_many(const char *, ContainerType &, const char *, const char *, utils::cascade_type) {} template void on_has_many(const char *, ContainerType &, utils::cascade_type) {} private: size_t index_ = 0; std::vector &columns_; fk_column_generator fk_column_generator_; }; template void column_generator::on_primary_key(const char *id, V &x, typename std::enable_if::value && !std::is_same::value>::type*) { on_attribute(id, x, { utils::constraints::PRIMARY_KEY }); } template void column_generator::on_attribute(const char *id, Type &x, const utils::field_attributes &attr) { columns_.emplace_back(id, x, attr); } } #endif //QUERY_COLUMN_GENERATOR_HPP