query/include/matador/sql/column_generator.hpp

113 lines
3.7 KiB
C++

#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 <vector>
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<column> &columns);
public:
~column_generator() = default;
template < class Type >
static std::vector<column> generate()
{
std::vector<column> 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<std::is_integral<V>::value && !std::is_same<bool, V>::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<typename Type>
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<class ContainerType>
void on_has_many(const char *, ContainerType &, const char *, const char *, utils::cascade_type) {}
template<class ContainerType>
void on_has_many(const char *, ContainerType &, utils::cascade_type) {}
private:
size_t index_ = 0;
std::vector<column> &columns_;
fk_column_generator fk_column_generator_;
};
template<typename V>
void column_generator::on_primary_key(const char *id, V &x, typename std::enable_if<std::is_integral<V>::value && !std::is_same<bool, V>::value>::type*)
{
on_attribute(id, x, { utils::constraints::PRIMARY_KEY });
}
template<typename Type>
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