fix #787

prisma · Jun 5, 2020 · f78e4ad · f78e4ad
1 parent 4ec10da
commit f78e4ad
Show file tree

Hide file tree

Showing 9 changed files with 52 additions and 13 deletions.
diff --git a/libs/datamodel/core/src/json/dmmf/mod.rs b/libs/datamodel/core/src/json/dmmf/mod.rs
@@ -57,6 +57,14 @@ pub struct Model {
     pub documentation: Option<String>,
     pub id_fields: Vec<String>,
     pub unique_fields: Vec<Vec<String>>,
+    pub unique_indexes: Vec<UniqueIndex>,
+}
+
+#[serde(rename_all = "camelCase")]
+#[derive(Debug, serde::Serialize, serde::Deserialize)]
+pub struct UniqueIndex {
+    pub name: Option<String>,
+    pub fields: Vec<String>,
 }
 
 #[serde(rename_all = "camelCase")]

diff --git a/libs/datamodel/core/src/json/dmmf/to_dmmf.rs b/libs/datamodel/core/src/json/dmmf/to_dmmf.rs
@@ -72,6 +72,20 @@ fn model_to_dmmf(model: &dml::Model) -> Model {
                 }
             })
             .collect(),
+        unique_indexes: model
+            .indices
+            .iter()
+            .filter_map(|i| {
+                if i.tpe == IndexType::Unique {
+                    Some(UniqueIndex {
+                        name: i.name.clone(),
+                        fields: i.fields.clone(),
+                    })
+                } else {
+                    None
+                }
+            })
+            .collect(),
     }
 }
 

diff --git a/libs/datamodel/core/tests/render_to_dmmf/files/functions.json b/libs/datamodel/core/tests/render_to_dmmf/files/functions.json
@@ -48,7 +48,8 @@
       ],
       "isGenerated": false,
       "idFields": [],
-      "uniqueFields": []
+      "uniqueFields": [],
+      "uniqueIndexes": []
     }
   ]
 }
diff --git a/libs/datamodel/core/tests/render_to_dmmf/files/general.json b/libs/datamodel/core/tests/render_to_dmmf/files/general.json
@@ -114,7 +114,8 @@
       ],
       "isGenerated": false,
       "idFields": [],
-      "uniqueFields": []
+      "uniqueFields": [],
+      "uniqueIndexes": []
     },
     {
       "name": "Profile",
@@ -184,7 +185,8 @@
       ],
       "isGenerated": false,
       "idFields": [],
-      "uniqueFields": []
+      "uniqueFields": [],
+      "uniqueIndexes": []
     },
     {
       "name": "Post",
@@ -326,7 +328,8 @@
         "title",
         "createdAt"
       ],
-      "uniqueFields": []
+      "uniqueFields": [],
+      "uniqueIndexes": []
     },
     {
       "name": "Category",
@@ -392,7 +395,8 @@
       ],
       "isGenerated": false,
       "idFields": [],
-      "uniqueFields": []
+      "uniqueFields": [],
+      "uniqueIndexes": []
     },
     {
       "name": "PostToCategory",
@@ -498,7 +502,11 @@
       ],
       "isGenerated": false,
       "idFields": [],
-      "uniqueFields": []
+      "uniqueFields": [["postTitle", "categoryId"]],
+      "uniqueIndexes": [{
+        "name": "MyUniqueIndex",
+        "fields": ["postTitle", "categoryId"]
+      }]
     },
     {
       "name": "A",
@@ -555,7 +563,8 @@
       ],
       "isGenerated": false,
       "idFields": [],
-      "uniqueFields": []
+      "uniqueFields": [],
+      "uniqueIndexes": []
     },
     {
       "name": "B",
@@ -595,7 +604,8 @@
       ],
       "isGenerated": false,
       "idFields": [],
-      "uniqueFields": []
+      "uniqueFields": [],
+      "uniqueIndexes": []
     }
   ]
 }
diff --git a/libs/datamodel/core/tests/render_to_dmmf/files/general.prisma b/libs/datamodel/core/tests/render_to_dmmf/files/general.prisma
@@ -50,6 +50,7 @@ model PostToCategory {
   post          Post     @relation(fields: [postTitle, postCreatedAt], references: [title, createdAt])
   category      Category @relation(fields: [categoryId], references: [id])
 
+  @@unique([postTitle, categoryId], name: "MyUniqueIndex")
   @@map("post_to_category")
 }
 

diff --git a/libs/datamodel/core/tests/render_to_dmmf/files/source.json b/libs/datamodel/core/tests/render_to_dmmf/files/source.json
@@ -48,7 +48,8 @@
       ],
       "isGenerated": false,
       "idFields": [],
-      "uniqueFields": []
+      "uniqueFields": [],
+      "uniqueIndexes": []
     }
   ]
 }
diff --git a/libs/datamodel/core/tests/render_to_dmmf/files/source_with_comments.json b/libs/datamodel/core/tests/render_to_dmmf/files/source_with_comments.json
@@ -37,7 +37,8 @@
       "isGenerated": false,
       "documentation": "My author model.",
       "idFields": [],
-      "uniqueFields": []
+      "uniqueFields": [],
+      "uniqueIndexes": []
     }
   ]
 }
diff --git a/libs/datamodel/core/tests/render_to_dmmf/files/source_with_generator.json b/libs/datamodel/core/tests/render_to_dmmf/files/source_with_generator.json
@@ -35,7 +35,8 @@
       ],
       "isGenerated": false,
       "idFields": [],
-      "uniqueFields": []
+      "uniqueFields": [],
+      "uniqueIndexes": []
     }
   ]
 }
diff --git a/libs/datamodel/core/tests/render_to_dmmf/files/without_relation_name.json b/libs/datamodel/core/tests/render_to_dmmf/files/without_relation_name.json
@@ -39,7 +39,8 @@
       ],
       "isGenerated": false,
       "idFields": [],
-      "uniqueFields": []
+      "uniqueFields": [],
+      "uniqueIndexes": []
     },
     {
       "name": "Post",
@@ -96,7 +97,8 @@
       ],
       "isGenerated": false,
       "idFields": [],
-      "uniqueFields": []
+      "uniqueFields": [],
+      "uniqueIndexes": []
     }
   ]
 }