Referential actions in the Prisma Schema Language

.envrc

@@ -1,11 +1,11 @@
 export SERVER_ROOT=$(pwd)
 export RUST_LOG_FORMAT=devel
 export RUST_BACKTRACE=1
-export RUST_LOG=query_engine=debug,query_core=trace,query_connector=debug,sql_query_connector=debug,prisma_models=debug,engineer=info,sql_introspection_connector=debug
+export RUST_LOG=query_engine_tests=trace,query_engine=debug,query_core=trace,query_connector=debug,sql_query_connector=debug,prisma_models=debug,engineer=info,sql_introspection_connector=debug
 export SKIP_CONNECTORS="vitess_5_7,vitess_8_0"
 
 # Controls Scala test kit verbosity. Levels are trace, debug, info, error, warning
-export LOG_LEVEL=debug
+export LOG_LEVEL=trace
 
 # Controls Scala test kit test complexity ("simple", "complex")
 export TEST_MODE=simple

datamodel_v2.prisma

@@ -1,115 +1,40 @@
-datasource chinook {
-  provider = "sqlite"
-  url      = "file:./db/Chinook.db?connection_limit=1&socket_timeout=20"
+datasource db {
+  provider = "postgres"
+  url      = "postgresql://postgres:prisma@localhost:5432"
 }
 
 generator js {
   provider        = "prisma-client-js"
-  previewFeatures = ["microsoftSqlServer", "mongodb", "orderByRelation", "napi", "selectRelationCount", "orderByAggregateGroup"]
+  previewFeatures = ["microsoftSqlServer", "mongodb", "orderByRelation", "napi", "selectRelationCount", "orderByAggregateGroup", "referentialActions"]
 }
 
-model Album {
-  id       Int     @id @map("AlbumId")
-  Title    String  @default("TestDefaultTitle")
-  ArtistId Int
-  Tracks   Track[]
-  Artist   Artist  @relation(fields: [ArtistId], references: [id])
-}
-
-model Track {
-  id           Int           @id @map("TrackId")
-  Name         String
-  Composer     String?
-  Milliseconds Int
-  UnitPrice    Float
-  AlbumId      Int?
-  GenreId      Int?
-  MediaTypeId  Int
-  MediaType    MediaType     @relation(fields: [MediaTypeId], references: [id])
-  Genre        Genre?        @relation(fields: [GenreId], references: [id])
-  Album        Album?        @relation(fields: [AlbumId], references: [id])
-  InvoiceLines InvoiceLine[]
-}
-
-model MediaType {
-  id    Int     @id @map("MediaTypeId")
-  Name  String?
-  Track Track[]
-}
+model A {
+  id  String  @id
+  gql String?
 
-model Genre {
-  id     Int     @id @map("GenreId")
-  Name   String?
-  Tracks Track[]
-}
+  bs B[]
 
-model Artist {
-  id     Int     @id @map("ArtistId")
-  Name   String?
-  Albums Album[]
+  c_id String
+  c    C      @relation(fields: [c_id], onDelete: Cascade, references: [id])
 }
 
-model Customer {
-  id           Int       @id @map("CustomerId")
-  FirstName    String
-  LastName     String
-  Company      String?
-  Address      String?
-  City         String?
-  State        String?
-  Country      String?
-  PostalCode   String?
-  Phone        String?
-  Fax          String?
-  Email        String
-  SupportRepId Int?
-  SupportRep   Employee? @relation(fields: [SupportRepId], references: [id])
-  Invoices     Invoice[]
-}
+model B {
+  id  String  @id
+  gql String?
 
-model Employee {
-  id         Int        @id @map("EmployeeId")
-  FirstName  String
-  LastName   String
-  Title      String?
-  BirthDate  DateTime?
-  HireDate   DateTime?
-  Address    String?
-  City       String?
-  State      String?
-  Country    String?
-  PostalCode String?
-  Phone      String?
-  Fax        String?
-  Email      String?
-  Customers  Customer[]
+  a_id String
+  a    A      @relation(fields: [a_id], references: [id], onDelete: Cascade)
 }
 
-model Invoice {
-  id                Int           @id @map("InvoiceId")
-  InvoiceDate       DateTime
-  BillingAddress    String?
-  BillingCity       String?
-  BillingState      String?
-  BillingCountry    String?
-  BillingPostalCode String?
-  Total             Float
-  CustomerId        Int
-  Customer          Customer      @relation(fields: [CustomerId], references: [id])
-  Lines             InvoiceLine[]
-}
+model C {
+  id  String  @id
+  gql String?
+  ds  D[]     @relation(onDelete: Cascade)
 
-model InvoiceLine {
-  id        Int     @id @map("InvoiceLineId")
-  UnitPrice Float
-  Quantity  Int
-  InvoiceId Int
-  TrackId   Int
-  Invoice   Invoice @relation(fields: [InvoiceId], references: [id])
-  Track     Track   @relation(fields: [TrackId], references: [id])
+  a A?
 }
 
-model Playlist {
-  id   Int     @id @map("PlaylistId")
-  Name String?
+model D {
+  id String @id
+  cs C[]
 }

introspection-engine/connectors/sql-introspection-connector/src/calculate_datamodel.rs

@@ -1,11 +1,10 @@
-use crate::commenting_out_guardrails::commenting_out_guardrails;
-use crate::introspection::introspect;
 use crate::introspection_helpers::*;
 use crate::prisma_1_defaults::*;
 use crate::re_introspection::enrich;
 use crate::sanitize_datamodel_names::{sanitization_leads_to_duplicate_names, sanitize_datamodel_names};
 use crate::version_checker::VersionChecker;
 use crate::SqlIntrospectionResult;
+use crate::{commenting_out_guardrails::commenting_out_guardrails, introspection::introspect};
 use datamodel::Datamodel;
 use introspection_connector::IntrospectionResult;
 use quaint::connector::SqlFamily;
@@ -63,7 +62,7 @@ mod tests {
     use super::*;
     use datamodel::{
         dml, Datamodel, DefaultValue as DMLDefault, Field, FieldArity, FieldType, Model, NativeTypeInstance,
-        OnDeleteStrategy, RelationField, RelationInfo, ScalarField, ScalarType, ValueGenerator,
+        ReferentialAction, RelationField, RelationInfo, ScalarField, ScalarType, ValueGenerator,
     };
     use native_types::{NativeType, PostgresType};
     use pretty_assertions::assert_eq;
@@ -470,12 +469,15 @@ mod tests {
                         Field::RelationField(RelationField::new(
                             "User",
                             FieldArity::List,
+                            FieldArity::List,
                             RelationInfo {
                                 to: "User".to_string(),
                                 fields: vec![],
                                 references: vec![],
                                 name: "CityToUser".to_string(),
-                                on_delete: OnDeleteStrategy::None,
+                                on_delete: None,
+                                on_update: None,
+                                legacy_referential_actions: false,
                             },
                         )),
                     ],
@@ -557,17 +559,26 @@ mod tests {
                             is_commented_out: false,
                             is_ignored: false,
                         }),
-                        Field::RelationField(RelationField::new(
-                            "City",
-                            FieldArity::Required,
-                            RelationInfo {
+                        Field::RelationField(RelationField {
+                            name: "City".into(),
+                            arity: FieldArity::Required,
+                            referential_arity: FieldArity::Required,
+                            documentation: None,
+                            is_generated: false,
+                            is_commented_out: false,
+                            is_ignored: false,
+                            supports_restrict_action: Some(true),
+                            virtual_referential_actions: None,
+                            relation_info: RelationInfo {
                                 name: "CityToUser".to_string(),
                                 to: "City".to_string(),
                                 fields: vec!["city_id".to_string(), "city_name".to_string()],
                                 references: vec!["id".to_string(), "name".to_string()],
-                                on_delete: OnDeleteStrategy::None,
+                                on_delete: Some(ReferentialAction::NoAction),
+                                on_update: Some(ReferentialAction::NoAction),
+                                legacy_referential_actions: false,
                             },
-                        )),
+                        }),
                     ],
                     is_generated: false,
                     indices: vec![],
@@ -862,12 +873,15 @@ mod tests {
                         Field::RelationField(RelationField::new(
                             "User",
                             FieldArity::List,
+                            FieldArity::List,
                             RelationInfo {
                                 to: "User".to_string(),
                                 fields: vec![],
                                 references: vec![],
                                 name: "CityToUser".to_string(),
-                                on_delete: OnDeleteStrategy::None,
+                                on_delete: None,
+                                on_update: None,
+                                legacy_referential_actions: false,
                             },
                         )),
                     ],
@@ -918,17 +932,26 @@ mod tests {
                                 }),
                             ),
                         )),
-                        Field::RelationField(RelationField::new(
-                            "City",
-                            FieldArity::Required,
-                            RelationInfo {
+                        Field::RelationField(RelationField {
+                            name: "City".into(),
+                            arity: FieldArity::Required,
+                            referential_arity: FieldArity::Required,
+                            documentation: None,
+                            is_generated: false,
+                            is_commented_out: false,
+                            is_ignored: false,
+                            supports_restrict_action: Some(true),
+                            virtual_referential_actions: None,
+                            relation_info: RelationInfo {
                                 name: "CityToUser".to_string(),
                                 to: "City".to_string(),
                                 fields: vec!["city_id".to_string()],
                                 references: vec!["id".to_string()],
-                                on_delete: OnDeleteStrategy::None,
+                                on_delete: Some(ReferentialAction::NoAction),
+                                on_update: Some(ReferentialAction::NoAction),
+                                legacy_referential_actions: false,
                             },
-                        )),
+                        }),
                     ],
                     is_generated: false,
                     indices: vec![],

introspection-engine/connectors/sql-introspection-connector/src/introspection.rs

@@ -41,7 +41,11 @@ pub fn introspect(
         for foreign_key in &foreign_keys_copy {
             version_check.has_inline_relations(table);
             version_check.uses_on_delete(foreign_key, table);
-            let relation_field = calculate_relation_field(schema, table, foreign_key)?;
+
+            let mut relation_field = calculate_relation_field(schema, table, foreign_key)?;
+
+            relation_field.supports_restrict_action(!sql_family.is_mssql());
+
             model.add_field(Field::RelationField(relation_field));
         }