164 lines
6.7 KiB
C++
164 lines
6.7 KiB
C++
#ifndef QUERY_COLUMN_DEFINITION_HPP
|
|
#define QUERY_COLUMN_DEFINITION_HPP
|
|
|
|
#include "matador/utils/basic_types.hpp"
|
|
#include "matador/utils/default_type_traits.hpp"
|
|
#include "matador/utils/field_attributes.hpp"
|
|
|
|
#include <memory>
|
|
|
|
namespace matador::object {
|
|
|
|
enum class null_option_type : uint8_t {
|
|
NULLABLE, NOT_NULL
|
|
};
|
|
|
|
struct attribute_options {
|
|
utils::field_attributes attributes;
|
|
null_option_type null_option{null_option_type::NOT_NULL};
|
|
int index{-1};
|
|
};
|
|
|
|
class attribute_definition {
|
|
public:
|
|
explicit attribute_definition(const char *name); // NOLINT(*-explicit-constructor)
|
|
explicit attribute_definition(std::string name); // NOLINT(*-explicit-constructor)
|
|
|
|
attribute_definition(const attribute_definition&) = default;
|
|
attribute_definition& operator=(const attribute_definition&) = default;
|
|
attribute_definition(attribute_definition&&) noexcept = default;
|
|
attribute_definition& operator=(attribute_definition&&) noexcept = default;
|
|
|
|
attribute_definition() = default;
|
|
attribute_definition(std::string name,
|
|
utils::basic_type type,
|
|
const utils::field_attributes&,
|
|
null_option_type null_opt,
|
|
int index = 0);
|
|
attribute_definition(std::string name,
|
|
utils::basic_type type,
|
|
int index,
|
|
const std::shared_ptr<attribute_definition> &ref_column,
|
|
const utils::field_attributes& attr,
|
|
null_option_type null_opt);
|
|
|
|
attribute_definition(std::string name,
|
|
utils::basic_type type,
|
|
const std::shared_ptr<attribute_definition> &ref_column = {});
|
|
attribute_definition(std::string name,
|
|
utils::basic_type type,
|
|
std::string table_name,
|
|
const attribute_options& options,
|
|
const std::shared_ptr<attribute_definition> &ref_column = {});
|
|
|
|
[[nodiscard]] const std::string& name() const;
|
|
void name(const std::string& n);
|
|
[[nodiscard]] std::string full_name() const;
|
|
[[nodiscard]] int index() const;
|
|
[[nodiscard]] const utils::field_attributes& attributes() const;
|
|
[[nodiscard]] utils::field_attributes& attributes();
|
|
[[nodiscard]] bool is_nullable() const;
|
|
[[nodiscard]] utils::basic_type type() const;
|
|
[[nodiscard]] const std::string& table_name() const;
|
|
void table_name(const std::string& name);
|
|
void change_type(utils::basic_type type, const utils::field_attributes &attr = utils::null_attributes);
|
|
template < typename Type >
|
|
void change_type(const utils::field_attributes &attr = utils::null_attributes) {
|
|
type_ = utils::data_type_traits<Type>::type(attr.size());
|
|
}
|
|
[[nodiscard]] std::shared_ptr<attribute_definition> reference_column() const;
|
|
[[nodiscard]] bool is_foreign_reference() const;
|
|
|
|
[[nodiscard]] bool is_integer() const;
|
|
[[nodiscard]] bool is_floating_point() const;
|
|
[[nodiscard]] bool is_bool() const;
|
|
[[nodiscard]] bool is_string() const;
|
|
[[nodiscard]] bool is_varchar() const;
|
|
[[nodiscard]] bool is_date() const;
|
|
[[nodiscard]] bool is_time() const;
|
|
[[nodiscard]] bool is_blob() const;
|
|
[[nodiscard]] bool is_null() const;
|
|
|
|
template< typename Type >
|
|
[[nodiscard]] bool is_type_of() const {
|
|
return type() == utils::data_type_traits<Type>::type(attributes().size());
|
|
}
|
|
|
|
private:
|
|
std::string name_;
|
|
std::string table_name_;
|
|
attribute_options options_;
|
|
utils::basic_type type_{utils::basic_type::type_null};
|
|
|
|
std::shared_ptr<attribute_definition> reference_column_;
|
|
};
|
|
|
|
/**
|
|
* User defined literal to have a shortcut creating a column object
|
|
* @param name Name of the column
|
|
* @param type
|
|
* @param attr Length of the column name
|
|
* @param null_opt
|
|
* @return A column object with a given name
|
|
*/
|
|
attribute_definition make_column(const std::string &name, utils::basic_type type, utils::field_attributes attr = utils::null_attributes, null_option_type null_opt = null_option_type::NOT_NULL);
|
|
|
|
template < typename Type >
|
|
attribute_definition make_column(const std::string &name, utils::field_attributes attr = utils::null_attributes, null_option_type null_opt = null_option_type::NOT_NULL)
|
|
{
|
|
return make_column(name, utils::data_type_traits<Type>::type(0), attr, null_opt);
|
|
}
|
|
template <>
|
|
attribute_definition make_column<std::string>(const std::string &name, utils::field_attributes attr, null_option_type null_opt);
|
|
|
|
template < typename Type >
|
|
attribute_definition make_pk_column(const std::string &name, size_t size = 0)
|
|
{
|
|
return make_column<Type>(name, { size, utils::constraints::PRIMARY_KEY });
|
|
}
|
|
|
|
template <>
|
|
attribute_definition make_pk_column<std::string>(const std::string &name, size_t size);
|
|
|
|
template < typename Type >
|
|
attribute_definition make_fk_column(const std::string &name, size_t size, const std::shared_ptr<attribute_definition> &ref_column) {
|
|
return {name, utils::data_type_traits<Type>::type(size), ref_column, { size, utils::constraints::FOREIGN_KEY }};
|
|
}
|
|
|
|
template < typename Type >
|
|
[[maybe_unused]] attribute_definition make_fk_column(const std::string &name, const std::shared_ptr<attribute_definition> &ref_column)
|
|
{
|
|
return {name, utils::data_type_traits<Type>::type(0), 0, ref_column, { 0, utils::constraints::FOREIGN_KEY }, null_option_type::NOT_NULL};
|
|
}
|
|
|
|
template <>
|
|
[[maybe_unused]] attribute_definition make_fk_column<std::string>(const std::string &name, size_t size, const std::shared_ptr<attribute_definition> &ref_column);
|
|
|
|
template < typename Type >
|
|
[[maybe_unused]] attribute_definition make_fk_column(const std::string &name, size_t size, const std::string &ref_table_name, const std::string &ref_column_name) {
|
|
return {
|
|
name, utils::data_type_traits<Type>::type(size), 0,
|
|
std::make_shared<attribute_definition>(ref_column_name, ref_table_name, utils::data_type_traits<Type>::type(size), utils::constraints::FOREIGN_KEY),
|
|
{ 0, utils::constraints::FOREIGN_KEY }, null_option_type::NOT_NULL
|
|
};
|
|
}
|
|
|
|
template < typename Type >
|
|
[[maybe_unused]] attribute_definition make_fk_column(const std::string &name, const std::string &ref_table_name, const std::string &ref_column_name) {
|
|
return {
|
|
name, utils::data_type_traits<Type>::type(0), 0,
|
|
std::make_shared<attribute_definition>(ref_column_name, utils::data_type_traits<Type>::type(0), ref_table_name, attribute_options{utils::constraints::FOREIGN_KEY}),
|
|
{ 0, utils::constraints::FOREIGN_KEY }, null_option_type::NOT_NULL
|
|
};
|
|
}
|
|
|
|
template <>
|
|
[[maybe_unused]] attribute_definition make_fk_column<std::string>(const std::string &name, const std::string &ref_table_name, const std::string &ref_column_name);
|
|
|
|
template <>
|
|
[[maybe_unused]] attribute_definition make_fk_column<std::string>(const std::string &name, size_t size, const std::string &ref_table_name, const std::string &ref_column_name);
|
|
|
|
}
|
|
|
|
#endif //QUERY_COLUMN_DEFINITION_HPP
|