Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(relation-mode): remove emulated NoAction for postgres and sqlite #3274

Merged
merged 16 commits into from
Oct 11, 2022
Merged
Show file tree
Hide file tree
Changes from 10 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
8 changes: 6 additions & 2 deletions psl/builtin-connectors/src/cockroach_datamodel_connector.rs
Original file line number Diff line number Diff line change
Expand Up @@ -133,10 +133,14 @@ impl Connector for CockroachDatamodelConnector {
63
}

fn referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
fn referential_actions(&self, _relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
jkomyno marked this conversation as resolved.
Show resolved Hide resolved
use ReferentialAction::*;

relation_mode.allowed_referential_actions(NoAction | Restrict | Cascade | SetNull | SetDefault)
NoAction | Restrict | Cascade | SetNull | SetDefault
}

fn emulated_referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
relation_mode.allowed_emulated_referential_actions_default()
}

fn scalar_type_for_native_type(&self, native_type: serde_json::Value) -> ScalarType {
Expand Down
8 changes: 6 additions & 2 deletions psl/builtin-connectors/src/mongodb.rs
Original file line number Diff line number Diff line change
Expand Up @@ -54,8 +54,12 @@ impl Connector for MongoDbDatamodelConnector {
&[ConstraintScope::ModelKeyIndex]
}

fn referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
relation_mode.allowed_referential_actions(BitFlags::empty())
fn referential_actions(&self, _relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
BitFlags::empty()
}

fn emulated_referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
relation_mode.allowed_emulated_referential_actions_default()
}

fn validate_model(&self, model: ModelWalker<'_>, errors: &mut Diagnostics) {
Expand Down
8 changes: 6 additions & 2 deletions psl/builtin-connectors/src/mssql_datamodel_connector.rs
Original file line number Diff line number Diff line change
Expand Up @@ -171,10 +171,14 @@ impl Connector for MsSqlDatamodelConnector {
128
}

fn referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
fn referential_actions(&self, _relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
use ReferentialAction::*;

relation_mode.allowed_referential_actions(NoAction | Cascade | SetNull | SetDefault)
NoAction | Cascade | SetNull | SetDefault
}

fn emulated_referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
relation_mode.allowed_emulated_referential_actions_default()
}

fn scalar_type_for_native_type(&self, native_type: serde_json::Value) -> ScalarType {
Expand Down
8 changes: 6 additions & 2 deletions psl/builtin-connectors/src/mysql_datamodel_connector.rs
Original file line number Diff line number Diff line change
Expand Up @@ -160,10 +160,14 @@ impl Connector for MySqlDatamodelConnector {
64
}

fn referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
fn referential_actions(&self, _relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
use ReferentialAction::*;

relation_mode.allowed_referential_actions(Restrict | Cascade | SetNull | NoAction | SetDefault)
Restrict | Cascade | SetNull | NoAction | SetDefault
}

fn emulated_referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
relation_mode.allowed_emulated_referential_actions_default()
}
jkomyno marked this conversation as resolved.
Show resolved Hide resolved

fn scalar_type_for_native_type(&self, native_type: serde_json::Value) -> ScalarType {
Expand Down
10 changes: 8 additions & 2 deletions psl/builtin-connectors/src/postgres_datamodel_connector.rs
Original file line number Diff line number Diff line change
Expand Up @@ -154,10 +154,16 @@ impl Connector for PostgresDatamodelConnector {
63
}

fn referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
fn referential_actions(&self, _relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
use ReferentialAction::*;

relation_mode.allowed_referential_actions(NoAction | Restrict | Cascade | SetNull | SetDefault)
NoAction | Restrict | Cascade | SetNull | SetDefault
}

fn emulated_referential_actions(&self, _relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
use ReferentialAction::*;

Restrict | SetNull | Cascade
}

fn scalar_type_for_native_type(&self, native_type: serde_json::Value) -> ScalarType {
Expand Down
10 changes: 8 additions & 2 deletions psl/builtin-connectors/src/sqlite_datamodel_connector.rs
Original file line number Diff line number Diff line change
Expand Up @@ -43,10 +43,16 @@ impl Connector for SqliteDatamodelConnector {
10000
}

fn referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
fn referential_actions(&self, _relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
use ReferentialAction::*;

relation_mode.allowed_referential_actions(SetNull | SetDefault | Cascade | Restrict | NoAction)
SetNull | SetDefault | Cascade | Restrict | NoAction
}

fn emulated_referential_actions(&self, _relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
use ReferentialAction::*;

Restrict | SetNull | Cascade
}

fn scalar_type_for_native_type(&self, _native_type: serde_json::Value) -> ScalarType {
Expand Down
8 changes: 7 additions & 1 deletion psl/psl-core/src/datamodel_connector.rs
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,9 @@ pub trait Connector: Send + Sync {
/// The referential actions supported by the connector.
fn referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction>;
jkomyno marked this conversation as resolved.
Show resolved Hide resolved

/// The referential actions supported when using relationMode = "prisma" by the connector.
fn emulated_referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction>;
jkomyno marked this conversation as resolved.
Show resolved Hide resolved

fn supports_composite_types(&self) -> bool {
self.has_capability(ConnectorCapability::CompositeTypes)
}
Expand All @@ -90,7 +93,10 @@ pub trait Connector: Send + Sync {
}

fn supports_referential_action(&self, relation_mode: &RelationMode, action: ReferentialAction) -> bool {
self.referential_actions(relation_mode).contains(action)
match relation_mode {
RelationMode::ForeignKeys => self.referential_actions(relation_mode).contains(action),
RelationMode::Prisma => self.emulated_referential_actions(relation_mode).contains(action),
}
}

/// This is used by the query engine schema builder.
Expand Down
4 changes: 4 additions & 0 deletions psl/psl-core/src/datamodel_connector/empty_connector.rs
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,10 @@ impl Connector for EmptyDatamodelConnector {
BitFlags::all()
}

fn emulated_referential_actions(&self, relation_mode: &RelationMode) -> BitFlags<ReferentialAction> {
relation_mode.allowed_emulated_referential_actions_default()
}

fn capabilities(&self) -> &'static [ConnectorCapability] {
&[
ConnectorCapability::AutoIncrement,
Expand Down
13 changes: 2 additions & 11 deletions psl/psl-core/src/datamodel_connector/relation_mode.rs
Original file line number Diff line number Diff line change
Expand Up @@ -16,19 +16,10 @@ pub enum RelationMode {
}

impl RelationMode {
/// Returns either the given actions if foreign keys are used, or the
/// allowed emulated actions if referential integrity happens in Prisma.
pub fn allowed_referential_actions(
&self,
from_connector: BitFlags<ReferentialAction>,
) -> BitFlags<ReferentialAction> {
pub fn allowed_emulated_referential_actions_default(self) -> BitFlags<ReferentialAction> {
use ReferentialAction::*;

match self {
Self::ForeignKeys => from_connector,
// The emulated modes should be listed here.
Self::Prisma => Restrict | SetNull | NoAction | Cascade,
}
Restrict | SetNull | NoAction | Cascade
}

pub fn is_prisma(&self) -> bool {
Expand Down
Original file line number Diff line number Diff line change
@@ -1,9 +1,11 @@
use super::{database_name::validate_db_name, names::Names};
use crate::{
ast::{self, WithName, WithSpan},
datamodel_connector::RelationMode,
diagnostics::DatamodelError,
validate::validation_pipeline::context::Context,
};
use enumflags2::BitFlags;
use itertools::Itertools;
use parser_database::{
walkers::{ModelWalker, RelationFieldWalker, RelationName},
Expand Down Expand Up @@ -140,39 +142,79 @@ pub(super) fn ignored_related_model(field: RelationFieldWalker<'_>, ctx: &mut Co
pub(super) fn referential_actions(field: RelationFieldWalker<'_>, ctx: &mut Context<'_>) {
let connector = ctx.connector;
let relation_mode = ctx.relation_mode;
let msg = |action: ReferentialAction| {
let allowed_values = connector
.referential_actions(&relation_mode)
.iter()
.map(|f| format!("`{}`", f.as_str()))
.join(", ");

fn fmt_allowed_actions(allowed_actions: BitFlags<ReferentialAction>) -> String {
allowed_actions.iter().map(|f| format!("`{}`", f.as_str())).join(", ")
}

// validation template for relationMode = "foreignKeys"
let msg_foreign_keys = |action: ReferentialAction| {
let allowed_actions = connector.referential_actions(&relation_mode);

format!(
"Invalid referential action: `{}`. Allowed values: ({})",
action.as_str(),
allowed_values,
fmt_allowed_actions(allowed_actions),
)
};

// validation template for relationMode = "prisma"
let msg_prisma = |action: ReferentialAction| {
let allowed_actions = connector.emulated_referential_actions(&relation_mode);

let additional_info = match action {
ReferentialAction::NoAction => {
if ctx
.connector
.supports_referential_action(&relation_mode, ReferentialAction::Restrict)
tomhoule marked this conversation as resolved.
Show resolved Hide resolved
{
Some(format!(
". `{}` is not implemented for {} when using `relationMode = \"prisma\"`, you could try using `{}`, which behaves the same if you do not need to defer constraint checks in a transaction. Learn more at https://pris.ly/d/relationMode",
jkomyno marked this conversation as resolved.
Show resolved Hide resolved
ReferentialAction::NoAction.as_str(),
connector.name(),
ReferentialAction::Restrict.as_str(),
))
} else {
None
}
}
_ => None,
};
let additional_info = additional_info.unwrap_or_default();

format!(
"Invalid referential action: `{}`. Allowed values: ({}){additional_info}",
action.as_str(),
fmt_allowed_actions(allowed_actions),
)
};

let msg_template = |action: ReferentialAction| -> String {
match relation_mode {
RelationMode::ForeignKeys => msg_foreign_keys(action),
RelationMode::Prisma => msg_prisma(action),
}
};

if let Some(on_delete) = field.explicit_on_delete() {
if !ctx.connector.supports_referential_action(&ctx.relation_mode, on_delete) {
let span = field
.ast_field()
.span_for_argument("relation", "onDelete")
.unwrap_or_else(|| field.ast_field().span());
let span = field
.ast_field()
.span_for_argument("relation", "onDelete")
.unwrap_or_else(|| field.ast_field().span());

ctx.push_error(DatamodelError::new_validation_error(&msg(on_delete), span));
if !ctx.connector.supports_referential_action(&ctx.relation_mode, on_delete) {
ctx.push_error(DatamodelError::new_validation_error(&msg_template(on_delete), span));
}
}

if let Some(on_update) = field.explicit_on_update() {
if !ctx.connector.supports_referential_action(&ctx.relation_mode, on_update) {
let span = field
.ast_field()
.span_for_argument("relation", "onUpdate")
.unwrap_or_else(|| field.ast_field().span());
let span = field
.ast_field()
.span_for_argument("relation", "onUpdate")
.unwrap_or_else(|| field.ast_field().span());

ctx.push_error(DatamodelError::new_validation_error(&msg(on_update), span));
if !ctx.connector.supports_referential_action(&ctx.relation_mode, on_update) {
ctx.push_error(DatamodelError::new_validation_error(&msg_template(on_update), span));
}
}
}
Expand Down