78 lines
2.9 KiB
C++
78 lines
2.9 KiB
C++
#include "matador/query/schema.hpp"
|
|
|
|
#include "matador/query/query.hpp"
|
|
#include "matador/query/manual_pk_generator.hpp"
|
|
#include "matador/query/identity_pk_generator.hpp"
|
|
#include "matador/query/sequence_pk_generator.hpp"
|
|
#include "matador/query/table_pk_generator.hpp"
|
|
|
|
#include "matador/object/repository_node.hpp"
|
|
|
|
#include "matador/sql/backend_provider.hpp"
|
|
#include "matador/sql/connection_pool.hpp"
|
|
#include "matador/sql/dialect.hpp"
|
|
|
|
namespace matador::query {
|
|
|
|
utils::result<void, utils::error> schema::drop_table(const std::string &table_name, const sql::connection &conn) {
|
|
auto result = query::drop()
|
|
.table(table_name)
|
|
.execute(conn);
|
|
if (result.is_error()) {
|
|
return utils::failure(result.err());
|
|
}
|
|
|
|
return utils::ok<void>();
|
|
}
|
|
|
|
utils::result<std::vector<object::attribute>, utils::error>
|
|
schema::describe_table(const std::string &table_name, const sql::connection &conn) {
|
|
return utils::ok(conn.describe(table_name).release());
|
|
}
|
|
|
|
utils::result<bool, utils::error> schema::table_exists(const std::string &table_name, const sql::connection &conn) const {
|
|
return conn.exists(repo_.name(), table_name);
|
|
}
|
|
|
|
void schema::dump(std::ostream &os) const {
|
|
repo_.dump(os);
|
|
}
|
|
|
|
schema::iterator schema::insert_table(const std::type_index &ti, const object::repository_node &node, const utils::generator_type generator_type) {
|
|
std::vector<table_column> columns;
|
|
for (const auto &attr: node.info().attributes()) {
|
|
columns.emplace_back(nullptr, attr.name(), attr.type(), attr.attributes());
|
|
}
|
|
std::unique_ptr<abstract_pk_generator> pk_generator;
|
|
switch (generator_type) {
|
|
case utils::generator_type::Identity:
|
|
pk_generator = std::make_unique<identity_pk_generator>();
|
|
break;
|
|
case utils::generator_type::Sequence:
|
|
pk_generator = std::make_unique<sequence_pk_generator>(node.name() + "_pk_seq");
|
|
break;
|
|
case utils::generator_type::Table:
|
|
pk_generator = std::make_unique<table_pk_generator>("sequence_table", node.name());
|
|
break;
|
|
default:
|
|
pk_generator = std::make_unique<manual_pk_generator>();
|
|
}
|
|
return schema_nodes_.insert({ti, schema_node{table(node.name(), columns), std::move(pk_generator), node}}).first;
|
|
}
|
|
|
|
basic_schema::iterator schema::insert_relation_table(const std::type_index &ti, const object::repository_node &node) {
|
|
std::vector<table_column> columns;
|
|
const auto* attr = node.info().object()->join_attribute();
|
|
if (attr == nullptr) {
|
|
return schema_nodes_.end();
|
|
}
|
|
columns.emplace_back(nullptr, attr->name(), attr->type(), attr->attributes());
|
|
attr = node.info().object()->inverse_join_attribute();
|
|
if (attr == nullptr) {
|
|
return schema_nodes_.end();
|
|
}
|
|
columns.emplace_back(nullptr, attr->name(), attr->type(), attr->attributes());
|
|
return schema_nodes_.insert({ti, schema_node{table(node.name(), columns, node.info().object()->join_attribute()->name(), node.info().object()->inverse_join_attribute()->name()), nullptr, node}}).first;
|
|
}
|
|
} // namespace matador::query
|