Show selection in update nodes in graphviz visualization

prisma · Nov 30, 2023 · 23801aa · 23801aa
1 parent 7523165
commit 23801aa
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/query-engine/core/src/query_ast/write.rs b/query-engine/core/src/query_ast/write.rs
@@ -206,7 +206,11 @@ impl ToGraphviz for WriteQuery {
         match self {
             Self::CreateRecord(q) => format!("CreateRecord(model: {}, args: {:?})", q.model.name(), q.args),
             Self::CreateManyRecords(q) => format!("CreateManyRecord(model: {})", q.model.name()),
-            Self::UpdateRecord(q) => format!("UpdateRecord(model: {})", q.model().name(),),
+            Self::UpdateRecord(q) => format!(
+                "UpdateRecord(model: {}, selection: {:?})",
+                q.model().name(),
+                q.selected_fields()
+            ),
             Self::DeleteRecord(q) => format!("DeleteRecord: {}, {:?}", q.model.name(), q.record_filter),
             Self::UpdateManyRecords(q) => format!("UpdateManyRecords(model: {}, args: {:?})", q.model.name(), q.args),
             Self::DeleteManyRecords(q) => format!("DeleteManyRecords: {}", q.model.name()),