98 lines
2.6 KiB
C++
98 lines
2.6 KiB
C++
#ifndef QUERY_COLUMN_NAME_GENERATOR_HPP
|
|
#define QUERY_COLUMN_NAME_GENERATOR_HPP
|
|
|
|
#include "matador/utils/access.hpp"
|
|
#include "matador/utils/field_attributes.hpp"
|
|
#include "matador/utils/foreign_attributes.hpp"
|
|
|
|
#include "matador/sql/column.hpp"
|
|
#include "matador/sql/schema.hpp"
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <stack>
|
|
|
|
namespace matador::sql {
|
|
|
|
class column_name_generator
|
|
{
|
|
private:
|
|
column_name_generator(std::vector<column> &column_infos, const sql::schema &ts, const std::string &table_name);
|
|
|
|
public:
|
|
~column_name_generator() = default;
|
|
|
|
template < class Type >
|
|
static std::vector<column> generate(const sql::schema &ts)
|
|
{
|
|
const auto info = ts.info<Type>();
|
|
if (!info) {
|
|
return {};
|
|
}
|
|
std::vector<column> columns;
|
|
column_name_generator gen(columns, ts, info.value().name);
|
|
Type obj;
|
|
matador::utils::access::process(gen, obj);
|
|
return std::move(columns);
|
|
}
|
|
|
|
template < class V >
|
|
void on_primary_key(const char *id, V &, typename std::enable_if<std::is_integral<V>::value && !std::is_same<bool, V>::value>::type* = 0)
|
|
{
|
|
push(id);
|
|
}
|
|
void on_primary_key(const char *id, std::string &, size_t);
|
|
void on_revision(const char *id, unsigned long long &/*rev*/);
|
|
|
|
template<typename Type>
|
|
void on_attribute(const char *id, Type &, const utils::field_attributes &/*attr*/ = utils::null_attributes)
|
|
{
|
|
push(id);
|
|
}
|
|
|
|
template<class Pointer>
|
|
void on_belongs_to(const char *id, Pointer &, const utils::foreign_attributes &/*attr*/)
|
|
{
|
|
push(id);
|
|
}
|
|
template<class Pointer>
|
|
void on_has_one(const char *id, Pointer &, const utils::foreign_attributes &/*attr*/)
|
|
{
|
|
push(id);
|
|
}
|
|
template<class ContainerType>
|
|
void on_has_many(const char *, ContainerType &, const char *, const char *, const utils::foreign_attributes &attr)
|
|
{
|
|
if (attr.fetch() == fetch_type::LAZY) {
|
|
return;
|
|
}
|
|
const auto info = table_schema_.info<typename ContainerType::value_type::value_type>();
|
|
if (!info) {
|
|
return;
|
|
}
|
|
|
|
table_name_stack_.push(info.value().name);
|
|
typename ContainerType::value_type::value_type obj;
|
|
matador::utils::access::process(*this, obj);
|
|
table_name_stack_.pop();
|
|
}
|
|
template<class ContainerType>
|
|
void on_has_many(const char *id, ContainerType &c, const utils::foreign_attributes &attr)
|
|
{
|
|
on_has_many(id, c, "", "", attr);
|
|
}
|
|
|
|
private:
|
|
void push(const std::string &column_name);
|
|
|
|
private:
|
|
std::stack<std::string> table_name_stack_;
|
|
std::vector<column> &column_infos_;
|
|
const sql::schema &table_schema_;
|
|
int column_index{0};
|
|
};
|
|
|
|
}
|
|
|
|
#endif //QUERY_COLUMN_NAME_GENERATOR_HPP
|