Ensure a bunch of unsupported capabilities excluded during compile time

prisma · Dec 22, 2023 · afc35d7 · afc35d7
1 parent 5d218b4
commit afc35d7
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/query-engine/schema/Cargo.toml b/query-engine/schema/Cargo.toml
@@ -8,6 +8,7 @@ query-structure = { path = "../query-structure" }
 psl.workspace = true
 rustc-hash = "1.1.0"
 once_cell = "1"
+enumflags2 = "0.7"
 
 [dev-dependencies]
 codspeed-criterion-compat = "1.1.0"

diff --git a/query-engine/schema/src/query_schema.rs b/query-engine/schema/src/query_schema.rs
@@ -14,6 +14,23 @@ enum Operation {
 
 type LazyField = Box<dyn for<'a> Fn(&'a QuerySchema) -> OutputField<'a> + Send + Sync>;
 
+const CAPABILITIES: ConnectorCapabilities = enumflags2::make_bitflags!(ConnectorCapability::{
+    AnyId |
+    AutoIncrement |
+    CompoundIds |
+    SqlQueryRaw |
+    RelationFieldsInArbitraryOrder |
+    UpdateableId |
+    ImplicitManyToManyRelation |
+    DecimalType |
+    BackwardCompatibleQueryRaw |
+    OrderByNullsFirstLast |
+    SupportsTxIsolationSerializable |
+    NativeUpsert |
+    FilteredInlineChildNestedToOneDisconnect |
+    RowIn
+});
+
 /// The query schema defines which operations (query/mutations) are possible on a database, based
 /// on a Prisma schema.
 ///
@@ -101,7 +118,7 @@ impl QuerySchema {
     }
 
     pub fn has_capability(&self, capability: ConnectorCapability) -> bool {
-        self.connector.has_capability(capability)
+        CAPABILITIES.contains(capability)
     }
 
     pub fn capabilities(&self) -> ConnectorCapabilities {