129 lines
4.6 KiB
C++
129 lines
4.6 KiB
C++
#ifndef QUERY_COLUMN_GENERATOR_HPP
|
|
#define QUERY_COLUMN_GENERATOR_HPP
|
|
|
|
#include "matador/sql/column.hpp"
|
|
#include "matador/sql/data_type_traits.hpp"
|
|
|
|
#include "matador/utils/access.hpp"
|
|
#include "matador/utils/field_attributes.hpp"
|
|
|
|
#include <typeindex>
|
|
#include <vector>
|
|
|
|
namespace matador::sql {
|
|
|
|
class table_repository;
|
|
|
|
class fk_column_generator
|
|
{
|
|
public:
|
|
fk_column_generator() = default;
|
|
|
|
template<class Type>
|
|
column generate(const char *id, Type &x, const std::string &ref_table, const std::string &ref_column)
|
|
{
|
|
utils::access::process(*this, x);
|
|
return column{id, type_, 0, ref_table, ref_column, { utils::constraints::FOREIGN_KEY }, null_option::NOT_NULL};
|
|
}
|
|
|
|
template<typename ValueType>
|
|
void on_primary_key(const char *, ValueType &/*pk*/, typename std::enable_if<std::is_integral<ValueType>::value && !std::is_same<bool, ValueType>::value>::type* = 0)
|
|
{
|
|
type_ = data_type_traits<ValueType>::builtin_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 < class Type >
|
|
void on_attribute(const char * /*id*/, Type &/*x*/, const utils::field_attributes &/*attr*/ = utils::null_attributes) {}
|
|
void on_attribute(const char * /*id*/, char * /*x*/, const utils::field_attributes &/*attr*/ = utils::null_attributes) {}
|
|
template<class Pointer>
|
|
void on_belongs_to(const char * /*id*/, Pointer &/*x*/, utils::cascade_type) {}
|
|
template<class Pointer>
|
|
void on_has_one(const char * /*id*/, Pointer &/*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:
|
|
data_type_t type_{};
|
|
};
|
|
|
|
class column_generator
|
|
{
|
|
private:
|
|
column_generator(std::vector<column> &columns, const table_repository &repo);
|
|
|
|
public:
|
|
~column_generator() = default;
|
|
|
|
template < class Type >
|
|
static std::vector<column> generate(const table_repository &repo)
|
|
{
|
|
std::vector<column> columns;
|
|
column_generator gen(columns, repo);
|
|
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);
|
|
|
|
template<typename Type>
|
|
void on_attribute(const char *id, std::optional<Type> &x, const utils::field_attributes &attr = utils::null_attributes);
|
|
|
|
template<class Pointer>
|
|
void on_belongs_to(const char *id, Pointer &x, utils::cascade_type)
|
|
{
|
|
const auto [ref_table, ref_column] = determine_foreign_ref(std::type_index(typeid(typename Pointer::value_type)));
|
|
columns_.push_back(fk_column_generator_.generate(id, *x, ref_table, ref_column));
|
|
}
|
|
template<class Pointer>
|
|
void on_has_one(const char *id, Pointer &x, utils::cascade_type)
|
|
{
|
|
const auto [ref_table, ref_column] = determine_foreign_ref(std::type_index(typeid(typename Pointer::value_type)));
|
|
columns_.push_back(fk_column_generator_.generate(id, *x, ref_table, ref_column));
|
|
}
|
|
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:
|
|
std::pair<std::string, std::string> determine_foreign_ref(const std::type_index &ti);
|
|
|
|
private:
|
|
size_t index_ = 0;
|
|
std::vector<column> &columns_;
|
|
const table_repository &repo_;
|
|
|
|
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, null_option::NOT_NULL);
|
|
}
|
|
|
|
template<typename Type>
|
|
void column_generator::on_attribute(const char *id, std::optional<Type> &x, const utils::field_attributes &attr)
|
|
{
|
|
columns_.emplace_back(id, data_type_traits<Type>::builtin_type(attr.size()), attr, null_option::NULLABLE);
|
|
}
|
|
|
|
}
|
|
#endif //QUERY_COLUMN_GENERATOR_HPP
|