use super::{InformationSchema, SchemaQueryBuilder};
use crate::sqlx_types::mysql::MySqlRow;
use sea_query::{Expr, Iden, Order, Query, SeaRc, SelectStatement};
#[derive(Debug, sea_query::Iden)]
pub enum KeyColumnUsageFields {
ConstraintSchema,
ConstraintName,
TableSchema,
TableName,
ColumnName,
OrdinalPosition,
PositionInUniqueConstraint,
ReferencedTableSchema,
ReferencedTableName,
ReferencedColumnName,
}
#[derive(Debug, sea_query::Iden)]
pub enum ReferentialConstraintsFields {
ConstraintSchema,
ConstraintName,
UniqueConstraintSchema,
UniqueConstraintName,
UpdateRule,
DeleteRule,
TableName,
ReferencedTableName,
}
#[derive(Debug, Default)]
pub struct ForeignKeyQueryResult {
pub constraint_name: String,
pub column_name: String,
pub referenced_table_name: String,
pub referenced_column_name: String,
pub update_rule: String,
pub delete_rule: String,
}
impl SchemaQueryBuilder {
pub fn query_foreign_key(
&self,
schema: SeaRc<dyn Iden>,
table: SeaRc<dyn Iden>,
) -> SelectStatement {
type Schema = InformationSchema;
type Key = KeyColumnUsageFields;
type Ref = ReferentialConstraintsFields;
Query::select()
.columns(vec![
(Schema::KeyColumnUsage, Key::ConstraintName),
(Schema::KeyColumnUsage, Key::ColumnName),
(Schema::KeyColumnUsage, Key::ReferencedTableName),
(Schema::KeyColumnUsage, Key::ReferencedColumnName),
])
.columns(vec![
(Schema::ReferentialConstraints, Ref::UpdateRule),
(Schema::ReferentialConstraints, Ref::DeleteRule),
])
.from((Schema::Schema, Schema::KeyColumnUsage))
.inner_join(
(Schema::Schema, Schema::ReferentialConstraints),
Expr::col((Schema::KeyColumnUsage, Key::ConstraintSchema))
.equals((Schema::ReferentialConstraints, Ref::ConstraintSchema))
.and(
Expr::col((Schema::KeyColumnUsage, Key::ConstraintName))
.equals((Schema::ReferentialConstraints, Ref::ConstraintName)),
),
)
.and_where(
Expr::col((Schema::KeyColumnUsage, Key::ConstraintSchema)).eq(schema.to_string()),
)
.and_where(Expr::col((Schema::KeyColumnUsage, Key::TableName)).eq(table.to_string()))
.and_where(Expr::col((Schema::KeyColumnUsage, Key::ReferencedTableName)).is_not_null())
.and_where(Expr::col((Schema::KeyColumnUsage, Key::ReferencedColumnName)).is_not_null())
.order_by(Key::ConstraintName, Order::Asc)
.order_by(Key::OrdinalPosition, Order::Asc)
.take()
}
}
#[cfg(feature = "sqlx-mysql")]
impl From<&MySqlRow> for ForeignKeyQueryResult {
fn from(row: &MySqlRow) -> Self {
use crate::sqlx_types::Row;
Self {
constraint_name: row.get(0),
column_name: row.get(1),
referenced_table_name: row.get(2),
referenced_column_name: row.get(3),
update_rule: row.get(4),
delete_rule: row.get(5),
}
}
}
#[cfg(not(feature = "sqlx-mysql"))]
impl From<&MySqlRow> for ForeignKeyQueryResult {
fn from(_: &MySqlRow) -> Self {
Self::default()
}
}